p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

BB tests: added the ability for BB tests to "register". If they need registration to run monkeys, BB tests selects passwordless option

This commit is contained in:
VakarisZ 2021-05-20 08:42:51 +03:00
parent 6eb377858d
commit 488143b1d3
1 changed files with 22 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
envs/monkey_zoo/blackbox/island_client/monkey_island_requests.py
View File

@ -12,6 +12,10 @@ NO_AUTH_CREDS = "1234567890!@#$%^&*()_nothing_up_my_sleeve_1234567890!@#$%^&*()"
LOGGER = logging.getLogger(__name__) LOGGER = logging.getLogger(__name__)
class AuthenticationFailedError(Exception):
pass
# noinspection PyArgumentList # noinspection PyArgumentList
class MonkeyIslandRequests(object): class MonkeyIslandRequests(object):
def __init__(self, server_address): def __init__(self, server_address):
@ -43,6 +47,9 @@ class MonkeyIslandRequests(object):
def try_get_jwt_from_server(self): def try_get_jwt_from_server(self):
try: try:
return self.get_jwt_from_server() return self.get_jwt_from_server()
except AuthenticationFailedError:
self.try_set_island_to_no_password()
return self.get_jwt_from_server()
except requests.ConnectionError as err: except requests.ConnectionError as err:
LOGGER.error( LOGGER.error(
"Unable to connect to island, aborting! Error information: {}. Server: {}".format( "Unable to connect to island, aborting! Error information: {}. Server: {}".format(
@ -51,6 +58,21 @@ class MonkeyIslandRequests(object):
) )
assert False assert False
def get_jwt_from_server(self):
resp = requests.post( # noqa: DUO123
self.addr + "api/auth",
json={"username": NO_AUTH_CREDS, "password": NO_AUTH_CREDS},
verify=False,
)
if resp.status_code == 401:
raise AuthenticationFailedError
return resp.json()["access_token"]
def try_set_island_to_no_password(self):
requests.patch( # noqa: DUO123
self.addr + "api/environment", json={"server_config": "standard"}, verify=False
)
class _Decorators: class _Decorators:
@classmethod @classmethod
def refresh_jwt_token(cls, request_function): def refresh_jwt_token(cls, request_function):
@ -62,14 +84,6 @@ class MonkeyIslandRequests(object):
return request_function_wrapper return request_function_wrapper
def get_jwt_from_server(self):
resp = requests.post( # noqa: DUO123
self.addr + "api/auth",
json={"username": NO_AUTH_CREDS, "password": NO_AUTH_CREDS},
verify=False,
)
return resp.json()["access_token"]
@_Decorators.refresh_jwt_token @_Decorators.refresh_jwt_token
def get(self, url, data=None): def get(self, url, data=None):
return requests.get( # noqa: DUO123 return requests.get( # noqa: DUO123