Change: implement get_step

1 files changed, 16 insertions, 1 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 5d050c8..5549694 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -102,7 +102,12 @@ impl Roadmap {
     }
 
     /// Get a step, given its name.
-    pub fn get_step<'a>(&'a self, _name: &str) -> Option<&'a mut Step> {
+    pub fn get_step<'a>(&'a mut self, name: &str) -> Option<&'a mut Step> {
+        for step in self.steps.iter_mut() {
+            if step.name() == name {
+                return Some(step);
+            }
+        }
         None
     }
 
@@ -174,4 +179,14 @@ mod tests {
         assert_eq!(roadmap.step_names().len(), 1);
         assert_eq!(roadmap.step_names(), vec!["first"]);
     }
+
+    #[test]
+    fn get_step_from_roadmap() {
+        let mut roadmap = Roadmap::new();
+        let first = Step::new("first", "the first step");
+        roadmap.add_step(&first).unwrap();
+        let gotit = roadmap.get_step("first").unwrap();
+        assert_eq!(gotit.name(), "first");
+        assert_eq!(gotit.label(), "the first step");
+    }
 }