def test_invalid_template(self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Resources": {
"ServerlessFunction": {
"Type": "AWS::Serverless::Function",
"Properties": {
"Handler": "index.handler",
"CodeUri": "s3://lambda-code.zip",
"Timeout": 60
},
}
},
}
managed_policy_mock = Mock()
managed_policy_mock.load.return_value = {"PolicyName": "FakePolicy"}
validator = SamTemplateValidator(template,
managed_policy_mock,
region="us-east-1")
with self.assertRaises(InvalidSamDocumentException):
validator.is_valid()
def test_replace_local_codeuri_when_no_codeuri_given(self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Resources": {
"ServerlessApi": {
"Type": "AWS::Serverless::Api",
"Properties": {
"StageName": "Prod"
}
},
"ServerlessFunction": {
"Type": "AWS::Serverless::Function",
"Properties": {
"Handler": "index.handler",
"Runtime": "nodejs6.10",
"Timeout": 60
},
},
},
}
managed_policy_mock = Mock()
validator = SamTemplateValidator(template, managed_policy_mock)
validator._replace_local_codeuri()
# check template
tempalte_resources = validator.sam_template.get("Resources")
self.assertEqual(
tempalte_resources.get("ServerlessFunction").get("Properties").get(
"CodeUri"), "s3://bucket/value")
def test_valid_template_with_DefinitionBody_for_api(self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Resources": {
"ServerlessApi": {
"Type": "AWS::Serverless::Api",
"Properties": {
"StageName": "Prod",
"DefinitionBody": {
"swagger": "2.0"
}
},
}
},
}
managed_policy_mock = Mock()
managed_policy_mock.load.return_value = {"PolicyName": "FakePolicy"}
validator = SamTemplateValidator(template,
managed_policy_mock,
region="us-east-1")
# Should not throw an exception
validator.is_valid()
def test_is_valid_raises_exception(self, sam_parser, sam_translator):
managed_policy_mock = Mock()
managed_policy_mock.load.return_value = {"policy": "SomePolicy"}
template = {"a": "b"}
parser = Mock()
sam_parser.Parser.return_value = parser
translate_mock = Mock()
translate_mock.translate.side_effect = InvalidDocumentException(
[Exception("message")])
sam_translator.return_value = translate_mock
validator = SamTemplateValidator(template, managed_policy_mock)
with self.assertRaises(InvalidSamDocumentException):
validator.is_valid()
sam_translator.assert_called_once_with(
managed_policy_map={"policy": "SomePolicy"},
sam_parser=parser,
plugins=[])
translate_mock.translate.assert_called_once_with(sam_template=template,
parameter_values={})
sam_parser.Parser.assert_called_once()
def test_DefinitionUri_does_not_get_added_to_template_when_DefinitionBody_given(
self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Resources": {
"ServerlessApi": {
"Type": "AWS::Serverless::Api",
"Properties": {
"StageName": "Prod",
"DefinitionBody": {
"swagger": {}
}
},
}
},
}
managed_policy_mock = Mock()
validator = SamTemplateValidator(template, managed_policy_mock)
validator._replace_local_codeuri()
tempalte_resources = validator.sam_template.get("Resources")
self.assertNotIn(
"DefinitionUri",
tempalte_resources.get("ServerlessApi").get("Properties"))
self.assertIn(
"DefinitionBody",
tempalte_resources.get("ServerlessApi").get("Properties"))
def test_DefinitionUri_does_not_get_added_to_template_when_DefinitionBody_given(self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Resources": {
"ServerlessApi": {
"Type": "AWS::Serverless::Api",
"Properties": {
"StageName": "Prod",
"DefinitionBody": {
"swagger": {}
}
}
}
}
}
managed_policy_mock = Mock()
validator = SamTemplateValidator(template, managed_policy_mock)
validator._replace_local_codeuri()
tempalte_resources = validator.sam_template.get("Resources")
self.assertNotIn("DefinitionUri", tempalte_resources.get("ServerlessApi").get("Properties"))
self.assertIn("DefinitionBody", tempalte_resources.get("ServerlessApi").get("Properties"))
def test_replace_local_codeuri_when_no_codeuri_given(self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Resources": {
"ServerlessApi": {
"Type": "AWS::Serverless::Api",
"Properties": {
"StageName": "Prod",
}
},
"ServerlessFunction": {
"Type": "AWS::Serverless::Function",
"Properties": {
"Handler": "index.handler",
"Runtime": "nodejs6.10",
"Timeout": 60
}
}
}
}
managed_policy_mock = Mock()
validator = SamTemplateValidator(template, managed_policy_mock)
validator._replace_local_codeuri()
# check template
tempalte_resources = validator.sam_template.get("Resources")
self.assertEquals(tempalte_resources.get("ServerlessApi").get("Properties").get("DefinitionUri"),
"s3://bucket/value")
self.assertEquals(tempalte_resources.get("ServerlessFunction").get("Properties").get("CodeUri"),
"s3://bucket/value")
def test_dont_replace_local_codeuri_when_no_codeuri_given_packagetype_image(
self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Resources": {
"ServerlessApi": {
"Type": "AWS::Serverless::Api",
"Properties": {
"StageName": "Prod"
}
},
"ServerlessFunction": {
"Type": "AWS::Serverless::Function",
"Properties": {
"PackageType": IMAGE,
"ImageUri": "myimage:latest",
"Timeout": 60
},
},
},
}
managed_policy_mock = Mock()
validator = SamTemplateValidator(template, managed_policy_mock)
validator._replace_local_codeuri()
# check template
template_resources = validator.sam_template.get("Resources")
self.assertEqual(
template_resources.get("ServerlessFunction").get("Properties").get(
"CodeUri", "NotPresent"), "NotPresent")
def test_update_to_s3_url_with_dict(self):
property_value = {"CodeUri": {"Bucket": "mybucket-name", "Key": "swagger", "Version": 121212}}
SamTemplateValidator._update_to_s3_uri("CodeUri", property_value)
self.assertEqual(
property_value.get("CodeUri"), {"Bucket": "mybucket-name", "Key": "swagger", "Version": 121212}
)
def test_valid_template_with_local_code_for_function(self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Resources": {
"ServerlessFunction": {
"Type": "AWS::Serverless::Function",
"Properties": {
"Handler": "index.handler",
"CodeUri": "./",
"Runtime": "nodejs6.10",
"Timeout": 60
},
}
},
}
managed_policy_mock = Mock()
managed_policy_mock.load.return_value = {"PolicyName": "FakePolicy"}
validator = SamTemplateValidator(template,
managed_policy_mock,
region="us-east-1")
# Should not throw an exception
validator.is_valid()
def test_is_valid_returns_true(self, sam_parser, sam_translator,
boto_session_patch):
managed_policy_mock = Mock()
managed_policy_mock.load.return_value = {"policy": "SomePolicy"}
template = {"a": "b"}
parser = Mock()
sam_parser.Parser.return_value = parser
boto_session_mock = Mock()
boto_session_patch.return_value = boto_session_mock
translate_mock = Mock()
translate_mock.translate.return_value = {"c": "d"}
sam_translator.return_value = translate_mock
validator = SamTemplateValidator(template,
managed_policy_mock,
profile="profile",
region="region")
# Should not throw an Exception
validator.is_valid()
boto_session_patch.assert_called_once_with(profile_name="profile",
region_name="region")
sam_translator.assert_called_once_with(
managed_policy_map={"policy": "SomePolicy"},
sam_parser=parser,
plugins=[],
boto_session=boto_session_mock)
translate_mock.translate.assert_called_once_with(sam_template=template,
parameter_values={})
sam_parser.Parser.assert_called_once()
def test_valid_template_with_s3_object_passed(self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Resources": {
"ServerlessApi": {
"Type": "AWS::Serverless::Api",
"Properties": {
"StageName": "Prod",
"DefinitionUri": {
"Bucket": "mybucket-name",
"Key": "swagger",
"Version": 121212
},
},
},
"ServerlessFunction": {
"Type": "AWS::Serverless::Function",
"Properties": {
"Handler": "index.handler",
"CodeUri": {
"Bucket": "mybucket-name",
"Key": "code.zip",
"Version": 121212
},
"Runtime": "nodejs6.10",
"Timeout": 60,
},
},
},
}
managed_policy_mock = Mock()
managed_policy_mock.load.return_value = {"PolicyName": "FakePolicy"}
validator = SamTemplateValidator(template, managed_policy_mock)
# Should not throw an exception
validator.is_valid()
# validate the CodeUri was not changed
self.assertEqual(
validator.sam_template.get("Resources").get("ServerlessApi").get(
"Properties").get("DefinitionUri"),
{
"Bucket": "mybucket-name",
"Key": "swagger",
"Version": 121212
},
)
self.assertEqual(
validator.sam_template.get("Resources").get(
"ServerlessFunction").get("Properties").get("CodeUri"),
{
"Bucket": "mybucket-name",
"Key": "code.zip",
"Version": 121212
},
)
def test_valid_api_request_model_template(self, template_path):
with open(template_path) as f:
template = yamlhelper.yaml_parse(f.read())
managed_policy_mock = Mock()
managed_policy_mock.load.return_value = {"PolicyName": "FakePolicy"}
validator = SamTemplateValidator(template, managed_policy_mock)
# Should not throw an exception
validator.is_valid()
def test_dont_replace_codeuri_when_global_code_uri_given__both_packagetype(
self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Globals": {
"Function": {
"CodeUri": "s3://globalcodeuri",
}
},
"Resources": {
"ServerlessApi": {
"Type": "AWS::Serverless::Api",
"Properties": {
"StageName": "Prod"
}
},
"ServerlessFunctionImage": {
"Type": "AWS::Serverless::Function",
"Properties": {
"PackageType": IMAGE,
"ImageUri": "myimage:latest",
"Timeout": 60
},
},
"ServerlessFunctionZip": {
"Type": "AWS::Serverless::Function",
"Properties": {
"Handler": "index.handler",
"Runtime": "nodejs6.10",
"Timeout": 60
},
},
},
}
managed_policy_mock = Mock()
validator = SamTemplateValidator(template, managed_policy_mock)
validator._replace_local_codeuri()
# check template
template_resources = validator.sam_template.get("Resources")
self.assertEqual(
template_resources.get("ServerlessFunctionImage").get(
"Properties").get("CodeUri", "NotPresent"),
"NotPresent",
)
# Globals not set since they cant apply to both Zip and Image based packagetypes.
self.assertEqual(
template_resources.get("ServerlessFunctionZip").get(
"Properties").get("CodeUri"), "s3://bucket/value")
def test_replace_local_codeuri_with_no_resources(self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Resources": {},
}
managed_policy_mock = Mock()
validator = SamTemplateValidator(template, managed_policy_mock)
validator._replace_local_codeuri()
# check template
self.assertEqual(validator.sam_template.get("Resources"), {})
def test_valid_template_with_local_code_for_layer_version(self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Resources": {
"ServerlessLayerVersion": {"Type": "AWS::Serverless::LayerVersion", "Properties": {"ContentUri": "./"}}
},
}
managed_policy_mock = Mock()
managed_policy_mock.load.return_value = {"PolicyName": "FakePolicy"}
validator = SamTemplateValidator(template, managed_policy_mock)
# Should not throw an exception
validator.is_valid()
def test_replace_local_codeuri_with_no_resources(self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Resources": {}
}
managed_policy_mock = Mock()
validator = SamTemplateValidator(template, managed_policy_mock)
validator._replace_local_codeuri()
# check template
self.assertEquals(validator.sam_template.get("Resources"), {})
def test_valid_template_with_s3_object_passed(self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Resources": {
"ServerlessApi": {
"Type": "AWS::Serverless::Api",
"Properties": {
"StageName": "Prod",
"DefinitionUri": {
"Bucket": "mybucket-name",
"Key": "swagger",
"Version": 121212
}
}
},
"ServerlessFunction": {
"Type": "AWS::Serverless::Function",
"Properties": {
"Handler": "index.handler",
"CodeUri": {
"Bucket": "mybucket-name",
"Key": "code.zip",
"Version": 121212
},
"Runtime": "nodejs6.10",
"Timeout": 60
}
}
}
}
managed_policy_mock = Mock()
managed_policy_mock.load.return_value = {"PolicyName": "FakePolicy"}
validator = SamTemplateValidator(template, managed_policy_mock)
# Should not throw an exception
validator.is_valid()
# validate the CodeUri was not changed
self.assertEquals(validator.sam_template.get("Resources").get("ServerlessApi").get("Properties").get("DefinitionUri"), {"Bucket": "mybucket-name", "Key": "swagger", "Version": 121212})
self.assertEquals(validator.sam_template.get("Resources").get("ServerlessFunction").get("Properties").get("CodeUri"), {"Bucket": "mybucket-name", "Key": "code.zip", "Version": 121212})
def test_init(self):
managed_policy_mock = Mock()
template = {"a": "b"}
validator = SamTemplateValidator(template, managed_policy_mock)
self.assertEqual(validator.managed_policy_loader, managed_policy_mock)
self.assertEqual(validator.sam_template, template)
# check to see if SamParser was created
self.assertIsNotNone(validator.sam_parser)
def test_valid_template_with_local_code_for_layer_version(self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Resources": {
"ServerlessLayerVersion": {
"Type": "AWS::Serverless::LayerVersion",
"Properties": {
"ContentUri": "./"
}
}
}
}
managed_policy_mock = Mock()
managed_policy_mock.load.return_value = {"PolicyName": "FakePolicy"}
validator = SamTemplateValidator(template, managed_policy_mock)
# Should not throw an exception
validator.is_valid()
def test_valid_template_with_DefinitionBody_for_api(self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Resources": {
"ServerlessApi": {
"Type": "AWS::Serverless::Api",
"Properties": {
"StageName": "Prod",
"DefinitionBody": {"swagger": "2.0"}
}
}
}
}
managed_policy_mock = Mock()
managed_policy_mock.load.return_value = {"PolicyName": "FakePolicy"}
validator = SamTemplateValidator(template, managed_policy_mock)
# Should not throw an exception
validator.is_valid()
def test_is_valid_returns_true(self, sam_parser, sam_translator):
managed_policy_mock = Mock()
managed_policy_mock.load.return_value = {"policy": "SomePolicy"}
template = {"a": "b"}
parser = Mock()
sam_parser.Parser.return_value = parser
translate_mock = Mock()
translate_mock.translate.return_value = {"c": "d"}
sam_translator.return_value = translate_mock
validator = SamTemplateValidator(template, managed_policy_mock)
# Should not throw an Exception
validator.is_valid()
sam_translator.assert_called_once_with(managed_policy_map={"policy": "SomePolicy"},
sam_parser=parser,
plugins=[])
translate_mock.translate.assert_called_once_with(sam_template=template, parameter_values={})
sam_parser.Parser.assert_called_once()
def test_is_valid_raises_exception(self, sam_parser, sam_translator):
managed_policy_mock = Mock()
managed_policy_mock.load.return_value = {"policy": "SomePolicy"}
template = {"a": "b"}
parser = Mock()
sam_parser.Parser.return_value = parser
translate_mock = Mock()
translate_mock.translate.side_effect = InvalidDocumentException([Exception('message')])
sam_translator.return_value = translate_mock
validator = SamTemplateValidator(template, managed_policy_mock)
with self.assertRaises(InvalidSamDocumentException):
validator.is_valid()
sam_translator.assert_called_once_with(managed_policy_map={"policy": "SomePolicy"},
sam_parser=parser,
plugins=[])
translate_mock.translate.assert_called_once_with(sam_template=template, parameter_values={})
sam_parser.Parser.assert_called_once()
def test_is_valid_returns_true(self, sam_parser, sam_translator):
managed_policy_mock = Mock()
managed_policy_mock.load.return_value = {"policy": "SomePolicy"}
template = {"a": "b"}
parser = Mock()
sam_parser.Parser.return_value = parser
translate_mock = Mock()
translate_mock.translate.return_value = {"c": "d"}
sam_translator.return_value = translate_mock
validator = SamTemplateValidator(template, managed_policy_mock)
# Should not throw an Exception
validator.is_valid()
sam_translator.assert_called_once_with(
managed_policy_map={"policy": "SomePolicy"}, sam_parser=parser, plugins=[]
)
translate_mock.translate.assert_called_once_with(sam_template=template, parameter_values={})
sam_parser.Parser.assert_called_once()
def test_invalid_template(self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Resources": {
"ServerlessFunction": {
"Type": "AWS::Serverless::Function",
"Properties": {
"Handler": "index.handler",
"CodeUri": "s3://lambda-code.zip",
"Timeout": 60
}
}
}
}
managed_policy_mock = Mock()
managed_policy_mock.load.return_value = {"PolicyName": "FakePolicy"}
validator = SamTemplateValidator(template, managed_policy_mock)
with self.assertRaises(InvalidSamDocumentException):
validator.is_valid()
def test_valid_template(self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Resources": {
"ServerlessFunction": {
"Type": "AWS::Serverless::Function",
"Properties": {
"Handler": "index.handler",
"CodeUri": "s3://fake-bucket/lambda-code.zip",
"Runtime": "nodejs6.10",
"Timeout": 60
}
}
}
}
managed_policy_mock = Mock()
managed_policy_mock.load.return_value = {"PolicyName": "FakePolicy"}
validator = SamTemplateValidator(template, managed_policy_mock)
# Should not throw an exception
validator.is_valid()
def test_valid_template(self):
template = {
"AWSTemplateFormatVersion": "2010-09-09",
"Transform": "AWS::Serverless-2016-10-31",
"Resources": {
"ServerlessFunction": {
"Type": "AWS::Serverless::Function",
"Properties": {
"Handler": "index.handler",
"CodeUri": "s3://fake-bucket/lambda-code.zip",
"Runtime": "nodejs6.10",
"Timeout": 60
}
}
}
}
managed_policy_mock = Mock()
managed_policy_mock.load.return_value = {"PolicyName": "FakePolicy"}
validator = SamTemplateValidator(template, managed_policy_mock)
# Should not throw an exception
validator.is_valid()
def test_uri_is_not_s3_uri(self):
self.assertFalse(SamTemplateValidator.is_s3_uri("www.amazon.com"))
def test_update_to_s3_url_with_s3_uri(self):
property_value = {"CodeUri": "s3://bucket/key/version"}
SamTemplateValidator._update_to_s3_uri("CodeUri", property_value)
self.assertEquals(property_value.get("CodeUri"), "s3://bucket/key/version")
def test_update_to_s3_url_with_dict(self):
property_value = {"CodeUri": {"Bucket": "mybucket-name", "Key": "swagger", "Version": 121212}}
SamTemplateValidator._update_to_s3_uri("CodeUri", property_value)
self.assertEquals(property_value.get("CodeUri"),
{"Bucket": "mybucket-name", "Key": "swagger", "Version": 121212})
def test_update_to_s3_uri_with_non_s3_uri(self):
property_value = {"CodeUri": "somevalue"}
SamTemplateValidator._update_to_s3_uri("CodeUri", property_value)
self.assertEquals(property_value.get("CodeUri"), "s3://bucket/value")
def test_int_is_not_s3_uri(self):
self.assertFalse(SamTemplateValidator.is_s3_uri(100))
def test_uri_is_not_s3_uri(self):
self.assertFalse(SamTemplateValidator.is_s3_uri("www.amazon.com"))
def test_uri_is_s3_uri(self):
self.assertTrue(SamTemplateValidator.is_s3_uri("s3://bucket/key"))
def test_update_to_s3_uri_with_non_s3_uri(self):
property_value = {"CodeUri": "somevalue"}
SamTemplateValidator._update_to_s3_uri("CodeUri", property_value)
self.assertEqual(property_value.get("CodeUri"), "s3://bucket/value")
def test_int_is_not_s3_uri(self):
self.assertFalse(SamTemplateValidator.is_s3_uri(100))
def test_uri_is_s3_uri(self):
self.assertTrue(SamTemplateValidator.is_s3_uri("s3://bucket/key"))
def test_update_to_s3_url_with_s3_uri(self):
property_value = {"CodeUri": "s3://bucket/key/version"}
SamTemplateValidator._update_to_s3_uri("CodeUri", property_value)
self.assertEqual(property_value.get("CodeUri"),
"s3://bucket/key/version")
