103 lines
3.0 KiB
Python
103 lines
3.0 KiB
Python
import json
|
|
|
|
import moto.server as server
|
|
from moto import mock_aws
|
|
|
|
"""
|
|
Test the different server responses
|
|
"""
|
|
|
|
|
|
@mock_aws
|
|
def test_create_identity_pool():
|
|
backend = server.create_backend_app("cognito-identity")
|
|
test_client = backend.test_client()
|
|
|
|
res = test_client.post(
|
|
"/",
|
|
data={"IdentityPoolName": "test", "AllowUnauthenticatedIdentities": True},
|
|
headers={
|
|
"X-Amz-Target": "com.amazonaws.cognito.identity.model.AWSCognitoIdentityService.CreateIdentityPool"
|
|
},
|
|
)
|
|
|
|
json_data = json.loads(res.data.decode("utf-8"))
|
|
assert json_data["IdentityPoolName"] == "test"
|
|
|
|
|
|
@mock_aws
|
|
def test_get_id():
|
|
backend = server.create_backend_app("cognito-identity")
|
|
test_client = backend.test_client()
|
|
|
|
res = test_client.post(
|
|
"/",
|
|
data={"IdentityPoolName": "test", "AllowUnauthenticatedIdentities": True},
|
|
headers={
|
|
"X-Amz-Target": "com.amazonaws.cognito.identity.model.AWSCognitoIdentityService.CreateIdentityPool"
|
|
},
|
|
)
|
|
|
|
json_data = json.loads(res.data.decode("utf-8"))
|
|
res = test_client.post(
|
|
"/",
|
|
data=json.dumps(
|
|
{
|
|
"AccountId": "someaccount",
|
|
"IdentityPoolId": json_data["IdentityPoolId"],
|
|
"Logins": {"someurl": "12345"},
|
|
}
|
|
),
|
|
headers={
|
|
"X-Amz-Target": "com.amazonaws.cognito.identity.model.AWSCognitoIdentityService.GetId"
|
|
},
|
|
)
|
|
|
|
json_data = json.loads(res.data.decode("utf-8"))
|
|
assert ":" in json_data["IdentityId"]
|
|
|
|
|
|
@mock_aws
|
|
def test_list_identities():
|
|
backend = server.create_backend_app("cognito-identity")
|
|
test_client = backend.test_client()
|
|
|
|
res = test_client.post(
|
|
"/",
|
|
data={"IdentityPoolName": "test", "AllowUnauthenticatedIdentities": True},
|
|
headers={
|
|
"X-Amz-Target": "com.amazonaws.cognito.identity.model.AWSCognitoIdentityService.CreateIdentityPool"
|
|
},
|
|
)
|
|
|
|
json_data = json.loads(res.data.decode("utf-8"))
|
|
identity_pool_id = json_data["IdentityPoolId"]
|
|
res = test_client.post(
|
|
"/",
|
|
data=json.dumps(
|
|
{
|
|
"AccountId": "someaccount",
|
|
"IdentityPoolId": identity_pool_id,
|
|
"Logins": {"someurl": "12345"},
|
|
}
|
|
),
|
|
headers={
|
|
"X-Amz-Target": "com.amazonaws.cognito.identity.model.AWSCognitoIdentityService.GetId"
|
|
},
|
|
)
|
|
|
|
json_data = json.loads(res.data.decode("utf-8"))
|
|
identity_id = json_data["IdentityId"]
|
|
|
|
res = test_client.post(
|
|
"/",
|
|
data=json.dumps({"IdentityPoolId": identity_pool_id}),
|
|
headers={
|
|
"X-Amz-Target": "com.amazonaws.cognito.identity.model.AWSCognitoIdentityService.ListIdentities"
|
|
},
|
|
)
|
|
|
|
json_data = json.loads(res.data.decode("utf-8"))
|
|
assert "IdentityPoolId" in json_data and "Identities" in json_data
|
|
assert identity_id in [x["IdentityId"] for x in json_data["Identities"]]
|