test: add tests for print function with --format

Author: ankddev

src/commands.rs

@@ -243,6 +243,15 @@ mod tests {
         unsafe { env::remove_var("TEST_PRINT_RUN") };
     }
 
+    #[test]
+    fn test_run_command_print_with_format() {
+        unsafe { env::set_var("TEST_PRINT_RUN", "test_value") };
+        with_captured_output(|| {
+            run_command(&Commands::Print(PrintArgs { format: Some("{name} = {value}".to_owned()) }));
+        });
+        unsafe { env::remove_var("TEST_PRINT_RUN") };
+    }
+
     #[test]
     fn test_run_command_delete() {
         unsafe { env::set_var("TEST_DELETE_RUN", "test_value") };

src/main.rs

@@ -66,6 +66,12 @@ mod tests {
         assert_eq!(args.command, Commands::Print(PrintArgs { format: None }));
     }
 
+    #[test]
+    fn test_print_command_with_format() {
+        let args = Cli::parse_from(["envfetch", "print", "--format", "{name}: \"{value}\""]);
+        assert_eq!(args.command, Commands::Print(PrintArgs { format: Some("{name}: \"{value}\"".to_owned()) }));
+    }
+
     #[test]
     fn test_set_command_simple() {
         let args = Cli::parse_from(["envfetch", "set", "VAR", "VALUE", "--", "npm run"]);

tests/cli.rs

@@ -101,6 +101,18 @@ fn print_success() -> Result<(), Box<dyn std::error::Error>> {
     Ok(())
 }
 
+#[test]
+/// Test for print command
+fn print_with_format_success() -> Result<(), Box<dyn std::error::Error>> {
+    let mut cmd = Command::cargo_bin("envfetch")?;
+    unsafe { env::set_var("PRINT_TEST", "Print") };
+    cmd.arg("print").arg("--format").arg("{name}: {value}")
+        .assert()
+        .success()
+        .stdout(predicate::str::contains("PRINT_TEST: Print"));
+    Ok(())
+}
+
 #[test]
 /// Test for delete command if specified process is successful
 fn delete_command_success() -> Result<(), Box<dyn std::error::Error>> {