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

UT: Rename RequestsGetArgument -> MockRequestsGetResponsePerServerArgument

This commit is contained in:
Shreya Malviya 2022-09-07 18:12:24 +05:30 committed by Mike Salvatore
parent 0239a1be6b
commit cd91b3e42a
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
monkey/tests/unit_tests/infection_monkey/test_control.py
View File

@ -19,7 +19,7 @@ class MockConnectionError:
raise requests.exceptions.ConnectionError
class RequestsGetArgument:
class MockRequestsGetResponsePerServerArgument:
def __init__(self, *args, **kwargs):
if SERVER_1 in args[0]:
MockConnectionError()
@ -55,7 +55,9 @@ def test_control_find_server__no_available_relays(monkeypatch, servers):
def test_control_find_server__control_message_sent_to_necessary_relays(monkeypatch, servers):
mock_connect = MagicMock()
mock_send = MagicMock()
monkeypatch.setattr("infection_monkey.control.requests.get", RequestsGetArgument)
monkeypatch.setattr(
"infection_monkey.control.requests.get", MockRequestsGetResponsePerServerArgument
)
monkeypatch.setattr("infection_monkey.control.socket.socket.connect", mock_connect)
monkeypatch.setattr("infection_monkey.control.socket.socket.send", mock_send)