Added tests for task1

hw9/task1/src/lib.rs

@@ -22,7 +22,7 @@ pub fn parse(message: &str) -> Command {
     }
 }
 
-#[derive(Debug)]
+#[derive(Debug, PartialEq)]
 pub enum Command {
     Stage(String),
     Control(String),

hw9/task1/tests/task1.rs

@@ -0,0 +1,40 @@
+#[cfg(test)]
+
+mod tests {
+
+    extern crate task1;
+
+    use self::task1::Command;
+    use self::task1::parse;
+
+    #[test]
+    fn empty_returns_correct_command() {
+        assert_eq!(parse("\n"), Command::EmptyError)
+    }
+
+    #[test]
+    fn not_known_command_returns_correct_command() {
+        assert_eq!(parse("Hello"), Command::NotFoundError)
+    }
+
+    #[test]
+    fn stage_returns_correct_command() {
+        assert_eq!(parse("STAGE Hello"), Command::Stage("Hello".to_string()))
+    }
+
+    #[test]
+    fn control_returns_correct_command() {
+        assert_eq!(parse("CONTROL Hello"), Command::Control("Hello".to_string()))
+    }
+
+    #[test]
+    fn retrieve_returns_correct_command() {
+        assert_eq!(parse("RETRIEVE"), Command::Retrieve)
+    }
+
+    #[test]
+    fn retrieve_with_arguments_returns_correct_command() {
+        assert_eq!(parse("RETRIEVE Hello"), Command::Retrieve)
+    }
+
+}