test(deep_causality_ast): increased test coverage.

Signed-off-by: Marvin Hansen <[email protected]>

Author: marvin-hansen

deep_causality_ast/tests/BUILD.bazel

@@ -1,9 +1,9 @@
 load("@rules_rust//rust:defs.bzl", "rust_test_suite")
 
 rust_test_suite(
-    name = "tests",
+    name = "const_tree",
     srcs = glob([
-        "*_tests.rs",
+        "const_tree/*_tests.rs",
     ]),
     tags = [
         "unit-test",

deep_causality_ast/tests/const_tree_tests.rs renamed to deep_causality_ast/tests/const_tree/const_tree_tests.rs

@@ -283,6 +283,21 @@ fn test_into_map() {
     assert_eq!(*tree_clone.children()[0].value(), 20);
 }
 
+#[test]
+fn test_consuming_iterator_shared_arc() {
+    let tree = ConstTree::with_children(1, vec![ConstTree::new(2), ConstTree::new(3)]);
+    let tree_clone1 = tree.clone();
+    let tree_clone2 = tree.clone(); // Ensure multiple references
+
+    // Call into_iter on one of the clones. This should trigger the Err branch.
+    let values: Vec<_> = tree_clone1.into_iter().collect();
+    assert_eq!(values, vec![1, 2, 3]);
+
+    // Verify that the other clone is still valid.
+    assert_eq!(*tree_clone2.value(), 1);
+    assert_eq!(tree_clone2.children().len(), 2);
+}
+
 #[test]
 fn test_join() {
     // Create a tree of trees: ConstTree<ConstTree<i32>>

deep_causality_ast/tests/const_tree/mod.rs

@@ -0,0 +1,7 @@
+/*
+ * SPDX-License-Identifier: MIT
+ * Copyright (c) "2025" . The DeepCausality Authors and Contributors. All Rights Reserved.
+ */
+
+#[cfg(test)]
+mod const_tree_tests;

deep_causality_ast/tests/mod.rs

@@ -2,5 +2,5 @@
  * SPDX-License-Identifier: MIT
  * Copyright (c) "2025" . The DeepCausality Authors and Contributors. All Rights Reserved.
  */
-#[cfg(test)]
-mod const_tree_tests;
+
+mod const_tree;