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

UT: Fix broken HTTPIslandAPIClient.get_agent_binary() tests

This commit is contained in:
Shreya Malviya 2022-09-20 18:45:39 +05:30
parent 2bd64a3175
commit 7884f96a9c
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
monkey/tests/unit_tests/infection_monkey/island_api_client/test_http_island_api_client.py
View File

@ -2,6 +2,7 @@ import pytest
import requests import requests
import requests_mock import requests_mock
from common import OperatingSystem
from infection_monkey.island_api_client import ( from infection_monkey.island_api_client import (
HTTPIslandAPIClient, HTTPIslandAPIClient,
IslandAPIConnectionError, IslandAPIConnectionError,
@ -137,7 +138,7 @@ def test_island_api_client__get_agent_binary(actual_error, expected_error):
with pytest.raises(expected_error): with pytest.raises(expected_error):
m.get(ISLAND_GET_AGENT_BINARY_URI, exc=actual_error) m.get(ISLAND_GET_AGENT_BINARY_URI, exc=actual_error)
island_api_client.get_agent_binary(os_name=WINDOWS) island_api_client.get_agent_binary(operating_system=OperatingSystem.WINDOWS)
@pytest.mark.parametrize( @pytest.mark.parametrize(
@ -154,4 +155,4 @@ def test_island_api_client__get_agent_binary_status_code(status_code, expected_e
with pytest.raises(expected_error): with pytest.raises(expected_error):
m.get(ISLAND_GET_AGENT_BINARY_URI, status_code=status_code) m.get(ISLAND_GET_AGENT_BINARY_URI, status_code=status_code)
island_api_client.get_agent_binary(os_name=WINDOWS) island_api_client.get_agent_binary(operating_system=OperatingSystem.WINDOWS)