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

APIGateway - Generate API key value when no value provided (#1713)

Browse Source
This commit is contained in:
Aidan Fewster 2018-07-13 12:53:00 +01:00 committed by Terry Cain
parent c3b690114c
commit c20e8568e0
2 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
moto/apigateway/models.py
View File

@ -300,11 +300,7 @@ class ApiKey(BaseModel, dict):
generateDistinctId=False, value=None, stageKeys=None, customerId=None): generateDistinctId=False, value=None, stageKeys=None, customerId=None):
super(ApiKey, self).__init__() super(ApiKey, self).__init__()
self['id'] = create_id() self['id'] = create_id()
if generateDistinctId: self['value'] = value if value else ''.join(random.sample(string.ascii_letters + string.digits, 40))
# 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['name'] = name
self['customerId'] = customerId self['customerId'] = customerId
self['description'] = description self['description'] = description

2
tests/test_apigateway/test_apigateway.py
View File

@ -981,7 +981,7 @@ def test_api_keys():
apikey['value'].should.equal(apikey_value) apikey['value'].should.equal(apikey_value)
apikey_name = 'TESTKEY2' apikey_name = 'TESTKEY2'
payload = {'name': apikey_name, 'generateDistinctId': True} payload = {'name': apikey_name }
response = client.create_api_key(**payload) response = client.create_api_key(**payload)
apikey_id = response['id'] apikey_id = response['id']
apikey = client.get_api_key(apiKey=apikey_id) apikey = client.get_api_key(apiKey=apikey_id)