Add CodeCommit::Repository to cfn package command

ConnorKirk · ConnorKirk · commit af217abfff0e · 2020-11-23T15:44:29.000Z
This commit adds the CodeCommitRepositoryS3Resource. This enables
`aws cloudformation package` to work with the CodeCommit repository
resource type
diff --git a/awscli/customizations/cloudformation/artifact_exporter.py b/awscli/customizations/cloudformation/artifact_exporter.py
@@ -496,6 +496,18 @@ class GlueJobCommandScriptLocationResource(Resource):
     # Note the PROPERTY_NAME includes a '.' implying it's nested.
     PROPERTY_NAME = "Command.ScriptLocation"
 
+class CodeCommitRepositoryS3Resource(ResourceWithS3UrlDict):
+    """
+    Represents CodeCommit::Repository resource.
+    """
+    RESOURCE_TYPE = "AWS::CodeCommit::Repository"
+    PROPERTY_NAME = "Code.S3"
+    BUCKET_NAME_PROPERTY = "Bucket"
+    OBJECT_KEY_PROPERTY = "Key"
+    VERSION_PROPERTY = "ObjectVersion"
+    # Don't package the directory if S3 is omitted.
+    PACKAGE_NULL_PROPERTY = False
+    FORCE_ZIP = True
 
 RESOURCES_EXPORT_LIST = [
     ServerlessFunctionResource,
@@ -513,7 +525,8 @@ class GlueJobCommandScriptLocationResource(Resource):
     ServerlessLayerVersionResource,
     LambdaLayerVersionResource,
     GlueJobCommandScriptLocationResource,
-    StepFunctionsStateMachineDefinitionResource
+    StepFunctionsStateMachineDefinitionResource,
+    CodeCommitRepositoryS3Resource
 ]
 
 METADATA_EXPORT_LIST = [
diff --git a/awscli/examples/cloudformation/_package_description.rst b/awscli/examples/cloudformation/_package_description.rst
@@ -23,6 +23,7 @@ This command can upload local artifacts referenced in the following places:
     - ``TemplateURL`` property for the ``AWS::CloudFormation::Stack`` resource
     - ``Command.ScriptLocation`` property for the ``AWS::Glue::Job`` resource
     - ``DefinitionS3Location`` property for the ``AWS::StepFunctions::StateMachine`` resource
+    - ``S3`` property for the ``AWS::CodeCommit::Repository`` resource
 
 
 To specify a local artifact in your template, specify a path to a local file or folder,
diff --git a/tests/unit/customizations/cloudformation/test_artifact_exporter.py b/tests/unit/customizations/cloudformation/test_artifact_exporter.py
@@ -28,7 +28,8 @@
     AppSyncFunctionConfigurationRequestTemplateResource, \
     AppSyncFunctionConfigurationResponseTemplateResource, \
     GlueJobCommandScriptLocationResource, \
-    StepFunctionsStateMachineDefinitionResource
+    StepFunctionsStateMachineDefinitionResource, \
+    CodeCommitRepositoryS3Resource
 
 
 def test_is_s3_url():
@@ -158,6 +159,16 @@ def test_all_resources_export():
                 "Bucket": "foo", "Key": "bar", "Version": "baz"
             }
         },
+        {
+            "class": CodeCommitRepositoryS3Resource,
+            "expected_result": {
+                "S3": {
+                    "Bucket": "foo",
+                    "Key": "bar", 
+                    "ObjectVersion": "baz"
+                }
+            }
+        },
     ]
 
     with patch("awscli.customizations.cloudformation.artifact_exporter.upload_local_artifacts") as upload_local_artifacts_mock:
