p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

UT: Parametrize errors in tests

This commit is contained in:
Kekoa Kaaikala 2022-09-19 20:54:26 +00:00
parent 51ce974c43
commit 3e96a30a09
1 changed files with 8 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
monkey/tests/unit_tests/infection_monkey/master/test_control_channel.py
View File

@ -38,19 +38,11 @@ def test_control_channel__register_agent(control_channel, island_api_client):
assert island_api_client.register_agent.called_once() assert island_api_client.register_agent.called_once()
def test_control_channel__register_agent_raises_on_connection_error( @pytest.mark.parametrize("api_error", [IslandAPIConnectionError, IslandAPITimeoutError])
control_channel, island_api_client def test_control_channel__register_agent_raises_error(
control_channel, island_api_client, api_error
): ):
island_api_client.register_agent.side_effect = IslandAPIConnectionError() island_api_client.register_agent.side_effect = api_error()
with pytest.raises(IslandCommunicationError):
control_channel.register_agent()
def test_control_channel__register_agent_raises_on_timeout_error(
control_channel, island_api_client
):
island_api_client.register_agent.side_effect = IslandAPITimeoutError()
with pytest.raises(IslandCommunicationError): with pytest.raises(IslandCommunicationError):
control_channel.register_agent() control_channel.register_agent()
@ -61,37 +53,11 @@ def test_control_channel__should_agent_stop(control_channel, island_api_client):
assert island_api_client.should_agent_stop.called_once() assert island_api_client.should_agent_stop.called_once()
def test_control_channel__should_agent_stop_raises_on_connection_error( @pytest.mark.parametrize("api_error", CONTROL_CHANNEL_API_ERRORS)
control_channel, island_api_client def test_control_channel__should_agent_stop_raises_error(
control_channel, island_api_client, api_error
): ):
island_api_client.should_agent_stop.side_effect = IslandAPIConnectionError() island_api_client.should_agent_stop.side_effect = api_error()
with pytest.raises(IslandCommunicationError):
control_channel.should_agent_stop()
def test_control_channel__should_agent_stop_raises_on_timeout_error(
control_channel, island_api_client
):
island_api_client.should_agent_stop.side_effect = IslandAPITimeoutError()
with pytest.raises(IslandCommunicationError):
control_channel.should_agent_stop()
def test_control_channel__should_agent_stop_raises_on_request_error(
control_channel, island_api_client
):
island_api_client.should_agent_stop.side_effect = IslandAPIRequestError()
with pytest.raises(IslandCommunicationError):
control_channel.should_agent_stop()
def test_control_channel__should_agent_stop_raises_on_request_failed_error(
control_channel, island_api_client
):
island_api_client.should_agent_stop.side_effect = IslandAPIRequestFailedError()
with pytest.raises(IslandCommunicationError): with pytest.raises(IslandCommunicationError):
control_channel.should_agent_stop() control_channel.should_agent_stop()