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

BB: Set island mode to Custom

This commit is contained in:
Ilija Lazoroski 2022-08-19 13:57:50 +02:00 committed by Mike Salvatore
parent 0e78129515
commit 789ca96bb0
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
envs/monkey_zoo/blackbox/island_client/monkey_island_client.py
View File

@ -35,9 +35,18 @@ class MonkeyIslandClient(object):
@avoid_race_condition @avoid_race_condition
def import_config(self, test_configuration: TestConfiguration): def import_config(self, test_configuration: TestConfiguration):
self._set_island_mode()
self._import_config(test_configuration) self._import_config(test_configuration)
self._import_credentials(test_configuration.propagation_credentials) 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 @avoid_race_condition
def _import_config(self, test_configuration: TestConfiguration): def _import_config(self, test_configuration: TestConfiguration):
response = self.requests.put_json( response = self.requests.put_json(