p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

UT: Simplify parametrize logic in test_find_server()

This commit is contained in:
Shreya Malviya 2022-09-08 14:53:01 +05:30 committed by Mike Salvatore
parent fac179bbda
commit e1759a7906
1 changed files with 3 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -16,23 +16,11 @@ servers = [SERVER_1, SERVER_2, SERVER_3, SERVER_4]
@pytest.mark.parametrize(
"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}),
],
),
(None, [(server, {"exc": requests.exceptions.ConnectionError}) for server in servers]),
(
SERVER_2,
[
(SERVER_1, {"exc": requests.exceptions.ConnectionError}),
(SERVER_2, {"text": ""}),
(SERVER_3, {"text": ""}),
(SERVER_4, {"text": ""}),
],
[(SERVER_1, {"exc": requests.exceptions.ConnectionError})]
+ [(server, {"text": ""}) for server in servers[1:]],
),
],
)