Merge branch 'liw/andbut' into 'master'

Add: add support for "and" and "but" keywords

See merge request larswirzenius/subplot!3

1 files changed, 19 insertions, 5 deletions

diff --git a/src/steps.rs b/src/steps.rs
index 66ab453..ccf7694 100644
--- a/src/steps.rs
+++ b/src/steps.rs
@@ -37,13 +37,18 @@ impl ScenarioStep {
     }
 
     /// Construct a step from a line in a scenario.
-    pub fn new_from_str(text: &str) -> Result<ScenarioStep> {
+    ///
+    /// If the step uses the "and" or "but" keyword, use the default
+    /// step kind instead.
+    pub fn new_from_str(text: &str, default: StepKind) -> Result<ScenarioStep> {
         let mut words = text.split_whitespace();
 
         let kind = match words.next() {
             Some("given") => StepKind::Given,
             Some("when") => StepKind::When,
             Some("then") => StepKind::Then,
+            Some("and") => default,
+            Some("but") => default,
             _ => return Err(Error::UnknownStepKind),
         };
 
@@ -99,28 +104,37 @@ mod test {
 
     #[test]
     fn parses_given() {
-        let step = ScenarioStep::new_from_str("given I am Tomjon").unwrap();
+        let step = ScenarioStep::new_from_str("given I am Tomjon", StepKind::Then).unwrap();
         assert_eq!(step.kind(), StepKind::Given);
         assert_eq!(step.text(), "I am Tomjon");
     }
 
     #[test]
     fn parses_given_with_extra_spaces() {
-        let step = ScenarioStep::new_from_str("   given     I   am   Tomjon   ").unwrap();
+        let step =
+            ScenarioStep::new_from_str("   given     I   am   Tomjon   ", StepKind::When).unwrap();
         assert_eq!(step.kind(), StepKind::Given);
         assert_eq!(step.text(), "I am Tomjon");
     }
 
     #[test]
     fn parses_when() {
-        let step = ScenarioStep::new_from_str("when I declare myself king").unwrap();
+        let step =
+            ScenarioStep::new_from_str("when I declare myself king", StepKind::Given).unwrap();
         assert_eq!(step.kind(), StepKind::When);
         assert_eq!(step.text(), "I declare myself king");
     }
 
     #[test]
     fn parses_then() {
-        let step = ScenarioStep::new_from_str("then everyone accepts it").unwrap();
+        let step = ScenarioStep::new_from_str("then everyone accepts it", StepKind::Given).unwrap();
+        assert_eq!(step.kind(), StepKind::Then);
+        assert_eq!(step.text(), "everyone accepts it");
+    }
+
+    #[test]
+    fn parses_and() {
+        let step = ScenarioStep::new_from_str("and everyone accepts it", StepKind::Then).unwrap();
         assert_eq!(step.kind(), StepKind::Then);
         assert_eq!(step.text(), "everyone accepts it");
     }