102 lines
3.3 KiB
Python
102 lines
3.3 KiB
Python
from pathlib import Path
|
|
import toml
|
|
from click.testing import CliRunner
|
|
from ctf.utils import load_config, active_categories
|
|
from ctf.forensics import forensics_group, inspect
|
|
|
|
# Define the persistent test environment path
|
|
TEST_ENV = Path("tests/env")
|
|
|
|
def test_load_config_static():
|
|
"""
|
|
Verifies load_config using the persistent test file.
|
|
"""
|
|
config_file = TEST_ENV / "config.toml"
|
|
test_data = {
|
|
"Competition": {"name": "PersistentComp"},
|
|
"Enviroment": {"data_dir": str(TEST_ENV.absolute())}
|
|
}
|
|
config_file.parent.mkdir(parents=True, exist_ok=True)
|
|
with open(config_file, "w") as f:
|
|
toml.dump(test_data, f)
|
|
|
|
loaded_data = load_config(str(config_file))
|
|
assert loaded_data["Competition"]["name"] == "PersistentComp"
|
|
|
|
def test_active_categories_static():
|
|
"""
|
|
Verifies active_categories using the pre-created 'comp1' folder.
|
|
"""
|
|
comp_dir = TEST_ENV / "comp1"
|
|
comp_dir.mkdir(parents=True, exist_ok=True)
|
|
(comp_dir / "web").mkdir(exist_ok=True)
|
|
(comp_dir / "pwn").mkdir(exist_ok=True)
|
|
|
|
cats = active_categories(comp_dir)
|
|
cat_names = [c.name for c in cats]
|
|
|
|
assert "web" in cat_names
|
|
assert "pwn" in cat_names
|
|
|
|
# --- Direct Function Callback Tests ---
|
|
|
|
def test_inspect_callback_success():
|
|
"""
|
|
Verifies that calling the inspect command's callback directly returns
|
|
the correct FileMetadata dataclass with expected values.
|
|
"""
|
|
test_file = TEST_ENV / "callback_test.txt"
|
|
with open(test_file, "wb") as f:
|
|
f.write(b"Hello World")
|
|
|
|
# Using inspect.callback to invoke the underlying undecorated function
|
|
meta = inspect.callback(str(test_file))
|
|
|
|
assert meta.filename == "callback_test.txt"
|
|
assert meta.size == 11
|
|
assert meta.magic == "48656C6C" # Hex for "Hell"
|
|
assert meta.extension == ".txt"
|
|
|
|
def test_inspect_callback_empty_file():
|
|
"""
|
|
Tests the boundary case of an empty (0-byte) file.
|
|
Verifies that the parser handles it gracefully without raising exceptions.
|
|
"""
|
|
test_file = TEST_ENV / "empty_test.txt"
|
|
with open(test_file, "wb") as f:
|
|
pass
|
|
|
|
meta = inspect.callback(str(test_file))
|
|
|
|
assert meta.filename == "empty_test.txt"
|
|
assert meta.size == 0
|
|
assert meta.magic == "" # Empty string as no bytes could be read
|
|
assert meta.extension == ".txt"
|
|
|
|
# --- CLI Integration Tests ---
|
|
|
|
def test_forensics_inspect_cli_success():
|
|
"""
|
|
Verifies that 'forensics inspect' successfully runs via the CLI and prints
|
|
the metadata in a formatted table.
|
|
"""
|
|
runner = CliRunner()
|
|
test_file = TEST_ENV / "test_file.png"
|
|
with open(test_file, "wb") as f:
|
|
f.write(b"\x89PNG\r\n\x1a\n")
|
|
|
|
result = runner.invoke(forensics_group, ["inspect", str(test_file)])
|
|
assert result.exit_code == 0
|
|
assert "Metadata: test_file.png" in result.output
|
|
assert "Size" in result.output
|
|
assert "89504E47" in result.output # Hex representation of PNG signature
|
|
|
|
def test_forensics_inspect_cli_missing_file():
|
|
"""
|
|
Verifies that the CLI fails gracefully when a non-existent file path is specified.
|
|
"""
|
|
runner = CliRunner()
|
|
result = runner.invoke(forensics_group, ["inspect", "non_existent_file.png"])
|
|
assert result.exit_code != 0
|
|
assert "does not exist" in result.output
|