mirror of
https://github.com/valitydev/salt.git
synced 2024-11-08 01:18:58 +00:00
unit tests for boto apigateway present/absent state.
This commit is contained in:
Winston Liu
parent
16a2439a47
commit
07d144eb6b
905
tests/unit/states/boto_apigateway_test.py
Normal file
905
tests/unit/states/boto_apigateway_test.py
Normal file
@ -0,0 +1,905 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
# Import Python libs
|
||||
from __future__ import absolute_import
|
||||
from distutils.version import LooseVersion # pylint: disable=import-error,no-name-in-module
|
||||
import os
|
||||
import datetime
|
||||
from dateutil.tz import tzlocal
|
||||
|
||||
# Import Salt Testing libs
|
||||
from salttesting.unit import skipIf, TestCase
|
||||
from salttesting.mock import NO_MOCK, NO_MOCK_REASON, patch
|
||||
from salttesting.helpers import ensure_in_syspath
|
||||
|
||||
ensure_in_syspath('../../')
|
||||
|
||||
# Import Salt libs
|
||||
import salt.config
|
||||
import salt.loader
|
||||
|
||||
# Import 3rd-party libs
|
||||
import logging
|
||||
from tempfile import NamedTemporaryFile
|
||||
import yaml
|
||||
|
||||
# Import Mock libraries
|
||||
from salttesting.mock import NO_MOCK, NO_MOCK_REASON, MagicMock, patch
|
||||
|
||||
# pylint: disable=import-error,no-name-in-module
|
||||
from unit.modules.boto_apigateway_test import BotoApiGatewayTestCaseMixin
|
||||
|
||||
# Import 3rd-party libs
|
||||
try:
|
||||
import boto3
|
||||
from botocore.exceptions import ClientError
|
||||
HAS_BOTO = True
|
||||
except ImportError:
|
||||
HAS_BOTO = False
|
||||
|
||||
# pylint: enable=import-error,no-name-in-module
|
||||
|
||||
# the boto_apigateway module relies on the connect_to_region() method
|
||||
# which was added in boto 2.8.0
|
||||
# https://github.com/boto/boto/commit/33ac26b416fbb48a60602542b4ce15dcc7029f12
|
||||
required_boto3_version = '1.2.1'
|
||||
|
||||
region = 'us-east-1'
|
||||
access_key = 'GKTADJGHEIQSXMKKRBJ08H'
|
||||
secret_key = 'askdjghsdfjkghWupUjasdflkdfklgjsdfjajkghs'
|
||||
conn_parameters = {'region': region, 'key': access_key, 'keyid': secret_key, 'profile': {}}
|
||||
error_message = 'An error occurred (101) when calling the {0} operation: Test-defined error'
|
||||
error_content = {
|
||||
'Error': {
|
||||
'Code': 101,
|
||||
'Message': "Test-defined error"
|
||||
}
|
||||
}
|
||||
|
||||
api_ret = dict(description=u'{\n "context": "See deployment or stage description",\n "provisioned_by": "Salt boto_apigateway.present State"\n}',
|
||||
createdDate=datetime.datetime(2015, 11, 17, 16, 33, 50, tzinfo=tzlocal()),
|
||||
id=u'vni0vq8wzi',
|
||||
name=u'unit test api')
|
||||
|
||||
no_apis_ret = {u'items': []}
|
||||
|
||||
apis_ret = {u'items': [api_ret]}
|
||||
|
||||
mock_model_ret = dict(contentType=u'application/json',
|
||||
description=u'mock model',
|
||||
id=u'123abc',
|
||||
name=u'mock model',
|
||||
schema=(u'{\n'
|
||||
u' "$schema": "http://json-schema.org/draft-04/schema#",\n'
|
||||
u' "properties": {\n'
|
||||
u' "field": {\n'
|
||||
u' "type": "string"\n'
|
||||
u' }\n'
|
||||
u' }\n'
|
||||
u'}'))
|
||||
|
||||
models_ret = {u'items': [dict(contentType=u'application/json',
|
||||
description=u'Error',
|
||||
id=u'50nw8r',
|
||||
name=u'Error',
|
||||
schema=(u'{\n'
|
||||
u' "$schema": "http://json-schema.org/draft-04/schema#",\n'
|
||||
u' "properties": {\n'
|
||||
u' "code": {\n'
|
||||
u' "format": "int32",\n'
|
||||
u' "type": "integer"\n'
|
||||
u' },\n'
|
||||
u' "fields": {\n'
|
||||
u' "type": "string"\n'
|
||||
u' },\n'
|
||||
u' "message": {\n'
|
||||
u' "type": "string"\n'
|
||||
u' }\n'
|
||||
u' },\n'
|
||||
u' "title": "Error Schema",\n'
|
||||
u' "type": "object"\n'
|
||||
u'}')),
|
||||
dict(contentType=u'application/json',
|
||||
description=u'User',
|
||||
id=u'terlnw',
|
||||
name=u'User',
|
||||
schema=(u'{\n'
|
||||
u' "$schema": "http://json-schema.org/draft-04/schema#",\n'
|
||||
u' "properties": {\n'
|
||||
u' "password": {\n'
|
||||
u' "description": "A password for the new user",\n'
|
||||
u' "type": "string"\n'
|
||||
u' },\n'
|
||||
u' "username": {\n'
|
||||
u' "description": "A unique username for the user",\n'
|
||||
u' "type": "string"\n'
|
||||
u' }\n'
|
||||
u' },\n'
|
||||
u' "title": "User Schema",\n'
|
||||
u' "type": "object"\n'
|
||||
u'}'))]}
|
||||
|
||||
root_resources_ret = {u'items': [dict(id=u'bgk0rk8rqb',
|
||||
path=u'/')]}
|
||||
|
||||
resources_ret = {u'items': [dict(id=u'bgk0rk8rqb',
|
||||
path=u'/'),
|
||||
dict(id=u'hrdwo2',
|
||||
parentId=u'bgk0rk8rqb',
|
||||
path=u'/api',
|
||||
pathPart=u'api'),
|
||||
dict(id=u'9waiaz',
|
||||
parentId=u'hrdwo2',
|
||||
path=u'/api/users',
|
||||
pathPart=u'users',
|
||||
resourceMethods={u'POST': {}})]}
|
||||
|
||||
no_resources_ret = {u'items': []}
|
||||
|
||||
stage1_deployment1_ret = dict(cacheClusterEnabled=False,
|
||||
cacheClusterSize=0.5,
|
||||
cacheClusterStatus='NOT_AVAILABLE',
|
||||
createdDate=datetime.datetime(2015, 11, 17, 16, 33, 50, tzinfo=tzlocal()),
|
||||
deploymentId=u'kobnrb',
|
||||
description=(u'{\n'
|
||||
u' "current_deployment_label": {\n'
|
||||
u' "api_name": "unit test api",\n'
|
||||
u' "swagger_file": "temp-swagger-sample.yaml",\n'
|
||||
u' "swagger_file_md5sum": "4fb17e43bab3a96e7f2410a1597cd0a5",\n'
|
||||
u' "swagger_info_object": {\n'
|
||||
u' "description": "salt boto apigateway unit test service",\n'
|
||||
u' "title": "salt boto apigateway unit test service",\n'
|
||||
u' "version": "0.0.0"\n'
|
||||
u' }\n'
|
||||
u' }\n'
|
||||
u'}'),
|
||||
lastUpdatedDate=datetime.datetime(2015, 11, 17, 16, 33, 50, tzinfo=tzlocal()),
|
||||
methodSettings=dict(),
|
||||
stageName='test',
|
||||
variables=dict())
|
||||
|
||||
stage1_deployment2_ret = dict(cacheClusterEnabled=False,
|
||||
cacheClusterSize=0.5,
|
||||
cacheClusterStatus='NOT_AVAILABLE',
|
||||
createdDate=datetime.datetime(2015, 11, 17, 16, 33, 50, tzinfo=tzlocal()),
|
||||
deploymentId=u'kobnrc',
|
||||
description=(u'{\n'
|
||||
u' "current_deployment_label": {\n'
|
||||
u' "api_name": "unit test api",\n'
|
||||
u' "swagger_file": "temp-swagger-sample.yaml",\n'
|
||||
u' "swagger_file_md5sum": "5fd538c4336ed5c54b4bf39ddf97c661",\n'
|
||||
u' "swagger_info_object": {\n'
|
||||
u' "description": "salt boto apigateway unit test service",\n'
|
||||
u' "title": "salt boto apigateway unit test service",\n'
|
||||
u' "version": "0.0.2"\n'
|
||||
u' }\n'
|
||||
u' }\n'
|
||||
u'}'),
|
||||
lastUpdatedDate=datetime.datetime(2015, 11, 17, 16, 33, 50, tzinfo=tzlocal()),
|
||||
methodSettings=dict(),
|
||||
stageName='test',
|
||||
variables=dict())
|
||||
|
||||
stage2_ret = dict(cacheClusterEnabled=False,
|
||||
cacheClusterSize=0.5,
|
||||
cacheClusterStatus='NOT_AVAILABLE',
|
||||
createdDate=datetime.datetime(2015, 11, 17, 16, 33, 50, tzinfo=tzlocal()),
|
||||
deploymentId=u'kobnrb',
|
||||
description=(u'{\n'
|
||||
u' "current_deployment_label": {\n'
|
||||
u' "api_name": "unit test api",\n'
|
||||
u' "swagger_file": "temp-swagger-sample.yaml",\n'
|
||||
u' "swagger_file_md5sum": "4fb17e43bab3a96e7f2410a1597cd0a5",\n'
|
||||
u' "swagger_info_object": {\n'
|
||||
u' "description": "salt boto apigateway unit test service",\n'
|
||||
u' "title": "salt boto apigateway unit test service",\n'
|
||||
u' "version": "0.0.0"\n'
|
||||
u' }\n'
|
||||
u' }\n'
|
||||
u'}'),
|
||||
lastUpdatedDate=datetime.datetime(2015, 11, 17, 16, 33, 50, tzinfo=tzlocal()),
|
||||
methodSettings=dict(),
|
||||
stageName='dev',
|
||||
variables=dict())
|
||||
|
||||
stages_stage2_ret = {u'item': [stage2_ret]}
|
||||
|
||||
no_stages_ret = {u'item': []}
|
||||
|
||||
deployment1_ret = dict(createdDate=datetime.datetime(2015, 11, 17, 16, 33, 50, tzinfo=tzlocal()),
|
||||
description=(u'{\n'
|
||||
u' "api_name": "unit test api",\n'
|
||||
u' "swagger_file": "temp-swagger-sample.yaml",\n'
|
||||
u' "swagger_file_md5sum": "4fb17e43bab3a96e7f2410a1597cd0a5",\n'
|
||||
u' "swagger_info_object": {\n'
|
||||
u' "description": "salt boto apigateway unit test service",\n'
|
||||
u' "title": "salt boto apigateway unit test service",\n'
|
||||
u' "version": "0.0.0"\n'
|
||||
u' }\n'
|
||||
u'}'),
|
||||
id=u'kobnrb')
|
||||
|
||||
deployment2_ret = dict(createdDate=datetime.datetime(2015, 11, 17, 16, 33, 50, tzinfo=tzlocal()),
|
||||
description=(u'{\n'
|
||||
u' "api_name": "unit test api",\n'
|
||||
u' "swagger_file": "temp-swagger-sample.yaml",\n'
|
||||
u' "swagger_file_md5sum": "5fd538c4336ed5c54b4bf39ddf97c661",\n'
|
||||
u' "swagger_info_object": {\n'
|
||||
u' "description": "salt boto apigateway unit test service",\n'
|
||||
u' "title": "salt boto apigateway unit test service",\n'
|
||||
u' "version": "0.0.2"\n'
|
||||
u' }\n'
|
||||
u'}'),
|
||||
id=u'kobnrc')
|
||||
|
||||
deployments_ret = {u'items': [deployment1_ret, deployment2_ret]}
|
||||
|
||||
function_ret = dict(FunctionName='unit_test_api_api_users_post',
|
||||
Runtime='python2.7',
|
||||
Role=None,
|
||||
Handler='handler',
|
||||
Description='abcdefg',
|
||||
Timeout=5,
|
||||
MemorySize=128,
|
||||
CodeSha256='abcdef',
|
||||
CodeSize=199,
|
||||
FunctionArn='arn:lambda:us-east-1:1234:Something',
|
||||
LastModified='yes')
|
||||
|
||||
method_integration_response_200_ret = dict(responseParameters={'method.response.header.Access-Control-Allow-Origin':'*'},
|
||||
responseTemplates={},
|
||||
selectionPattern='.*',
|
||||
statusCode='200')
|
||||
|
||||
method_integration_ret = dict(cacheKeyParameters={},
|
||||
cacheNamespace='9waiaz',
|
||||
credentials='arn:aws:iam::1234:role/apigatewayrole',
|
||||
httpMethod='POST',
|
||||
integrationResponses={'200': method_integration_response_200_ret},
|
||||
requestParameters={},
|
||||
requestTemplates={'application/json': ('#set($inputRoot = $input.path(\'$\'))'
|
||||
'{'
|
||||
'"header-params" : {#set ($map = $input.params().header)#foreach( $param in $map.entrySet() )"$param.key" : "$param.value" #if( $foreach.hasNext ), #end#end},'
|
||||
'"query-params" : {#set ($map = $input.params().querystring)#foreach( $param in $map.entrySet() )"$param.key" : "$param.value" #if( $foreach.hasNext ), #end#end},'
|
||||
'"path-params" : {#set ($map = $input.params().path)#foreach( $param in $map.entrySet() )"$param.key" : "$param.value" #if( $foreach.hasNext ), #end#end},'
|
||||
'"body-params" : $input.json(\'$\')'
|
||||
'}')},
|
||||
type='AWS',
|
||||
uri=('arn:aws:apigateway:us-west-2:'
|
||||
'lambda:path/2015-03-31/functions/arn:aws:lambda:us-west-2:1234567:'
|
||||
'function:unit_test_api_api_users_post/invocations'))
|
||||
|
||||
method_response_200_ret = dict(responseModels={'application/json': 'User'},
|
||||
responseParameters={'method.response.header.Access-Control-Allow-Origin': False},
|
||||
statusCode='200')
|
||||
|
||||
method_ret = dict(apiKeyRequired=False,
|
||||
authorizationType='None',
|
||||
httpMethod='POST',
|
||||
methodIntegration=method_integration_ret,
|
||||
methodResponses={'200': method_response_200_ret},
|
||||
requestModels={'application/json': 'User'},
|
||||
requestParameters={})
|
||||
|
||||
log = logging.getLogger(__name__)
|
||||
|
||||
opts = salt.config.DEFAULT_MINION_OPTS
|
||||
context = {}
|
||||
utils = salt.loader.utils(opts, whitelist=['boto3'], context=context)
|
||||
serializers = salt.loader.serializers(opts)
|
||||
funcs = salt.loader.minion_mods(opts, context=context, utils=utils, whitelist=['boto_apigateway'])
|
||||
salt_states = salt.loader.states(opts=opts, functions=funcs, utils=utils, whitelist=['boto_apigateway'], serializers=serializers)
|
||||
|
||||
|
||||
def _has_required_boto():
|
||||
'''
|
||||
Returns True/False boolean depending on if Boto is installed and correct
|
||||
version.
|
||||
'''
|
||||
if not HAS_BOTO:
|
||||
return False
|
||||
elif LooseVersion(boto3.__version__) < LooseVersion(required_boto3_version):
|
||||
return False
|
||||
else:
|
||||
return True
|
||||
|
||||
class TempSwaggerFile(object):
|
||||
_tmp_swagger_dict = {'info': {'version': '0.0.0',
|
||||
'description': 'salt boto apigateway unit test service',
|
||||
'title': 'salt boto apigateway unit test service'},
|
||||
'paths': {'/users': {'post': {'responses':
|
||||
{'200':
|
||||
{'headers': {'Access-Control-Allow-Origin': {'type': 'string'}},
|
||||
'description': 'The username of the new user',
|
||||
'schema': {'$ref': '#/definitions/User'}}},
|
||||
'parameters': [{'in': 'body',
|
||||
'description': 'New user details.',
|
||||
'name': 'NewUser',
|
||||
'schema': {'$ref': '#/definitions/User'}}],
|
||||
'produces': ['application/json'],
|
||||
'description': 'Creates a new user.\n',
|
||||
'tags': ['Auth'],
|
||||
'consumes': ['application/json'],
|
||||
'summary': 'Registers a new user'}}},
|
||||
'schemes': ['https'],
|
||||
'produces': ['application/json'],
|
||||
'basePath': '/api',
|
||||
'host': 'rm06h9oac4.execute-api.us-west-2.amazonaws.com',
|
||||
'definitions': {'User': {'properties':
|
||||
{'username': {'type': 'string',
|
||||
'description': 'A unique username for the user'},
|
||||
'password': {'type': 'string',
|
||||
'description': 'A password for the new user'}}},
|
||||
'Error': {'properties':
|
||||
{'fields': {'type': 'string'},
|
||||
'message': {'type': 'string'},
|
||||
'code': {'type': 'integer',
|
||||
'format': 'int32'}}}},
|
||||
'swagger': '2.0'}
|
||||
|
||||
def __enter__(self):
|
||||
self.swaggerfile = 'temp-swagger-sample.yaml'
|
||||
with salt.utils.fopen(self.swaggerfile, 'w') as f:
|
||||
f.write(yaml.dump(self.swaggerdict))
|
||||
return self.swaggerfile
|
||||
|
||||
def __exit__(self, type, value, traceback):
|
||||
os.remove(self.swaggerfile)
|
||||
|
||||
def __init__(self, create_invalid_file=False):
|
||||
if create_invalid_file:
|
||||
self.swaggerdict = TempSwaggerFile._tmp_swagger_dict.copy()
|
||||
# add an invalid top level key
|
||||
self.swaggerdict['invalid_key'] = 'invalid'
|
||||
# remove one of the required keys 'schemes'
|
||||
self.swaggerdict.pop('schemes', None)
|
||||
# set swagger version to an unsupported verison 3.0
|
||||
self.swaggerdict['swagger'] = '3.0'
|
||||
# missing info object
|
||||
self.swaggerdict.pop('info', None)
|
||||
else:
|
||||
self.swaggerdict = TempSwaggerFile._tmp_swagger_dict
|
||||
|
||||
class BotoApiGatewayStateTestCaseBase(TestCase):
|
||||
conn = None
|
||||
|
||||
# Set up MagicMock to replace the boto3 session
|
||||
def setUp(self):
|
||||
context.clear()
|
||||
|
||||
self.patcher = patch('boto3.session.Session')
|
||||
self.addCleanup(self.patcher.stop)
|
||||
mock_session = self.patcher.start()
|
||||
|
||||
session_instance = mock_session.return_value
|
||||
self.conn = MagicMock()
|
||||
session_instance.client.return_value = self.conn
|
||||
|
||||
|
||||
@skipIf(HAS_BOTO is False, 'The boto module must be installed.')
|
||||
@skipIf(_has_required_boto() is False, 'The boto3 module must be greater than'
|
||||
' or equal to version {0}'
|
||||
.format(required_boto3_version))
|
||||
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
||||
class BotoApiGatewayFunctionTestCase(BotoApiGatewayStateTestCaseBase, BotoApiGatewayTestCaseMixin):
|
||||
'''
|
||||
TestCase for salt.modules.boto_apigateway state.module
|
||||
'''
|
||||
|
||||
def test_present_when_swagger_file_is_invalid(self):
|
||||
'''
|
||||
Tests present when the swagger file is invalid.
|
||||
'''
|
||||
result = {}
|
||||
with TempSwaggerFile(create_invalid_file=True) as swagger_file:
|
||||
result = salt_states['boto_apigateway.present'](
|
||||
'api present',
|
||||
'unit test api',
|
||||
swagger_file,
|
||||
'test',
|
||||
False,
|
||||
'arn:aws:iam::1234:role/apigatewayrole',
|
||||
**conn_parameters)
|
||||
|
||||
self.assertFalse(result.get('result', True))
|
||||
|
||||
def test_present_when_stage_is_already_at_desired_deployment(self):
|
||||
'''
|
||||
Tests scenario where no action will be taken since we're already
|
||||
at desired state
|
||||
'''
|
||||
self.conn.get_rest_apis.return_value = apis_ret
|
||||
self.conn.get_deployment.return_value = deployment1_ret
|
||||
result = {}
|
||||
with TempSwaggerFile() as swagger_file:
|
||||
result = salt_states['boto_apigateway.present'](
|
||||
'api present',
|
||||
'unit test api',
|
||||
swagger_file,
|
||||
'test',
|
||||
False,
|
||||
'arn:aws:iam::1234:role/apigatewayrole',
|
||||
**conn_parameters)
|
||||
|
||||
self.assertTrue(result.get('abort', False) and result.get('result', False))
|
||||
|
||||
def test_present_when_stage_exists_and_is_to_associate_to_existing_deployment(self):
|
||||
'''
|
||||
Tests scenario where we merely reassociate a stage to a pre-existing
|
||||
deployments
|
||||
'''
|
||||
self.conn.get_rest_apis.return_value = apis_ret
|
||||
self.conn.get_deployment.return_value = deployment2_ret
|
||||
self.conn.get_deployments.return_value = deployments_ret
|
||||
self.conn.get_stage.return_value = stage1_deployment2_ret
|
||||
self.conn.update_stage.return_value = stage1_deployment1_ret
|
||||
|
||||
# should never get to the following calls
|
||||
self.conn.create_stage.side_effect = ClientError(error_content, 'create_stage')
|
||||
self.conn.create_deployment.side_effect = ClientError(error_content, 'create_deployment')
|
||||
|
||||
result = {}
|
||||
with TempSwaggerFile() as swagger_file:
|
||||
result = salt_states['boto_apigateway.present'](
|
||||
'api present',
|
||||
'unit test api',
|
||||
swagger_file,
|
||||
'test',
|
||||
False,
|
||||
'arn:aws:iam::1234:role/apigatewayrole',
|
||||
**conn_parameters)
|
||||
|
||||
self.assertTrue(result.get('publish', False) and
|
||||
result.get('result', False) and
|
||||
not result.get('abort') and
|
||||
bool(result.get('changes', {}).get('new', [{}])[0]))
|
||||
|
||||
def test_present_when_stage_is_to_associate_to_new_deployment(self):
|
||||
'''
|
||||
Tests creation of a new api/model/resource given nothing has been created previously
|
||||
'''
|
||||
# no api existed
|
||||
self.conn.get_rest_apis.return_value = no_apis_ret
|
||||
# create top level api
|
||||
self.conn.create_rest_api.return_value = api_ret
|
||||
# no models existed in the newly created api
|
||||
self.conn.get_model.side_effect = ClientError(error_content, 'get_model')
|
||||
# create model return values
|
||||
self.conn.create_model.return_value = mock_model_ret
|
||||
# various paths/resources already created
|
||||
self.conn.get_resources.return_value = resources_ret
|
||||
# the following should never be called
|
||||
self.conn.create_resource.side_effect = ClientError(error_content, 'create_resource')
|
||||
|
||||
# create api method for POST
|
||||
self.conn.put_method.return_value = method_ret
|
||||
# create api method integration for POST
|
||||
self.conn.put_integration.return_value = method_integration_ret
|
||||
# create api method response for POST/200
|
||||
self.conn.put_method_response.return_value = method_response_200_ret
|
||||
# create api method integration response for POST
|
||||
self.conn.put_intgration_response.return_value = method_integration_response_200_ret
|
||||
|
||||
result = {}
|
||||
with patch.dict(funcs, {'boto_lambda.describe_function': MagicMock(return_value={'function': function_ret})}):
|
||||
with TempSwaggerFile() as swagger_file:
|
||||
result = salt_states['boto_apigateway.present'](
|
||||
'api present',
|
||||
'unit test api',
|
||||
swagger_file,
|
||||
'test',
|
||||
False,
|
||||
'arn:aws:iam::1234:role/apigatewayrole',
|
||||
**conn_parameters)
|
||||
|
||||
self.assertTrue(result.get('result', False) and
|
||||
result.get('abort') is None)
|
||||
|
||||
def test_present_when_stage_associating_to_new_deployment_errored_on_api_creation(self):
|
||||
'''
|
||||
Tests creation of a new api/model/resource given nothing has been created previously,
|
||||
and we failed on creating the top level api object.
|
||||
'''
|
||||
# no api existed
|
||||
self.conn.get_rest_apis.return_value = no_apis_ret
|
||||
# create top level api
|
||||
self.conn.create_rest_api.side_effect = ClientError(error_content, 'create_rest_api')
|
||||
|
||||
result = {}
|
||||
with TempSwaggerFile() as swagger_file:
|
||||
result = salt_states['boto_apigateway.present'](
|
||||
'api present',
|
||||
'unit test api',
|
||||
swagger_file,
|
||||
'test',
|
||||
False,
|
||||
'arn:aws:iam::1234:role/apigatewayrole',
|
||||
**conn_parameters)
|
||||
|
||||
self.assertTrue(result.get('abort', False) and
|
||||
not result.get('result', True) and
|
||||
result.get('comment', '').find('create_rest_api') != -1)
|
||||
|
||||
def test_present_when_stage_associating_to_new_deployment_errored_on_model_creation(self):
|
||||
'''
|
||||
Tests creation of a new api/model/resource given nothing has been created previously,
|
||||
and we failed on creating the models after successful creation of top level api object.
|
||||
'''
|
||||
# no api existed
|
||||
self.conn.get_rest_apis.return_value = no_apis_ret
|
||||
# create top level api
|
||||
self.conn.create_rest_api.return_value = api_ret
|
||||
# no models existed in the newly created api
|
||||
self.conn.get_model.side_effect = ClientError(error_content, 'get_model')
|
||||
# create model return error
|
||||
self.conn.create_model.side_effect = ClientError(error_content, 'create_model')
|
||||
|
||||
result = {}
|
||||
with TempSwaggerFile() as swagger_file:
|
||||
result = salt_states['boto_apigateway.present'](
|
||||
'api present',
|
||||
'unit test api',
|
||||
swagger_file,
|
||||
'test',
|
||||
False,
|
||||
'arn:aws:iam::1234:role/apigatewayrole',
|
||||
**conn_parameters)
|
||||
|
||||
self.assertTrue(result.get('abort', False) and
|
||||
not result.get('result', True) and
|
||||
result.get('comment', '').find('create_model') != -1)
|
||||
|
||||
def test_present_when_stage_associating_to_new_deployment_errored_on_resource_creation(self):
|
||||
'''
|
||||
Tests creation of a new api/model/resource given nothing has been created previously,
|
||||
and we failed on creating the resource (paths) after successful creation of top level api/model
|
||||
objects.
|
||||
'''
|
||||
# no api existed
|
||||
self.conn.get_rest_apis.return_value = no_apis_ret
|
||||
# create top level api
|
||||
self.conn.create_rest_api.return_value = api_ret
|
||||
# no models existed in the newly created api
|
||||
self.conn.get_model.side_effect = ClientError(error_content, 'get_model')
|
||||
# create model return values
|
||||
self.conn.create_model.return_value = mock_model_ret
|
||||
# get resources has nothing intiially except to the root path '/'
|
||||
self.conn.get_resources.return_value = root_resources_ret
|
||||
# create resources return error
|
||||
self.conn.create_resource.side_effect = ClientError(error_content, 'create_resource')
|
||||
result = {}
|
||||
with TempSwaggerFile() as swagger_file:
|
||||
result = salt_states['boto_apigateway.present'](
|
||||
'api present',
|
||||
'unit test api',
|
||||
swagger_file,
|
||||
'test',
|
||||
False,
|
||||
'arn:aws:iam::1234:role/apigatewayrole',
|
||||
**conn_parameters)
|
||||
self.assertTrue(result.get('abort', False) and
|
||||
not result.get('result', True) and
|
||||
result.get('comment', '').find('create_resource') != -1)
|
||||
|
||||
def test_present_when_stage_associating_to_new_deployment_errored_on_put_method(self):
|
||||
'''
|
||||
Tests creation of a new api/model/resource given nothing has been created previously,
|
||||
and we failed on adding a post method to the resource after successful creation of top level
|
||||
api, model, resource objects.
|
||||
'''
|
||||
# no api existed
|
||||
self.conn.get_rest_apis.return_value = no_apis_ret
|
||||
# create top level api
|
||||
self.conn.create_rest_api.return_value = api_ret
|
||||
# no models existed in the newly created api
|
||||
self.conn.get_model.side_effect = ClientError(error_content, 'get_model')
|
||||
# create model return values
|
||||
self.conn.create_model.return_value = mock_model_ret
|
||||
# various paths/resources already created
|
||||
self.conn.get_resources.return_value = resources_ret
|
||||
# the following should never be called
|
||||
self.conn.create_resource.side_effect = ClientError(error_content, 'create_resource')
|
||||
|
||||
# create api method for POST
|
||||
self.conn.put_method.side_effect = ClientError(error_content, 'put_method')
|
||||
|
||||
result = {}
|
||||
with patch.dict(funcs, {'boto_lambda.describe_function': MagicMock(return_value={'function': function_ret})}):
|
||||
with TempSwaggerFile() as swagger_file:
|
||||
result = salt_states['boto_apigateway.present'](
|
||||
'api present',
|
||||
'unit test api',
|
||||
swagger_file,
|
||||
'test',
|
||||
False,
|
||||
'arn:aws:iam::1234:role/apigatewayrole',
|
||||
**conn_parameters)
|
||||
|
||||
self.assertTrue(result.get('abort', False) and
|
||||
not result.get('result', True) and
|
||||
result.get('comment', '').find('put_method') != -1)
|
||||
|
||||
def test_present_when_stage_associating_to_new_deployment_errored_on_lambda_function_lookup(self):
|
||||
'''
|
||||
Tests creation of a new api/model/resource given nothing has been created previously,
|
||||
and we failed on adding a post method due to a lamda look up failure after successful
|
||||
creation of top level api, model, resource objects.
|
||||
'''
|
||||
# no api existed
|
||||
self.conn.get_rest_apis.return_value = no_apis_ret
|
||||
# create top level api
|
||||
self.conn.create_rest_api.return_value = api_ret
|
||||
# no models existed in the newly created api
|
||||
self.conn.get_model.side_effect = ClientError(error_content, 'get_model')
|
||||
# create model return values
|
||||
self.conn.create_model.return_value = mock_model_ret
|
||||
# various paths/resources already created
|
||||
self.conn.get_resources.return_value = resources_ret
|
||||
# the following should never be called
|
||||
self.conn.create_resource.side_effect = ClientError(error_content, 'create_resource')
|
||||
# create api method for POST
|
||||
self.conn.put_method.return_value = method_ret
|
||||
# create api method integration for POST
|
||||
self.conn.put_integration.side_effect = ClientError(error_content, 'put_integration should not be invoked')
|
||||
|
||||
result = {}
|
||||
with patch.dict(funcs, {'boto_lambda.describe_function': MagicMock(return_value={'error': 'no such lambda'})}):
|
||||
with TempSwaggerFile() as swagger_file:
|
||||
result = salt_states['boto_apigateway.present'](
|
||||
'api present',
|
||||
'unit test api',
|
||||
swagger_file,
|
||||
'test',
|
||||
False,
|
||||
'arn:aws:iam::1234:role/apigatewayrole',
|
||||
**conn_parameters)
|
||||
|
||||
self.assertTrue(not result.get('result', True) and
|
||||
result.get('comment', '').find('put_integration should not be invoked') == -1 and
|
||||
result.get('comment', '').find('not find lambda function') != -1)
|
||||
|
||||
def test_present_when_stage_associating_to_new_deployment_errored_on_put_integration(self):
|
||||
'''
|
||||
Tests creation of a new api/model/resource given nothing has been created previously,
|
||||
and we failed on adding an integration for the post method to the resource after
|
||||
successful creation of top level api, model, resource objects.
|
||||
'''
|
||||
# no api existed
|
||||
self.conn.get_rest_apis.return_value = no_apis_ret
|
||||
# create top level api
|
||||
self.conn.create_rest_api.return_value = api_ret
|
||||
# no models existed in the newly created api
|
||||
self.conn.get_model.side_effect = ClientError(error_content, 'get_model')
|
||||
# create model return values
|
||||
self.conn.create_model.return_value = mock_model_ret
|
||||
# various paths/resources already created
|
||||
self.conn.get_resources.return_value = resources_ret
|
||||
# the following should never be called
|
||||
self.conn.create_resource.side_effect = ClientError(error_content, 'create_resource')
|
||||
|
||||
# create api method for POST
|
||||
self.conn.put_method.return_value = method_ret
|
||||
# create api method integration for POST
|
||||
self.conn.put_integration.side_effect = ClientError(error_content, 'put_integration')
|
||||
|
||||
result = {}
|
||||
with patch.dict(funcs, {'boto_lambda.describe_function': MagicMock(return_value={'function': function_ret})}):
|
||||
with TempSwaggerFile() as swagger_file:
|
||||
result = salt_states['boto_apigateway.present'](
|
||||
'api present',
|
||||
'unit test api',
|
||||
swagger_file,
|
||||
'test',
|
||||
False,
|
||||
'arn:aws:iam::1234:role/apigatewayrole',
|
||||
**conn_parameters)
|
||||
|
||||
self.assertTrue(result.get('abort', False) and
|
||||
not result.get('result', True) and
|
||||
result.get('comment', '').find('put_integration') != -1)
|
||||
|
||||
def test_present_when_stage_associating_to_new_deployment_errored_on_put_method_response(self):
|
||||
'''
|
||||
Tests creation of a new api/model/resource given nothing has been created previously,
|
||||
and we failed on adding a method response for the post method to the resource after
|
||||
successful creation of top level api, model, resource objects.
|
||||
'''
|
||||
# no api existed
|
||||
self.conn.get_rest_apis.return_value = no_apis_ret
|
||||
# create top level api
|
||||
self.conn.create_rest_api.return_value = api_ret
|
||||
# no models existed in the newly created api
|
||||
self.conn.get_model.side_effect = ClientError(error_content, 'get_model')
|
||||
# create model return values
|
||||
self.conn.create_model.return_value = mock_model_ret
|
||||
# various paths/resources already created
|
||||
self.conn.get_resources.return_value = resources_ret
|
||||
# the following should never be called
|
||||
self.conn.create_resource.side_effect = ClientError(error_content, 'create_resource')
|
||||
|
||||
# create api method for POST
|
||||
self.conn.put_method.return_value = method_ret
|
||||
# create api method integration for POST
|
||||
self.conn.put_integration.return_value = method_integration_ret
|
||||
# create api method response for POST/200
|
||||
self.conn.put_method_response.side_effect = ClientError(error_content, 'put_method_response')
|
||||
|
||||
result = {}
|
||||
with patch.dict(funcs, {'boto_lambda.describe_function': MagicMock(return_value={'function': function_ret})}):
|
||||
with TempSwaggerFile() as swagger_file:
|
||||
result = salt_states['boto_apigateway.present'](
|
||||
'api present',
|
||||
'unit test api',
|
||||
swagger_file,
|
||||
'test',
|
||||
False,
|
||||
'arn:aws:iam::1234:role/apigatewayrole',
|
||||
**conn_parameters)
|
||||
|
||||
self.assertTrue(result.get('abort', False) and
|
||||
not result.get('result', True) and
|
||||
result.get('comment', '').find('put_method_response') != -1)
|
||||
|
||||
def test_present_when_stage_associating_to_new_deployment_errored_on_put_integration_response(self):
|
||||
'''
|
||||
Tests creation of a new api/model/resource given nothing has been created previously,
|
||||
and we failed on adding an integration response for the post method to the resource after
|
||||
successful creation of top level api, model, resource objects.
|
||||
'''
|
||||
# no api existed
|
||||
self.conn.get_rest_apis.return_value = no_apis_ret
|
||||
# create top level api
|
||||
self.conn.create_rest_api.return_value = api_ret
|
||||
# no models existed in the newly created api
|
||||
self.conn.get_model.side_effect = ClientError(error_content, 'get_model')
|
||||
# create model return values
|
||||
self.conn.create_model.return_value = mock_model_ret
|
||||
# various paths/resources already created
|
||||
self.conn.get_resources.return_value = resources_ret
|
||||
# the following should never be called
|
||||
self.conn.create_resource.side_effect = ClientError(error_content, 'create_resource')
|
||||
|
||||
# create api method for POST
|
||||
self.conn.put_method.return_value = method_ret
|
||||
# create api method integration for POST
|
||||
self.conn.put_integration.return_value = method_integration_ret
|
||||
# create api method response for POST/200
|
||||
self.conn.put_method_response.return_value = method_response_200_ret
|
||||
# create api method integration response for POST
|
||||
self.conn.put_integration_response.side_effect = ClientError(error_content, 'put_integration_response')
|
||||
|
||||
result = {}
|
||||
with patch.dict(funcs, {'boto_lambda.describe_function': MagicMock(return_value={'function': function_ret})}):
|
||||
with TempSwaggerFile() as swagger_file:
|
||||
result = salt_states['boto_apigateway.present'](
|
||||
'api present',
|
||||
'unit test api',
|
||||
swagger_file,
|
||||
'test',
|
||||
False,
|
||||
'arn:aws:iam::1234:role/apigatewayrole',
|
||||
**conn_parameters)
|
||||
|
||||
self.assertTrue(result.get('abort', False) and
|
||||
not result.get('result', True) and
|
||||
result.get('comment', '').find('put_integration_response') != -1)
|
||||
|
||||
def test_absent_when_stage_is_invalid(self):
|
||||
'''
|
||||
Tests scenario where the stagename doesn't exist
|
||||
'''
|
||||
self.conn.get_rest_apis.return_value = apis_ret
|
||||
self.conn.get_stage.return_value = stage1_deployment1_ret
|
||||
self.conn.delete_stage.side_effect = ClientError(error_content, 'delete_stage')
|
||||
|
||||
result = salt_states['boto_apigateway.absent'](
|
||||
'api present',
|
||||
'unit test api',
|
||||
'no_such_stage',
|
||||
nuke_api=False,
|
||||
**conn_parameters)
|
||||
|
||||
self.assertTrue(result.get('abort', False))
|
||||
|
||||
def test_absent_when_stage_is_valid_and_only_one_stage_is_associated_to_deployment(self):
|
||||
'''
|
||||
Tests scenario where the stagename exists
|
||||
'''
|
||||
self.conn.get_rest_apis.return_value = apis_ret
|
||||
self.conn.get_stage.return_value = stage1_deployment1_ret
|
||||
self.conn.delete_stage.return_value = {'ResponseMetadata': {'HTTPStatusCode': 200, 'RequestId': '2d31072c-9d15-11e5-9977-6d9fcfda9c0a'}}
|
||||
self.conn.get_stages.return_value = no_stages_ret
|
||||
self.conn.delete_deployment.return_value = {'ResponseMetadata': {'HTTPStatusCode': 200, 'RequestId': '2d31072c-9d15-11e5-9977-6d9fcfda9c0a'}}
|
||||
|
||||
result = salt_states['boto_apigateway.absent'](
|
||||
'api present',
|
||||
'unit test api',
|
||||
'test',
|
||||
nuke_api=False,
|
||||
**conn_parameters)
|
||||
|
||||
self.assertTrue(result.get('result', False))
|
||||
|
||||
def test_absent_when_stage_is_valid_and_two_stages_are_associated_to_deployment(self):
|
||||
'''
|
||||
Tests scenario where the stagename exists and there are two stages associated with same deployment
|
||||
'''
|
||||
self.conn.get_rest_apis.return_value = apis_ret
|
||||
self.conn.get_stage.return_value = stage1_deployment1_ret
|
||||
self.conn.delete_stage.return_value = {'ResponseMetadata': {'HTTPStatusCode': 200, 'RequestId': '2d31072c-9d15-11e5-9977-6d9fcfda9c0a'}}
|
||||
self.conn.get_stages.return_value = stages_stage2_ret
|
||||
|
||||
result = salt_states['boto_apigateway.absent'](
|
||||
'api present',
|
||||
'unit test api',
|
||||
'test',
|
||||
nuke_api=False,
|
||||
**conn_parameters)
|
||||
|
||||
self.assertTrue(result.get('result', False))
|
||||
|
||||
def test_absent_when_failing_to_delete_a_deployment_no_longer_associated_with_any_stages(self):
|
||||
'''
|
||||
Tests scenario where stagename exists and is deleted, but a failure occurs when trying to delete
|
||||
the deployment which is no longer associated to any other stages
|
||||
'''
|
||||
self.conn.get_rest_apis.return_value = apis_ret
|
||||
self.conn.get_stage.return_value = stage1_deployment1_ret
|
||||
self.conn.delete_stage.return_value = {'ResponseMetadata': {'HTTPStatusCode': 200, 'RequestId': '2d31072c-9d15-11e5-9977-6d9fcfda9c0a'}}
|
||||
self.conn.get_stages.return_value = no_stages_ret
|
||||
self.conn.delete_deployment.side_effect = ClientError(error_content, 'delete_deployment')
|
||||
|
||||
result = salt_states['boto_apigateway.absent'](
|
||||
'api present',
|
||||
'unit test api',
|
||||
'test',
|
||||
nuke_api=False,
|
||||
**conn_parameters)
|
||||
|
||||
self.assertTrue(result.get('abort', False))
|
||||
|
||||
|
||||
def test_absent_when_nuke_api_and_no_more_stages_deployments_remain(self):
|
||||
'''
|
||||
Tests scenario where the stagename exists and there are two stages associated with same deployment,
|
||||
though nuke_api is requested, due to remaining deployments, we will not call the delete_rest_api call.
|
||||
'''
|
||||
self.conn.get_rest_apis.return_value = apis_ret
|
||||
self.conn.get_stage.return_value = stage1_deployment1_ret
|
||||
self.conn.delete_stage.return_value = {'ResponseMetadata': {'HTTPStatusCode': 200, 'RequestId': '2d31072c-9d15-11e5-9977-6d9fcfda9c0a'}}
|
||||
self.conn.get_stages.return_value = no_stages_ret
|
||||
self.conn.get_deployments.return_value = deployments_ret
|
||||
self.conn.delete_rest_api.return_value = {'ResponseMetadata': {'HTTPStatusCode': 200, 'RequestId': '2d31072c-9d15-11e5-9977-6d9fcfda9c0a'}}
|
||||
|
||||
result = salt_states['boto_apigateway.absent'](
|
||||
'api present',
|
||||
'unit test api',
|
||||
'test',
|
||||
nuke_api=True,
|
||||
**conn_parameters)
|
||||
|
||||
self.assertTrue(result.get('result', False) and
|
||||
result.get('abort') != True and
|
||||
result.get('changes', {}).get('new', [{}])[0].get('delete_api', {}).get('deleted'))
|
||||
|
||||
def test_absent_when_nuke_api_and_other_stages_deployments_exist(self):
|
||||
'''
|
||||
Tests scenario where the stagename exists and there are two stages associated with same deployment,
|
||||
though nuke_api is requested, due to remaining deployments, we will not call the delete_rest_api call.
|
||||
'''
|
||||
self.conn.get_rest_apis.return_value = apis_ret
|
||||
self.conn.get_stage.return_value = stage1_deployment1_ret
|
||||
self.conn.delete_stage.return_value = {'ResponseMetadata': {'HTTPStatusCode': 200, 'RequestId': '2d31072c-9d15-11e5-9977-6d9fcfda9c0a'}}
|
||||
self.conn.get_stages.return_value = stages_stage2_ret
|
||||
self.conn.get_deployments.return_value = deployments_ret
|
||||
self.conn.delete_rest_api.side_effect = ClientError(error_content, 'unexpected_api_delete')
|
||||
|
||||
result = salt_states['boto_apigateway.absent'](
|
||||
'api present',
|
||||
'unit test api',
|
||||
'test',
|
||||
nuke_api=True,
|
||||
**conn_parameters)
|
||||
|
||||
self.assertTrue(result.get('result', False) and
|
||||
result.get('abort') != True)
|
||||
Loading…
Reference in New Issue
Block a user