From 9324a771d36e6cd80601eac20b394323d9748858 Mon Sep 17 00:00:00 2001
From: Colby Swandale <996377+colby-swandale@users.noreply.github.com>
Date: Wed, 9 Oct 2024 18:03:15 +1100
Subject: [PATCH] Add reindex trait to rubygems & version factories

---
 test/factories/rubygem.rb | 6 ++++++
 test/factories/version.rb | 6 ++++++
 2 files changed, 12 insertions(+)

diff --git a/test/factories/rubygem.rb b/test/factories/rubygem.rb
index 78224af58d9..357f3efd665 100644
--- 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
index 520cdb82c8e..dba960ef4d8 100644
--- 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