From 789ca96bb0aaa0bf92f1792fee18b2b9e1ee3b2b Mon Sep 17 00:00:00 2001 From: Ilija Lazoroski Date: Fri, 19 Aug 2022 13:57:50 +0200 Subject: [PATCH] BB: Set island mode to Custom --- .../blackbox/island_client/monkey_island_client.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/envs/monkey_zoo/blackbox/island_client/monkey_island_client.py b/envs/monkey_zoo/blackbox/island_client/monkey_island_client.py index 8cadac06e..03ab56d22 100644 --- a/envs/monkey_zoo/blackbox/island_client/monkey_island_client.py +++ b/envs/monkey_zoo/blackbox/island_client/monkey_island_client.py @@ -35,9 +35,18 @@ class MonkeyIslandClient(object): @avoid_race_condition def import_config(self, test_configuration: TestConfiguration): + self._set_island_mode() self._import_config(test_configuration) self._import_credentials(test_configuration.propagation_credentials) + @avoid_race_condition + def _set_island_mode(self): + if self.requests.put_json("api/island/mode", json="advanced").ok: + LOGGER.info("Setting island mode to Custom.") + else: + LOGGER.error("Failed to set island mode") + assert False + @avoid_race_condition def _import_config(self, test_configuration: TestConfiguration): response = self.requests.put_json(