UT: Fix bug in test mock

This commit is contained in:
Mike Salvatore 2022-09-30 13:16:04 -04:00
parent 2811009019
commit 067a143f2c
1 changed files with 1 additions and 1 deletions

View File

@ -267,7 +267,7 @@ def test_machine_not_upserted_if_existing_machine_has_os(
):
machine_with_os = TARGET_MACHINE
machine_with_os.operating_system = OperatingSystem.WINDOWS
machine_repository.get_machine_by_ip = MagicMock(return_value=machine_with_os)
machine_repository.get_machines_by_ip = MagicMock(return_value=[machine_with_os])
scan_event_handler.handle_ping_scan_event(PING_SCAN_EVENT)