cyberbuff
32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
from atomic_red_team.new_atomic import create_or_append_atomic
|
|
|
|
|
|
def test_create_or_append_atomic_creates_new_technique_file(tmp_path):
|
|
atomics_dir = tmp_path / "atomics"
|
|
|
|
output_path = create_or_append_atomic("t1001", atomics_dir=atomics_dir)
|
|
|
|
assert output_path == atomics_dir / "T1001" / "T1001.yaml"
|
|
assert "attack_technique: T1001" in output_path.read_text()
|
|
assert "atomic_tests:" in output_path.read_text()
|
|
|
|
|
|
def test_create_or_append_atomic_appends_atomic_test_to_existing_file(tmp_path):
|
|
atomics_dir = tmp_path / "atomics"
|
|
technique_dir = atomics_dir / "T1001"
|
|
technique_dir.mkdir(parents=True)
|
|
output_path = technique_dir / "T1001.yaml"
|
|
output_path.write_text(
|
|
"attack_technique: T1001\n"
|
|
"display_name: Existing Technique\n"
|
|
"atomic_tests:\n"
|
|
"- name: Existing test\n"
|
|
)
|
|
|
|
create_or_append_atomic("T1001", atomics_dir=atomics_dir)
|
|
|
|
output = output_path.read_text()
|
|
assert "- name: Existing test" in output
|
|
assert output.count("- name: TODO") == 1
|
|
assert output.count("atomic_tests:") == 1
|