Fix CLI tests

duckysmacky · Aug 21, 2024 · 048f16d · 048f16d
1 parent 710d5c5
commit 048f16d
Show file tree

Hide file tree

Showing 5 changed files with 14 additions and 17 deletions.
diff --git a/test-files/test/image.png b/test-files/test/image.png
diff --git a/test-files/test/word.txt b/test-files/test/word.txt
diff --git a/test-files/test/words.txt b/test-files/test/words.txt
diff --git a/tests/cli_tests.rs b/tests/cli_tests.rs
@@ -19,6 +19,13 @@ fn decrypt_command(path: &PathBuf) -> Output {
         .expect("Failed to execute the \"unbox\" command")
 }
 
+fn create_profile_command() -> Output {
+    Command::new("lockbox")
+        .args(&["-v", "-p", PASSWORD, "profile", "create", "test"])
+        .output()
+        .expect("Failed to execute the \"profile create\" command")
+}
+
 fn format_output(text: &str, output: Output) -> String {
     format!("{}\nSTDOUT: {:?}\nSTDERR: {:?}\nSTATUS CODE: {}\n", text, output.stdout, output.stderr, output.status)
 }
@@ -31,6 +38,9 @@ fn test_word_encryption() {
     let test_file = Path::new(TEST_DIR).join(file_name);
     let original_file = Path::new(ORIGINAL_DIR).join(file_name);
 
+    let output = create_profile_command();
+    assert!(output.status.success(), "{}", format_output("Profile creation failed", output));
+
     let output = encrypt_command(&test_file);
     assert!(output.status.success(), "{}", format_output("Encryption failed", output));
 
@@ -52,6 +62,9 @@ fn test_words_encryption() {
     let test_file = Path::new(TEST_DIR).join(file_name);
     let original_file = Path::new(ORIGINAL_DIR).join(file_name);
 
+    let output = create_profile_command();
+    assert!(output.status.success(), "{}", format_output("Profile creation failed", output));
+
     let output = encrypt_command(&test_file);
     assert!(output.status.success(), "{}", format_output("Encryption failed", output));
 

diff --git a/tests/common/mod.rs b/tests/common/mod.rs
@@ -1,7 +1,7 @@
 use std::{env, fs};
 use std::path::{Path, PathBuf};
 
-pub const PASSWORD: &str = "test123"; // just a password for testing
+pub const PASSWORD: &str = "test_password"; // just a password for testing
 #[allow(dead_code)]
 pub const ORIGINAL_DIR: &str = "test-files/original";
 #[allow(dead_code)]