First public commit

tests/run_log.rs

@@ -0,0 +1,80 @@
+use senbei::job;
+use std::path::Path;
+
+fn list_logs(dir: &Path) -> Vec<std::path::PathBuf> {
+    std::fs::read_dir(dir)
+        .into_iter()
+        .flatten()
+        .filter_map(|e| e.ok())
+        .map(|e| e.path())
+        .filter(|p| {
+            p.file_name()
+                .and_then(|n| n.to_str())
+                .map(|n| n.starts_with("senbei-") && n.ends_with(".log"))
+                .unwrap_or(false)
+        })
+        .collect()
+}
+
+#[test]
+fn run_file_no_log_creates_no_logfile() {
+    let td = tempfile::tempdir().unwrap();
+    let input = td.path().join("not_crackproof.bin");
+    std::fs::write(&input, b"not a pe").unwrap();
+    let out = td.path().join("out");
+    let s = job::run_file_v(&input, Some(&out), 2, false, true).unwrap();
+    assert_eq!(s.errors, 1);
+    // With no_log, no senbei-*.log under out (even if the dir was created).
+    assert!(list_logs(&out).is_empty());
+}
+
+#[test]
+fn run_file_writes_log_under_out_with_header_footer() {
+    let td = tempfile::tempdir().unwrap();
+    let input = td.path().join("not_crackproof.bin");
+    std::fs::write(&input, b"not a pe").unwrap();
+    let out = td.path().join("out");
+    let s = job::run_file_v(&input, Some(&out), 2, false, false).unwrap();
+    assert_eq!(s.errors, 1);
+    let logs = list_logs(&out);
+    assert_eq!(logs.len(), 1, "expected one log under out, got {logs:?}");
+    let text = std::fs::read_to_string(&logs[0]).unwrap();
+    assert!(text.contains("Senbei "), "header version: {text}");
+    assert!(text.contains("started "), "{text}");
+    assert!(text.contains("input "), "{text}");
+    assert!(text.contains("out "), "{text}");
+    assert!(text.contains("ERR "), "{text}");
+    assert!(text.contains("done in "), "{text}");
+    assert!(text.contains("summary:"), "{text}");
+}
+
+#[test]
+fn run_file_default_out_root_is_parent_unpack() {
+    let td = tempfile::tempdir().unwrap();
+    let input = td.path().join("not_crackproof.bin");
+    std::fs::write(&input, b"not a pe").unwrap();
+    let _ = job::run_file_v(&input, None, 2, false, false).unwrap();
+    let unpack = td.path().join("unpack");
+    assert!(unpack.is_dir());
+    assert_eq!(list_logs(&unpack).len(), 1);
+    // log must NOT be next to input's parent root without unpack
+    assert!(list_logs(td.path()).is_empty());
+}
+
+#[test]
+fn run_folder_log_lives_under_out_not_root() {
+    let td = tempfile::tempdir().unwrap();
+    // empty tree: 0 candidates still creates log under unpack
+    let s = job::run_folder_v(td.path(), None, 2, false, false).unwrap();
+    assert_eq!(s.unpacked, 0);
+    let unpack = td.path().join("unpack");
+    assert!(unpack.is_dir());
+    assert_eq!(list_logs(&unpack).len(), 1);
+    assert!(
+        list_logs(td.path()).is_empty(),
+        "log must not sit on input root"
+    );
+    let text = std::fs::read_to_string(&list_logs(&unpack)[0]).unwrap();
+    assert!(text.contains("done in "));
+    assert!(text.contains("summary:"));
+}