Basic support for AWS Gateway apikeys (#1494)

* Basic support for AWS Gateway apikeys

* Adds missing -- much needed tests

* Fixes issue with master merge

* Fixes linter

* Fixes tests
This commit is contained in:
Javier Buzzi 2018-03-21 17:11:49 +01:00 committed by Jack Danger
commit d55ff20685
4 changed files with 93 additions and 2 deletions

View file

@ -1,6 +1,8 @@
from __future__ import absolute_import
from __future__ import unicode_literals
import random
import string
import requests
import time
@ -291,6 +293,25 @@ class Stage(BaseModel, dict):
raise Exception('Patch operation "%s" not implemented' % op['op'])
class ApiKey(BaseModel, dict):
def __init__(self, name=None, description=None, enabled=True,
generateDistinctId=False, value=None, stageKeys=None, customerId=None):
super(ApiKey, self).__init__()
self['id'] = create_id()
if generateDistinctId:
# Best guess of what AWS does internally
self['value'] = ''.join(random.sample(string.ascii_letters + string.digits, 40))
else:
self['value'] = value
self['name'] = name
self['customerId'] = customerId
self['description'] = description
self['enabled'] = enabled
self['createdDate'] = self['lastUpdatedDate'] = int(time.time())
self['stageKeys'] = stageKeys
class RestAPI(BaseModel):
def __init__(self, id, region_name, name, description):
@ -386,6 +407,7 @@ class APIGatewayBackend(BaseBackend):
def __init__(self, region_name):
super(APIGatewayBackend, self).__init__()
self.apis = {}
self.keys = {}
self.region_name = region_name
def reset(self):
@ -539,6 +561,21 @@ class APIGatewayBackend(BaseBackend):
api = self.get_rest_api(function_id)
return api.delete_deployment(deployment_id)
def create_apikey(self, payload):
key = ApiKey(**payload)
self.keys[key['value']] = key
return key
def get_apikeys(self):
return list(self.keys.values())
def get_apikey(self, value):
return self.keys[value]
def delete_apikey(self, value):
self.keys.pop(value)
return {}
apigateway_backends = {}
# Not available in boto yet