Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #1545 from haidaraM/fixApiKeyIdentification

Browse Source 
Fix ApiGateway key identification
This commit is contained in:
Steve Pulec 2018-04-12 18:58:47 -04:00 committed by GitHub
commit 5f840aadcc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
moto/apigateway/models.py
View File

@ -563,17 +563,17 @@ class APIGatewayBackend(BaseBackend):
def create_apikey(self, payload):
key = ApiKey(**payload)
self.keys[key['value']] = key
self.keys[key['id']] = key
return key
def get_apikeys(self):
return list(self.keys.values())
def get_apikey(self, value):
return self.keys[value]
def get_apikey(self, api_key_id):
return self.keys[api_key_id]
def delete_apikey(self, value):
self.keys.pop(value)
def delete_apikey(self, api_key_id):
self.keys.pop(api_key_id)
return {}

8
tests/test_apigateway/test_apigateway.py
View File

@ -976,22 +976,22 @@ def test_api_keys():
apikey_name = 'TESTKEY1'
payload = {'value': apikey_value, 'name': apikey_name}
response = client.create_api_key(**payload)
apikey = client.get_api_key(apiKey=payload['value'])
apikey = client.get_api_key(apiKey=response['id'])
apikey['name'].should.equal(apikey_name)
apikey['value'].should.equal(apikey_value)
apikey_name = 'TESTKEY2'
payload = {'name': apikey_name, 'generateDistinctId': True}
response = client.create_api_key(**payload)
apikey = client.get_api_key(apiKey=response['value'])
apikey_id = response['id']
apikey = client.get_api_key(apiKey=apikey_id)
apikey['name'].should.equal(apikey_name)
len(apikey['value']).should.equal(40)
apikey_value = apikey['value']
response = client.get_api_keys()
len(response['items']).should.equal(2)
client.delete_api_key(apiKey=apikey_value)
client.delete_api_key(apiKey=apikey_id)
response = client.get_api_keys()
len(response['items']).should.equal(1)