avltree: Add sdb_avltree_iter_peek_next().

[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 9be5e0960821626be4e842350da7314723306015..433f67d7ea42d89151aa0b4f4336696f2e7a7213 100644 (file)
--- a/t/unit/utils/avltree_test.c
+++ b/t/unit/utils/avltree_test.c
@@ -209,11 +209,17 @@ START_TEST(test_iter)
 
        for (i = 0; i < SDB_STATIC_ARRAY_LEN(test_data); ++i) {
                char expected_name[] = { (char)('a' + (int)i), '\0' };
+               sdb_object_t *expected_obj;
 
                _Bool c = sdb_avltree_iter_has_next(iter);
                fail_unless(c, "sdb_avltree_iter_has_next(<iter[%zu]>) = false; "
                                "expected: true", i);
 
+               expected_obj = sdb_avltree_iter_peek_next(iter);
+               fail_unless(expected_obj != NULL,
+                               "sdb_avltree_iter_peek_next(<iter[%zu]>) = NULL; "
+                               "expected: <obj>", i);
+
                obj = sdb_avltree_iter_get_next(iter);
                fail_unless(obj != NULL,
                                "sdb_avltree_iter_get_next(<iter[%zu]>) = NULL; "
@@ -221,11 +227,18 @@ START_TEST(test_iter)
                fail_unless(!strcmp(obj->name, expected_name),
                                "sdb_avltree_iter[%zu] = %s; expected: %s",
                                i, obj->name, expected_name);
+
+               fail_unless(obj == expected_obj,
+                               "sdb_avltree_iter_get_next(<iter[%zu]>) = %p; "
+                               "expected: %p (from peek())", i, obj, expected_obj);
        }
 
        check = sdb_avltree_iter_has_next(iter) != 0;
        fail_unless(check == 0, "sdb_avltree_iter_has_next(<iter>) = true; "
                        "expected: false");
+       obj = sdb_avltree_iter_peek_next(iter);
+       fail_unless(obj == NULL,
+                       "sdb_avltree_iter_peek_next(<iter>) = <obj>; expected: NULL");
        obj = sdb_avltree_iter_get_next(iter);
        fail_unless(obj == NULL,
                        "sdb_avltree_iter_get_next(<iter>) = <obj>; expected: NULL");