diff -r a39e410a05e6 -r 6456036bbb37 tests/test_tree.c --- a/tests/test_tree.c Wed Mar 20 23:35:32 2024 +0100 +++ b/tests/test_tree.c Wed Apr 03 21:22:23 2024 +0200 @@ -644,6 +644,196 @@ } } +CX_TEST(test_tree_visitor_continue) { + tree_node root = {0}; + tree_node a = {0}; + tree_node b = {0}; + tree_node c = {0}; + tree_node aa = {0}; + tree_node ab = {0}; + tree_node ba = {0}; + tree_node ca = {0}; + tree_node cb = {0}; + tree_node cc = {0}; + tree_node cba = {0}; + + tree_node* expected_order[] = { + &root, + &c, &b, &a, + &ba, &ab, &aa + }; + tree_node* actual_order[16]; // reserve more space in case s.t. goes wrong + + cx_tree_link(&root, &a, tree_node_layout); + cx_tree_link(&root, &b, tree_node_layout); + cx_tree_link(&root, &c, tree_node_layout); + cx_tree_link(&a, &aa, tree_node_layout); + cx_tree_link(&a, &ab, tree_node_layout); + cx_tree_link(&b, &ba, tree_node_layout); + cx_tree_link(&c, &ca, tree_node_layout); + cx_tree_link(&c, &cb, tree_node_layout); + cx_tree_link(&c, &cc, tree_node_layout); + cx_tree_link(&cb, &cba, tree_node_layout); + CX_TEST_DO { + CxTreeVisitor iter = cx_tree_visitor(&root, tree_child_list); + unsigned chk = 0; + cx_foreach(tree_node*, node, iter) { + CX_TEST_ASSERT(node->data == 0); + node->data++; + actual_order[chk] = node; + chk++; + CX_TEST_ASSERT(node == iter.node); + if (node == &root) { + CX_TEST_ASSERT(iter.depth == 1); + } else if (node == &a || node == &b || node == &c) { + CX_TEST_ASSERT(iter.depth == 2); + } else { + CX_TEST_ASSERT(iter.depth == 3); + } + if (node == &c) { + cxTreeVisitorContinue(iter); + } + } + CX_TEST_ASSERT(iter.counter == 7); + CX_TEST_ASSERT(chk == 7); + for (unsigned i = 0 ; i < chk ; i++) { + CX_TEST_ASSERT(actual_order[i] == expected_order[i]); + CX_TEST_ASSERT(actual_order[i]->data == 1); + } + CX_TEST_ASSERT(ca.data == 0); + CX_TEST_ASSERT(cb.data == 0); + CX_TEST_ASSERT(cc.data == 0); + CX_TEST_ASSERT(cba.data == 0); + CX_TEST_ASSERT(iter.queue_next == NULL); + CX_TEST_ASSERT(iter.queue_last == NULL); + } +} + +CX_TEST(test_tree_iterator_continue) { + tree_node root = {0}; + tree_node a = {0}; + tree_node b = {0}; + tree_node c = {0}; + tree_node aa = {0}; + tree_node ab = {0}; + tree_node ba = {0}; + tree_node ca = {0}; + tree_node cb = {0}; + tree_node cc = {0}; + tree_node cba = {0}; + + tree_node* expected_order[] = { + &root, + &c, + &b, &ba, + &a, &ab, &aa, + }; + tree_node* actual_order[16]; // reserve more space in case s.t. goes wrong + + cx_tree_link(&root, &a, tree_node_layout); + cx_tree_link(&root, &b, tree_node_layout); + cx_tree_link(&root, &c, tree_node_layout); + cx_tree_link(&a, &aa, tree_node_layout); + cx_tree_link(&a, &ab, tree_node_layout); + cx_tree_link(&b, &ba, tree_node_layout); + cx_tree_link(&c, &ca, tree_node_layout); + cx_tree_link(&c, &cb, tree_node_layout); + cx_tree_link(&c, &cc, tree_node_layout); + cx_tree_link(&cb, &cba, tree_node_layout); + CX_TEST_DO { + CxTreeIterator iter = cx_tree_iterator(&root, false, tree_child_list); + unsigned chk = 0; + cx_foreach(tree_node*, node, iter) { + CX_TEST_ASSERT(node->data == 0); + node->data++; + actual_order[chk] = node; + chk++; + CX_TEST_ASSERT(node == iter.node); + CX_TEST_ASSERT(!iter.exiting); + if (node == &root) { + CX_TEST_ASSERT(iter.depth == 1); + } else if (node == &a || node == &b || node == &c) { + CX_TEST_ASSERT(iter.depth == 2); + } else { + CX_TEST_ASSERT(iter.depth == 3); + } + if (node == &c) { + cxTreeIteratorContinue(iter); + } + } + CX_TEST_ASSERT(iter.counter == 7); + CX_TEST_ASSERT(chk == 7); + for (unsigned i = 0 ; i < chk ; i++) { + CX_TEST_ASSERT(actual_order[i] == expected_order[i]); + CX_TEST_ASSERT(actual_order[i]->data == 1); + } + CX_TEST_ASSERT(ca.data == 0); + CX_TEST_ASSERT(cb.data == 0); + CX_TEST_ASSERT(cc.data == 0); + CX_TEST_ASSERT(cba.data == 0); + CX_TEST_ASSERT(iter.stack == NULL); + } +} + +CX_TEST(test_tree_iterator_continue_with_exit) { + tree_node root = {0}; + tree_node a = {0}; + tree_node b = {0}; + tree_node c = {0}; + tree_node aa = {0}; + tree_node ab = {0}; + tree_node ba = {0}; + tree_node ca = {0}; + tree_node cb = {0}; + tree_node cc = {0}; + tree_node cba = {0}; + + cx_tree_link(&root, &a, tree_node_layout); + cx_tree_link(&root, &b, tree_node_layout); + cx_tree_link(&root, &c, tree_node_layout); + cx_tree_link(&a, &aa, tree_node_layout); + cx_tree_link(&a, &ab, tree_node_layout); + cx_tree_link(&b, &ba, tree_node_layout); + cx_tree_link(&c, &ca, tree_node_layout); + cx_tree_link(&c, &cb, tree_node_layout); + cx_tree_link(&c, &cc, tree_node_layout); + cx_tree_link(&cb, &cba, tree_node_layout); + CX_TEST_DO { + CxTreeIterator iter = cx_tree_iterator(&root, true, tree_child_list); + unsigned chk = 0; + cx_foreach(tree_node*, node, iter) { + CX_TEST_ASSERT(iter.exiting || node->data == 0); + node->data++; + chk++; + CX_TEST_ASSERT(node == iter.node); + if (node == &root) { + CX_TEST_ASSERT(iter.depth == 1); + } else if (node == &a || node == &b || node == &c) { + CX_TEST_ASSERT(iter.depth == 2); + } else { + CX_TEST_ASSERT(iter.depth == 3); + } + if (node == &c) { + cxTreeIteratorContinue(iter); + } + } + CX_TEST_ASSERT(iter.counter == 7); + CX_TEST_ASSERT(chk == 14); + CX_TEST_ASSERT(iter.stack == NULL); + CX_TEST_ASSERT(root.data == 2); + CX_TEST_ASSERT(a.data == 2); + CX_TEST_ASSERT(b.data == 2); + CX_TEST_ASSERT(c.data == 2); + CX_TEST_ASSERT(aa.data == 2); + CX_TEST_ASSERT(ab.data == 2); + CX_TEST_ASSERT(ba.data == 2); + CX_TEST_ASSERT(ca.data == 0); + CX_TEST_ASSERT(cb.data == 0); + CX_TEST_ASSERT(cc.data == 0); + CX_TEST_ASSERT(cba.data == 0); + } +} + CxTestSuite *cx_test_suite_tree_low_level(void) { CxTestSuite *suite = cx_test_suite_new("tree (low level)"); @@ -660,6 +850,9 @@ cx_test_register(suite, test_tree_visitor); cx_test_register(suite, test_tree_visitor_no_children); cx_test_register(suite, test_tree_visitor_no_branching); + cx_test_register(suite, test_tree_visitor_continue); + cx_test_register(suite, test_tree_iterator_continue); + cx_test_register(suite, test_tree_iterator_continue_with_exit); return suite; }