tests: more test cases for CLI.

VirusTotal · Jan 24, 2025 · b4e8ad9 · b4e8ad9
1 parent 37f1061
commit b4e8ad9
Show file tree

Hide file tree

Showing 3 changed files with 87 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/cli/Cargo.toml b/cli/Cargo.toml
@@ -75,4 +75,5 @@ wild = "2.2.1"
 
 [dev-dependencies]
 assert_cmd = "2.0.16"
+assert_fs = "1.1.2"
 predicates = { workspace = true }
diff --git a/cli/src/tests/mod.rs b/cli/src/tests/mod.rs
@@ -1,4 +1,6 @@
 use assert_cmd::Command;
+use assert_fs::prelude::*;
+use assert_fs::TempDir;
 use predicates::prelude::*;
 
 #[test]
@@ -273,6 +275,29 @@ fn cli_scan_compiled_rules() {
         .failure()
         .code(1)
         .stderr("error: can\'t use namespace with \'--compiled-rules\'\n");
+
+    let temp_dir = TempDir::new().unwrap();
+    let input_file = temp_dir.child("rule.yar");
+
+    input_file.write_str("rule test { condition: true }").unwrap();
+
+    Command::cargo_bin("yr")
+        .unwrap()
+        .arg("compile")
+        .arg("-o")
+        .arg(input_file.with_extension("yarc"))
+        .arg(input_file.path())
+        .assert()
+        .success();
+
+    Command::cargo_bin("yr")
+        .unwrap()
+        .arg("scan")
+        .arg("--compiled-rules")
+        .arg(input_file.with_extension("yarc"))
+        .arg("src/tests/testdata/dummy.file")
+        .assert()
+        .success();
 }
 
 #[test]
@@ -313,3 +338,48 @@ fn cli_issue_280() {
         .assert()
         .success();
 }
+
+#[cfg(feature = "debug-cmd")]
+#[test]
+fn cli_debug_ast() {
+    Command::cargo_bin("yr")
+        .unwrap()
+        .arg("debug")
+        .arg("ast")
+        .arg("src/tests/testdata/foo.yar")
+        .assert()
+        .success();
+}
+
+#[cfg(feature = "debug-cmd")]
+#[test]
+fn cli_debug_cst() {
+    Command::cargo_bin("yr")
+        .unwrap()
+        .arg("debug")
+        .arg("cst")
+        .arg("src/tests/testdata/foo.yar")
+        .assert()
+        .success();
+}
+
+#[cfg(feature = "debug-cmd")]
+#[test]
+fn cli_debug_wasm() {
+    let temp_dir = TempDir::new().unwrap();
+    let input_file = temp_dir.child("rule.yar");
+
+    input_file.write_str("rule test { condition: true }").unwrap();
+
+    Command::cargo_bin("yr")
+        .unwrap()
+        .arg("debug")
+        .arg("wasm")
+        .arg(input_file.path())
+        .assert()
+        .success();
+
+    if !input_file.with_extension("wasm").exists() {
+        panic!("`yr debug wasm` didn't create .wasm file")
+    }
+}