Add: unit test for adding step to roadmap

1 files changed, 9 insertions, 0 deletions

diff --git a/src/lib.rs b/src/lib.rs
index aa1d1eb..5d050c8 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -165,4 +165,13 @@ mod tests {
         let roadmap = Roadmap::new();
         assert_eq!(roadmap.step_names().len(), 0);
     }
+
+    #[test]
+    fn add_step_to_roadmap() {
+        let mut roadmap = Roadmap::new();
+        let first = Step::new("first", "the first step");
+        roadmap.add_step(&first).unwrap();
+        assert_eq!(roadmap.step_names().len(), 1);
+        assert_eq!(roadmap.step_names(), vec!["first"]);
+    }
 }