misc: Improve tree leaves test

Col-E · Sep 14, 2024 · 2c1e1c6 · 2c1e1c6
1 parent 1ccfb2d
commit 2c1e1c6
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/test/java/software/coley/collections/TreeTest.java b/src/test/java/software/coley/collections/TreeTest.java
@@ -7,7 +7,7 @@
 import java.util.List;
 import java.util.Set;
 
-import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.*;
 
 /**
  * Tests for {@link Tree}
@@ -146,7 +146,11 @@ public void test_tree_gets_all_leaves() {
 		tree.putTree("key4", "value4");
 		tree.putTree("key5", "value5");
 		// There are 6 leaves, as key1 has two children
-		Set<Tree<String, String>> foo = tree.getAllLeaves();
-		assertEquals(6, foo.size());
+		Set<Tree<String, String>> leaves = tree.getAllLeaves();
+		assertEquals(6, leaves.size());
+		// Double check parent/child relation
+		Tree<String, String> key1a = sub1.get("key1-a");
+		assertSame(sub1, key1a.getParent());
+		leaves.forEach(t -> assertNotNull(t.getParent()));
 	}
 }