container_of: Add container_of_or_null()

[ccan] / ccan / container_of / test / run.c

diff --git a/ccan/container_of/test/run.c b/ccan/container_of/test/run.c
index 5da440a1e5d09ca05c3c36f03d04a149d694a9ab..18207f60521f418d75df8cee15293c062acc1a21 100644 (file)
--- a/ccan/container_of/test/run.c
+++ b/ccan/container_of/test/run.c
@@ -12,9 +12,13 @@ int main(int argc, char *argv[])
        int *intp = &foo.a;
        char *charp = &foo.b;
 
-       plan_tests(8);
+       plan_tests(12);
        ok1(container_of(intp, struct foo, a) == &foo);
        ok1(container_of(charp, struct foo, b) == &foo);
+       ok1(container_of_or_null(intp, struct foo, a) == &foo);
+       ok1(container_of_or_null(charp, struct foo, b) == &foo);
+       ok1(container_of_or_null((int *)NULL, struct foo, a) == NULL);
+       ok1(container_of_or_null((char *)NULL, struct foo, b) == NULL);
        ok1(container_of_var(intp, &foo, a) == &foo);
        ok1(container_of_var(charp, &foo, b) == &foo);