Files
ghidra-cli/tests/stats_tests.rs
T
Alexander KiselevandClaude Opus 4.6 ae2d882f26 fix: add --project/--program to BatchArgs and fix arg order in tests
BatchArgs was missing --project and --program fields, so batch commands
couldn't specify which project to use. Also fixed argument ordering in
all test files to place --project/--program after the subcommand name,
matching clap's per-subcommand argument parsing.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-02-05 13:19:16 -08:00

99 lines
2.5 KiB
Rust

//! Tests for stats command.
use assert_cmd::Command;
use predicates::prelude::*;
use serial_test::serial;
#[macro_use]
mod common;
use common::{ensure_test_project, DaemonTestHarness};
const TEST_PROJECT: &str = "stats-test";
const TEST_PROGRAM: &str = "sample_binary";
#[test]
#[serial]
fn test_stats_normal() {
require_ghidra!();
ensure_test_project(TEST_PROJECT, TEST_PROGRAM);
let harness =
DaemonTestHarness::new(TEST_PROJECT, TEST_PROGRAM).expect("Failed to start daemon");
Command::cargo_bin("ghidra")
.unwrap()
.arg("stats")
.arg("--project")
.arg(TEST_PROJECT)
.arg("--program")
.arg(TEST_PROGRAM)
.assert()
.success()
.stdout(predicate::str::contains("stats"))
.stdout(predicate::str::contains("functions"))
.stdout(predicate::str::contains("symbols"));
drop(harness);
}
#[test]
#[serial]
fn test_stats_has_all_fields() {
require_ghidra!();
ensure_test_project(TEST_PROJECT, TEST_PROGRAM);
let harness =
DaemonTestHarness::new(TEST_PROJECT, TEST_PROGRAM).expect("Failed to start daemon");
Command::cargo_bin("ghidra")
.unwrap()
.arg("stats")
.arg("--project")
.arg(TEST_PROJECT)
.arg("--program")
.arg(TEST_PROGRAM)
.assert()
.success()
.stdout(predicate::str::contains("functions"))
.stdout(predicate::str::contains("symbols"))
.stdout(predicate::str::contains("strings"))
.stdout(predicate::str::contains("imports"))
.stdout(predicate::str::contains("exports"))
.stdout(predicate::str::contains("memory_size"))
.stdout(predicate::str::contains("sections"))
.stdout(predicate::str::contains("data_types"));
drop(harness);
}
#[test]
#[serial]
fn test_stats_json_format() {
require_ghidra!();
ensure_test_project(TEST_PROJECT, TEST_PROGRAM);
let harness =
DaemonTestHarness::new(TEST_PROJECT, TEST_PROGRAM).expect("Failed to start daemon");
let output = Command::cargo_bin("ghidra")
.unwrap()
.arg("stats")
.arg("--project")
.arg(TEST_PROJECT)
.arg("--program")
.arg(TEST_PROGRAM)
.assert()
.success()
.get_output()
.stdout
.clone();
let output_str = String::from_utf8_lossy(&output);
assert!(
serde_json::from_str::<serde_json::Value>(&output_str).is_ok(),
"Output should be valid JSON"
);
drop(harness);
}