Adds support for raw output to b3sum

1 files changed, 21 insertions, 1 deletions

diff --git a/b3sum/tests/test.rs b/b3sum/tests/test.rs
index 92e1c4c..2c55d2b 100644
--- a/b3sum/tests/test.rs
+++ b/b3sum/tests/test.rs
@@ -11,7 +11,19 @@ pub fn b3sum_exe() -> PathBuf {
 fn test_hash_one() {
     let expected = blake3::hash(b"foo").to_hex();
     let output = cmd!(b3sum_exe()).stdin_bytes("foo").read().unwrap();
-    assert_eq!(&*expected, &*output);
+    assert_eq!(&*expected, output);
+}
+
+#[test]
+fn test_hash_one_raw() {
+    let expected = blake3::hash(b"foo").as_bytes().to_owned();
+    let mut stdout = Vec::new();
+    let mut output_reader = cmd!(b3sum_exe(), "--raw")
+        .stdin_bytes("foo")
+        .reader()
+        .unwrap();
+    output_reader.read_to_end(&mut stdout).unwrap();
+    assert_eq!(expected, stdout.as_slice());
 }
 
 #[test]
@@ -93,3 +105,11 @@ fn test_length_without_value_is_an_error() {
         .run();
     assert!(result.is_err());
 }
+
+#[test]
+fn test_raw_with_multi_files_is_an_error() {
+    let result = cmd!(b3sum_exe(), "--raw", "file1", "file2")
+        .stderr_capture()
+        .run();
+    assert!(result.is_err());
+}