p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

UT: Fix failing tests

This commit is contained in:
Kekoa Kaaikala 2022-09-29 16:26:27 +00:00
parent d811c6548c
commit 9a880123da
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
monkey/tests/unit_tests/monkey_island/cc/agent_event_handlers/test_handle_ping_scan_event.py
View File

@ -125,7 +125,7 @@ class error_machine_by_ip:
return machines return machines
def test_handle_ping_scan_event__upserts_machine( def test_handle_ping_scan_event__target_machine_not_exists(
handler: handle_ping_scan_event, handler: handle_ping_scan_event,
machine_repository: IMachineRepository, machine_repository: IMachineRepository,
): ):
@ -133,13 +133,13 @@ def test_handle_ping_scan_event__upserts_machine(
machine_repository.get_machines_by_ip = MagicMock(side_effect=UnknownRecordError) machine_repository.get_machines_by_ip = MagicMock(side_effect=UnknownRecordError)
handler(EVENT) handler(EVENT)
expected_machine = TARGET_MACHINE.copy() expected_machine = Machine(id=SEED_ID, network_interfaces=[IPv4Interface(EVENT.target)])
expected_machine.operating_system = OperatingSystem.LINUX expected_machine.operating_system = EVENT.os
machine_repository.upsert_machine.assert_called_with(expected_machine) machine_repository.upsert_machine.assert_called_with(expected_machine)
def test_handle_ping_scan_event__machine_already_exists( def test_handle_ping_scan_event__target_machine_already_exists(
handler: handle_ping_scan_event, handler: handle_ping_scan_event,
machine_repository: IMachineRepository, machine_repository: IMachineRepository,
): ):