Fix: unit test to match current behaviour

1 files changed, 5 insertions, 3 deletions

diff --git a/src/lib.rs b/src/lib.rs
index f345882..4403633 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -306,16 +306,18 @@ mod tests {
     #[test]
     fn simple_dot() {
         let mut roadmap = Roadmap::new();
-        let first = Step::new("first", "");
+        let mut first = Step::new("first", "");
+        first.set_status("ready");
         let mut second = Step::new("second", "");
         second.add_dependency("first");
+        second.set_status("goal");
         roadmap.add_step(&first).unwrap();
         roadmap.add_step(&second).unwrap();
         assert_eq!(
             roadmap.as_dot(999).unwrap(),
             "digraph \"roadmap\" {
-first [label=\"\" fillcolor=\"#ffffff\" shape=\"ellipse\"];
-second [label=\"\" fillcolor=\"#00eeeeee\" shape=\"diamond\"];
+first [label=\"\" style=filled fillcolor=\"#ffffff\" shape=\"ellipse\"];
+second [label=\"\" style=filled fillcolor=\"#00eeee\" shape=\"diamond\"];
 first -> second;
 }
 "