mirror of
https://github.com/maxDcb/C2TeamServer
synced 2026-06-06 16:14:27 +00:00
maxdcb
23 lines
855 B
Python
23 lines
855 B
Python
from __future__ import annotations
|
|
|
|
from C2Client.assistant_agent.tools.loader import load_tool_specs
|
|
|
|
|
|
def test_tool_loader_loads_unique_json_tool_specs():
|
|
specs = load_tool_specs()
|
|
names = [spec.name for spec in specs]
|
|
|
|
assert "assemblyExec" in names
|
|
assert "dcomExec" in names
|
|
assert "ls" in names
|
|
assert "screenShot" in names
|
|
assert "run" in names
|
|
assert "winRm" in names
|
|
assert len(names) == len(set(names))
|
|
assert all(spec.description for spec in specs)
|
|
assert all(spec.command_template for spec in specs)
|
|
assert all(spec.source_path.suffix == ".json" for spec in specs)
|
|
assert all(spec.parameters["type"] == "object" for spec in specs)
|
|
assert all("beacon_hash" in spec.parameters["required"] for spec in specs)
|
|
assert all("listener_hash" in spec.parameters["required"] for spec in specs)
|