116 lines
3.7 KiB
Python
116 lines
3.7 KiB
Python
|
from collections import OrderedDict
|
||
|
|
import json
|
||
|
|
import yaml
|
||
|
|
import os
|
||
|
|
import boto3
|
||
|
|
from nose.tools import raises
|
||
|
|
import botocore
|
||
|
|
|
||
|
|
|
||
|
|
from moto.cloudformation.exceptions import ValidationError
|
||
|
|
from moto.cloudformation.models import FakeStack
|
||
|
|
from moto.cloudformation.parsing import resource_class_from_type, parse_condition, Export
|
||
|
|
from moto.sqs.models import Queue
|
||
|
|
from moto.s3.models import FakeBucket
|
||
|
|
from moto.cloudformation.utils import yaml_tag_constructor
|
||
|
|
from boto.cloudformation.stack import Output
|
||
|
|
from moto import mock_cloudformation, mock_s3, mock_sqs, mock_ec2
|
||
|
|
|
||
|
|
json_template = {
|
||
|
|
"AWSTemplateFormatVersion": "2010-09-09",
|
||
|
|
"Description": "Stack 1",
|
||
|
|
"Resources": {
|
||
|
|
"EC2Instance1": {
|
||
|
|
"Type": "AWS::EC2::Instance",
|
||
|
|
"Properties": {
|
||
|
|
"ImageId": "ami-d3adb33f",
|
||
|
|
"KeyName": "dummy",
|
||
|
|
"InstanceType": "t2.micro",
|
||
|
|
"Tags": [
|
||
|
|
{
|
||
|
|
"Key": "Description",
|
||
|
|
"Value": "Test tag"
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"Key": "Name",
|
||
|
|
"Value": "Name tag for tests"
|
||
|
|
}
|
||
|
|
]
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
# One resource is required
|
||
|
|
json_bad_template = {
|
||
|
|
"AWSTemplateFormatVersion": "2010-09-09",
|
||
|
|
"Description": "Stack 1"
|
||
|
|
}
|
||
|
|
|
||
|
|
dummy_template_json = json.dumps(json_template)
|
||
|
|
dummy_bad_template_json = json.dumps(json_bad_template)
|
||
|
|
|
||
|
|
|
||
|
|
@mock_cloudformation
|
||
|
|
def test_boto3_json_validate_successful():
|
||
|
|
cf_conn = boto3.client('cloudformation', region_name='us-east-1')
|
||
|
|
response = cf_conn.validate_template(
|
||
|
|
TemplateBody=dummy_template_json,
|
||
|
|
)
|
||
|
|
assert response['Description'] == "Stack 1"
|
||
|
|
assert response['Parameters'] == []
|
||
|
|
assert response['ResponseMetadata']['HTTPStatusCode'] == 200
|
||
|
|
|
||
|
|
@mock_cloudformation
|
||
|
|
def test_boto3_json_invalid_missing_resource():
|
||
|
|
cf_conn = boto3.client('cloudformation', region_name='us-east-1')
|
||
|
|
try:
|
||
|
|
cf_conn.validate_template(
|
||
|
|
TemplateBody=dummy_bad_template_json,
|
||
|
|
)
|
||
|
|
assert False
|
||
|
|
except botocore.exceptions.ClientError as e:
|
||
|
|
assert str(e) == 'An error occurred (ValidationError) when calling the ValidateTemplate operation: Stack' \
|
||
|
|
' with id Missing top level item Resources to file module does not exist'
|
||
|
|
assert True
|
||
|
|
|
||
|
|
|
||
|
|
yaml_template = """
|
||
|
|
AWSTemplateFormatVersion: '2010-09-09'
|
||
|
|
Description: Simple CloudFormation Test Template
|
||
|
|
Resources:
|
||
|
|
S3Bucket:
|
||
|
|
Type: AWS::S3::Bucket
|
||
|
|
Properties:
|
||
|
|
AccessControl: PublicRead
|
||
|
|
BucketName: cf-test-bucket-1
|
||
|
|
"""
|
||
|
|
|
||
|
|
yaml_bad_template = """
|
||
|
|
AWSTemplateFormatVersion: '2010-09-09'
|
||
|
|
Description: Simple CloudFormation Test Template
|
||
|
|
"""
|
||
|
|
|
||
|
|
@mock_cloudformation
|
||
|
|
def test_boto3_yaml_validate_successful():
|
||
|
|
cf_conn = boto3.client('cloudformation', region_name='us-east-1')
|
||
|
|
response = cf_conn.validate_template(
|
||
|
|
TemplateBody=yaml_template,
|
||
|
|
)
|
||
|
|
assert response['Description'] == "Simple CloudFormation Test Template"
|
||
|
|
assert response['Parameters'] == []
|
||
|
|
assert response['ResponseMetadata']['HTTPStatusCode'] == 200
|
||
|
|
|
||
|
|
@mock_cloudformation
|
||
|
|
def test_boto3_yaml_invalid_missing_resource():
|
||
|
|
cf_conn = boto3.client('cloudformation', region_name='us-east-1')
|
||
|
|
try:
|
||
|
|
cf_conn.validate_template(
|
||
|
|
TemplateBody=yaml_bad_template,
|
||
|
|
)
|
||
|
|
assert False
|
||
|
|
except botocore.exceptions.ClientError as e:
|
||
|
|
assert str(e) == 'An error occurred (ValidationError) when calling the ValidateTemplate operation: Stack' \
|
||
|
|
' with id Missing top level item Resources to file module does not exist'
|
||
|
|
assert True
|