UT: Fix ExploitingInterceptingTelemetryMessenger tests

This commit is contained in:
Ilija Lazoroski 2022-09-01 13:58:14 +02:00
parent eee06467f6
commit c9b7f924a3
1 changed files with 3 additions and 3 deletions

View File

@ -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