def test_sign_in_url(user_pool_id, user_pool_client_id,
user_pool_client_secret, region, domain):
cognito = CognitoService(
user_pool_id,
user_pool_client_id,
user_pool_client_secret,
"http://redirect/url",
region,
domain,
)
assert (cognito.get_sign_in_url() == f"https://{domain}"
"/login?response_type=code&"
"client_id=545isk1een1lvilb9en643g3vd&"
"redirect_uri=http%3A//redirect/url&"
"state=dc0de448b88af41d1cd06387ac2d5102")
def test_get_user_info(
user_pool_id, user_pool_client_id, user_pool_client_secret, region, domain, \
user_endpoint_request, \
test_access_token, \
test_user
):
cognito = CognitoService(
user_pool_id,
user_pool_client_id,
user_pool_client_secret,
"http://redirect/url",
region,
domain,
)
user = cognito.get_user_info(access_token=test_access_token,
requests_client=user_endpoint_request)
assert all(v == user[k]
for k, v in test_user.items()) and len(user) == len(test_user)
def test_exchange_code_for_token(
user_pool_id,
user_pool_client_id,
user_pool_client_secret,
region,
domain,
token_endpoint_request,
):
cognito = CognitoService(
user_pool_id,
user_pool_client_id,
user_pool_client_secret,
"http://redirect/url",
region,
domain,
)
token = cognito.exchange_code_for_token(
code="test_code", requests_client=token_endpoint_request)
assert token == "test_access_token"
def test_base_url(user_pool_id, user_pool_client_id, user_pool_client_secret,
region, domain):
cognito = CognitoService(
user_pool_id,
user_pool_client_id,
user_pool_client_secret,
"redirect",
region,
domain,
)
assert cognito.domain == "https://" + domain
def cognito_service_factory(
user_pool_id,
user_pool_client_id,
user_pool_client_secret,
redirect_url,
region,
domain,
):
return CognitoService(
user_pool_id,
user_pool_client_id,
user_pool_client_secret,
redirect_url,
region,
domain,
)