Add reindex trait to rubygems & version factories

rubygems · Oct 9, 2024 · 9324a77 · 9324a77
1 parent c86e1c1
commit 9324a77
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/test/factories/rubygem.rb b/test/factories/rubygem.rb
@@ -8,6 +8,12 @@
 
     name
 
+    trait :reindex do
+      after(:create) do |rubygem, _evaluator|
+        rubygem.reindex(refresh: true)
+      end
+    end
+
     after(:build) do |rubygem, evaluator|
       if evaluator.linkset
         rubygem.linkset = evaluator.linkset

diff --git a/test/factories/version.rb b/test/factories/version.rb
@@ -20,6 +20,12 @@
     sha256 { "tdQEXD9Gb6kf4sxqvnkjKhpXzfEE96JucW4KHieJ33g=" }
     spec_sha256 { Digest::SHA2.base64digest("#{rubygem.name}-#{number}-#{platform}") }
 
+    trait :reindex do
+      after(:create) do |version|
+        version.rubygem.reindex(refresh: true)
+      end
+    end
+
     trait :yanked do
       indexed { false }
     end