def test_get_tokens_exception(self):
with XRDSSTTest() as app:
with mock.patch('xrdsst.api.tokens_api.TokensApi.get_tokens',
side_effect=ApiException):
token_controller = TokenController()
token_controller.app = app
token_controller.load_config = (lambda: self.ss_config)
for security_server in self.ss_config["security_server"]:
configuration = token_controller.create_api_config(
security_server, self.ss_config)
token_controller.remote_get_tokens(configuration)
self.assertRaises(ApiException)
def step_token_login(self):
with XRDSSTTest() as app:
token_controller = TokenController()
token_controller.app = app
for security_server in self.config["security_server"]:
configuration = token_controller.create_api_config(security_server, self.config)
response = token_controller.remote_get_tokens(configuration)
assert len(response) > 0
assert response[0].logged_in is False
token_controller.remote_token_login(configuration, security_server)
response = token_controller.remote_get_tokens(configuration)
assert len(response) > 0
assert response[0].logged_in is True
def step_token_init_keys(self):
with XRDSSTTest() as app:
token_controller = TokenController()
token_controller.app = app
for security_server in self.config["security_server"]:
configuration = token_controller.create_api_config(security_server, self.config)
response = token_controller.remote_get_tokens(configuration)
assert len(response) > 0
assert len(response[0].keys) == 0
token_controller.remote_token_add_keys_with_csrs(configuration, security_server)
if "clients" in security_server:
for client in security_server["clients"]:
if client["member_class"] != security_server["owner_member_class"] or client["member_code"] != \
security_server["owner_member_code"]:
token_controller.remote_token_add_signing_key_new_member(configuration, security_server, client)
response = token_controller.remote_get_tokens(configuration)
assert len(response) > 0
assert len(response[0].keys) == 3
auth_key_label = security_server['name'] + '-default-auth-key'
sign_key_label = security_server['name'] + '-default-sign-key'
assert str(response[0].keys[0].label) == auth_key_label
assert str(response[0].keys[1].label) == sign_key_label
def test_get_tokens(self):
with XRDSSTTest() as app:
with mock.patch('xrdsst.api.tokens_api.TokensApi.get_tokens',
return_value=TokenTestData.token_list_response):
token_controller = TokenController()
token_controller.app = app
token_controller.load_config = (lambda: self.ss_config)
for security_server in self.ss_config["security_server"]:
configuration = token_controller.create_api_config(
security_server, self.ss_config)
response = token_controller.remote_get_tokens(
configuration)
assert response == TokenTestData.token_list_response