From ff1c32bb33f45e4441801d9861c481af2ffeb941 Mon Sep 17 00:00:00 2001
From: Akos Horvath <koma@koma.sh>
Date: Fri, 29 Dec 2023 15:15:49 +0100
Subject: [PATCH] add test for wm_treenode_all_keyroots

---
 src/tests.c | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/src/tests.c b/src/tests.c
index 7c2c4af..601cf5c 100644
--- a/src/tests.c
+++ b/src/tests.c
@@ -196,6 +196,24 @@ static void test_wm_treenode_all_lmds(void **state)
     wm_ptrarray_free(&ret);
 }
 
+static void test_wm_treenode_all_keyroots(void **state)
+{
+    TreeNode *node1 = (TreeNode*)*state;
+
+    PtrArray keyroots = wm_treenode_all_keyroots(node1);
+    TreeNode **keyroot_nodes = (TreeNode**)keyroots.ptrs;
+
+    assert_int_equal(keyroots.size, 5);
+
+    assert_int_equal(keyroot_nodes[0]->id, 5);
+    assert_int_equal(keyroot_nodes[1]->id, 6);
+    assert_int_equal(keyroot_nodes[2]->id, 8);
+    assert_int_equal(keyroot_nodes[3]->id, 3);
+    assert_int_equal(keyroot_nodes[4]->id, 1);
+
+    wm_ptrarray_free(&keyroots);
+}
+
 static int test2_setup(void **state)
 {
     TreeNode *node1 = wm_treenode_new(NODE_VERTICAL, NULL);
@@ -299,6 +317,7 @@ int main(void)
         cmocka_unit_test(test_wm_treenode_lmd),
         cmocka_unit_test(test_wm_is_treenode_keyroot),
         cmocka_unit_test(test_wm_treenode_all_lmds),
+        cmocka_unit_test(test_wm_treenode_all_keyroots),
     };
 
     cmocka_run_group_tests(test_group1, NULL, NULL);