114 lines
3.2 KiB
Python
114 lines
3.2 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
import boto3
|
|
|
|
from moto import mock_aws
|
|
from moto.core.utils import unix_time
|
|
|
|
|
|
@mock_aws
|
|
def test_create_api_key_simple():
|
|
client = boto3.client("appsync", region_name="eu-west-1")
|
|
|
|
api_id = client.create_graphql_api(name="api1", authenticationType="API_KEY")[
|
|
"graphqlApi"
|
|
]["apiId"]
|
|
resp = client.create_api_key(apiId=api_id)
|
|
|
|
assert "apiKey" in resp
|
|
api_key = resp["apiKey"]
|
|
|
|
assert "id" in api_key
|
|
assert "description" not in api_key
|
|
assert "expires" in api_key
|
|
assert "deletes" in api_key
|
|
|
|
|
|
@mock_aws
|
|
def test_create_api_key():
|
|
client = boto3.client("appsync", region_name="ap-southeast-1")
|
|
tomorrow = datetime.now() + timedelta(days=1)
|
|
tomorrow_in_secs = int(unix_time(tomorrow))
|
|
|
|
api_id = client.create_graphql_api(name="api1", authenticationType="API_KEY")[
|
|
"graphqlApi"
|
|
]["apiId"]
|
|
resp = client.create_api_key(
|
|
apiId=api_id, description="my first api key", expires=tomorrow_in_secs
|
|
)
|
|
|
|
assert "apiKey" in resp
|
|
api_key = resp["apiKey"]
|
|
|
|
assert "id" in api_key
|
|
assert api_key["description"] == "my first api key"
|
|
assert api_key["expires"] == tomorrow_in_secs
|
|
assert api_key["deletes"] == tomorrow_in_secs
|
|
|
|
|
|
@mock_aws
|
|
def test_delete_api_key():
|
|
client = boto3.client("appsync", region_name="us-east-1")
|
|
|
|
api_id = client.create_graphql_api(name="api1", authenticationType="API_KEY")[
|
|
"graphqlApi"
|
|
]["apiId"]
|
|
api_key_id = client.create_api_key(apiId=api_id)["apiKey"]["id"]
|
|
|
|
client.delete_api_key(apiId=api_id, id=api_key_id)
|
|
|
|
resp = client.list_api_keys(apiId=api_id)
|
|
assert len(resp["apiKeys"]) == 0
|
|
|
|
|
|
@mock_aws
|
|
def test_list_api_keys_unknown_api():
|
|
client = boto3.client("appsync", region_name="ap-southeast-1")
|
|
resp = client.list_api_keys(apiId="unknown")
|
|
assert resp["apiKeys"] == []
|
|
|
|
|
|
@mock_aws
|
|
def test_list_api_keys_empty():
|
|
client = boto3.client("appsync", region_name="ap-southeast-1")
|
|
api_id = client.create_graphql_api(name="api1", authenticationType="API_KEY")[
|
|
"graphqlApi"
|
|
]["apiId"]
|
|
|
|
resp = client.list_api_keys(apiId=api_id)
|
|
assert resp["apiKeys"] == []
|
|
|
|
|
|
@mock_aws
|
|
def test_list_api_keys():
|
|
client = boto3.client("appsync", region_name="ap-southeast-1")
|
|
api_id = client.create_graphql_api(name="api1", authenticationType="API_KEY")[
|
|
"graphqlApi"
|
|
]["apiId"]
|
|
client.create_api_key(apiId=api_id)
|
|
client.create_api_key(apiId=api_id, description="my first api key")
|
|
resp = client.list_api_keys(apiId=api_id)
|
|
assert len(resp["apiKeys"]) == 2
|
|
|
|
|
|
@mock_aws
|
|
def test_update_api_key():
|
|
client = boto3.client("appsync", region_name="eu-west-1")
|
|
|
|
api_id = client.create_graphql_api(name="api1", authenticationType="API_KEY")[
|
|
"graphqlApi"
|
|
]["apiId"]
|
|
|
|
original = client.create_api_key(apiId=api_id, description="my first api key")[
|
|
"apiKey"
|
|
]
|
|
|
|
updated = client.update_api_key(
|
|
apiId=api_id, id=original["id"], description="my second api key"
|
|
)["apiKey"]
|
|
|
|
assert updated["id"] == original["id"]
|
|
assert updated["description"] == "my second api key"
|
|
assert updated["expires"] == original["expires"]
|
|
assert updated["deletes"] == original["deletes"]
|