Updated tests to test for newlines in Commands.

hw9/task1/tests/task1.rs

@@ -45,4 +45,18 @@ mod tests {
         assert_eq!(parse("RETRIEVE Hello\n"), Ok(Command::Retrieve))
     }
 
+    #[test]
+    fn stage_with_newline_returns_correct_command() {
+        assert_eq!(parse("STAGE 123\n456"), Ok(Command::Stage("123".to_string())))
+    }
+
+    #[test]
+    fn control_with_newline_returns_correct_command() {
+        assert_eq!(parse("Control 123\n456"), Ok(Command::Control("123".to_string())))
+    }
+
+    #[test]
+    fn retrieve_with_newline_returns_correct_command() {
+        assert_eq!(parse("RETRIEVE 123\n456"), Ok(Command::Retrieve))
+    }
 }