catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

APIGateway: Added API for usage plans

Browse source
This commit is contained in:
Aidan Fewster 2018-07-10 14:58:02 +01:00
commit 43e430560c
No known key found for this signature in database
GPG key ID: 73BB656D17B1F302
5 changed files with 134 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

45
tests/test_apigateway/test_server.py
View file

@ -1,5 +1,6 @@
from __future__ import unicode_literals
import sure # noqa
import json
import moto.server as server
@ -9,8 +10,50 @@ Test the different server responses
def test_list_apis():
backend = server.create_backend_app("apigateway")
backend = server.create_backend_app('apigateway')
test_client = backend.test_client()
res = test_client.get('/restapis')
res.data.should.equal(b'{"item": []}')
def test_usage_plans_apis():
backend = server.create_backend_app('apigateway')
test_client = backend.test_client()
'''
List usage plans (expect empty)
'''
res = test_client.get('/usageplans')
json.loads(res.data)["item"].should.have.length_of(0)
'''
Create usage plan
'''
res = test_client.post('/usageplans', data=json.dumps({'name': 'test'}))
created_plan = json.loads(res.data)
created_plan['name'].should.equal('test')
'''
List usage plans (expect 1 plan)
'''
res = test_client.get('/usageplans')
json.loads(res.data)["item"].should.have.length_of(1)
'''
Get single usage plan
'''
res = test_client.get('/usageplans/{0}'.format(created_plan["id"]))
fetched_plan = json.loads(res.data)
fetched_plan.should.equal(created_plan)
'''
Delete usage plan
'''
res = test_client.delete('/usageplans/{0}'.format(created_plan["id"]))
res.data.should.equal(b'{}')
'''
List usage plans (expect empty again)
'''
res = test_client.get('/usageplans')
json.loads(res.data)["item"].should.have.length_of(0)