summaryrefslogtreecommitdiff
path: root/test/test_rbtree.cc
diff options
context:
space:
mode:
authorRichard Qin <richard.qin.x@gmail.com>2026-07-10 11:31:46 +0800
committerRichard Qin <richard.qin.x@gmail.com>2026-07-10 11:31:46 +0800
commite90b72ec3082588b296956730bd2930cf8d5d671 (patch)
tree36ff6916af9b4176ea5a9662219d11e721f0b691 /test/test_rbtree.cc
parent1767d390fbf3d0c65389de46caf0b6d43ef3b6ad (diff)
downloadlume-e90b72ec3082588b296956730bd2930cf8d5d671.tar.gz
lume-e90b72ec3082588b296956730bd2930cf8d5d671.tar.xz
refactor: promote new architecture to repository root
Archive the deprecated implementation under lume-legacy, make the new kernel the default root-level build, and update repository entry points without changing kernel behavior.
Diffstat (limited to 'test/test_rbtree.cc')
-rw-r--r--test/test_rbtree.cc119
1 files changed, 119 insertions, 0 deletions
diff --git a/test/test_rbtree.cc b/test/test_rbtree.cc
new file mode 100644
index 0000000..27f049a
--- /dev/null
+++ b/test/test_rbtree.cc
@@ -0,0 +1,119 @@
+/* SPDX-License-Identifier: GPL-2.0-or-later */
+/* Copyright (C) 2026 Richard Qin */
+
+#include <lume/selftest.h>
+#include <lume/rbtree.h>
+
+struct TestNode {
+ int key;
+ struct rb_node rb;
+};
+
+static bool node_less(struct rb_node *node, const struct rb_node *parent) {
+ TestNode *n = rb_entry(node, TestNode, rb);
+ const TestNode *p = reinterpret_cast<const TestNode*>(
+ reinterpret_cast<const char*>(parent) - __builtin_offsetof(TestNode, rb));
+ return n->key < p->key;
+}
+
+static int node_cmp(const void *key, const struct rb_node *node) {
+ int k = *static_cast<const int*>(key);
+ const TestNode *n = reinterpret_cast<const TestNode*>(
+ reinterpret_cast<const char*>(node) - __builtin_offsetof(TestNode, rb));
+ if (k < n->key) return -1;
+ if (k > n->key) return 1;
+ return 0;
+}
+
+void selftest_rbtree() {
+ st_begin("rbtree: basic insert, find, erase");
+ {
+ struct rb_root tree = RB_ROOT;
+ TestNode n1{10, {}}, n2{20, {}}, n3{5, {}};
+
+ rb_add(&n1.rb, &tree, node_less);
+ rb_add(&n2.rb, &tree, node_less);
+ rb_add(&n3.rb, &tree, node_less);
+
+ int search_key = 20;
+ struct rb_node *found = rb_find(&search_key, &tree, node_cmp);
+ ST_ASSERT(found == &n2.rb);
+
+ search_key = 99;
+ found = rb_find(&search_key, &tree, node_cmp);
+ ST_ASSERT(found == nullptr);
+
+ rb_erase(&n2.rb, &tree);
+ // Verify erased
+ search_key = 20;
+ found = rb_find(&search_key, &tree, node_cmp);
+ ST_ASSERT(found == nullptr);
+ }
+ st_pass();
+
+ st_begin("rbtree: ordered iteration");
+ {
+ struct rb_root tree = RB_ROOT;
+ TestNode nodes[5] = { {50, {}}, {10, {}}, {40, {}}, {20, {}}, {30, {}} };
+ for (int i=0; i<5; i++) {
+ rb_add(&nodes[i].rb, &tree, node_less);
+ }
+
+ int expected[] = {10, 20, 30, 40, 50};
+ int idx = 0;
+ struct rb_node *node = rb_first(&tree);
+ while (node) {
+ TestNode *n = rb_entry(node, TestNode, rb);
+ ST_ASSERT_EQ(n->key, expected[idx]);
+ idx++;
+ node = rb_next(node);
+ }
+ ST_ASSERT_EQ(idx, 5);
+ }
+ st_pass();
+
+ st_begin("rbtree: large scale insert/erase balance");
+ {
+ struct rb_root tree = RB_ROOT;
+ constexpr int N = 200;
+ static TestNode nodes[N]; // Static to avoid blowing up the BSP stack
+ for (int i=0; i<N; i++) {
+ nodes[i].key = (i * 17) % N; // Pseudo-random insertions
+ rb_add(&nodes[i].rb, &tree, node_less);
+ }
+
+ int c = 0;
+ for (struct rb_node *n = rb_first(&tree); n; n = rb_next(n)) c++;
+ ST_ASSERT_EQ(c, N);
+
+ for (int i=0; i<N; i+=2) {
+ rb_erase(&nodes[i].rb, &tree);
+ }
+ c = 0;
+ for (struct rb_node *n = rb_first(&tree); n; n = rb_next(n)) c++;
+ ST_ASSERT_EQ(c, N / 2);
+ }
+ st_pass();
+
+ st_begin("rbtree: empty tree operations");
+ {
+ struct rb_root tree = RB_ROOT;
+ ST_ASSERT(RB_EMPTY_ROOT(&tree));
+ ST_ASSERT(rb_first(&tree) == nullptr);
+ ST_ASSERT(rb_last(&tree) == nullptr);
+ }
+ st_pass();
+
+ st_begin("rbtree: rb_erase_init after effects");
+ {
+ struct rb_root tree = RB_ROOT;
+ TestNode n1{100, {}};
+ rb_add(&n1.rb, &tree, node_less);
+ ST_ASSERT(!RB_EMPTY_NODE(&n1.rb));
+
+ rb_erase_init(&n1.rb, &tree);
+ ST_ASSERT(RB_EMPTY_NODE(&n1.rb));
+ ST_ASSERT(RB_EMPTY_ROOT(&tree));
+ }
+ st_pass();
+}