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:
parent
cc14114afe
commit
d55ff20685
4 changed files with 93 additions and 2 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue