avltree: Added sdb_avltree_lookup().

[sysdb.git] / t / unit / utils / avltree_test.c

diff --git a/t/unit/utils/avltree_test.c b/t/unit/utils/avltree_test.c
index d978eff9da688fc4306728f060db34a9a9e04609..3526e4107df078c5fa2f3aa99913f25659bccbf3 100644 (file)
--- a/t/unit/utils/avltree_test.c
+++ b/t/unit/utils/avltree_test.c
@@ -47,21 +47,27 @@ teardown(void)
        tree = NULL;
 } /* teardown */
 
-/* 'a' - 'k' */
+/* 'a' thru 'o' */
 static sdb_object_t test_data[] = {
-       SSTRING_OBJ("d"),
-       SSTRING_OBJ("f"),
-       SSTRING_OBJ("e"),
-       SSTRING_OBJ("b"),
-       SSTRING_OBJ("a"),
-       SSTRING_OBJ("c"),
-       SSTRING_OBJ("g"),
-       SSTRING_OBJ("h"),
-       SSTRING_OBJ("i"),
-       SSTRING_OBJ("j"),
-       SSTRING_OBJ("k"),
+       SDB_OBJECT_STATIC("h"),
+       SDB_OBJECT_STATIC("j"),
+       SDB_OBJECT_STATIC("i"),
+       SDB_OBJECT_STATIC("f"),
+       SDB_OBJECT_STATIC("e"),
+       SDB_OBJECT_STATIC("g"),
+       SDB_OBJECT_STATIC("k"),
+       SDB_OBJECT_STATIC("l"),
+       SDB_OBJECT_STATIC("m"),
+       SDB_OBJECT_STATIC("n"),
+       SDB_OBJECT_STATIC("o"),
+       SDB_OBJECT_STATIC("d"),
+       SDB_OBJECT_STATIC("c"),
+       SDB_OBJECT_STATIC("b"),
+       SDB_OBJECT_STATIC("a"),
 };
 
+static char *unused_names[] = { "x", "y", "z" };
+
 static void
 populate(void)
 {
@@ -72,7 +78,7 @@ populate(void)
 
 START_TEST(test_null)
 {
-       sdb_object_t o1 = SSTRING_OBJ("obj");
+       sdb_object_t o1 = SDB_OBJECT_STATIC("obj");
        sdb_object_t *o2;
        sdb_avltree_iter_t *iter;
        int check;
@@ -126,6 +132,57 @@ START_TEST(test_insert)
                fail_unless(check == (int)i + 1,
                                "sdb_avltree_size(<tree>) = %d; expected: %zu",
                                check, i + 1);
+
+               fail_unless(sdb_avltree_valid(tree),
+                               "sdb_avltree_insert(<tree>, <%s>) left behind invalid tree",
+                               test_data[i].name);
+       }
+
+       /* and again ... now reporting errors because of duplicates */
+       for (i = 0; i < SDB_STATIC_ARRAY_LEN(test_data); ++i) {
+               int check;
+
+               check = sdb_avltree_insert(tree, &test_data[i]);
+               fail_unless(check < 0,
+                               "sdb_avltree_insert(<tree>, <%s>) = %d (redo); expected: <0",
+                               test_data[i].name, check);
+
+               check = (int)sdb_avltree_size(tree);
+               fail_unless(check == SDB_STATIC_ARRAY_LEN(test_data),
+                               "sdb_avltree_size(<tree>) = %d; expected: %zu",
+                               check, SDB_STATIC_ARRAY_LEN(test_data));
+       }
+}
+END_TEST
+
+START_TEST(test_lookup)
+{
+       size_t i;
+
+       populate();
+
+       for (i = 0; i < SDB_STATIC_ARRAY_LEN(test_data); ++i) {
+               sdb_object_t ref = SDB_OBJECT_STATIC(test_data[i].name);
+               sdb_object_t *obj;
+
+               obj = sdb_avltree_lookup(tree, &ref);
+               fail_unless(obj != NULL,
+                               "sdb_avltree_lookup(<tree>, <%s>) = NULL; "
+                               "expected: <obj>", ref.name);
+               fail_unless(obj == &test_data[i],
+                               "sdb_avltree_lookup(<tree>, <%s>) = %p (%s); "
+                               "expected: %p, (%s)", ref.name, obj, obj->name,
+                               &test_data[i], test_data[i].name);
+       }
+
+       for (i = 0; i < SDB_STATIC_ARRAY_LEN(unused_names); ++i) {
+               sdb_object_t ref = SDB_OBJECT_STATIC(unused_names[i]);
+               sdb_object_t *obj;
+
+               obj = sdb_avltree_lookup(tree, &ref);
+               fail_unless(obj == NULL,
+                               "sdb_avltree_lookup(<tree>, <%s> = %p (%s); "
+                               "expected: NULL", ref.name, obj, obj ? obj->name : "<nil>");
        }
 }
 END_TEST
@@ -171,6 +228,12 @@ START_TEST(test_iter)
                        "sdb_avltree_iter_get_next(<iter>) = <obj>; expected: NULL");
 
        sdb_avltree_iter_destroy(iter);
+
+       sdb_avltree_clear(tree);
+       check = sdb_avltree_size(tree);
+       fail_unless(check == 0,
+                       "sdb_avltree_clear(<tree>) left %zu nodes in the tree; "
+                       "expected: 0", check);
 }
 END_TEST
 
@@ -184,6 +247,7 @@ util_avltree_suite(void)
        tcase_add_checked_fixture(tc, setup, teardown);
        tcase_add_test(tc, test_null);
        tcase_add_test(tc, test_insert);
+       tcase_add_test(tc, test_lookup);
        tcase_add_test(tc, test_iter);
        suite_add_tcase(s, tc);