Merge branch 'sh-fix-geo-db-cleaning' into 'master'

Make Geo tracking database work with let_it_be

Closes #232749

See merge request gitlab-org/gitlab!38118

ee/spec/services/geo/blob_verification_secondary_service_spec.rb

@@ -14,7 +14,7 @@ RSpec.describe Geo::BlobVerificationSecondaryService, :geo do
   describe '#execute' do
     let_it_be(:package_file) { create(:conan_package_file, :conan_recipe_file, verification_checksum: '62fc1ec4ce60') }
 
-    let_it_be(:registry) { create(:geo_package_file_registry, :synced, package_file: package_file) }
+    let!(:registry) { create(:geo_package_file_registry, :synced, package_file: package_file) }
 
     subject(:service) { described_class.new(package_file.replicator) }
 

ee/spec/support/before_all_adapter.rb

+# frozen_string_literal: true
+
+module EE
+  class BeforeAllAdapter
+    def self.begin_transaction
+      TestProf::BeforeAll::Adapters::ActiveRecord.begin_transaction
+
+      ::Geo::BaseRegistry.connection.begin_transaction
+    end
+
+    def self.rollback_transaction
+      TestProf::BeforeAll::Adapters::ActiveRecord.rollback_transaction
+
+      ::Geo::BaseRegistry.connection.rollback_transaction
+    end
+  end
+end

ee/spec/support/db_cleaner.rb

@@ -13,6 +13,7 @@ module EE
     def setup_database_cleaner
       if ::Gitlab::Geo.geo_database_configured?
         ::DatabaseCleaner[:active_record, { connection: ::Geo::BaseRegistry }]
+        TestProf::BeforeAll.adapter = ::EE::BeforeAllAdapter
       end
 
       super