From 159d3fce93024ffa9b97605f768e9a116a040cc1 Mon Sep 17 00:00:00 2001
From: Fintan Halpenny <fintan.halpenny@gmail.com>
Date: Mon, 11 Nov 2024 13:47:59 +0000
Subject: [PATCH] dag: test contains

Test coverage for the `Dag::contains` method were not covered. Add tests for the
method to cover an empty and filled `Dag`.
---
 radicle-dag/src/lib.rs | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/radicle-dag/src/lib.rs b/radicle-dag/src/lib.rs
index 923e7274..10d2c1fa 100644
--- a/radicle-dag/src/lib.rs
+++ b/radicle-dag/src/lib.rs
@@ -1020,4 +1020,24 @@ mod tests {
         );
         assert_eq!(order, vec!["R", "A3", "A2", "B3", "A1", "B2", "B1", "C1"]);
     }
+
+    #[test]
+    fn test_contains() {
+        let mut dag = Dag::<u8, ()>::new();
+
+        assert!(!dag.contains(&0));
+
+        dag.node(0, ());
+        dag.node(1, ());
+        dag.dependency(0, 1);
+        dag.node(2, ());
+        dag.dependency(2, 1);
+        dag.dependency(2, 0);
+        dag.node(3, ());
+
+        for i in 0..4 {
+            assert!(dag.contains(&i));
+        }
+        assert!(!dag.contains(&4));
+    }
 }