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

UT: Use only Python3.7 features in test_aws_command_runner

This commit is contained in:
Mike Salvatore 2022-05-11 07:56:15 -04:00
parent b8006a6274
commit 680dbca574
1 changed files with 10 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
monkey/tests/unit_tests/monkey_island/cc/services/aws/test_aws_command_runner.py
View File

@ -149,45 +149,40 @@ def test_correct_instance_id(successful_mock_client):
start_infection_monkey_agent(successful_mock_client, INSTANCE_ID, "linux", ISLAND_IP) start_infection_monkey_agent(successful_mock_client, INSTANCE_ID, "linux", ISLAND_IP)
successful_mock_client.send_command.assert_called_once() successful_mock_client.send_command.assert_called_once()
assert successful_mock_client.send_command.call_args.kwargs["InstanceIds"] == [INSTANCE_ID] call_args_kwargs = successful_mock_client.send_command.call_args[1]
assert call_args_kwargs["InstanceIds"] == [INSTANCE_ID]
def test_linux_doc_name(successful_mock_client): def test_linux_doc_name(successful_mock_client):
start_infection_monkey_agent(successful_mock_client, INSTANCE_ID, "linux", ISLAND_IP) start_infection_monkey_agent(successful_mock_client, INSTANCE_ID, "linux", ISLAND_IP)
successful_mock_client.send_command.assert_called_once() successful_mock_client.send_command.assert_called_once()
assert ( call_args_kwargs = successful_mock_client.send_command.call_args[1]
successful_mock_client.send_command.call_args.kwargs["DocumentName"] == LINUX_DOCUMENT_NAME assert call_args_kwargs["DocumentName"] == LINUX_DOCUMENT_NAME
)
def test_windows_doc_name(successful_mock_client): def test_windows_doc_name(successful_mock_client):
start_infection_monkey_agent(successful_mock_client, INSTANCE_ID, "windows", ISLAND_IP) start_infection_monkey_agent(successful_mock_client, INSTANCE_ID, "windows", ISLAND_IP)
successful_mock_client.send_command.assert_called_once() successful_mock_client.send_command.assert_called_once()
assert ( call_args_kwargs = successful_mock_client.send_command.call_args[1]
successful_mock_client.send_command.call_args.kwargs["DocumentName"] assert call_args_kwargs["DocumentName"] == WINDOWS_DOCUMENT_NAME
== WINDOWS_DOCUMENT_NAME
)
def test_linux_command(successful_mock_client): def test_linux_command(successful_mock_client):
start_infection_monkey_agent(successful_mock_client, INSTANCE_ID, "linux", ISLAND_IP) start_infection_monkey_agent(successful_mock_client, INSTANCE_ID, "linux", ISLAND_IP)
successful_mock_client.send_command.assert_called_once() successful_mock_client.send_command.assert_called_once()
assert ( call_args_kwargs = successful_mock_client.send_command.call_args[1]
"wget" in successful_mock_client.send_command.call_args.kwargs["Parameters"]["commands"][0] assert "wget" in call_args_kwargs["Parameters"]["commands"][0]
)
def test_windows_command(successful_mock_client): def test_windows_command(successful_mock_client):
start_infection_monkey_agent(successful_mock_client, INSTANCE_ID, "windows", ISLAND_IP) start_infection_monkey_agent(successful_mock_client, INSTANCE_ID, "windows", ISLAND_IP)
successful_mock_client.send_command.assert_called_once() successful_mock_client.send_command.assert_called_once()
assert ( call_args_kwargs = successful_mock_client.send_command.call_args[1]
"DownloadFile" assert "DownloadFile" in call_args_kwargs["Parameters"]["commands"][0]
in successful_mock_client.send_command.call_args.kwargs["Parameters"]["commands"][0]
)
def test_multiple_status_queries(send_command_response, in_progress_response, success_response): def test_multiple_status_queries(send_command_response, in_progress_response, success_response):