Harmonize calls to SafePgMigrations.get_pg_version_num with ActiveRec…

…ord::Base.connection
doctolib · Dec 11, 2023 · 698c9a0 · 698c9a0
1 parent b32aaab
commit 698c9a0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/test/IdempotentStatements/change_column_null_test.rb b/test/IdempotentStatements/change_column_null_test.rb
@@ -59,7 +59,7 @@ def change
 
     def skip_if_unmet_requirements
       if Gem::Requirement.new('>6.1').satisfied_by?(Gem::Version.new(::ActiveRecord::VERSION::STRING)) &&
-         SafePgMigrations.get_pg_version_num(@connection) >= 120_000
+         SafePgMigrations.get_pg_version_num(ActiveRecord::Base.connection) >= 120_000
         return
       end
 

diff --git a/test/StatementInsurer/change_column_null_test.rb b/test/StatementInsurer/change_column_null_test.rb
@@ -127,7 +127,7 @@ def skip_if_unmet_requirements!
 
     def met_requirements?
       Gem::Requirement.new('>6.1').satisfied_by?(Gem::Version.new(::ActiveRecord::VERSION::STRING)) &&
-        SafePgMigrations.get_pg_version_num(@connection) >= 120_000
+        SafePgMigrations.get_pg_version_num(ActiveRecord::Base.connection) >= 120_000
     end
   end
 end