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

Agent: Call raise_for_status() in HTTPIslandAPIClient

This commit is contained in:
Mike Salvatore 2022-09-19 20:05:19 -04:00
parent a724758caa
commit 107a15b5f0
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
monkey/infection_monkey/island_api_client/http_island_api_client.py
View File

@ -46,23 +46,25 @@ class HTTPIslandAPIClient(IIslandAPIClient):
@handle_island_errors @handle_island_errors
def __init__(self, island_server: str): def __init__(self, island_server: str):
requests.get( # noqa: DUO123 response = requests.get( # noqa: DUO123
f"https://{island_server}/api?action=is-up", f"https://{island_server}/api?action=is-up",
verify=False, verify=False,
timeout=MEDIUM_REQUEST_TIMEOUT, timeout=MEDIUM_REQUEST_TIMEOUT,
) )
response.raise_for_status()
self._island_server = island_server self._island_server = island_server
self._api_url = f"https://{self._island_server}/api" self._api_url = f"https://{self._island_server}/api"
@handle_island_errors @handle_island_errors
def send_log(self, log_contents: str): def send_log(self, log_contents: str):
requests.post( # noqa: DUO123 response = requests.post( # noqa: DUO123
f"{self._api_url}/log", f"{self._api_url}/log",
json=log_contents, json=log_contents,
verify=False, verify=False,
timeout=MEDIUM_REQUEST_TIMEOUT, timeout=MEDIUM_REQUEST_TIMEOUT,
) )
response.raise_for_status()
@handle_island_errors @handle_island_errors
def get_pba_file(self, filename: str): def get_pba_file(self, filename: str):
@ -71,5 +73,6 @@ class HTTPIslandAPIClient(IIslandAPIClient):
verify=False, verify=False,
timeout=LONG_REQUEST_TIMEOUT, timeout=LONG_REQUEST_TIMEOUT,
) )
response.raise_for_status()
return response.content return response.content