224 lines
7.3 KiB
Python
224 lines
7.3 KiB
Python
# (c) 2017 Red Hat Inc.
|
|
#
|
|
# This file is part of Ansible
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
|
|
# Make coding more python3-ish
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
import pytest
|
|
|
|
from units.utils.amazon_placebo_fixtures import placeboify, maybe_sleep
|
|
from ansible.modules.cloud.amazon import cloudformation as cfn_module
|
|
|
|
basic_yaml_tpl = """
|
|
---
|
|
AWSTemplateFormatVersion: '2010-09-09'
|
|
Description: 'Basic template that creates an S3 bucket'
|
|
Resources:
|
|
MyBucket:
|
|
Type: "AWS::S3::Bucket"
|
|
Outputs:
|
|
TheName:
|
|
Value:
|
|
!Ref MyBucket
|
|
"""
|
|
|
|
bad_json_tpl = """{
|
|
"AWSTemplateFormatVersion": "2010-09-09",
|
|
"Description": "Broken template, no comma here ->"
|
|
"Resources": {
|
|
"MyBucket": {
|
|
"Type": "AWS::S3::Bucket"
|
|
}
|
|
}
|
|
}"""
|
|
|
|
failing_yaml_tpl = """
|
|
---
|
|
AWSTemplateFormatVersion: 2010-09-09
|
|
Resources:
|
|
ECRRepo:
|
|
Type: AWS::ECR::Repository
|
|
Properties:
|
|
RepositoryPolicyText:
|
|
Version: 3000-10-17 # <--- invalid version
|
|
Statement:
|
|
- Effect: Allow
|
|
Action:
|
|
- 'ecr:*'
|
|
Principal:
|
|
AWS: !Sub arn:${AWS::Partition}:iam::${AWS::AccountId}:root
|
|
"""
|
|
|
|
default_events_limit = 10
|
|
|
|
|
|
class FakeModule(object):
|
|
def __init__(self, **kwargs):
|
|
self.params = kwargs
|
|
|
|
def fail_json(self, *args, **kwargs):
|
|
self.exit_args = args
|
|
self.exit_kwargs = kwargs
|
|
raise Exception('FAIL')
|
|
|
|
def exit_json(self, *args, **kwargs):
|
|
self.exit_args = args
|
|
self.exit_kwargs = kwargs
|
|
raise Exception('EXIT')
|
|
|
|
|
|
def test_invalid_template_json(placeboify):
|
|
connection = placeboify.client('cloudformation')
|
|
params = {
|
|
'StackName': 'ansible-test-wrong-json',
|
|
'TemplateBody': bad_json_tpl,
|
|
}
|
|
m = FakeModule(disable_rollback=False)
|
|
with pytest.raises(Exception) as exc_info:
|
|
cfn_module.create_stack(m, params, connection, default_events_limit)
|
|
pytest.fail('Expected malformed JSON to have caused the call to fail')
|
|
|
|
assert exc_info.match('FAIL')
|
|
assert "ValidationError" in m.exit_kwargs['msg']
|
|
|
|
|
|
def test_client_request_token_s3_stack(maybe_sleep, placeboify):
|
|
connection = placeboify.client('cloudformation')
|
|
params = {
|
|
'StackName': 'ansible-test-client-request-token-yaml',
|
|
'TemplateBody': basic_yaml_tpl,
|
|
'ClientRequestToken': '3faf3fb5-b289-41fc-b940-44151828f6cf',
|
|
}
|
|
m = FakeModule(disable_rollback=False)
|
|
result = cfn_module.create_stack(m, params, connection, default_events_limit)
|
|
assert result['changed']
|
|
assert len(result['events']) > 1
|
|
# require that the final recorded stack state was CREATE_COMPLETE
|
|
# events are retrieved newest-first, so 0 is the latest
|
|
assert 'CREATE_COMPLETE' in result['events'][0]
|
|
connection.delete_stack(StackName='ansible-test-client-request-token-yaml')
|
|
|
|
|
|
def test_basic_s3_stack(maybe_sleep, placeboify):
|
|
connection = placeboify.client('cloudformation')
|
|
params = {
|
|
'StackName': 'ansible-test-basic-yaml',
|
|
'TemplateBody': basic_yaml_tpl
|
|
}
|
|
m = FakeModule(disable_rollback=False)
|
|
result = cfn_module.create_stack(m, params, connection, default_events_limit)
|
|
assert result['changed']
|
|
assert len(result['events']) > 1
|
|
# require that the final recorded stack state was CREATE_COMPLETE
|
|
# events are retrieved newest-first, so 0 is the latest
|
|
assert 'CREATE_COMPLETE' in result['events'][0]
|
|
connection.delete_stack(StackName='ansible-test-basic-yaml')
|
|
|
|
|
|
def test_delete_nonexistent_stack(maybe_sleep, placeboify):
|
|
connection = placeboify.client('cloudformation')
|
|
result = cfn_module.stack_operation(connection, 'ansible-test-nonexist', 'DELETE', default_events_limit)
|
|
assert result['changed']
|
|
assert 'Stack does not exist.' in result['log']
|
|
|
|
|
|
def test_get_nonexistent_stack(placeboify):
|
|
connection = placeboify.client('cloudformation')
|
|
assert cfn_module.get_stack_facts(connection, 'ansible-test-nonexist') is None
|
|
|
|
|
|
def test_missing_template_body():
|
|
m = FakeModule()
|
|
with pytest.raises(Exception) as exc_info:
|
|
cfn_module.create_stack(
|
|
module=m,
|
|
stack_params={},
|
|
cfn=None,
|
|
events_limit=default_events_limit
|
|
)
|
|
pytest.fail('Expected module to have failed with no template')
|
|
|
|
assert exc_info.match('FAIL')
|
|
assert not m.exit_args
|
|
assert "Either 'template', 'template_body' or 'template_url' is required when the stack does not exist." == m.exit_kwargs['msg']
|
|
|
|
|
|
def test_disable_rollback_and_on_failure_defined():
|
|
m = FakeModule(
|
|
on_create_failure='DELETE',
|
|
disable_rollback=True,
|
|
)
|
|
with pytest.raises(Exception) as exc_info:
|
|
cfn_module.create_stack(
|
|
module=m,
|
|
stack_params={'TemplateBody': ''},
|
|
cfn=None,
|
|
events_limit=default_events_limit
|
|
)
|
|
pytest.fail('Expected module to fail with both on_create_failure and disable_rollback defined')
|
|
|
|
assert exc_info.match('FAIL')
|
|
assert not m.exit_args
|
|
assert "You can specify either 'on_create_failure' or 'disable_rollback', but not both." == m.exit_kwargs['msg']
|
|
|
|
|
|
def test_on_create_failure_delete(maybe_sleep, placeboify):
|
|
m = FakeModule(
|
|
on_create_failure='DELETE',
|
|
disable_rollback=False,
|
|
)
|
|
connection = placeboify.client('cloudformation')
|
|
params = {
|
|
'StackName': 'ansible-test-on-create-failure-delete',
|
|
'TemplateBody': failing_yaml_tpl
|
|
}
|
|
result = cfn_module.create_stack(m, params, connection, default_events_limit)
|
|
assert result['changed']
|
|
assert result['failed']
|
|
assert len(result['events']) > 1
|
|
# require that the final recorded stack state was DELETE_COMPLETE
|
|
# events are retrieved newest-first, so 0 is the latest
|
|
assert 'DELETE_COMPLETE' in result['events'][0]
|
|
|
|
|
|
def test_on_create_failure_rollback(maybe_sleep, placeboify):
|
|
m = FakeModule(
|
|
on_create_failure='ROLLBACK',
|
|
disable_rollback=False,
|
|
)
|
|
connection = placeboify.client('cloudformation')
|
|
params = {
|
|
'StackName': 'ansible-test-on-create-failure-rollback',
|
|
'TemplateBody': failing_yaml_tpl
|
|
}
|
|
result = cfn_module.create_stack(m, params, connection, default_events_limit)
|
|
assert result['changed']
|
|
assert result['failed']
|
|
assert len(result['events']) > 1
|
|
# require that the final recorded stack state was ROLLBACK_COMPLETE
|
|
# events are retrieved newest-first, so 0 is the latest
|
|
assert 'ROLLBACK_COMPLETE' in result['events'][0]
|
|
connection.delete_stack(StackName=params['StackName'])
|
|
|
|
|
|
def test_on_create_failure_do_nothing(maybe_sleep, placeboify):
|
|
m = FakeModule(
|
|
on_create_failure='DO_NOTHING',
|
|
disable_rollback=False,
|
|
)
|
|
connection = placeboify.client('cloudformation')
|
|
params = {
|
|
'StackName': 'ansible-test-on-create-failure-do-nothing',
|
|
'TemplateBody': failing_yaml_tpl
|
|
}
|
|
result = cfn_module.create_stack(m, params, connection, default_events_limit)
|
|
assert result['changed']
|
|
assert result['failed']
|
|
assert len(result['events']) > 1
|
|
# require that the final recorded stack state was CREATE_FAILED
|
|
# events are retrieved newest-first, so 0 is the latest
|
|
assert 'CREATE_FAILED' in result['events'][0]
|
|
connection.delete_stack(StackName=params['StackName'])
|