ID

0dd5aed

date

2021-11-27 07:53:24+00:00

author

Alex Chan <alex@alexwlchan.net>

parent

0cae65c

message

test that it fails on an invalid count

changed files

1 file, 15 additions

• src/main.rs

Changed files

src/main.rs (5299) → src/main.rs (5867)

diff --git a/src/main.rs b/src/main.rs
index 593ae54..f9fc051 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -150,4 +150,19 @@ mod tests {
 
         assert_eq!(str::from_utf8(&output.stderr).unwrap(), "");
     }
+
+    #[test]
+    fn it_fails_if_you_pass_an_invalid_count() {
+        let mut cmd = Command::cargo_bin("dominant_colours").unwrap();
+        let output = cmd
+            .args(&["./src/tests/red.png", "--count=NaN"])
+            .unwrap_err()
+            .as_output()
+            .unwrap()
+            .to_owned();
+
+        assert_eq!(output.status.code().unwrap(), 1);
+        assert_eq!(str::from_utf8(&output.stdout).unwrap(), "");
+        assert_eq!(str::from_utf8(&output.stderr).unwrap(), "error: Invalid value: The argument 'NaN' isn't a valid value\n");
+    }
 }