From 7884f96a9cb4032028b9f2f32a37b73487e800c5 Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Tue, 20 Sep 2022 18:45:39 +0530 Subject: [PATCH] UT: Fix broken HTTPIslandAPIClient.get_agent_binary() tests --- .../island_api_client/test_http_island_api_client.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/monkey/tests/unit_tests/infection_monkey/island_api_client/test_http_island_api_client.py b/monkey/tests/unit_tests/infection_monkey/island_api_client/test_http_island_api_client.py index b18f3990b..22ad87161 100644 --- a/monkey/tests/unit_tests/infection_monkey/island_api_client/test_http_island_api_client.py +++ b/monkey/tests/unit_tests/infection_monkey/island_api_client/test_http_island_api_client.py @@ -2,6 +2,7 @@ import pytest import requests import requests_mock +from common import OperatingSystem from infection_monkey.island_api_client import ( HTTPIslandAPIClient, IslandAPIConnectionError, @@ -137,7 +138,7 @@ def test_island_api_client__get_agent_binary(actual_error, expected_error): with pytest.raises(expected_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( @@ -154,4 +155,4 @@ def test_island_api_client__get_agent_binary_status_code(status_code, expected_e with pytest.raises(expected_error): 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)