diff --git a/monkey/tests/unit_tests/infection_monkey/telemetry/messengers/test_exploit_intercepting_telemetry_messenger.py b/monkey/tests/unit_tests/infection_monkey/telemetry/messengers/test_exploit_intercepting_telemetry_messenger.py index 7e9e1e7fd..0758a5e4d 100644 --- a/monkey/tests/unit_tests/infection_monkey/telemetry/messengers/test_exploit_intercepting_telemetry_messenger.py +++ b/monkey/tests/unit_tests/infection_monkey/telemetry/messengers/test_exploit_intercepting_telemetry_messenger.py @@ -31,7 +31,7 @@ def test_generic_telemetry(TestTelem): assert mock_telemetry_messenger.send_telemetry.called assert not mock_tunnel.set_wait_for_exploited_machines.called - assert not mock_relay.on_potential_new_user.called + assert not mock_relay.add_potential_user.called def test_propagation_successful_exploit_telemetry(): @@ -48,7 +48,7 @@ def test_propagation_successful_exploit_telemetry(): assert mock_telemetry_messenger.send_telemetry.called assert mock_tunnel.set_wait_for_exploited_machines.called - assert mock_relay.on_potential_new_user.called + assert mock_relay.add_potential_user.called def test_propagation_failed_exploit_telemetry(): @@ -65,4 +65,4 @@ def test_propagation_failed_exploit_telemetry(): assert mock_telemetry_messenger.send_telemetry.called assert not mock_tunnel.set_wait_for_exploited_machines.called - assert not mock_relay.on_potential_new_user.called + assert not mock_relay.add_potential_user.called