mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 17:09:03 +00:00
1559 lines
83 KiB
Python
1559 lines
83 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Import Python libs
|
|
from __future__ import absolute_import
|
|
import logging
|
|
import os
|
|
import datetime
|
|
import random
|
|
import string
|
|
|
|
# Import Salt Testing libs
|
|
from tests.support.mixins import LoaderModuleMockMixin
|
|
from tests.support.unit import skipIf, TestCase
|
|
from tests.support.mock import NO_MOCK, NO_MOCK_REASON, MagicMock, patch
|
|
|
|
# Import Salt libs
|
|
import salt.config
|
|
import salt.loader
|
|
from salt.utils.versions import LooseVersion
|
|
|
|
# pylint: disable=import-error,no-name-in-module
|
|
from tests.unit.modules.test_boto_apigateway import BotoApiGatewayTestCaseMixin
|
|
|
|
# Import 3rd-party libs
|
|
import yaml
|
|
from salt.ext.six.moves import range
|
|
try:
|
|
import boto3
|
|
import botocore
|
|
from botocore.exceptions import ClientError
|
|
HAS_BOTO = True
|
|
except ImportError:
|
|
HAS_BOTO = False
|
|
|
|
|
|
# Import Salt Libs
|
|
import salt.states.boto_apigateway as boto_apigateway
|
|
|
|
# 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'
|
|
required_botocore_version = '1.4.49'
|
|
|
|
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),
|
|
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'9waiaz',
|
|
parentId=u'bgk0rk8rqb',
|
|
path=u'/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),
|
|
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),
|
|
methodSettings=dict(),
|
|
stageName='test',
|
|
variables=dict())
|
|
|
|
stage1_deployment1_vars_ret = dict(cacheClusterEnabled=False,
|
|
cacheClusterSize=0.5,
|
|
cacheClusterStatus='NOT_AVAILABLE',
|
|
createdDate=datetime.datetime(2015, 11, 17, 16, 33, 50),
|
|
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),
|
|
methodSettings=dict(),
|
|
stageName='test',
|
|
variables={'var1': 'val1'})
|
|
|
|
stage1_deployment2_ret = dict(cacheClusterEnabled=False,
|
|
cacheClusterSize=0.5,
|
|
cacheClusterStatus='NOT_AVAILABLE',
|
|
createdDate=datetime.datetime(2015, 11, 17, 16, 33, 50),
|
|
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),
|
|
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),
|
|
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),
|
|
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),
|
|
description=(u'{\n'
|
|
u' "api_name": "unit test api",\n'
|
|
u' "swagger_file": "temp-swagger-sample.yaml",\n'
|
|
u' "swagger_file_md5sum": "99cd44f491f25a8e8bb468b0a8e23a0a",\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),
|
|
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_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={})
|
|
|
|
throttle_rateLimit = 10.0
|
|
association_stage_1 = {'apiId': 'apiId1', 'stage': 'stage1'}
|
|
association_stage_2 = {'apiId': 'apiId1', 'stage': 'stage2'}
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
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
|
|
|
|
|
|
def _has_required_botocore():
|
|
'''
|
|
Returns True/False boolean depending on if botocore supports usage plan
|
|
'''
|
|
if not HAS_BOTO:
|
|
return False
|
|
elif LooseVersion(botocore.__version__) < LooseVersion(required_botocore_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, objtype, 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, LoaderModuleMockMixin):
|
|
conn = None
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.opts = salt.config.DEFAULT_MINION_OPTS
|
|
cls.opts['grains'] = salt.loader.grains(cls.opts)
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
del cls.opts
|
|
|
|
def setup_loader_modules(self):
|
|
context = {}
|
|
utils = salt.loader.utils(self.opts, whitelist=['boto', 'boto3'], context=context)
|
|
serializers = salt.loader.serializers(self.opts)
|
|
self.funcs = salt.loader.minion_mods(self.opts, context=context, utils=utils, whitelist=['boto_apigateway'])
|
|
self.salt_states = salt.loader.states(opts=self.opts, functions=self.funcs, utils=utils, whitelist=['boto_apigateway'], serializers=serializers)
|
|
return {
|
|
boto_apigateway: {
|
|
'__opts__': self.opts,
|
|
'__utils__': utils,
|
|
'__salt__': self.funcs,
|
|
'__states__': self.salt_states,
|
|
'__serializers__': serializers,
|
|
}
|
|
}
|
|
|
|
# Set up MagicMock to replace the boto3 session
|
|
def setUp(self):
|
|
self.addCleanup(delattr, self, 'funcs')
|
|
self.addCleanup(delattr, self, 'salt_states')
|
|
# connections keep getting cached from prior tests, can't find the
|
|
# correct context object to clear it. So randomize the cache key, to prevent any
|
|
# cache hits
|
|
conn_parameters['key'] = ''.join(random.choice(string.ascii_lowercase + string.digits) for _ in range(50))
|
|
|
|
patcher = patch('boto3.session.Session')
|
|
self.addCleanup(patcher.stop)
|
|
mock_session = patcher.start()
|
|
|
|
session_instance = mock_session.return_value
|
|
self.conn = MagicMock()
|
|
self.addCleanup(delattr, self, 'conn')
|
|
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 BotoApiGatewayTestCase(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 = self.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
|
|
self.conn.get_stage.return_value = stage1_deployment1_ret
|
|
self.conn.update_stage.side_effect = ClientError(error_content, 'update_stage should not be called')
|
|
result = {}
|
|
with TempSwaggerFile() as swagger_file:
|
|
result = self.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('abort'))
|
|
self.assertTrue(result.get('current'))
|
|
self.assertIs(result.get('result'), True)
|
|
self.assertNotIn('update_stage should not be called', result.get('comment', ''))
|
|
|
|
def test_present_when_stage_is_already_at_desired_deployment_and_needs_stage_variables_update(self):
|
|
'''
|
|
Tests scenario where the deployment is current except for the need to update stage variables
|
|
from {} to {'var1':'val1'}
|
|
'''
|
|
self.conn.get_rest_apis.return_value = apis_ret
|
|
self.conn.get_deployment.return_value = deployment1_ret
|
|
self.conn.get_stage.return_value = stage1_deployment1_ret
|
|
self.conn.update_stage.return_value = stage1_deployment1_vars_ret
|
|
result = {}
|
|
with TempSwaggerFile() as swagger_file:
|
|
result = self.salt_states['boto_apigateway.present'](
|
|
'api present',
|
|
'unit test api',
|
|
swagger_file,
|
|
'test',
|
|
False,
|
|
'arn:aws:iam::1234:role/apigatewayrole',
|
|
stage_variables={'var1': 'val1'},
|
|
**conn_parameters)
|
|
|
|
self.assertFalse(result.get('abort'))
|
|
self.assertTrue(result.get('current'))
|
|
self.assertIs(result.get('result'), True)
|
|
|
|
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 = self.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'))
|
|
self.assertIs(result.get('result'), True)
|
|
self.assertFalse(result.get('abort'))
|
|
self.assertTrue(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(self.funcs, {'boto_lambda.describe_function': MagicMock(return_value={'function': function_ret})}):
|
|
with TempSwaggerFile() as swagger_file:
|
|
result = self.salt_states['boto_apigateway.present'](
|
|
'api present',
|
|
'unit test api',
|
|
swagger_file,
|
|
'test',
|
|
False,
|
|
'arn:aws:iam::1234:role/apigatewayrole',
|
|
**conn_parameters)
|
|
|
|
self.assertIs(result.get('result'), True)
|
|
self.assertIs(result.get('abort'), 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 = self.salt_states['boto_apigateway.present'](
|
|
'api present',
|
|
'unit test api',
|
|
swagger_file,
|
|
'test',
|
|
False,
|
|
'arn:aws:iam::1234:role/apigatewayrole',
|
|
**conn_parameters)
|
|
|
|
self.assertIs(result.get('abort'), True)
|
|
self.assertIs(result.get('result'), False)
|
|
self.assertIn('create_rest_api', result.get('comment', ''))
|
|
|
|
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 = self.salt_states['boto_apigateway.present'](
|
|
'api present',
|
|
'unit test api',
|
|
swagger_file,
|
|
'test',
|
|
False,
|
|
'arn:aws:iam::1234:role/apigatewayrole',
|
|
**conn_parameters)
|
|
|
|
self.assertIs(result.get('abort'), True)
|
|
self.assertIs(result.get('result'), False)
|
|
self.assertIn('create_model', result.get('comment', ''))
|
|
|
|
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 = self.salt_states['boto_apigateway.present'](
|
|
'api present',
|
|
'unit test api',
|
|
swagger_file,
|
|
'test',
|
|
False,
|
|
'arn:aws:iam::1234:role/apigatewayrole',
|
|
**conn_parameters)
|
|
self.assertIs(result.get('abort'), True)
|
|
self.assertIs(result.get('result'), False)
|
|
self.assertIn('create_resource', result.get('comment', ''))
|
|
|
|
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(self.funcs, {'boto_lambda.describe_function': MagicMock(return_value={'function': function_ret})}):
|
|
with TempSwaggerFile() as swagger_file:
|
|
result = self.salt_states['boto_apigateway.present'](
|
|
'api present',
|
|
'unit test api',
|
|
swagger_file,
|
|
'test',
|
|
False,
|
|
'arn:aws:iam::1234:role/apigatewayrole',
|
|
**conn_parameters)
|
|
|
|
self.assertIs(result.get('abort'), True)
|
|
self.assertIs(result.get('result'), False)
|
|
self.assertIn('put_method', result.get('comment', ''))
|
|
|
|
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(self.funcs, {'boto_lambda.describe_function': MagicMock(return_value={'error': 'no such lambda'})}):
|
|
with TempSwaggerFile() as swagger_file:
|
|
result = self.salt_states['boto_apigateway.present'](
|
|
'api present',
|
|
'unit test api',
|
|
swagger_file,
|
|
'test',
|
|
False,
|
|
'arn:aws:iam::1234:role/apigatewayrole',
|
|
**conn_parameters)
|
|
|
|
self.assertIs(result.get('result'), False)
|
|
self.assertNotIn('put_integration should not be invoked', result.get('comment', ''))
|
|
self.assertIn('not find lambda function', result.get('comment', ''))
|
|
|
|
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(self.funcs, {'boto_lambda.describe_function': MagicMock(return_value={'function': function_ret})}):
|
|
with TempSwaggerFile() as swagger_file:
|
|
result = self.salt_states['boto_apigateway.present'](
|
|
'api present',
|
|
'unit test api',
|
|
swagger_file,
|
|
'test',
|
|
False,
|
|
'arn:aws:iam::1234:role/apigatewayrole',
|
|
**conn_parameters)
|
|
|
|
self.assertIs(result.get('abort'), True)
|
|
self.assertIs(result.get('result'), False)
|
|
self.assertIn('put_integration', result.get('comment', ''))
|
|
|
|
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(self.funcs, {'boto_lambda.describe_function': MagicMock(return_value={'function': function_ret})}):
|
|
with TempSwaggerFile() as swagger_file:
|
|
result = self.salt_states['boto_apigateway.present'](
|
|
'api present',
|
|
'unit test api',
|
|
swagger_file,
|
|
'test',
|
|
False,
|
|
'arn:aws:iam::1234:role/apigatewayrole',
|
|
**conn_parameters)
|
|
|
|
self.assertIs(result.get('abort'), True)
|
|
self.assertIs(result.get('result'), False)
|
|
self.assertIn('put_method_response', result.get('comment', ''))
|
|
|
|
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(self.funcs, {'boto_lambda.describe_function': MagicMock(return_value={'function': function_ret})}):
|
|
with TempSwaggerFile() as swagger_file:
|
|
result = self.salt_states['boto_apigateway.present'](
|
|
'api present',
|
|
'unit test api',
|
|
swagger_file,
|
|
'test',
|
|
False,
|
|
'arn:aws:iam::1234:role/apigatewayrole',
|
|
**conn_parameters)
|
|
|
|
self.assertIs(result.get('abort'), True)
|
|
self.assertIs(result.get('result'), False)
|
|
self.assertIn('put_integration_response', result.get('comment', ''))
|
|
|
|
def test_absent_when_rest_api_does_not_exist(self):
|
|
'''
|
|
Tests scenario where the given api_name does not exist, absent state should return True
|
|
with no changes.
|
|
'''
|
|
self.conn.get_rest_apis.return_value = apis_ret
|
|
self.conn.get_stage.side_effect = ClientError(error_content, 'get_stage should not be called')
|
|
|
|
result = self.salt_states['boto_apigateway.absent'](
|
|
'api present',
|
|
'no_such_rest_api',
|
|
'no_such_stage',
|
|
nuke_api=False,
|
|
**conn_parameters)
|
|
|
|
self.assertIs(result.get('result'), True)
|
|
self.assertNotIn('get_stage should not be called', result.get('comment', ''))
|
|
self.assertEqual(result.get('changes'), {})
|
|
|
|
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 = self.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 = self.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 = self.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 = self.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 no stages associated with same deployment,
|
|
the api would be deleted.
|
|
'''
|
|
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 = self.salt_states['boto_apigateway.absent'](
|
|
'api present',
|
|
'unit test api',
|
|
'test',
|
|
nuke_api=True,
|
|
**conn_parameters)
|
|
|
|
self.assertIs(result.get('result'), True)
|
|
self.assertIsNot(result.get('abort'), True)
|
|
self.assertIs(result.get('changes', {}).get('new', [{}])[0].get('delete_api', {}).get('deleted'), True)
|
|
|
|
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 = self.salt_states['boto_apigateway.absent'](
|
|
'api present',
|
|
'unit test api',
|
|
'test',
|
|
nuke_api=True,
|
|
**conn_parameters)
|
|
|
|
self.assertIs(result.get('result'), True)
|
|
self.assertIsNot(result.get('abort'), True)
|
|
|
|
|
|
@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(_has_required_botocore() is False,
|
|
'The botocore module must be greater than'
|
|
' or equal to version {0}'.format(required_botocore_version))
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class BotoApiGatewayUsagePlanTestCase(BotoApiGatewayStateTestCaseBase, BotoApiGatewayTestCaseMixin):
|
|
'''
|
|
TestCase for salt.modules.boto_apigateway state.module, usage_plans portion
|
|
'''
|
|
|
|
def test_usage_plan_present_if_describe_fails(self, *args):
|
|
'''
|
|
Tests correct error processing for describe_usage_plan failure
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'error': 'error'})}):
|
|
result = boto_apigateway.usage_plan_present('name', 'plan_name', **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'Failed to describe existing usage plans')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_present_if_there_is_no_such_plan_and_test_option_is_set(self, *args):
|
|
'''
|
|
TestCse for salt.modules.boto_apigateway state.module, checking that if __opts__['test'] is set
|
|
and usage plan does not exist, correct diagnostic will be returned
|
|
'''
|
|
with patch.dict(boto_apigateway.__opts__, {'test': True}):
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': []})}):
|
|
result = boto_apigateway.usage_plan_present('name', 'plan_name', **conn_parameters)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'a new usage plan plan_name would be created')
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], None)
|
|
|
|
def test_usage_plan_present_if_create_usage_plan_fails(self, *args):
|
|
'''
|
|
Tests behavior for the case when creating a new usage plan fails
|
|
'''
|
|
with patch.dict(boto_apigateway.__opts__, {'test': False}):
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': []}),
|
|
'boto_apigateway.create_usage_plan': MagicMock(return_value={'error': 'error'})}):
|
|
result = boto_apigateway.usage_plan_present('name', 'plan_name', **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'Failed to create a usage plan plan_name, error')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_present_if_plan_is_there_and_needs_no_updates(self, *args):
|
|
'''
|
|
Tests behavior for the case when plan is present and needs no updates
|
|
'''
|
|
with patch.dict(boto_apigateway.__opts__, {'test': False}):
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': [{
|
|
'id': 'planid',
|
|
'name': 'planname'
|
|
}]}),
|
|
'boto_apigateway.update_usage_plan': MagicMock()}):
|
|
result = boto_apigateway.usage_plan_present('name', 'plan_name', **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], True)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'usage plan plan_name is already in a correct state')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
self.assertTrue(boto_apigateway.__salt__['boto_apigateway.update_usage_plan'].call_count == 0)
|
|
|
|
def test_usage_plan_present_if_plan_is_there_and_needs_updates_but_test_is_set(self, *args):
|
|
'''
|
|
Tests behavior when usage plan needs to be updated by tests option is set
|
|
'''
|
|
with patch.dict(boto_apigateway.__opts__, {'test': True}):
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': [{
|
|
'id': 'planid',
|
|
'name': 'planname',
|
|
'throttle': {'rateLimit': 10.0}
|
|
}]}),
|
|
'boto_apigateway.update_usage_plan': MagicMock()}):
|
|
result = boto_apigateway.usage_plan_present('name', 'plan_name', **conn_parameters)
|
|
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'a new usage plan plan_name would be updated')
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], None)
|
|
self.assertTrue(boto_apigateway.__salt__['boto_apigateway.update_usage_plan'].call_count == 0)
|
|
|
|
def test_usage_plan_present_if_plan_is_there_and_needs_updates_but_update_fails(self, *args):
|
|
'''
|
|
Tests error processing for the case when updating an existing usage plan fails
|
|
'''
|
|
with patch.dict(boto_apigateway.__opts__, {'test': False}):
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': [{
|
|
'id': 'planid',
|
|
'name': 'planname',
|
|
'throttle': {'rateLimit': 10.0}
|
|
}]}),
|
|
'boto_apigateway.update_usage_plan': MagicMock(return_value={'error': 'error'})}):
|
|
result = boto_apigateway.usage_plan_present('name', 'plan_name', **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'Failed to update a usage plan plan_name, error')
|
|
|
|
def test_usage_plan_present_if_plan_has_been_created(self, *args):
|
|
'''
|
|
Tests successful case for creating a new usage plan
|
|
'''
|
|
with patch.dict(boto_apigateway.__opts__, {'test': False}):
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(side_effect=[{'plans': []}, {'plans': [{'id': 'id'}]}]),
|
|
'boto_apigateway.create_usage_plan': MagicMock(return_value={'created': True})}):
|
|
result = boto_apigateway.usage_plan_present('name', 'plan_name', **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], True)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'A new usage plan plan_name has been created')
|
|
self.assertEqual(result['changes']['old'], {'plan': None})
|
|
self.assertEqual(result['changes']['new'], {'plan': {'id': 'id'}})
|
|
|
|
def test_usage_plan_present_if_plan_has_been_updated(self, *args):
|
|
'''
|
|
Tests successful case for updating a usage plan
|
|
'''
|
|
with patch.dict(boto_apigateway.__opts__, {'test': False}):
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(side_effect=[{'plans': [{'id': 'id'}]},
|
|
{'plans': [{'id': 'id',
|
|
'throttle': {'rateLimit': throttle_rateLimit}}]}]),
|
|
'boto_apigateway.update_usage_plan': MagicMock(return_value={'updated': True})}):
|
|
result = boto_apigateway.usage_plan_present('name', 'plan_name', throttle={'rateLimit': throttle_rateLimit}, **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], True)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'usage plan plan_name has been updated')
|
|
self.assertEqual(result['changes']['old'], {'plan': {'id': 'id'}})
|
|
self.assertEqual(result['changes']['new'], {'plan': {'id': 'id', 'throttle': {'rateLimit': throttle_rateLimit}}})
|
|
|
|
def test_usage_plan_present_if_ValueError_is_raised(self, *args):
|
|
'''
|
|
Tests error processing for the case when ValueError is raised when creating a usage plan
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(side_effect=ValueError('error'))}):
|
|
result = boto_apigateway.usage_plan_present('name', 'plan_name', throttle={'rateLimit': throttle_rateLimit}, **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], "('error',)")
|
|
|
|
def test_usage_plan_present_if_IOError_is_raised(self, *args):
|
|
'''
|
|
Tests error processing for the case when IOError is raised when creating a usage plan
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(side_effect=IOError('error'))}):
|
|
result = boto_apigateway.usage_plan_present('name', 'plan_name', throttle={'rateLimit': throttle_rateLimit}, **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], "('error',)")
|
|
|
|
def test_usage_plan_absent_if_describe_fails(self, *args):
|
|
'''
|
|
Tests correct error processing for describe_usage_plan failure
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'error': 'error'})}):
|
|
result = {}
|
|
|
|
result = boto_apigateway.usage_plan_absent('name', 'plan_name', **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'Failed to describe existing usage plans')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_absent_if_plan_is_not_present(self, *args):
|
|
'''
|
|
Tests behavior for the case when the plan that needs to be absent does not exist
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': []})}):
|
|
result = {}
|
|
|
|
result = boto_apigateway.usage_plan_absent('name', 'plan_name', **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], True)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'Usage plan plan_name does not exist already')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_absent_if_plan_is_present_but_test_option_is_set(self, *args):
|
|
'''
|
|
Tests behavior for the case when usage plan needs to be deleted by tests option is set
|
|
'''
|
|
with patch.dict(boto_apigateway.__opts__, {'test': True}):
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': [{'id': 'id'}]})}):
|
|
result = {}
|
|
|
|
result = boto_apigateway.usage_plan_absent('name', 'plan_name', **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], None)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'Usage plan plan_name exists and would be deleted')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_absent_if_plan_is_present_but_delete_fails(self, *args):
|
|
'''
|
|
Tests correct error processing when deleting a usage plan fails
|
|
'''
|
|
with patch.dict(boto_apigateway.__opts__, {'test': False}):
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': [{'id': 'id'}]}),
|
|
'boto_apigateway.delete_usage_plan': MagicMock(return_value={'error': 'error'})}):
|
|
result = boto_apigateway.usage_plan_absent('name', 'plan_name', **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'Failed to delete usage plan plan_name, {\'error\': \'error\'}')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_absent_if_plan_has_been_deleted(self, *args):
|
|
'''
|
|
Tests successful case for deleting a usage plan
|
|
'''
|
|
with patch.dict(boto_apigateway.__opts__, {'test': False}):
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': [{'id': 'id'}]}),
|
|
'boto_apigateway.delete_usage_plan': MagicMock(return_value={'deleted': True})}):
|
|
result = boto_apigateway.usage_plan_absent('name', 'plan_name', **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], True)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'Usage plan plan_name has been deleted')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {'new': {'plan': None}, 'old': {'plan': {'id': 'id'}}})
|
|
|
|
def test_usage_plan_absent_if_ValueError_is_raised(self, *args):
|
|
'''
|
|
Tests correct error processing for the case when ValueError is raised when deleting a usage plan
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(side_effect=ValueError('error'))}):
|
|
result = boto_apigateway.usage_plan_absent('name', 'plan_name', **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], "('error',)")
|
|
|
|
def test_usage_plan_absent_if_IOError_is_raised(self, *args):
|
|
'''
|
|
Tests correct error processing for the case when IOError is raised when deleting a usage plan
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(side_effect=IOError('error'))}):
|
|
result = boto_apigateway.usage_plan_absent('name', 'plan_name', **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], "('error',)")
|
|
|
|
|
|
@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(_has_required_botocore() is False,
|
|
'The botocore module must be greater than'
|
|
' or equal to version {0}'.format(required_botocore_version))
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class BotoApiGatewayUsagePlanAssociationTestCase(BotoApiGatewayStateTestCaseBase, BotoApiGatewayTestCaseMixin):
|
|
'''
|
|
TestCase for salt.modules.boto_apigateway state.module, usage_plans_association portion
|
|
'''
|
|
|
|
def test_usage_plan_association_present_if_describe_fails(self, *args):
|
|
'''
|
|
Tests correct error processing for describe_usage_plan failure
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'error': 'error'})}):
|
|
result = boto_apigateway.usage_plan_association_present('name', 'plan_name', [association_stage_1], **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'Failed to describe existing usage plans')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_association_present_if_plan_is_not_present(self, *args):
|
|
'''
|
|
Tests correct error processing if a plan for which association has been requested is not present
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': []})}):
|
|
result = boto_apigateway.usage_plan_association_present('name', 'plan_name', [association_stage_1], **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'Usage plan plan_name does not exist')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_association_present_if_multiple_plans_with_the_same_name_exist(self, *args):
|
|
'''
|
|
Tests correct error processing for the case when multiple plans with the same name exist
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': [{'id': 'id1'},
|
|
{'id': 'id2'}]})}):
|
|
result = boto_apigateway.usage_plan_association_present('name', 'plan_name', [association_stage_1], **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'There are multiple usage plans with the same name - it is not supported')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_association_present_if_association_already_exists(self, *args):
|
|
'''
|
|
Tests the behavior for the case when requested association is already present
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': [{'id': 'id1',
|
|
'apiStages':
|
|
[association_stage_1]}]})}):
|
|
result = boto_apigateway.usage_plan_association_present('name', 'plan_name', [association_stage_1], **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], True)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'Usage plan is already asssociated to all api stages')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_association_present_if_update_fails(self, *args):
|
|
'''
|
|
Tests correct error processing for the case when adding associations fails
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': [{'id': 'id1',
|
|
'apiStages':
|
|
[association_stage_1]}]}),
|
|
'boto_apigateway.attach_usage_plan_to_apis': MagicMock(return_value={'error': 'error'})}):
|
|
result = boto_apigateway.usage_plan_association_present('name', 'plan_name', [association_stage_2], **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertTrue(result['comment'].startswith('Failed to associate a usage plan'))
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_association_present_success(self, *args):
|
|
'''
|
|
Tests successful case for adding usage plan associations to a given api stage
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': [{'id': 'id1',
|
|
'apiStages':
|
|
[association_stage_1]}]}),
|
|
'boto_apigateway.attach_usage_plan_to_apis': MagicMock(return_value={'result': {'apiStages': [association_stage_1,
|
|
association_stage_2]}})}):
|
|
result = boto_apigateway.usage_plan_association_present('name', 'plan_name', [association_stage_2], **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], True)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'successfully associated usage plan to apis')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {'new': [association_stage_1, association_stage_2], 'old': [association_stage_1]})
|
|
|
|
def test_usage_plan_association_present_if_value_error_is_thrown(self, *args):
|
|
'''
|
|
Tests correct error processing for the case when IOError is raised while trying to set usage plan associations
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(side_effect=ValueError('error'))}):
|
|
result = boto_apigateway.usage_plan_association_present('name', 'plan_name', [], **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], "('error',)")
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_association_present_if_io_error_is_thrown(self, *args):
|
|
'''
|
|
Tests correct error processing for the case when IOError is raised while trying to set usage plan associations
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(side_effect=IOError('error'))}):
|
|
result = boto_apigateway.usage_plan_association_present('name', 'plan_name', [], **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], "('error',)")
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_association_absent_if_describe_fails(self, *args):
|
|
'''
|
|
Tests correct error processing for describe_usage_plan failure
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'error': 'error'})}):
|
|
result = boto_apigateway.usage_plan_association_absent('name', 'plan_name', [association_stage_1], **conn_parameters)
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'Failed to describe existing usage plans')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_association_absent_if_plan_is_not_present(self, *args):
|
|
'''
|
|
Tests error processing for the case when plan for which associations need to be modified is not present
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': []})}):
|
|
result = boto_apigateway.usage_plan_association_absent('name', 'plan_name', [association_stage_1], **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'Usage plan plan_name does not exist')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_association_absent_if_multiple_plans_with_the_same_name_exist(self, *args):
|
|
'''
|
|
Tests the case when there are multiple plans with the same name but different Ids
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': [{'id': 'id1'},
|
|
{'id': 'id2'}]})}):
|
|
result = boto_apigateway.usage_plan_association_absent('name', 'plan_name', [association_stage_1], **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'There are multiple usage plans with the same name - it is not supported')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_association_absent_if_plan_has_no_associations(self, *args):
|
|
'''
|
|
Tests the case when the plan has no associations at all
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': [{'id': 'id1', 'apiStages': []}]})}):
|
|
result = boto_apigateway.usage_plan_association_absent('name', 'plan_name', [association_stage_1], **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], True)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'Usage plan plan_name has no associated stages already')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_association_absent_if_plan_has_no_specific_association(self, *args):
|
|
'''
|
|
Tests the case when requested association is not present already
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': [{'id': 'id1', 'apiStages': [association_stage_1]}]})}):
|
|
result = boto_apigateway.usage_plan_association_absent('name', 'plan_name', [association_stage_2], **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], True)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'Usage plan is already not asssociated to any api stages')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_association_absent_if_detaching_association_fails(self, *args):
|
|
'''
|
|
Tests correct error processing when detaching the usage plan from the api function is called
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': [{'id': 'id1', 'apiStages': [association_stage_1,
|
|
association_stage_2]}]}),
|
|
'boto_apigateway.detach_usage_plan_from_apis': MagicMock(return_value={'error': 'error'})}):
|
|
result = boto_apigateway.usage_plan_association_absent('name', 'plan_name', [association_stage_2], **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertTrue(result['comment'].startswith('Failed to disassociate a usage plan plan_name from the apis'))
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_association_absent_success(self, *args):
|
|
'''
|
|
Tests successful case of disaccosiation the usage plan from api stages
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(return_value={'plans': [{'id': 'id1', 'apiStages': [association_stage_1,
|
|
association_stage_2]}]}),
|
|
'boto_apigateway.detach_usage_plan_from_apis': MagicMock(return_value={'result': {'apiStages': [association_stage_1]}})}):
|
|
result = boto_apigateway.usage_plan_association_absent('name', 'plan_name', [association_stage_2], **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], True)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], 'successfully disassociated usage plan from apis')
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {'new': [association_stage_1], 'old': [association_stage_1, association_stage_2]})
|
|
|
|
def test_usage_plan_association_absent_if_ValueError_is_raised(self, *args):
|
|
'''
|
|
Tests correct error processing for the case where ValueError is raised while trying to remove plan associations
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(side_effect=ValueError('error'))}):
|
|
result = boto_apigateway.usage_plan_association_absent('name', 'plan_name', [association_stage_1], **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], "('error',)")
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|
|
|
|
def test_usage_plan_association_absent_if_IOError_is_raised(self, *args):
|
|
'''
|
|
Tests correct error processing for the case where IOError exception is raised while trying to remove plan associations
|
|
'''
|
|
with patch.dict(boto_apigateway.__salt__, {'boto_apigateway.describe_usage_plans': MagicMock(side_effect=IOError('error'))}):
|
|
result = boto_apigateway.usage_plan_association_absent('name', 'plan_name', [association_stage_1], **conn_parameters)
|
|
|
|
self.assertIn('result', result)
|
|
self.assertEqual(result['result'], False)
|
|
self.assertIn('comment', result)
|
|
self.assertEqual(result['comment'], "('error',)")
|
|
self.assertIn('changes', result)
|
|
self.assertEqual(result['changes'], {})
|