p17860254
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

UT: Implement test_handle_scan_data__upserts_node

This commit is contained in:
Kekoa Kaaikala 2022-09-27 20:53:56 +00:00
parent ab32daa0e0
commit ce8219aa6d
1 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
monkey/tests/unit_tests/monkey_island/cc/agent_event_handlers/test_handle_scan_data.py
View File

@ -81,13 +81,24 @@ def test_handle_scan_data__upserts_machine(
expected_machine.operating_system = OperatingSystem.LINUX
assert machine_repository.upsert_machine.called_with(expected_machine)
assert node_repository.upsert_communication.called_with(
MACHINE.id, expected_machine.id, CommunicationType.SCANNED
def test_handle_scan_data__upserts_node(
handler: handle_scan_data,
machine_repository: IMachineRepository,
node_repository: INodeRepository,
):
event = PingScanEvent(
source=AGENT_ID,
target=IPv4Address("10.10.10.1"),
scan_data=PingScanData(True, OperatingSystem.LINUX),
)
machine_repository.get_machine_by_id = MagicMock(return_value=STORED_MACHINE)
handler(event)
def test_handle_scan_data__upserts_node(handler: handle_scan_data):
pass
assert node_repository.upsert_communication.called_with(
MACHINE.id, STORED_MACHINE.id, CommunicationType.SCANNED
)
def test_handle_scan_data__node_not_upserted_if_no_matching_agent(handler: handle_scan_data):