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

UT: Simplify test logic in test_find_server()

This commit is contained in:
Shreya Malviya 2022-09-08 14:48:03 +05:30 committed by Mike Salvatore
parent 45d1cc78c1
commit fac179bbda
1 changed files with 24 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
monkey/tests/unit_tests/infection_monkey/network/relay/test_utils.py
View File

@ -14,14 +14,31 @@ servers = [SERVER_1, SERVER_2, SERVER_3, SERVER_4]
@pytest.mark.parametrize(
"expected_server,connection_error_servers,do_nothing_servers",
[(None, servers, []), (SERVER_2, [SERVER_1], [SERVER_2, SERVER_3, SERVER_4])],
"expected_server,server_response_pairs",
[
(
None,
[
(SERVER_1, {"exc": requests.exceptions.ConnectionError}),
(SERVER_2, {"exc": requests.exceptions.ConnectionError}),
(SERVER_3, {"exc": requests.exceptions.ConnectionError}),
(SERVER_4, {"exc": requests.exceptions.ConnectionError}),
],
),
(
SERVER_2,
[
(SERVER_1, {"exc": requests.exceptions.ConnectionError}),
(SERVER_2, {"text": ""}),
(SERVER_3, {"text": ""}),
(SERVER_4, {"text": ""}),
],
),
],
)
def test_find_server(expected_server, connection_error_servers, do_nothing_servers):
def test_find_server(expected_server, server_response_pairs):
with requests_mock.Mocker() as mock:
for server in connection_error_servers:
mock.get(f"https://{server}/api?action=is-up", exc=requests.exceptions.ConnectionError)
for server in do_nothing_servers:
mock.get(f"https://{server}/api?action=is-up", text="")
for server, response in server_response_pairs:
mock.get(f"https://{server}/api?action=is-up", **response)
assert find_server(servers) is expected_server