data_migrate 7.0.0 → 8.0.0.rc2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7dd1c0cc04a76e4332a3af19199beab36e329ecb2af1ab8d0fd706d0457e6f93
-  data.tar.gz: 1431e365d30537f284039aa1dd6dd7db12b24f9ec845fc3d0d3448adbec70c1c
+  metadata.gz: '09b6187a1a958297e53be31177698552df42454a2e075fcaf392add7fb9f9085'
+  data.tar.gz: 71af0e938b31d223f04399e771ff1dbbba8cac52b5560ce0305da7935e1c03a0
 SHA512:
-  metadata.gz: c46327a2feee44ac29106baefa254204ce4adfa914e849ae0f63c6b617369ad3c4c00b0ece39f94137b5df2547ceecf73f8d3716539b38476467cac11b9ce664
-  data.tar.gz: d635f7561555a22a3695f60bfcb51eeb93ed89cce158abaeba78f373dbb37746be42ff94fb9277f65d48964887cde7587e215bbda3ee565f3c3aee3e40af6d46
+  metadata.gz: dd9810087eaef6a8fb537401908d945007436b946759284fdfdd5836d9d7632518452039ec1432aab23743b54bd58401fa2e44572af34a87c47e924f7612c71f
+  data.tar.gz: 5b0aa4e76fc0641c3d609f4f434d91905c48a2a281952c9ebc0817aa71cea757bc5c68823351f83b1a2776e11c87085320bbb7a884cb75f29304f0235eab7453

data/.ruby-version

@@ -1 +1 @@
-2.6.1
+2.7.5

data/.travis.yml

@@ -1,17 +1,14 @@
 language: ruby
 rvm:
-  - 2.4.4
-  - 2.5.5
-  - 2.6.2
+  - 2.7
+  - 3.0
 script: bundle exec rspec
 gemfile:
-  - gemfiles/rails_5.1.gemfile
   - gemfiles/rails_5.2.gemfile
   - gemfiles/rails_6.0.gemfile
   - gemfiles/rails_6.1.gemfile
-matrix:
+  - gemfiles/rails_7.0.gemfile
+jobs:
   exclude:
-    - rvm: 2.4.4
-      gemfile: gemfiles/rails_6.0.gemfile
-    - rvm: 2.4.4
-      gemfile: gemfiles/rails_6.1.gemfile
+    - rvm: 3.0
+      gemfile: gemfiles/rails_5.2.gemfile

data/Appraisals

@@ -1,11 +1,3 @@
-appraise 'rails-5.0' do
-  gem 'rails', '5.0.7.2'
-end
-
-appraise 'rails-5.1' do
-  gem 'rails', '5.1.7'
-end
-
 appraise 'rails-5.2' do
   gem 'rails', '~> 5.2.3'
 end
@@ -17,3 +9,7 @@ end
 appraise 'rails-6.1' do
   gem 'rails', '~> 6.1.0'
 end
+
+appraise 'rails-7.0' do
+  gem 'rails', '~> 7.0'
+end

data/Changelog.md

@@ -1,9 +1,24 @@
 # Changelog
 
+## 8.0.0.rc2
+
+Bug fixes [gdott9](https://github.com/gdott9)
+
+## 8.0.0.rc1
+Add support for Rails 7
+Removed support for Rails versions below 5.2. Now are supported only versions Rails 5.2 and up
+
+## 7.0.2
+
+Remove magic comment in migration files [y-yagi](https://github.com/y-yagi)
+User frozen string [jonnay](https://github.com/jonnay)
+## 7.0.1
+Use SchemaMigration.migrations_paths in main rake task [lewhit](https://github.com/lewhit)
+
 ## 6.8.0
 
 Specify database name for migrations_paths [lewhit](https://github.com/lewhit)
-# 6.7.0
+## 6.7.0
 
 Add configuration for which database name is to be used for database migrations [lewhit](https://github.com/lewhit)
 Add tests for Rails 6.1 [lewhit](https://github.com/lewhit)

data/Gemfile

@@ -1,11 +1,4 @@
 source "http://rubygems.org"
 
-# Specify your gem's dependencies in data_migrate.gemspec
-%w[
-  activerecord
-  railties
-].each do |rails_gem|
-  gem rails_gem, '~> 6.0.0'
-end
 gem 'sqlite3', "~> 1.4"
 gemspec

data/README.md

@@ -36,7 +36,7 @@ table to track all migrations.
 
 ## Rails Support
 
-Support Rails 5.0 through 6.0
+Support Rails 5.2 through 7.0
 
 ### Important notes for older versions
 

data/gemfiles/{rails_5.1.gemfile → rails_7.0.gemfile}

@@ -2,7 +2,7 @@
 
 source "http://rubygems.org"
 
-gem "rails", "5.1.7"
 gem "sqlite3", "~> 1.4"
+gem "rails", "~> 7.0"
 
 gemspec path: "../"

data/lib/data_migrate/database_tasks.rb

@@ -23,6 +23,20 @@ module DataMigrate
         ENV["DATA_SCHEMA"] || File.join(schema_location, filename)
       end
 
+      def schema_dump_path(db_config, format = ActiveRecord.schema_format)
+        return ENV["DATA_SCHEMA"] if ENV["DATA_SCHEMA"]
+
+        filename = if db_config.primary?
+          schema_file_type(format)
+        else
+          [db_config.name, schema_file_type(format)].join("_")
+        end
+
+        return unless filename
+
+        File.dirname(filename) == schema_location ? filename : File.join(schema_location, filename)
+      end
+
       def schema_location
         db_dir
       end

data/lib/data_migrate/schema_dumper.rb

@@ -8,7 +8,7 @@ module DataMigrate
     private_class_method :new
 
     class << self
-      def dump(connection = ActiveRecord::Base.connection, stream = STDOUT)
+      def dump(connection = ActiveRecord::Base.connection, stream = $stdout)
         new(connection).dump(stream)
         stream
       end
@@ -17,10 +17,6 @@ module DataMigrate
     def dump(stream)
       define_params = @version ? "version: #{@version}" : ""
 
-      if stream.respond_to?(:external_encoding) && stream.external_encoding
-        stream.puts "# encoding: #{stream.external_encoding.name}"
-      end
-
       stream.puts "DataMigrate::Data.define(#{define_params})"
 
       stream

data/lib/data_migrate/status_service_five.rb

@@ -1,7 +1,7 @@
 module DataMigrate
   class StatusService
     class << self
-      def dump(connection = ActiveRecord::Base.connection, stream = STDOUT)
+      def dump(connection = ActiveRecord::Base.connection, stream = $stdout)
         new(connection).dump(stream)
         stream
       end
@@ -32,7 +32,11 @@ module DataMigrate
       end
 
       # output
-      stream.puts "\ndatabase: #{ActiveRecord::Base.connection_config[:database]}\n\n"
+      if ActiveRecord.version >= Gem::Version.new('7.0')
+        stream.puts "\ndatabase: #{ActiveRecord::Base.connection_db_config.configuration_hash[:database]}\n\n"
+      else
+        stream.puts "\ndatabase: #{ActiveRecord::Base.connection_config[:database]}\n\n"
+      end
       stream.puts "#{'Status'.center(8)}  #{'Migration ID'.ljust(14)}  Migration Name"
       stream.puts "-" * 50
       db_list.each do |status, version, name|

data/lib/data_migrate/tasks/data_migrate_tasks.rb

@@ -2,12 +2,17 @@ module DataMigrate
   module Tasks
     module DataMigrateTasks
       extend self
+
+      def schema_migrations_path
+        File.join('db', 'migrate')
+      end
+
       def migrations_paths
         @migrations_paths ||= DataMigrate.config.data_migrations_path
       end
 
       def dump
-        if ActiveRecord::Base.dump_schema_after_migration
+        if dump_schema_after_migration?
           filename = DataMigrate::DatabaseTasks.schema_file
           ActiveRecord::Base.establish_connection(DataMigrate.config.db_configuration) if DataMigrate.config.db_configuration
           File.open(filename, "w:utf-8") do |file|
@@ -17,14 +22,10 @@ module DataMigrate
       end
 
       def migrate
-        DataMigrate::DataMigrator.assure_data_schema_table
         target_version = ENV["VERSION"] ? ENV["VERSION"].to_i : nil
-        if (Rails::VERSION::MAJOR == 5 && Rails::VERSION::MINOR == 2) ||
-          Rails::VERSION::MAJOR == 6
-          DataMigrate::MigrationContext.new(migrations_paths).migrate(target_version)
-        else
-          DataMigrate::DataMigrator.migrate(migrations_paths, ENV["VERSION"] ? ENV["VERSION"].to_i : nil)
-        end
+
+        DataMigrate::DataMigrator.assure_data_schema_table
+        DataMigrate::MigrationContext.new(migrations_paths).migrate(target_version)
       end
 
       def abort_if_pending_migrations(migrations, message)
@@ -36,6 +37,90 @@ module DataMigrate
           abort message
         end
       end
+
+      def dump_schema_after_migration?
+        if ActiveRecord.respond_to?(:dump_schema_after_migration)
+          ActiveRecord.dump_schema_after_migration
+        else
+          ActiveRecord::Base.dump_schema_after_migration
+        end
+      end
+
+      def status
+        config = connect_to_database
+        return unless config
+
+        connection = ActiveRecord::Base.connection
+        puts "\ndatabase: #{config['database']}\n\n"
+        DataMigrate::StatusService.dump(connection)
+      end
+
+      def status_with_schema
+        config = connect_to_database
+        return unless config
+
+        db_list_data = ActiveRecord::Base.connection.select_values(
+          "SELECT version FROM #{DataMigrate::DataSchemaMigration.table_name}"
+        )
+        db_list_schema = ActiveRecord::Base.connection.select_values(
+          "SELECT version FROM #{ActiveRecord::SchemaMigration.schema_migrations_table_name}"
+        )
+        file_list = []
+
+        Dir.foreach(File.join(Rails.root, migrations_paths)) do |file|
+          # only files matching "20091231235959_some_name.rb" pattern
+          if match_data = /(\d{14})_(.+)\.rb/.match(file)
+            status = db_list_data.delete(match_data[1]) ? 'up' : 'down'
+            file_list << [status, match_data[1], match_data[2], 'data']
+          end
+        end
+
+        Dir.foreach(File.join(Rails.root, schema_migrations_path)) do |file|
+          # only files matching "20091231235959_some_name.rb" pattern
+          if match_data = /(\d{14})_(.+)\.rb/.match(file)
+            status = db_list_schema.delete(match_data[1]) ? 'up' : 'down'
+            file_list << [status, match_data[1], match_data[2], 'schema']
+          end
+        end
+
+        file_list.sort!{|a,b| "#{a[1]}_#{a[3] == 'data' ? 1 : 0}" <=> "#{b[1]}_#{b[3] == 'data' ? 1 : 0}" }
+
+        # output
+        puts "\ndatabase: #{config['database']}\n\n"
+        puts "#{"Status".center(8)} #{"Type".center(8)}  #{"Migration ID".ljust(14)} Migration Name"
+        puts "-" * 60
+        file_list.each do |file|
+          puts "#{file[0].center(8)} #{file[3].center(8)} #{file[1].ljust(14)}  #{file[2].humanize}"
+        end
+        db_list_schema.each do |version|
+          puts "#{'up'.center(8)}  #{version.ljust(14)}  *** NO SCHEMA FILE ***"
+        end
+        db_list_data.each do |version|
+          puts "#{'up'.center(8)}  #{version.ljust(14)}  *** NO DATA FILE ***"
+        end
+        puts
+      end
+
+      private
+
+      def connect_to_database
+        config = if ActiveRecord.version < Gem::Version.new('6.1')
+          ActiveRecord::Base.configurations[Rails.env || 'development']
+        else
+          ActiveRecord::Base.configurations.find_db_config(Rails.env || 'development').configuration_hash
+        end
+        ActiveRecord::Base.establish_connection(config)
+
+        unless DataMigrate::DataSchemaMigration.table_exists?
+          puts 'Data migrations table does not exist yet.'
+          config = nil
+        end
+        unless ActiveRecord::SchemaMigration.table_exists?
+          puts 'Schema migrations table does not exist yet.'
+          config = nil
+        end
+        config
+      end
     end
   end
 end

data/lib/data_migrate/version.rb

@@ -1,3 +1,3 @@
 module DataMigrate
-  VERSION = "7.0.0".freeze
+  VERSION = "8.0.0.rc2".freeze
 end

data/lib/data_migrate.rb

@@ -1,44 +1,22 @@
 # frozen_string_literal: true
 
-if Rails::VERSION::MAJOR == 6
-  require File.join(File.dirname(__FILE__), "data_migrate", "data_migrator_five")
-elsif Rails::VERSION::MAJOR == 5 && Rails::VERSION::MINOR == 2
-  require File.join(File.dirname(__FILE__), "data_migrate", "data_migrator_five")
-else
-  require File.join(File.dirname(__FILE__), "data_migrate", "data_migrator")
-end
-require File.join(File.dirname(__FILE__), "data_migrate",
-                  "data_schema_migration")
+require File.join(File.dirname(__FILE__), "data_migrate", "data_migrator_five")
+require File.join(File.dirname(__FILE__), "data_migrate", "data_schema_migration")
 require File.join(File.dirname(__FILE__), "data_migrate", "data_schema")
 require File.join(File.dirname(__FILE__), "data_migrate", "database_tasks")
 require File.join(File.dirname(__FILE__), "data_migrate", "schema_dumper")
-if Rails::VERSION::MAJOR == 6
-  require File.join(File.dirname(__FILE__), "data_migrate", "status_service_five")
-  require File.join(File.dirname(__FILE__), "data_migrate", "schema_migration_six")
-elsif Rails::VERSION::MAJOR == 5 &&  Rails::VERSION::MINOR == 2
-  require File.join(File.dirname(__FILE__), "data_migrate", "status_service_five")
-  require File.join(File.dirname(__FILE__), "data_migrate", "schema_migration_five")
-else
-  require File.join(File.dirname(__FILE__), "data_migrate", "status_service")
-  require File.join(File.dirname(__FILE__), "data_migrate", "schema_migration")
-end
-
-if Rails::VERSION::MAJOR == 6
-  require File.join(File.dirname(__FILE__), "data_migrate", "migration_context")
-  # require File.join(File.dirname(__FILE__), "data_migrate", "migration_five")
-elsif Rails::VERSION::MAJOR == 5
-  if Rails::VERSION::MINOR == 2
-    require File.join(File.dirname(__FILE__), "data_migrate", "migration_context")
-  else
-    require File.join(File.dirname(__FILE__), "data_migrate", "migration_five")
-  end
-else
-  require File.join(File.dirname(__FILE__), "data_migrate", "migration")
-end
+require File.join(File.dirname(__FILE__), "data_migrate", "status_service_five")
+require File.join(File.dirname(__FILE__), "data_migrate", "migration_context")
 require File.join(File.dirname(__FILE__), "data_migrate", "railtie")
 require File.join(File.dirname(__FILE__), "data_migrate", "tasks/data_migrate_tasks")
 require File.join(File.dirname(__FILE__), "data_migrate", "legacy_migrator")
 require File.join(File.dirname(__FILE__), "data_migrate", "config")
 
+if Rails::VERSION::MAJOR == 5
+  require File.join(File.dirname(__FILE__), "data_migrate", "schema_migration_five")
+else
+  require File.join(File.dirname(__FILE__), "data_migrate", "schema_migration_six")
+end
+
 module DataMigrate
 end

data/lib/generators/data_migration/data_migration_generator.rb

@@ -27,13 +27,7 @@ module DataMigrate
       end
 
       def migration_base_class_name
-        if ActiveRecord.version >= Gem::Version.new("5.0")
-          "ActiveRecord::Migration[#{ActiveRecord::Migration.current_version}]"
-        elsif ActiveRecord.version >= Gem::Version.new("5.2")
-          "DataMigrate::MigrationContext"
-        else
-          "ActiveRecord::Migration"
-        end
+        "ActiveRecord::Migration[#{ActiveRecord::Migration.current_version}]"
       end
 
       def data_migrations_file_path

data/lib/generators/data_migration/templates/data_migration.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 class <%= migration_class_name %> < <%= migration_base_class_name %>
   def up
   end

data/spec/data_migrate/database_tasks_spec.rb

@@ -5,16 +5,10 @@ require "spec_helper"
 describe DataMigrate::DatabaseTasks do
   let(:subject) { DataMigrate::DatabaseTasks }
   let(:migration_path) {
-    if Rails::VERSION::MAJOR == 6
-      "spec/db/migrate/6.0"
-    elsif Rails::VERSION::MAJOR == 5
-      if Rails::VERSION::MINOR == 2
-        "spec/db/migrate/5.2"
-      else
-        "spec/db/migrate/5.0"
-      end
+    if Rails::VERSION::MAJOR == 5
+      "spec/db/migrate/5.2"
     else
-      "spec/db/migrate/4.2"
+      "spec/db/migrate/6.0"
     end
   }
   let(:data_migrations_path) {
@@ -32,6 +26,7 @@ describe DataMigrate::DatabaseTasks do
     # Rails.application.config.paths["db"].first
     # @see https://github.com/rails/rails/blob/a7d49ef78c36df2d1ca876451f30915ada1079a5/activerecord/lib/active_record/tasks/database_tasks.rb#L54
     allow(subject).to receive(:db_dir).and_return("db")
+    allow(ActiveRecord::Tasks::DatabaseTasks).to receive(:db_dir).and_return("db")
   end
 
   before do

data/spec/data_migrate/migration_context_spec.rb

@@ -2,28 +2,20 @@ require "spec_helper"
 
 describe DataMigrate::DataMigrator do
   let(:context) {
-    if (Rails::VERSION::MAJOR == 6)
-      DataMigrate::MigrationContext.new("spec/db/data-6.0")
-    else
+    if (Rails::VERSION::MAJOR == 5)
       DataMigrate::MigrationContext.new("spec/db/data")
+    else
+      DataMigrate::MigrationContext.new("spec/db/data-6.0")
     end
   }
   let(:schema_context) {
-    if (Rails::VERSION::MAJOR == 6)
-      ActiveRecord::MigrationContext.new("spec/db/migrate/6.0", ActiveRecord::Base.connection.schema_migration)
-    else
+    if (Rails::VERSION::MAJOR == 5)
       ActiveRecord::MigrationContext.new("spec/db/migrate/5.2")
+    else
+      ActiveRecord::MigrationContext.new("spec/db/migrate/6.0", ActiveRecord::Base.connection.schema_migration)
     end
   }
 
-  before do
-    unless (Rails::VERSION::MAJOR == 5 and
-           Rails::VERSION::MINOR == 2) ||
-           Rails::VERSION::MAJOR == 6
-      skip("Tests are only applicable for Rails 5.2")
-    end
-  end
-
   after do
     begin
       ActiveRecord::Migration.drop_table("data_migrations")

data/spec/data_migrate/schema_migration_spec.rb

@@ -4,16 +4,10 @@ require "spec_helper"
 
 describe DataMigrate::SchemaMigration do
   let(:migration_path) {
-    if Rails::VERSION::MAJOR == 6
-      "spec/db/migrate/6.0"
-    elsif Rails::VERSION::MAJOR == 5
-      if Rails::VERSION::MINOR == 2
-        "spec/db/migrate/5.2"
-      else
-        "spec/db/migrate/5.0"
-      end
+    if Rails::VERSION::MAJOR == 5
+      "spec/db/migrate/5.2"
     else
-      "spec/db/migrate/4.2"
+      "spec/db/migrate/6.0"
     end
   }
 
@@ -72,22 +66,27 @@ describe DataMigrate::SchemaMigration do
   if Rails.version > '6'
     describe :migrations_paths do
       context 'when a db_name is configured' do
+        let(:config) { double(:config) }
         let(:paths) { ['spec/db/migrate/6.0', 'spec/db/components/migrate/6.0'] }
-
-        if Rails.version > '6.1'
-          before do
-            allow(ActiveRecord::Base.configurations.configs_for(env_name: 'test', name: 'primary')).to receive(:migrations_paths).and_return(paths)
-            DataMigrate.configure do |config|
-              config.spec_name = 'primary'
-            end
+        let(:config_options) do
+          if Rails.version > '6.1'
+            { env_name: 'test', name: 'primary' }
+          else
+            { env_name: 'test', spec_name: 'primary' }
           end
-        else
-          before do
-            allow(ActiveRecord::Base.configurations.configs_for(env_name: 'test', spec_name: 'primary')).to receive(:migrations_paths).and_return(paths)
-            DataMigrate.configure do |config|
-              config.spec_name = 'primary'
-            end
+        end
+
+        before do
+          DataMigrate.configure do |config|
+            config.spec_name = 'primary'
           end
+
+          allow(ActiveRecord::Base.configurations)
+            .to receive(:configs_for)
+            .with(config_options)
+            .and_return(config)
+
+          allow(config).to receive(:migrations_paths).and_return(paths)
         end
 
         it 'lists schema migration paths' do

data/spec/data_migrate/tasks/data_migrate_tasks_spec.rb

@@ -115,4 +115,50 @@ describe DataMigrate::Tasks::DataMigrateTasks do
       end
     end
   end
+
+  describe :status do
+    let(:db_config) do
+      {
+        adapter: "sqlite3",
+        database: "spec/db/test.db"
+      }
+    end
+
+    before do
+      if Rails::VERSION::MAJOR == 5
+        ActiveRecord::Base.configurations['test'] = db_config
+      else
+        hash_config = ActiveRecord::DatabaseConfigurations::HashConfig.new('test', 'test', db_config)
+        config_obj = ActiveRecord::DatabaseConfigurations.new([hash_config])
+        allow(ActiveRecord::Base).to receive(:configurations).and_return(config_obj)
+      end
+
+      allow(Rails).to receive(:root) { '.' }
+
+      if Rails::VERSION::MAJOR == 5
+        allow(DataMigrate::Tasks::DataMigrateTasks).to receive(:schema_migrations_path) { 'spec/db/migrate/5.2' }
+      else
+        allow(DataMigrate::Tasks::DataMigrateTasks).to receive(:schema_migrations_path) { 'spec/db/migrate/6.0' }
+      end
+
+      DataMigrate::Tasks::DataMigrateTasks.migrate
+    end
+
+    after do
+      ActiveRecord::Migration.drop_table("data_migrations")
+    end
+
+    it "should display data migration status" do
+      expect {
+        DataMigrate::Tasks::DataMigrateTasks.status
+      }.to output(/up     20091231235959  Some name/).to_stdout
+    end
+
+    it "should display schema and data migration status" do
+      expect {
+        DataMigrate::Tasks::DataMigrateTasks.status_with_schema
+      }.to output(match(/up      data   20091231235959  Some name/)
+        .and match(/down    schema  20131111111111  Late migration/)).to_stdout
+    end
+  end
 end

data/spec/db/data/partial_schema/data_schema.rb

@@ -1,2 +1 @@
-# encoding: UTF-8
 DataMigrate::Data.define(version: 20091231235959)

data/spec/db/data/partial_schema/test_data_schema.rb

@@ -1,2 +1 @@
-# encoding: UTF-8
 DataMigrate::Data.define(version: 20091231235959)

data/spec/db/data/schema/data_schema.rb

@@ -1,2 +1 @@
-# encoding: UTF-8
 DataMigrate::Data.define(version: 20171231235959)

data/spec/db/data/schema/test_data_schema.rb

@@ -1,2 +1 @@
-# encoding: UTF-8
 DataMigrate::Data.define(version: 20171231235959)

data/spec/generators/data_migration/data_migration_generator_spec.rb

@@ -8,8 +8,12 @@ describe DataMigrate::Generators::DataMigrationGenerator do
   describe :next_migration_number do
     it "next migration" do
       Timecop.freeze("2016-12-03 22:15:26 -0800") do
-        expect(ActiveRecord::Base).to receive(:timestamped_migrations) { true }
-        expect(subject.next_migration_number(1)).to eq("20161204061526")
+        if ActiveRecord.version >= Gem::Version.new('7.0')
+          expect(ActiveRecord).to receive(:timestamped_migrations) { true }
+        else
+          expect(ActiveRecord::Base).to receive(:timestamped_migrations) { true }
+        end
+        expect(subject.next_migration_number(1).to_s).to eq("20161204061526")
       end
     end
   end
@@ -17,11 +21,7 @@ describe DataMigrate::Generators::DataMigrationGenerator do
   describe :migration_base_class_name do
     let(:subject) { DataMigrate::Generators::DataMigrationGenerator.new(['my_migration']) }
     it "returns the correct base class name" do
-      if ActiveRecord.version >= Gem::Version.new('5.0')
-        expect(subject.send(:migration_base_class_name)).to eq("ActiveRecord::Migration[#{ActiveRecord::Migration.current_version}]")
-      else
-        expect(subject.send(:migration_base_class_name)).to eq('ActiveRecord::Migration')
-      end
+      expect(subject.send(:migration_base_class_name)).to eq("ActiveRecord::Migration[#{ActiveRecord::Migration.current_version}]")
     end
   end
 

data/spec/spec_helper.rb

@@ -20,23 +20,13 @@ RSpec.configure do |config|
     if example.metadata[:no_override]
     else
       @prev_data_migrations_path = DataMigrate.config.data_migrations_path
-      if Rails::VERSION::MAJOR == 6
+      if Rails::VERSION::MAJOR == 5
         DataMigrate.configure do |config|
-          config.data_migrations_path = "spec/db/6.0"
-        end
-      elsif Rails::VERSION::MAJOR == 5
-        if Rails::VERSION::MINOR == 2
-          DataMigrate.configure do |config|
-            config.data_migrations_path = "spec/db/data"
-          end
-        else
-          DataMigrate.configure do |config|
-            config.data_migrations_path = "spec/db/5.0"
-          end
+          config.data_migrations_path = "spec/db/data"
         end
       else
         DataMigrate.configure do |config|
-          config.data_migrations_path = "spec/db/5.0"
+          config.data_migrations_path = "spec/db/6.0"
         end
       end
     end