def test_set_user_id():
uid = "my_id"
attributes = {"attr_1": "v1", "attr_2": "v2", "attr_3": "v3"}
internal_response = InternalResponse(UserIdHashType.persistent)
internal_response.add_attributes(attributes)
internal_response.set_user_id(uid)
assert uid == internal_response.get_user_id()
def test_set_user_id():
uid = "my_id"
attributes = {"attr_1": "v1", "attr_2": "v2", "attr_3": "v3"}
internal_response = InternalResponse(UserIdHashType.persistent)
internal_response.add_attributes(attributes)
internal_response.set_user_id(uid)
assert uid == internal_response.get_user_id()
class TestAccountLinking():
@pytest.fixture(autouse=True)
def setup(self, signing_key_path):
self.account_linking_config = {
"enable": True,
"rest_uri": "https://localhost:8167",
"redirect": "https://localhost:8167/approve",
"endpoint": "handle_account_linking",
"sign_key": signing_key_path,
"verify_ssl": False
}
self.satosa_config = {
"BASE": "https://proxy.example.com",
"USER_ID_HASH_SALT": "qwerty",
"COOKIE_STATE_NAME": "SATOSA_SATE",
"STATE_ENCRYPTION_KEY": "ASDasd123",
"PLUGIN_PATH": "",
"BACKEND_MODULES": "",
"FRONTEND_MODULES": "",
"INTERNAL_ATTRIBUTES": {},
"ACCOUNT_LINKING": self.account_linking_config
}
self.callback_func = MagicMock()
self.context = Context()
state = State()
self.context.state = state
auth_info = AuthenticationInformation("auth_class_ref", "timestamp", "issuer")
self.internal_response = InternalResponse(auth_info=auth_info)
def test_disable_account_linking(self):
self.account_linking_config['enable'] = False
config = SATOSAConfig(self.satosa_config)
account_linking = AccountLinkingModule(config, self.callback_func)
account_linking.manage_al(None, None)
assert self.callback_func.called
@responses.activate
def test_store_existing_uuid_in_internal_attributes(self):
uuid = "uuid"
responses.add(
responses.GET,
"%s/get_id" % self.account_linking_config['rest_uri'],
status=200,
body=uuid,
content_type='text/html'
)
account_linking = AccountLinkingModule(
SATOSAConfig(self.satosa_config),
self.callback_func
)
account_linking.manage_al(self.context, self.internal_response)
assert self.internal_response.get_user_id() == uuid
@responses.activate
def test_account_link_does_not_exists(self):
ticket = "ticket"
responses.add(
responses.GET,
"%s/get_id" % self.account_linking_config['rest_uri'],
status=404,
body=ticket,
content_type='text/html'
)
account_linking = AccountLinkingModule(
SATOSAConfig(self.satosa_config),
self.callback_func
)
result = account_linking.manage_al(self.context, self.internal_response)
assert isinstance(result, Redirect)
assert self.account_linking_config["redirect"] in result.message
@responses.activate
def test_handle_failed_connection(self):
exception = requests.ConnectionError("No connection")
responses.add(responses.GET, "%s/get_id" % self.account_linking_config['rest_uri'],
body=exception)
account_linking = AccountLinkingModule(
SATOSAConfig(self.satosa_config),
self.callback_func
)
with pytest.raises(SATOSAAuthenticationError):
account_linking.manage_al(self.context, self.internal_response)
class TestAccountLinking():
@pytest.fixture(autouse=True)
def setup(self, signing_key_path):
self.account_linking_config = {
"enable": True,
"rest_uri": "https://localhost:8167",
"redirect": "https://localhost:8167/approve",
"endpoint": "handle_account_linking",
"sign_key": signing_key_path,
"verify_ssl": False
}
self.satosa_config = {
"BASE": "https://proxy.example.com",
"USER_ID_HASH_SALT": "qwerty",
"COOKIE_STATE_NAME": "SATOSA_SATE",
"STATE_ENCRYPTION_KEY": "ASDasd123",
"PLUGIN_PATH": "",
"BACKEND_MODULES": "",
"FRONTEND_MODULES": "",
"INTERNAL_ATTRIBUTES": {},
"ACCOUNT_LINKING": self.account_linking_config
}
self.callback_func = MagicMock()
self.context = Context()
state = State()
self.context.state = state
auth_info = AuthenticationInformation("auth_class_ref", "timestamp",
"issuer")
self.internal_response = InternalResponse(auth_info=auth_info)
def test_disable_account_linking(self):
self.account_linking_config['enable'] = False
config = SATOSAConfig(self.satosa_config)
account_linking = AccountLinkingModule(config, self.callback_func)
account_linking.manage_al(None, None)
assert self.callback_func.called
@responses.activate
def test_store_existing_uuid_in_internal_attributes(self):
uuid = "uuid"
responses.add(responses.GET,
"%s/get_id" % self.account_linking_config['rest_uri'],
status=200,
body=uuid,
content_type='text/html')
account_linking = AccountLinkingModule(
SATOSAConfig(self.satosa_config), self.callback_func)
account_linking.manage_al(self.context, self.internal_response)
assert self.internal_response.get_user_id() == uuid
@responses.activate
def test_account_link_does_not_exists(self):
ticket = "ticket"
responses.add(responses.GET,
"%s/get_id" % self.account_linking_config['rest_uri'],
status=404,
body=ticket,
content_type='text/html')
account_linking = AccountLinkingModule(
SATOSAConfig(self.satosa_config), self.callback_func)
result = account_linking.manage_al(self.context,
self.internal_response)
assert isinstance(result, Redirect)
assert self.account_linking_config["redirect"] in result.message
@responses.activate
def test_handle_failed_connection(self):
exception = requests.ConnectionError("No connection")
responses.add(responses.GET,
"%s/get_id" % self.account_linking_config['rest_uri'],
body=exception)
account_linking = AccountLinkingModule(
SATOSAConfig(self.satosa_config), self.callback_func)
with pytest.raises(SATOSAAuthenticationError):
account_linking.manage_al(self.context, self.internal_response)
