pass internal_metadata to ActiveRecord::Migrator for 7.1 or above

Author: CharlesDelannoy

test/test_helper.rb

@@ -34,12 +34,14 @@ def setup
     @connection.tables.each { |table| @connection.drop_table table, force: :cascade }
     # only instanciate ActiveRecord::SchemaMigration with the right connection for Activerecord 7.1 or above
     if Gem::Requirement.new('>=7.1.0').satisfied_by?(Gem::Version.new(::ActiveRecord::VERSION::STRING))
-      ActiveRecord::SchemaMigration.new(@connection).create_table
-      ActiveRecord::InternalMetadata.new(@connection).create_table
+      @schema_migration = ActiveRecord::SchemaMigration.new(@connection)
+      @internal_metadata = ActiveRecord::InternalMetadata.new(@connection)
     else
-      ActiveRecord::SchemaMigration.create_table
-      ActiveRecord::InternalMetadata.create_table
+      @schema_migration = ActiveRecord::SchemaMigration
+      @internal_metadata = ActiveRecord::InternalMetadata
     end
+    @schema_migration.create_table
+    @internal_metadata.create_table
     ActiveRecord::Migration.verbose = false
     @connection.execute("SET statement_timeout TO '70s'")
     @connection.execute("SET lock_timeout TO '70s'")
@@ -59,8 +61,10 @@ def run_migration(direction = :up)
     @migration.version = DUMMY_MIGRATION_VERSION
 
     migrator =
-      if Gem::Requirement.new('>=6.0.0').satisfied_by?(Gem::Version.new(::ActiveRecord::VERSION::STRING))
-        ActiveRecord::Migrator.new(direction, [@migration], ActiveRecord::SchemaMigration)
+      if Gem::Requirement.new('>=7.1.0').satisfied_by?(Gem::Version.new(::ActiveRecord::VERSION::STRING))
+        ActiveRecord::Migrator.new(direction, [@migration], @schema_migration, @internal_metadata)
+      elsif Gem::Requirement.new('>=6.0.0').satisfied_by?(Gem::Version.new(::ActiveRecord::VERSION::STRING))
+        ActiveRecord::Migrator.new(direction, [@migration], @schema_migration)
       else
         ActiveRecord::Migrator.new(direction, [@migration])
       end