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

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
monkey/tests/unit_tests/infection_monkey/telemetry/messengers/test_exploit_intercepting_telemetry_messenger.py
View File

@ -31,7 +31,7 @@ def test_generic_telemetry(TestTelem):
assert mock_telemetry_messenger.send_telemetry.called assert mock_telemetry_messenger.send_telemetry.called
assert not mock_tunnel.set_wait_for_exploited_machines.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(): 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_telemetry_messenger.send_telemetry.called
assert mock_tunnel.set_wait_for_exploited_machines.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(): def test_propagation_failed_exploit_telemetry():
@ -65,4 +65,4 @@ def test_propagation_failed_exploit_telemetry():
assert mock_telemetry_messenger.send_telemetry.called assert mock_telemetry_messenger.send_telemetry.called
assert not mock_tunnel.set_wait_for_exploited_machines.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