Fix specs for container repository tags

1 files changed, 11 insertions, 12 deletions

diff --git a/spec/factories/container_repositories.rb b/spec/factories/container_repositories.rb
index fbf6bf62dfd..295b3596ee9 100644
--- a/spec/factories/container_repositories.rb
+++ b/spec/factories/container_repositories.rb
@@ -1,22 +1,21 @@
 FactoryGirl.define do
   factory :container_repository do
-    name "test_container_image"
+    name 'test_container_image'
     project
 
     transient do
-      tags ['tag']
+      tags []
     end
 
-    after(:build) do |image, evaluator|
-      # if evaluator.tags.to_a.any?
-      #   allow(Gitlab.config.registry).to receive(:enabled).and_return(true)
-      #   allow(Auth::ContainerRegistryAuthenticationService)
-      #     .to receive(:full_access_token).and_return('token')
-      #   allow(image.client).to receive(:repository_tags).and_return({
-      #     name: image.name_with_namespace,
-      #     tags: evaluator.tags
-      #   })
-      # end
+    after(:build) do |repository, evaluator|
+      if evaluator.tags.any?
+        allow(repository.client)
+          .to receive(:repository_tags)
+          .and_return({
+            name: repository.path,
+            tags: evaluator.tags
+          })
+      end
     end
   end
 end