def test_sam_function_with_alias_cannot_be_list(self):
# Alias cannot be a list
with self.assertRaises(InvalidResourceException):
self.func_dict["Properties"]["AutoPublishAlias"] = ["a", "b"]
SamFunction.from_dict(logical_id="foo",
resource_dict=self.func_dict)
def test_sam_function_with_code_signer(self):
code_signing_config_arn = "code_signing_config_arn"
func = {
"Type": "AWS::Serverless::Function",
"Properties": {
"CodeUri": self.code_uri,
"Runtime": "nodejs12.x",
"Handler": "index.handler",
"CodeSigningConfigArn": code_signing_config_arn,
},
}
sam_func = SamFunction.from_dict(logical_id="foo", resource_dict=func)
kwargs = {}
kwargs["managed_policy_map"] = {"a": "b"}
kwargs["event_resources"] = []
kwargs["intrinsics_resolver"] = self.intrinsics_resolver_mock
self.intrinsics_resolver_mock.resolve_parameter_refs.return_value = {
"S3Bucket": "bucket",
"S3Key": "key",
"S3ObjectVersion": "version",
}
resources = sam_func.to_cloudformation(**kwargs)
lambda_functions = [r.to_dict() for r in resources if r.resource_type == LambdaFunction.resource_type]
self.assertEqual(len(lambda_functions), 1)
expected_code_signing_config_arn = lambda_functions[0]["foo"]["Properties"]["CodeSigningConfigArn"]
self.assertEqual(expected_code_signing_config_arn, code_signing_config_arn)
def test_sam_function_with_deployment_preference_instrinsic_ref_enabled_dict_parameter(self, get_resolved_alias_name_mock):
alias_name = "AliasName"
enabled = {"Ref": "MyEnabledFlag"}
deploy_preference_dict = {"Type": "LINEAR", "Enabled": enabled}
func = {
"Type": "AWS::Serverless::Function",
"Properties": {
"CodeUri": self.code_uri,
"Runtime": "nodejs4.3",
"Handler": "index.handler",
"AutoPublishAlias": alias_name,
"DeploymentPreference": deploy_preference_dict
}
}
sam_func = SamFunction.from_dict(logical_id="foo", resource_dict=func)
kwargs = dict()
kwargs["managed_policy_map"] = {"a": "b"}
kwargs["event_resources"] = []
kwargs["intrinsics_resolver"] = self.intrinsics_resolver_mock
deployment_preference_collection = self._make_deployment_preference_collection()
kwargs['deployment_preference_collection'] = deployment_preference_collection
self.intrinsics_resolver_mock.resolve_parameter_refs.return_value = {"key": "value"}
get_resolved_alias_name_mock.return_value = alias_name
with self.assertRaises(InvalidResourceException):
sam_func.to_cloudformation(**kwargs)
def test_sam_function_with_deployment_preference_missing_collection_raises_error(
self, get_resolved_alias_name_mock):
alias_name = "AliasName"
deploy_preference_dict = {"Type": "LINEAR"}
func = {
"Type": "AWS::Serverless::Function",
"Properties": {
"CodeUri": self.code_uri,
"Runtime": "nodejs4.3",
"Handler": "index.handler",
"AutoPublishAlias": alias_name,
"DeploymentPreference": deploy_preference_dict
}
}
sam_func = SamFunction.from_dict(logical_id="foo", resource_dict=func)
kwargs = dict()
kwargs["managed_policy_map"] = {"a": "b"}
kwargs["event_resources"] = []
kwargs["intrinsics_resolver"] = self.intrinsics_resolver_mock
kwargs["mappings_resolver"] = self.mappings_resolver_mock
self.intrinsics_resolver_mock.resolve_parameter_refs.return_value = {
"S3Bucket": "bucket",
"S3Key": "key",
"S3ObjectVersion": "version"
}
get_resolved_alias_name_mock.return_value = alias_name
with self.assertRaises(ValueError):
sam_func.to_cloudformation(**kwargs)
def test_sam_function_without_alias_allows_disabled_deployment_preference(self):
enabled = False
deploy_preference_dict = {"Enabled": enabled}
func = {
"Type": "AWS::Serverless::Function",
"Properties": {
"CodeUri": self.code_uri,
"Runtime": "nodejs4.3",
"Handler": "index.handler",
"DeploymentPreference": deploy_preference_dict
}
}
sam_func = SamFunction.from_dict(logical_id="foo", resource_dict=func)
kwargs = dict()
kwargs["managed_policy_map"] = {"a": "b"}
kwargs["event_resources"] = []
kwargs["intrinsics_resolver"] = self.intrinsics_resolver_mock
preference_collection = self._make_deployment_preference_collection()
preference_collection.get.return_value = DeploymentPreference.from_dict(sam_func.logical_id,
deploy_preference_dict)
kwargs['deployment_preference_collection'] = preference_collection
self.intrinsics_resolver_mock.resolve_parameter_refs.return_value = enabled
resources = sam_func.to_cloudformation(**kwargs)
self.intrinsics_resolver_mock.resolve_parameter_refs.assert_called_with(enabled)
# Function, IAM Role
self.assertEquals(len(resources), 2)
def test_sam_function_cannot_be_with_deployment_preference_without_alias(
self):
with self.assertRaises(InvalidResourceException):
func = {
"Type": "AWS::Serverless::Function",
"Properties": {
"CodeUri": self.code_uri,
"Runtime": "nodejs4.3",
"Handler": "index.handler",
"DeploymentPreference": {
"Type": "LINEAR"
}
}
}
sam_func = SamFunction.from_dict(logical_id="foo",
resource_dict=func)
kwargs = dict()
kwargs["intrinsics_resolver"] = self.intrinsics_resolver_mock
kwargs["mappings_resolver"] = self.mappings_resolver_mock
kwargs[
'deployment_preference_collection'] = self._make_deployment_preference_collection(
)
sam_func.to_cloudformation(**kwargs)
def test_sam_function_without_alias_allows_disabled_deployment_preference(
self):
enabled = False
deploy_preference_dict = {"Enabled": enabled}
func = {
"Type": "AWS::Serverless::Function",
"Properties": {
"CodeUri": self.code_uri,
"Runtime": "nodejs4.3",
"Handler": "index.handler",
"DeploymentPreference": deploy_preference_dict
}
}
sam_func = SamFunction.from_dict(logical_id="foo", resource_dict=func)
kwargs = dict()
kwargs["managed_policy_map"] = {"a": "b"}
kwargs["event_resources"] = []
kwargs["intrinsics_resolver"] = self.intrinsics_resolver_mock
kwargs["mappings_resolver"] = self.mappings_resolver_mock
preference_collection = self._make_deployment_preference_collection()
preference_collection.get.return_value = DeploymentPreference.from_dict(
sam_func.logical_id, deploy_preference_dict)
kwargs['deployment_preference_collection'] = preference_collection
self.intrinsics_resolver_mock.resolve_parameter_refs.return_value = enabled
resources = sam_func.to_cloudformation(**kwargs)
self.intrinsics_resolver_mock.resolve_parameter_refs.assert_called_with(
enabled)
# Function, IAM Role
self.assertEqual(len(resources), 2)
def test_sam_function_with_deployment_preference_intrinsic_findinmap_enabled_dict_parameter(
self, get_resolved_alias_name_mock):
alias_name = "AliasName"
enabled = {"Fn::FindInMap": ["FooMap", "FooKey", "Enabled"]}
deploy_preference_dict = {"Type": "LINEAR", "Enabled": enabled}
func = {
"Type": "AWS::Serverless::Function",
"Properties": {
"CodeUri": self.code_uri,
"Runtime": "nodejs4.3",
"Handler": "index.handler",
"AutoPublishAlias": alias_name,
"DeploymentPreference": deploy_preference_dict
}
}
sam_func = SamFunction.from_dict(logical_id="foo", resource_dict=func)
kwargs = dict()
kwargs["managed_policy_map"] = {"a": "b"}
kwargs["event_resources"] = []
kwargs["intrinsics_resolver"] = self.intrinsics_resolver_mock
kwargs["mappings_resolver"] = self.mappings_resolver_mock
deployment_preference_collection = self._make_deployment_preference_collection(
)
kwargs[
'deployment_preference_collection'] = deployment_preference_collection
self.intrinsics_resolver_mock.resolve_parameter_refs.return_value = {
"MyEnabledFlag": True
}
self.mappings_resolver_mock.resolve_parameter_refs.return_value = True
get_resolved_alias_name_mock.return_value = alias_name
sam_func.to_cloudformation(**kwargs)
self.assertTrue(sam_func.DeploymentPreference['Enabled'])
def test_sam_function_with_deployment_preference_missing_collection_raises_error(self, get_resolved_alias_name_mock):
alias_name = "AliasName"
deploy_preference_dict = {"Type": "LINEAR"}
func = {
"Type": "AWS::Serverless::Function",
"Properties": {
"CodeUri": self.code_uri,
"Runtime": "nodejs4.3",
"Handler": "index.handler",
"AutoPublishAlias": alias_name,
"DeploymentPreference": deploy_preference_dict
}
}
sam_func = SamFunction.from_dict(logical_id="foo", resource_dict=func)
kwargs = dict()
kwargs["managed_policy_map"] = {"a": "b"}
kwargs["event_resources"] = []
kwargs["intrinsics_resolver"] = self.intrinsics_resolver_mock
self.intrinsics_resolver_mock.resolve_parameter_refs.return_value = {"S3Bucket": "bucket", "S3Key": "key",
"S3ObjectVersion": "version"}
get_resolved_alias_name_mock.return_value = alias_name
with self.assertRaises(ValueError):
sam_func.to_cloudformation(**kwargs)
def test_sam_function_with_deployment_preference_instrinsic_ref_enabled_dict_parameter(self, get_resolved_alias_name_mock):
alias_name = "AliasName"
enabled = {"Ref": "MyEnabledFlag"}
deploy_preference_dict = {"Type": "LINEAR", "Enabled": enabled}
func = {
"Type": "AWS::Serverless::Function",
"Properties": {
"CodeUri": self.code_uri,
"Runtime": "nodejs4.3",
"Handler": "index.handler",
"AutoPublishAlias": alias_name,
"DeploymentPreference": deploy_preference_dict
}
}
sam_func = SamFunction.from_dict(logical_id="foo", resource_dict=func)
kwargs = dict()
kwargs["managed_policy_map"] = {"a": "b"}
kwargs["event_resources"] = []
kwargs["intrinsics_resolver"] = self.intrinsics_resolver_mock
deployment_preference_collection = self._make_deployment_preference_collection()
kwargs['deployment_preference_collection'] = deployment_preference_collection
self.intrinsics_resolver_mock.resolve_parameter_refs.return_value = {"key": "value"}
get_resolved_alias_name_mock.return_value = alias_name
with self.assertRaises(InvalidResourceException):
sam_func.to_cloudformation(**kwargs)
def test_sam_function_with_deployment_preference(
self, get_resolved_alias_name_mock):
deploy_preference_dict = {"Type": "LINEAR"}
alias_name = "AliasName"
func = {
"Type": "AWS::Serverless::Function",
"Properties": {
"CodeUri": self.code_uri,
"Runtime": "nodejs4.3",
"Handler": "index.handler",
"AutoPublishAlias": alias_name,
"DeploymentPreference": deploy_preference_dict
}
}
sam_func = SamFunction.from_dict(logical_id="foo", resource_dict=func)
kwargs = dict()
kwargs["managed_policy_map"] = {"a": "b"}
kwargs["event_resources"] = []
kwargs["intrinsics_resolver"] = self.intrinsics_resolver_mock
kwargs["mappings_resolver"] = self.mappings_resolver_mock
deployment_preference_collection = self._make_deployment_preference_collection(
)
kwargs[
'deployment_preference_collection'] = deployment_preference_collection
get_resolved_alias_name_mock.return_value = alias_name
self.intrinsics_resolver_mock.resolve_parameter_refs.return_value = {
"S3Bucket": "bucket",
"S3Key": "key",
"S3ObjectVersion": "version"
}
resources = sam_func.to_cloudformation(**kwargs)
deployment_preference_collection.update_policy.assert_called_once_with(
self.sam_func.logical_id)
deployment_preference_collection.add.assert_called_once_with(
self.sam_func.logical_id, deploy_preference_dict)
aliases = [
r.to_dict() for r in resources
if r.resource_type == LambdaAlias.resource_type
]
self.assertTrue("UpdatePolicy" in list(aliases[0].values())[0])
self.assertEqual(
list(aliases[0].values())[0]["UpdatePolicy"],
self.update_policy().to_dict())
def setUp(self):
self.intrinsics_resolver_mock = Mock()
self.intrinsics_resolver_mock.resolve = Mock()
self.mappings_resolver_mock = Mock()
self.mappings_resolver_mock.resolve = Mock()
self.code_uri = "s3://bucket/key?versionId=version"
self.func_dict = {
"Type": "AWS::Serverless::Function",
"Properties": {"CodeUri": self.code_uri, "Runtime": "nodejs12.x", "Handler": "index.handler"},
}
self.sam_func = SamFunction.from_dict(logical_id="foo", resource_dict=self.func_dict)
self.lambda_func = self._make_lambda_function(self.sam_func.logical_id)
self.lambda_version = self._make_lambda_version("VersionLogicalId", self.sam_func)
def test_sam_function_with_deployment_preference_intrinsic_ref_enabled_boolean_parameter(
self, get_resolved_alias_name_mock):
alias_name = "AliasName"
enabled = {"Ref": "MyEnabledFlag"}
deploy_preference_dict = {"Type": "LINEAR", "Enabled": enabled}
func = {
"Type": "AWS::Serverless::Function",
"Properties": {
"CodeUri": self.code_uri,
"Runtime": "nodejs12.x",
"Handler": "index.handler",
"AutoPublishAlias": alias_name,
"DeploymentPreference": deploy_preference_dict,
},
}
sam = SamFunction.from_dict(logical_id="foo", resource_dict=func)
kwargs = dict()
kwargs["managed_policy_map"] = {"a": "b"}
kwargs["event_resources"] = []
kwargs["intrinsics_resolver"] = self.intrinsics_resolver_mock
kwargs["mappings_resolver"] = self.mappings_resolver_mock
deployment_preference_collection = self._make_deployment_preference_collection(
)
kwargs[
"deployment_preference_collection"] = deployment_preference_collection
self.intrinsics_resolver_mock.resolve_parameter_refs.return_value = True
get_resolved_alias_name_mock.return_value = alias_name
resources = sam.to_cloudformation(**kwargs)
deployment_preference_collection.update_policy.assert_called_once_with(
self.sam_func.logical_id)
deployment_preference_collection.add.assert_called_once_with(
self.sam_func.logical_id, deploy_preference_dict)
self.intrinsics_resolver_mock.resolve_parameter_refs.assert_any_call(
enabled)
aliases = [
r.to_dict() for r in resources
if r.resource_type == LambdaAlias.resource_type
]
self.assertTrue("UpdatePolicy" in list(aliases[0].values())[0])
self.assertEqual(
list(aliases[0].values())[0]["UpdatePolicy"],
self.update_policy().to_dict())
def test_sam_function_with_alias(self, get_resolved_alias_name_mock):
alias_name = "AliasName"
func = {
"Type": "AWS::Serverless::Function",
"Properties": {
"CodeUri": self.code_uri,
"Runtime": "nodejs4.3",
"Handler": "index.handler",
"AutoPublishAlias": alias_name
}
}
sam_func = SamFunction.from_dict(logical_id="foo", resource_dict=func)
kwargs = {}
kwargs["managed_policy_map"] = {"a": "b"}
kwargs["event_resources"] = []
kwargs["intrinsics_resolver"] = self.intrinsics_resolver_mock
self.intrinsics_resolver_mock.resolve_parameter_refs.return_value = {
"S3Bucket": "bucket",
"S3Key": "key",
"S3ObjectVersion": "version"
}
get_resolved_alias_name_mock.return_value = alias_name
resources = sam_func.to_cloudformation(**kwargs)
# Function, Version, Alias, IAM Role
self.assertEqual(len(resources), 4)
aliases = [
r.to_dict() for r in resources
if r.resource_type == LambdaAlias.resource_type
]
versions = [
r.to_dict() for r in resources
if r.resource_type == LambdaVersion.resource_type
]
self.assertEqual(len(aliases), 1)
self.assertEqual(len(versions), 1)
alias = list(aliases[0].values())[0]["Properties"]
self.assertEqual(alias["Name"], alias_name)
# We don't need to do any deeper validation here because there is a separate SAM template -> CFN template conversion test
# that will care of validating all properties & connections
sam_func._get_resolved_alias_name.assert_called_once_with(
"AutoPublishAlias", alias_name, self.intrinsics_resolver_mock)
def setUp(self):
self.intrinsics_resolver_mock = Mock()
self.intrinsics_resolver_mock.resolve = Mock()
self.code_uri = "s3://bucket/key?versionId=version"
self.func_dict = {
"Type": "AWS::Serverless::Function",
"Properties": {
"CodeUri": self.code_uri,
"Runtime": "nodejs4.3",
"Handler": "index.handler"
}
}
self.sam_func = SamFunction.from_dict(logical_id="foo", resource_dict=self.func_dict)
self.lambda_func = self._make_lambda_function(self.sam_func.logical_id)
self.lambda_version = self._make_lambda_version("VersionLogicalId", self.sam_func)
def test_sam_function_cannot_be_with_deployment_preference_without_alias(self):
with self.assertRaises(InvalidResourceException):
func = {
"Type": "AWS::Serverless::Function",
"Properties": {
"CodeUri": self.code_uri,
"Runtime": "nodejs4.3",
"Handler": "index.handler",
"DeploymentPreference": {"Type": "LINEAR"}
}
}
sam_func = SamFunction.from_dict(logical_id="foo", resource_dict=func)
kwargs = dict()
kwargs["intrinsics_resolver"] = self.intrinsics_resolver_mock
kwargs['deployment_preference_collection'] = self._make_deployment_preference_collection()
sam_func.to_cloudformation(**kwargs)
def test_sam_function_with_disabled_deployment_preference_does_not_add_update_policy(
self, get_resolved_alias_name_mock):
alias_name = "AliasName"
enabled = False
deploy_preference_dict = {"Enabled": enabled}
func = {
"Type": "AWS::Serverless::Function",
"Properties": {
"CodeUri": self.code_uri,
"Runtime": "nodejs4.3",
"Handler": "index.handler",
"AutoPublishAlias": alias_name,
"DeploymentPreference": deploy_preference_dict
}
}
sam_func = SamFunction.from_dict(logical_id="foo", resource_dict=func)
kwargs = dict()
kwargs["managed_policy_map"] = {"a": "b"}
kwargs["event_resources"] = []
kwargs["intrinsics_resolver"] = self.intrinsics_resolver_mock
kwargs["mappings_resolver"] = self.mappings_resolver_mock
preference_collection = self._make_deployment_preference_collection()
preference_collection.get.return_value = DeploymentPreference.from_dict(
sam_func.logical_id, deploy_preference_dict)
kwargs['deployment_preference_collection'] = preference_collection
self.intrinsics_resolver_mock.resolve_parameter_refs.return_value = enabled
get_resolved_alias_name_mock.return_value = alias_name
resources = sam_func.to_cloudformation(**kwargs)
preference_collection.add.assert_called_once_with(
sam_func.logical_id, deploy_preference_dict)
preference_collection.get.assert_called_once_with(sam_func.logical_id)
self.intrinsics_resolver_mock.resolve_parameter_refs.assert_called_with(
enabled)
aliases = [
r.to_dict() for r in resources
if r.resource_type == LambdaAlias.resource_type
]
self.assertTrue("UpdatePolicy" not in list(aliases[0].values())[0])
def test_sam_function_with_alias(self, get_resolved_alias_name_mock):
alias_name = "AliasName"
func = {
"Type": "AWS::Serverless::Function",
"Properties": {
"CodeUri": self.code_uri,
"Runtime": "nodejs4.3",
"Handler": "index.handler",
"AutoPublishAlias": alias_name
}
}
sam_func = SamFunction.from_dict(logical_id="foo", resource_dict=func)
kwargs = {}
kwargs["managed_policy_map"] = {"a": "b"}
kwargs["event_resources"] = []
kwargs["intrinsics_resolver"] = self.intrinsics_resolver_mock
self.intrinsics_resolver_mock.resolve_parameter_refs.return_value = {"S3Bucket": "bucket", "S3Key": "key", "S3ObjectVersion": "version"}
get_resolved_alias_name_mock.return_value = alias_name
resources = sam_func.to_cloudformation(**kwargs)
# Function, Version, Alias, IAM Role
self.assertEquals(len(resources), 4)
aliases = [r.to_dict() for r in resources if r.resource_type == LambdaAlias.resource_type]
versions = [r.to_dict() for r in resources if r.resource_type == LambdaVersion.resource_type]
self.assertEquals(len(aliases), 1)
self.assertEquals(len(versions), 1)
alias = list(aliases[0].values())[0]["Properties"]
self.assertEquals(alias["Name"], alias_name)
# We don't need to do any deeper validation here because there is a separate SAM template -> CFN template conversion test
# that will care of validating all properties & connections
sam_func._get_resolved_alias_name.assert_called_once_with("AutoPublishAlias", alias_name, self.intrinsics_resolver_mock)
def test_sam_function_with_disabled_deployment_preference_does_not_add_update_policy(self, get_resolved_alias_name_mock):
alias_name = "AliasName"
enabled = False
deploy_preference_dict = {"Enabled": enabled}
func = {
"Type": "AWS::Serverless::Function",
"Properties": {
"CodeUri": self.code_uri,
"Runtime": "nodejs4.3",
"Handler": "index.handler",
"AutoPublishAlias": alias_name,
"DeploymentPreference": deploy_preference_dict
}
}
sam_func = SamFunction.from_dict(logical_id="foo", resource_dict=func)
kwargs = dict()
kwargs["managed_policy_map"] = {"a": "b"}
kwargs["event_resources"] = []
kwargs["intrinsics_resolver"] = self.intrinsics_resolver_mock
preference_collection = self._make_deployment_preference_collection()
preference_collection.get.return_value = DeploymentPreference.from_dict(sam_func.logical_id,
deploy_preference_dict)
kwargs['deployment_preference_collection'] = preference_collection
self.intrinsics_resolver_mock.resolve_parameter_refs.return_value = enabled
get_resolved_alias_name_mock.return_value = alias_name
resources = sam_func.to_cloudformation(**kwargs)
preference_collection.add.assert_called_once_with(sam_func.logical_id, deploy_preference_dict)
preference_collection.get.assert_called_once_with(sam_func.logical_id)
self.intrinsics_resolver_mock.resolve_parameter_refs.assert_called_with(enabled)
aliases = [r.to_dict() for r in resources if r.resource_type == LambdaAlias.resource_type]
self.assertTrue("UpdatePolicy" not in list(aliases[0].values())[0])
def test_sam_function_with_deployment_preference_intrinsic_ref_enabled_boolean_parameter(self, get_resolved_alias_name_mock):
alias_name = "AliasName"
enabled = {"Ref": "MyEnabledFlag"}
deploy_preference_dict = {"Type": "LINEAR", "Enabled": enabled}
func = {
"Type": "AWS::Serverless::Function",
"Properties": {
"CodeUri": self.code_uri,
"Runtime": "nodejs4.3",
"Handler": "index.handler",
"AutoPublishAlias": alias_name,
"DeploymentPreference": deploy_preference_dict
}
}
sam = SamFunction.from_dict(logical_id="foo", resource_dict=func)
kwargs = dict()
kwargs["managed_policy_map"] = {"a": "b"}
kwargs["event_resources"] = []
kwargs["intrinsics_resolver"] = self.intrinsics_resolver_mock
deployment_preference_collection = self._make_deployment_preference_collection()
kwargs['deployment_preference_collection'] = deployment_preference_collection
self.intrinsics_resolver_mock.resolve_parameter_refs.return_value = True
get_resolved_alias_name_mock.return_value = alias_name
resources = sam.to_cloudformation(**kwargs)
deployment_preference_collection.update_policy.assert_called_once_with(self.sam_func.logical_id)
deployment_preference_collection.add.assert_called_once_with(self.sam_func.logical_id,
deploy_preference_dict)
self.intrinsics_resolver_mock.resolve_parameter_refs.assert_called_with(enabled)
aliases = [r.to_dict() for r in resources if r.resource_type == LambdaAlias.resource_type]
self.assertTrue("UpdatePolicy" in list(aliases[0].values())[0])
self.assertEqual(list(aliases[0].values())[0]["UpdatePolicy"], self.update_policy().to_dict())
def test_sam_function_with_alias_cannot_be_list(self):
# Alias cannot be a list
with self.assertRaises(InvalidResourceException):
self.func_dict["Properties"]["AutoPublishAlias"] = ["a", "b"]
SamFunction.from_dict(logical_id="foo", resource_dict=self.func_dict)