Adding policy text to ecr_repository model.

k1LoW · Jul 10, 2020 · ffe713c · ffe713c
1 parent 75becba
commit ffe713c
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/lib/awspec/helper/finder/ecr.rb b/lib/awspec/helper/finder/ecr.rb
@@ -5,6 +5,10 @@ def find_ecr_repository(repository_name)
         res = ecr_client.describe_repositories(repository_names: [repository_name])
         res.repositories.first if res.repositories.count == 1
       end
+
+      def get_policy_text(repository_name)
+        res = ecr_client.get_repository_policy(repository_name: repository_name)[:policy_text]
+      end
     end
   end
 end
diff --git a/lib/awspec/type/ecr_repository.rb b/lib/awspec/type/ecr_repository.rb
@@ -14,5 +14,9 @@ def resource_via_client
     def id
       @id ||= resource_via_client.repository_name if resource_via_client
     end
+
+    def policy_text
+      @policy_text ||= get_policy_text(@display_name)
+    end
   end
 end
diff --git a/spec/type/ecr_repository_spec.rb b/spec/type/ecr_repository_spec.rb
@@ -7,4 +7,5 @@
   its(:registry_id) { should eq '123456789012' }
   its(:repository_name) { should eq 'my-ecr-repository' }
   its(:repository_arn) { should eq 'arn:aws:ecr:us-east-1:123456789012:repository/my-ecr-repository' }
+  its(:policy_text) { should eq 'RepositoryPolicyText' }
 end