Steve Pulec
commit
aebc7c8ea3
10 changed files with 1081 additions and 2 deletions
518
tests/test_apigateway/test_apigateway.py
Normal file
518
tests/test_apigateway/test_apigateway.py
Normal file
|
|
@ -0,0 +1,518 @@
|
|||
from __future__ import unicode_literals
|
||||
|
||||
from datetime import datetime
|
||||
from dateutil.tz import tzutc
|
||||
import boto3
|
||||
from freezegun import freeze_time
|
||||
import httpretty
|
||||
import requests
|
||||
import sure # noqa
|
||||
|
||||
from moto import mock_apigateway
|
||||
|
||||
|
||||
@freeze_time("2015-01-01")
|
||||
@mock_apigateway
|
||||
def test_create_and_get_rest_api():
|
||||
client = boto3.client('apigateway', region_name='us-west-2')
|
||||
|
||||
response = client.create_rest_api(
|
||||
name='my_api',
|
||||
description='this is my api',
|
||||
)
|
||||
api_id = response['id']
|
||||
|
||||
response = client.get_rest_api(
|
||||
restApiId=api_id
|
||||
)
|
||||
|
||||
response.pop('ResponseMetadata')
|
||||
response.should.equal({
|
||||
'id': api_id,
|
||||
'name': 'my_api',
|
||||
'description': 'this is my api',
|
||||
'createdDate': datetime(2015, 1, 1, tzinfo=tzutc())
|
||||
})
|
||||
|
||||
|
||||
@mock_apigateway
|
||||
def test_list_and_delete_apis():
|
||||
client = boto3.client('apigateway', region_name='us-west-2')
|
||||
|
||||
response = client.create_rest_api(
|
||||
name='my_api',
|
||||
description='this is my api',
|
||||
)
|
||||
api_id = response['id']
|
||||
client.create_rest_api(
|
||||
name='my_api2',
|
||||
description='this is my api2',
|
||||
)
|
||||
|
||||
response = client.get_rest_apis()
|
||||
len(response['items']).should.equal(2)
|
||||
|
||||
client.delete_rest_api(
|
||||
restApiId=api_id
|
||||
)
|
||||
|
||||
response = client.get_rest_apis()
|
||||
len(response['items']).should.equal(1)
|
||||
|
||||
|
||||
@mock_apigateway
|
||||
def test_create_resource():
|
||||
client = boto3.client('apigateway', region_name='us-west-2')
|
||||
response = client.create_rest_api(
|
||||
name='my_api',
|
||||
description='this is my api',
|
||||
)
|
||||
api_id = response['id']
|
||||
|
||||
resources = client.get_resources(restApiId=api_id)
|
||||
root_id = [resource for resource in resources['items'] if resource['path'] == '/'][0]['id']
|
||||
|
||||
root_resource = client.get_resource(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
)
|
||||
root_resource.should.equal({
|
||||
'path': '/',
|
||||
'id': root_id,
|
||||
'ResponseMetadata': {'HTTPStatusCode': 200},
|
||||
'resourceMethods': {
|
||||
'GET': {}
|
||||
}
|
||||
})
|
||||
|
||||
response = client.create_resource(
|
||||
restApiId=api_id,
|
||||
parentId=root_id,
|
||||
pathPart='/users',
|
||||
)
|
||||
|
||||
resources = client.get_resources(restApiId=api_id)['items']
|
||||
len(resources).should.equal(2)
|
||||
non_root_resource = [resource for resource in resources if resource['path'] != '/'][0]
|
||||
|
||||
response = client.delete_resource(
|
||||
restApiId=api_id,
|
||||
resourceId=non_root_resource['id']
|
||||
)
|
||||
|
||||
len(client.get_resources(restApiId=api_id)['items']).should.equal(1)
|
||||
|
||||
|
||||
@mock_apigateway
|
||||
def test_child_resource():
|
||||
client = boto3.client('apigateway', region_name='us-west-2')
|
||||
response = client.create_rest_api(
|
||||
name='my_api',
|
||||
description='this is my api',
|
||||
)
|
||||
api_id = response['id']
|
||||
|
||||
resources = client.get_resources(restApiId=api_id)
|
||||
root_id = [resource for resource in resources['items'] if resource['path'] == '/'][0]['id']
|
||||
|
||||
response = client.create_resource(
|
||||
restApiId=api_id,
|
||||
parentId=root_id,
|
||||
pathPart='users',
|
||||
)
|
||||
users_id = response['id']
|
||||
|
||||
response = client.create_resource(
|
||||
restApiId=api_id,
|
||||
parentId=users_id,
|
||||
pathPart='tags',
|
||||
)
|
||||
tags_id = response['id']
|
||||
|
||||
child_resource = client.get_resource(
|
||||
restApiId=api_id,
|
||||
resourceId=tags_id,
|
||||
)
|
||||
child_resource.should.equal({
|
||||
'path': '/users/tags',
|
||||
'pathPart': 'tags',
|
||||
'parentId': users_id,
|
||||
'id': tags_id,
|
||||
'ResponseMetadata': {'HTTPStatusCode': 200},
|
||||
'resourceMethods': {'GET': {}},
|
||||
})
|
||||
|
||||
|
||||
@mock_apigateway
|
||||
def test_create_method():
|
||||
client = boto3.client('apigateway', region_name='us-west-2')
|
||||
response = client.create_rest_api(
|
||||
name='my_api',
|
||||
description='this is my api',
|
||||
)
|
||||
api_id = response['id']
|
||||
|
||||
resources = client.get_resources(restApiId=api_id)
|
||||
root_id = [resource for resource in resources['items'] if resource['path'] == '/'][0]['id']
|
||||
|
||||
client.put_method(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
authorizationType='none',
|
||||
)
|
||||
|
||||
response = client.get_method(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET'
|
||||
)
|
||||
|
||||
response.should.equal({
|
||||
'httpMethod': 'GET',
|
||||
'authorizationType': 'none',
|
||||
'ResponseMetadata': {'HTTPStatusCode': 200}
|
||||
})
|
||||
|
||||
|
||||
@mock_apigateway
|
||||
def test_create_method_response():
|
||||
client = boto3.client('apigateway', region_name='us-west-2')
|
||||
response = client.create_rest_api(
|
||||
name='my_api',
|
||||
description='this is my api',
|
||||
)
|
||||
api_id = response['id']
|
||||
|
||||
resources = client.get_resources(restApiId=api_id)
|
||||
root_id = [resource for resource in resources['items'] if resource['path'] == '/'][0]['id']
|
||||
|
||||
client.put_method(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
authorizationType='none',
|
||||
)
|
||||
|
||||
response = client.get_method(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET'
|
||||
)
|
||||
|
||||
response = client.put_method_response(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
statusCode='200',
|
||||
)
|
||||
response.should.equal({
|
||||
'ResponseMetadata': {'HTTPStatusCode': 200},
|
||||
'statusCode': '200'
|
||||
})
|
||||
|
||||
response = client.get_method_response(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
statusCode='200',
|
||||
)
|
||||
response.should.equal({
|
||||
'ResponseMetadata': {'HTTPStatusCode': 200},
|
||||
'statusCode': '200'
|
||||
})
|
||||
|
||||
response = client.delete_method_response(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
statusCode='200',
|
||||
)
|
||||
response.should.equal({'ResponseMetadata': {'HTTPStatusCode': 200}})
|
||||
|
||||
|
||||
@mock_apigateway
|
||||
def test_integrations():
|
||||
client = boto3.client('apigateway', region_name='us-west-2')
|
||||
response = client.create_rest_api(
|
||||
name='my_api',
|
||||
description='this is my api',
|
||||
)
|
||||
api_id = response['id']
|
||||
|
||||
resources = client.get_resources(restApiId=api_id)
|
||||
root_id = [resource for resource in resources['items'] if resource['path'] == '/'][0]['id']
|
||||
|
||||
client.put_method(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
authorizationType='none',
|
||||
)
|
||||
|
||||
client.put_method_response(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
statusCode='200',
|
||||
)
|
||||
|
||||
response = client.put_integration(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
type='HTTP',
|
||||
uri='http://httpbin.org/robots.txt',
|
||||
)
|
||||
response.should.equal({
|
||||
'ResponseMetadata': {'HTTPStatusCode': 200},
|
||||
'httpMethod': 'GET',
|
||||
'integrationResponses': {
|
||||
'200': {
|
||||
'responseTemplates': {
|
||||
'application/json': None
|
||||
},
|
||||
'statusCode': 200
|
||||
}
|
||||
},
|
||||
'type': 'HTTP',
|
||||
'uri': 'http://httpbin.org/robots.txt'
|
||||
})
|
||||
|
||||
response = client.get_integration(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET'
|
||||
)
|
||||
response.should.equal({
|
||||
'ResponseMetadata': {'HTTPStatusCode': 200},
|
||||
'httpMethod': 'GET',
|
||||
'integrationResponses': {
|
||||
'200': {
|
||||
'responseTemplates': {
|
||||
'application/json': None
|
||||
},
|
||||
'statusCode': 200
|
||||
}
|
||||
},
|
||||
'type': 'HTTP',
|
||||
'uri': 'http://httpbin.org/robots.txt'
|
||||
})
|
||||
|
||||
response = client.get_resource(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
)
|
||||
response['resourceMethods']['GET']['methodIntegration'].should.equal({
|
||||
'httpMethod': 'GET',
|
||||
'integrationResponses': {
|
||||
'200': {
|
||||
'responseTemplates': {
|
||||
'application/json': None
|
||||
},
|
||||
'statusCode': 200
|
||||
}
|
||||
},
|
||||
'type': 'HTTP',
|
||||
'uri': 'http://httpbin.org/robots.txt'
|
||||
})
|
||||
|
||||
client.delete_integration(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET'
|
||||
)
|
||||
|
||||
response = client.get_resource(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
)
|
||||
response['resourceMethods']['GET'].shouldnt.contain("methodIntegration")
|
||||
|
||||
|
||||
@mock_apigateway
|
||||
def test_integration_response():
|
||||
client = boto3.client('apigateway', region_name='us-west-2')
|
||||
response = client.create_rest_api(
|
||||
name='my_api',
|
||||
description='this is my api',
|
||||
)
|
||||
api_id = response['id']
|
||||
|
||||
resources = client.get_resources(restApiId=api_id)
|
||||
root_id = [resource for resource in resources['items'] if resource['path'] == '/'][0]['id']
|
||||
|
||||
client.put_method(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
authorizationType='none',
|
||||
)
|
||||
|
||||
client.put_method_response(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
statusCode='200',
|
||||
)
|
||||
|
||||
response = client.put_integration(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
type='HTTP',
|
||||
uri='http://httpbin.org/robots.txt',
|
||||
)
|
||||
|
||||
response = client.put_integration_response(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
statusCode='200',
|
||||
selectionPattern='foobar',
|
||||
)
|
||||
response.should.equal({
|
||||
'statusCode': '200',
|
||||
'selectionPattern': 'foobar',
|
||||
'ResponseMetadata': {'HTTPStatusCode': 200},
|
||||
'responseTemplates': {
|
||||
'application/json': None
|
||||
}
|
||||
})
|
||||
|
||||
response = client.get_integration_response(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
statusCode='200',
|
||||
)
|
||||
response.should.equal({
|
||||
'statusCode': '200',
|
||||
'selectionPattern': 'foobar',
|
||||
'ResponseMetadata': {'HTTPStatusCode': 200},
|
||||
'responseTemplates': {
|
||||
'application/json': None
|
||||
}
|
||||
})
|
||||
|
||||
response = client.get_method(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
)
|
||||
response['methodIntegration']['integrationResponses'].should.equal({
|
||||
'200': {
|
||||
'responseTemplates': {
|
||||
'application/json': None
|
||||
},
|
||||
'selectionPattern': 'foobar',
|
||||
'statusCode': '200'
|
||||
}
|
||||
})
|
||||
|
||||
response = client.delete_integration_response(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
statusCode='200',
|
||||
)
|
||||
|
||||
response = client.get_method(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
)
|
||||
response['methodIntegration']['integrationResponses'].should.equal({})
|
||||
|
||||
|
||||
@mock_apigateway
|
||||
def test_deployment():
|
||||
client = boto3.client('apigateway', region_name='us-west-2')
|
||||
response = client.create_rest_api(
|
||||
name='my_api',
|
||||
description='this is my api',
|
||||
)
|
||||
api_id = response['id']
|
||||
|
||||
response = client.create_deployment(
|
||||
restApiId=api_id,
|
||||
stageName='staging',
|
||||
)
|
||||
deployment_id = response['id']
|
||||
|
||||
response = client.get_deployment(
|
||||
restApiId=api_id,
|
||||
deploymentId=deployment_id,
|
||||
)
|
||||
response.should.equal({
|
||||
'id': deployment_id,
|
||||
'ResponseMetadata': {'HTTPStatusCode': 200}
|
||||
})
|
||||
|
||||
response = client.get_deployments(
|
||||
restApiId=api_id,
|
||||
)
|
||||
response['items'].should.equal([
|
||||
{'id': deployment_id}
|
||||
])
|
||||
|
||||
response = client.delete_deployment(
|
||||
restApiId=api_id,
|
||||
deploymentId=deployment_id,
|
||||
)
|
||||
|
||||
response = client.get_deployments(
|
||||
restApiId=api_id,
|
||||
)
|
||||
len(response['items']).should.equal(0)
|
||||
|
||||
|
||||
@httpretty.activate
|
||||
@mock_apigateway
|
||||
def test_http_proxying_integration():
|
||||
httpretty.register_uri(
|
||||
httpretty.GET, "http://httpbin.org/robots.txt", body='a fake response'
|
||||
)
|
||||
|
||||
region_name = 'us-west-2'
|
||||
client = boto3.client('apigateway', region_name=region_name)
|
||||
response = client.create_rest_api(
|
||||
name='my_api',
|
||||
description='this is my api',
|
||||
)
|
||||
api_id = response['id']
|
||||
|
||||
resources = client.get_resources(restApiId=api_id)
|
||||
root_id = [resource for resource in resources['items'] if resource['path'] == '/'][0]['id']
|
||||
|
||||
client.put_method(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
authorizationType='none',
|
||||
)
|
||||
|
||||
client.put_method_response(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
statusCode='200',
|
||||
)
|
||||
|
||||
response = client.put_integration(
|
||||
restApiId=api_id,
|
||||
resourceId=root_id,
|
||||
httpMethod='GET',
|
||||
type='HTTP',
|
||||
uri='http://httpbin.org/robots.txt',
|
||||
)
|
||||
|
||||
stage_name = 'staging'
|
||||
client.create_deployment(
|
||||
restApiId=api_id,
|
||||
stageName=stage_name,
|
||||
)
|
||||
|
||||
deploy_url = "https://{api_id}.execute-api.{region_name}.amazonaws.com/{stage_name}".format(api_id=api_id, region_name=region_name, stage_name=stage_name)
|
||||
|
||||
requests.get(deploy_url).content.should.equal(b"a fake response")
|
||||
16
tests/test_apigateway/test_server.py
Normal file
16
tests/test_apigateway/test_server.py
Normal file
|
|
@ -0,0 +1,16 @@
|
|||
from __future__ import unicode_literals
|
||||
import sure # noqa
|
||||
|
||||
import moto.server as server
|
||||
|
||||
'''
|
||||
Test the different server responses
|
||||
'''
|
||||
|
||||
|
||||
def test_list_apis():
|
||||
backend = server.create_backend_app("apigateway")
|
||||
test_client = backend.test_client()
|
||||
|
||||
res = test_client.get('/restapis')
|
||||
res.data.should.equal(b'{"item": []}')
|
||||
Loading…
Add table
Add a link
Reference in a new issue