fix(aws-ecr): fix addToResourcePolicy (#737)

akiym · rix0rrr · commit eadbda598e18 · 2018-09-19T10:29:26.000+02:00
diff --git a/packages/@aws-cdk/aws-ecr/lib/repository.ts b/packages/@aws-cdk/aws-ecr/lib/repository.ts
@@ -53,7 +53,7 @@ export class Repository extends RepositoryRef {
         const resource = new cloudformation.RepositoryResource(this, 'Resource', {
             repositoryName: props.repositoryName,
             // It says "Text", but they actually mean "Object".
-            repositoryPolicyText: this.policyDocument,
+            repositoryPolicyText: new cdk.Token(() => this.policyDocument),
             lifecyclePolicy: new cdk.Token(() => this.renderLifecyclePolicy()),
         });
 
diff --git a/packages/@aws-cdk/aws-ecr/test/test.repository.ts b/packages/@aws-cdk/aws-ecr/test/test.repository.ts
@@ -160,6 +160,30 @@ export = {
             'Fn::ImportValue': 'RepoRepositoryArn7F2901C9'
         });
 
+        test.done();
+    },
+
+    'resource policy'(test: Test) {
+        // GIVEN
+        const stack = new cdk.Stack();
+        const repo = new ecr.Repository(stack, 'Repo');
+
+        // WHEN
+        repo.addToResourcePolicy(new cdk.PolicyStatement().addAction('*'));
+
+        // THEN
+        expect(stack).to(haveResource('AWS::ECR::Repository', {
+            RepositoryPolicyText: {
+                Statement: [
+                    {
+                        Action: "*",
+                        Effect: "Allow"
+                    }
+                ],
+                Version: "2012-10-17"
+            },
+        }));
+
         test.done();
     }
 };
