BB: Fix API call to kill all monkeys

This commit is contained in:
Mike Salvatore 2022-03-22 15:32:16 -04:00
parent 51cfb73ce0
commit 88422f9764
1 changed files with 8 additions and 3 deletions

View File

@ -1,6 +1,6 @@
import json
import logging
from time import sleep
import time
from typing import Union
from bson import json_util
@ -15,7 +15,7 @@ LOGGER = logging.getLogger(__name__)
def avoid_race_condition(func):
sleep(SLEEP_BETWEEN_REQUESTS_SECONDS)
time.sleep(SLEEP_BETWEEN_REQUESTS_SECONDS)
return func
@ -48,10 +48,15 @@ class MonkeyIslandClient(object):
@avoid_race_condition
def kill_all_monkeys(self):
if self.requests.get("api", {"action": "killall"}).ok:
response = self.requests.post_json(
"api/monkey_control/stop-all-agents", data={"kill_time": time.time()}
)
if response.ok:
LOGGER.info("Killing all monkeys after the test.")
else:
LOGGER.error("Failed to kill all monkeys.")
LOGGER.error(response.status_code)
LOGGER.error(response.content)
assert False
@avoid_race_condition