data_migrate 9.4.2 → 10.0.0.rc1

data/spec/data_migrate/database_tasks_spec.rb

@@ -8,6 +8,12 @@ describe DataMigrate::DatabaseTasks do
   let(:data_migrations_path) {
     DataMigrate.config.data_migrations_path
   }
+  let(:db_config) do
+    {
+      adapter: "sqlite3",
+      database: "spec/db/test.db"
+    }
+  end
 
   before do
     # In a normal Rails installation, db_dir would defer to
@@ -21,10 +27,14 @@ describe DataMigrate::DatabaseTasks do
     allow(DataMigrate::Tasks::DataMigrateTasks).to receive(:migrations_paths) {
       data_migrations_path
     }
-    ActiveRecord::Base.establish_connection({ adapter: "sqlite3", database: "spec/db/test.db" })
-    hash_config = ActiveRecord::DatabaseConfigurations::HashConfig.new('test', 'test', { adapter: "sqlite3", database: "spec/db/test.db" })
-    config_obj = ActiveRecord::DatabaseConfigurations.new([hash_config])
-    allow(ActiveRecord::Base).to receive(:configurations).and_return(config_obj)
+    ActiveRecord::Base.establish_connection(db_config)
+    if Gem::Dependency.new("rails", ">= 6.1").match?("rails", Gem.loaded_specs["rails"].version)
+      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)
+    else
+      ActiveRecord::Base.configurations[:test] = db_config
+    end
   end
 
   context "migrations" do
@@ -34,7 +44,7 @@ describe DataMigrate::DatabaseTasks do
     end
 
     before do
-      DataMigrate::RailsHelper.schema_migration.create_table
+      ActiveRecord::SchemaMigration.create_table
 
       allow(DataMigrate::SchemaMigration).to receive(:migrations_paths) {
         migration_path
@@ -59,47 +69,22 @@ describe DataMigrate::DatabaseTasks do
     end
 
     describe :forward do
+
       it "run forward default amount of times" do
         subject.forward
-        versions = DataMigrate::RailsHelper.data_schema_migration.normalized_versions
+        versions = DataMigrate::DataSchemaMigration.normalized_versions
         expect(versions.count).to eq(1)
       end
 
       it "run forward defined number of times" do
         subject.forward(2)
-        versions = DataMigrate::RailsHelper.data_schema_migration.normalized_versions
+        versions = DataMigrate::DataSchemaMigration.normalized_versions
         expect(versions.count).to eq(1)
         expect(versions.first).to eq "20091231235959"
-        versions = DataMigrate::RailsHelper.schema_migration.normalized_versions
+        versions = ActiveRecord::SchemaMigration.normalized_versions
         expect(versions.count).to eq(1)
         expect(versions.first).to eq "20131111111111"
       end
     end
-
-    if DataMigrate::RailsHelper.rails_version_equal_to_or_higher_than_7_0
-      describe :schema_dump_path do
-        before do
-          allow(ActiveRecord::Base).to receive(:configurations).and_return(ActiveRecord::DatabaseConfigurations.new([db_config]))
-        end
-
-        context "for primary database" do
-          let(:db_config) { ActiveRecord::DatabaseConfigurations::HashConfig.new("development", "primary", {} ) }
-
-          context "for :ruby db format" do
-            it 'returns the data schema path' do
-              allow(ActiveRecord).to receive(:schema_format).and_return(:ruby)
-              expect(subject.schema_dump_path(db_config)).to eq("db/data_schema.rb")
-            end
-          end
-
-          context "for :sql db format" do
-            it 'returns the data schema path' do
-              allow(ActiveRecord).to receive(:schema_format).and_return(:sql)
-              expect(subject.schema_dump_path(db_config, :sql)).to eq("db/data_schema.rb")
-            end
-          end
-        end
-      end
-    end
   end
 end

data/spec/data_migrate/legacy_migrator_spec.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+require "spec_helper"
+
+describe DataMigrate::LegacyMigrator do
+  let(:context) {
+    DataMigrate::MigrationContext.new("spec/db/data")
+  }
+
+  before do
+    ActiveRecord::SchemaMigration.create_table
+    DataMigrate::DataSchemaMigration.create_table
+  end
+
+  after do
+    ActiveRecord::Migration.drop_table("data_migrations") rescue nil
+    ActiveRecord::Migration.drop_table("schema_migrations") rescue nil
+  end
+
+  it "migrate legacy migrations to be in correct table" do
+    # simulate creation of legacy data migration when
+    # it was recorded in schema table
+    ActiveRecord::SchemaMigration.create(version: "20091231235959")
+
+    # create one migration in correct place
+    DataMigrate::DataSchemaMigration.create(version: "20171231235959")
+
+    migrated = DataMigrate::DataMigrator.new(:up, []).load_migrated
+    expect(migrated.count).to eq 1
+
+    DataMigrate::LegacyMigrator.new("spec/db/data").migrate
+
+    # after migacy migrator has been run, we should have records
+    # of both migrations
+    migrated = DataMigrate::DataMigrator .new(:up, []).load_migrated
+    expect(migrated.count).to eq 2
+  end
+end

data/spec/data_migrate/migration_context_spec.rb

@@ -5,17 +5,10 @@ require "spec_helper"
 describe DataMigrate::DataMigrator do
   let(:context) { DataMigrate::MigrationContext.new("spec/db/data") }
   let(:schema_context) { ActiveRecord::MigrationContext.new("spec/db/migrate", ActiveRecord::Base.connection.schema_migration) }
-  let(:db_config) do
-    {
-      adapter: "sqlite3",
-      database: "spec/db/test.db"
-    }
-  end
 
   before do
-    ActiveRecord::Base.establish_connection(db_config)
-    DataMigrate::RailsHelper.schema_migration.create_table
-    DataMigrate::RailsHelper.data_schema_migration.create_table
+    ActiveRecord::SchemaMigration.create_table
+    DataMigrate::DataSchemaMigration.create_table
   end
 
   after do
@@ -27,7 +20,7 @@ describe DataMigrate::DataMigrator do
     it "migrates existing file" do
       context.migrate(nil)
       context.migrations_status
-      versions = DataMigrate::RailsHelper.data_schema_migration.normalized_versions
+      versions = DataMigrate::DataSchemaMigration.normalized_versions
       expect(versions.count).to eq(2)
       expect(versions).to include("20091231235959")
       expect(versions).to include("20171231235959")
@@ -36,7 +29,7 @@ describe DataMigrate::DataMigrator do
     it "undo migration" do
       context.migrate(nil)
       context.run(:down, 20171231235959)
-      versions = DataMigrate::RailsHelper.data_schema_migration.normalized_versions
+      versions = DataMigrate::DataSchemaMigration.normalized_versions
       expect(versions.count).to eq(1)
       expect(versions).to include("20091231235959")
     end
@@ -53,7 +46,7 @@ describe DataMigrate::DataMigrator do
 
     it "runs a specific migration" do
       context.run(:up, 20171231235959)
-      versions = DataMigrate::RailsHelper.data_schema_migration.normalized_versions
+      versions = DataMigrate::DataSchemaMigration.normalized_versions
       expect(versions.count).to eq(1)
       expect(versions).to include("20171231235959")
     end
@@ -81,7 +74,7 @@ describe DataMigrate::DataMigrator do
       expect {
         context.rollback
       }.to output(/Undoing SuperUpdate/).to_stdout
-      versions = DataMigrate::RailsHelper.data_schema_migration.normalized_versions
+      versions = DataMigrate::DataSchemaMigration.normalized_versions
       expect(versions.count).to eq(1)
       expect(versions).to include("20091231235959")
     end
@@ -92,7 +85,7 @@ describe DataMigrate::DataMigrator do
       expect {
         context.rollback(2)
       }.to output(/Undoing SomeName/).to_stdout
-      versions = DataMigrate::RailsHelper.data_schema_migration.normalized_versions
+      versions = DataMigrate::DataSchemaMigration.normalized_versions
       expect(versions.count).to eq(0)
     end
 
@@ -101,7 +94,7 @@ describe DataMigrate::DataMigrator do
       expect {
         context.rollback(2)
       }.to output(/Undoing SomeName/).to_stdout
-      versions = DataMigrate::RailsHelper.data_schema_migration.normalized_versions
+      versions = DataMigrate::DataSchemaMigration.normalized_versions
       expect(versions.count).to eq(0)
     end
   end

data/spec/data_migrate/schema_dumper_spec.rb

@@ -9,12 +9,9 @@ describe DataMigrate::SchemaDumper do
   end
 
   before do
-    DataMigrate::RailsHelper.schema_migration.create_table
-    DataMigrate::RailsHelper.data_schema_migration.create_table
-
-    fixture_file_timestamps.map do |t|
-      DataMigrate::RailsHelper.data_schema_migration.create_version(t)
-    end
+    ActiveRecord::SchemaMigration.create_table
+    DataMigrate::DataSchemaMigration.create_table
+    DataMigrate::DataSchemaMigration.create(fixture_file_timestamps.map { |t| { version: t } })
   end
 
   after do

data/spec/data_migrate/schema_migration_spec.rb

@@ -8,17 +8,10 @@ describe DataMigrate::SchemaMigration do
   let(:fixture_file_timestamps) do
     %w[20091231235959 20101231235959 20111231235959]
   end
-  let(:db_config) do
-    {
-      adapter: "sqlite3",
-      database: "spec/db/test.db"
-    }
-  end
 
   before do
-    ActiveRecord::Base.establish_connection(db_config)
-    DataMigrate::RailsHelper.schema_migration.create_table
-    DataMigrate::RailsHelper.data_schema_migration.create_table
+    ActiveRecord::SchemaMigration.create_table
+    DataMigrate::DataSchemaMigration.create_table
   end
 
   after do
@@ -45,7 +38,7 @@ describe DataMigrate::SchemaMigration do
       expect {
         subject.run(:up, migration_path, 20202020202011)
       }.to output(/20202020202011 DbMigration: migrating/).to_stdout
-      versions = DataMigrate::RailsHelper.schema_migration.normalized_versions
+      versions = ActiveRecord::SchemaMigration.normalized_versions
       expect(versions.first).to eq("20202020202011")
     end
 
@@ -56,7 +49,7 @@ describe DataMigrate::SchemaMigration do
         subject.run(:down, migration_path, 20202020202011)
       }.to output(/Undoing DbMigration/).to_stdout
 
-      versions = DataMigrate::RailsHelper.schema_migration.normalized_versions
+      versions = ActiveRecord::SchemaMigration.normalized_versions
 
       expect(versions.count).to eq(0)
     end
@@ -68,9 +61,9 @@ describe DataMigrate::SchemaMigration do
       let(:paths) { ['spec/db/migrate', 'spec/db/migrate/other'] }
       let(:specification_name) { "primary" }
       let(:config_options) do
-        if Gem::Dependency.new("railties", "~> 6.1").match?("railties", Gem.loaded_specs["railties"].version)
+        if Gem::Dependency.new("rails", "~> 6.0").match?("rails", Gem.loaded_specs["rails"].version)
           { env_name: Rails.env, spec_name: specification_name }
-        else
+        elsif Gem::Dependency.new("rails", "~> 7.0").match?("rails", Gem.loaded_specs["rails"].version)
           { env_name: Rails.env, name: specification_name }
         end
       end

data/spec/data_migrate/status_service_spec.rb

@@ -7,7 +7,7 @@ describe DataMigrate::StatusService do
   let(:stream) { StringIO.new }
   let(:stream_data) { stream.read }
   let(:connection_db_config) do
-    if Gem::Dependency.new("railties", ">= 6.1").match?("railties", Gem.loaded_specs["railties"].version)
+    if Gem::Dependency.new("rails", ">= 6.1").match?("rails", Gem.loaded_specs["rails"].version)
       ActiveRecord::Base.connection_db_config
     else
       ActiveRecord::Base.configurations.configs_for.first
@@ -33,12 +33,9 @@ describe DataMigrate::StatusService do
     end
 
     before do
-      DataMigrate::RailsHelper.schema_migration.create_table
-      DataMigrate::RailsHelper.data_schema_migration.create_table
-
-      fixture_file_timestamps.map do |t|
-        DataMigrate::RailsHelper.data_schema_migration.create_version(t)
-      end
+      ActiveRecord::SchemaMigration.create_table
+      DataMigrate::DataSchemaMigration.create_table
+      DataMigrate::DataSchemaMigration.create(fixture_file_timestamps.map { |t| { version: t } })
 
       subject.dump(connection_db_config, stream)
       stream.rewind

data/spec/data_migrate/tasks/data_migrate_tasks_spec.rb

@@ -3,17 +3,17 @@
 require "spec_helper"
 
 describe DataMigrate::Tasks::DataMigrateTasks do
-  let(:db_config) do
-    {
-      adapter: "sqlite3",
-      database: "spec/db/test.db"
-    }
+  let(:connection_db_config) do
+    if Gem::Dependency.new("rails", ">= 6.1").match?("rails", Gem.loaded_specs["rails"].version)
+      ActiveRecord::Base.connection_db_config
+    else
+      ActiveRecord::Base.configurations.configs_for.first
+    end
   end
 
   before do
-    ActiveRecord::Base.establish_connection(db_config)
-    DataMigrate::RailsHelper.schema_migration.create_table
-    DataMigrate::RailsHelper.data_schema_migration.create_table
+    ActiveRecord::SchemaMigration.create_table
+    DataMigrate::DataSchemaMigration.create_table
   end
 
   after do
@@ -28,11 +28,11 @@ describe DataMigrate::Tasks::DataMigrateTasks do
     end
 
     context 'when not given a separate db config' do
-      it 'does not override the default connection' do
+      it 'does not override the default connection' do  
         expect(ActiveRecord::Base).not_to receive(:establish_connection)
         expect(DataMigrate::SchemaDumper).to receive(:dump)
 
-        DataMigrate::Tasks::DataMigrateTasks.dump
+        DataMigrate::Tasks::DataMigrateTasks.dump(connection_db_config)
       end
     end
 
@@ -56,7 +56,8 @@ describe DataMigrate::Tasks::DataMigrateTasks do
 
       it 'overrides the default connection' do
         expect(ActiveRecord::Base).to receive(:establish_connection).with(override_config)
-        DataMigrate::Tasks::DataMigrateTasks.dump
+
+        DataMigrate::Tasks::DataMigrateTasks.dump(connection_db_config)
       end
     end
   end
@@ -115,13 +116,13 @@ describe DataMigrate::Tasks::DataMigrateTasks do
 
     it "should display data migration status" do
       expect {
-        DataMigrate::Tasks::DataMigrateTasks.status
+        DataMigrate::Tasks::DataMigrateTasks.status(connection_db_config)
       }.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
+        DataMigrate::Tasks::DataMigrateTasks.status_with_schema(connection_db_config)
       }.to output(match(/up      data   20091231235959  Some name/)
         .and match(/down    schema  20131111111111  Late migration/)).to_stdout
     end

data/spec/db/data/20091231235959_some_name.rb

@@ -1,4 +1,4 @@
-class SomeName < ActiveRecord::Migration[6.1]
+class SomeName < ActiveRecord::Migration[6.0]
   def up
     puts "Doing data migration"
   end

data/spec/db/data/20171231235959_super_update.rb

@@ -1,4 +1,4 @@
-class SuperUpdate < ActiveRecord::Migration[6.1]
+class SuperUpdate < ActiveRecord::Migration[6.0]
   def up
     puts "Doing SuperUpdate"
   end

data/spec/db/migrate/20131111111111_late_migration.rb

@@ -1,4 +1,4 @@
-class LateMigration < ActiveRecord::Migration[6.1]
+class LateMigration < ActiveRecord::Migration[6.0]
   def up
     puts "Doing schema LateMigration"
   end

data/spec/db/migrate/20202020202011_db_migration.rb

@@ -1,4 +1,4 @@
-class DbMigration < ActiveRecord::Migration[6.1]
+class DbMigration < ActiveRecord::Migration[6.0]
   def up
     puts "Doing schema migration"
   end

data/tasks/databases.rake

@@ -5,16 +5,66 @@ require 'data_migrate/tasks/data_migrate_tasks'
 namespace :db do
   namespace :migrate do
     desc "Migrate the database data and schema (options: VERSION=x, VERBOSE=false)."
-    task :with_data => :load_config do
-      DataMigrate::DatabaseTasks.migrate_with_data
+    task :with_data => :environment do
+      original_db_config = ActiveRecord::Base.connection_db_config
+      ActiveRecord::Base.configurations.configs_for(env_name: ActiveRecord::Tasks::DatabaseTasks.env).each do |db_config|
+        ActiveRecord::Base.establish_connection(db_config)
+        DataMigrate::DataMigrator.assure_data_schema_table
+
+      ActiveRecord::Migration.verbose = ENV["VERBOSE"] ? ENV["VERBOSE"] == "true" : true
+      target_version = ENV["VERSION"] ? ENV["VERSION"].to_i : nil
+      migrations = []
+
+      if target_version.nil?
+        migrations = DataMigrate::DatabaseTasks.pending_migrations.map{ |m| m.merge(:direction =>:up) }
+      else
+        current_schema_version = ActiveRecord::Migrator.current_version
+        schema_migrations = if target_version > current_schema_version
+                              DataMigrate::DatabaseTasks.pending_schema_migrations.keep_if{ |m| m[:version] <= target_version }.map{ |m| m.merge(:direction =>:up) }
+                            elsif target_version < current_schema_version
+                              DataMigrate::DatabaseTasks.past_migrations.keep_if{ |m| m[:version] > target_version }.map{ |m| m.merge(:direction =>:down) }
+                            else # ==
+                              []
+                            end
+
+        current_data_version = DataMigrate::DataMigrator.current_version
+        data_migrations = if target_version > current_data_version
+                            DataMigrate::DatabaseTasks.pending_data_migrations.keep_if{ |m| m[:version] <= target_version }.map{ |m| m.merge(:direction =>:up) }
+                          elsif target_version < current_data_version
+                            DataMigrate::DatabaseTasks.past_migrations.keep_if{ |m| m[:version] > target_version }.map{ |m| m.merge(:direction =>:down) }
+                          else # ==
+                            []
+                          end
+        migrations = if schema_migrations.empty?
+                       data_migrations
+                     elsif data_migrations.empty?
+                       schema_migrations
+                     elsif target_version > current_data_version && target_version > current_schema_version
+                       DataMigrate::DatabaseTasks.sort_migrations data_migrations, schema_migrations
+                     elsif target_version < current_data_version && target_version < current_schema_version
+                       DataMigrate::DatabaseTasks.sort_migrations(data_migrations, schema_migrations).reverse
+                     elsif target_version > current_data_version && target_version < current_schema_version
+                       schema_migrations + data_migrations
+                     elsif target_version < current_data_version && target_version > current_schema_version
+                       schema_migrations + data_migrations
+                     end
+      end
+
+      migrations.each do |migration|
+        DataMigrate::DatabaseTasks.run_migration(migration, migration[:direction])
+      end
+      end
+
       Rake::Task["db:_dump"].invoke
       Rake::Task["data:dump"].invoke
+    ensure
+      ActiveRecord::Base.establish_connection(original_db_config)
     end
 
     namespace :redo do
       desc 'Rollbacks the database one migration and re migrate up (options: STEP=x, VERSION=x).'
       task :with_data => :environment do
-        DataMigrate::DataMigrator.create_data_schema_table
+      DataMigrate::DataMigrator.assure_data_schema_table
         if ENV["VERSION"]
           Rake::Task["db:migrate:down:with_data"].invoke
           Rake::Task["db:migrate:up:with_data"].invoke
@@ -30,7 +80,7 @@ namespace :db do
       task :with_data => :environment do
         version = ENV["VERSION"] ? ENV["VERSION"].to_i : nil
         raise "VERSION is required" unless version
-        DataMigrate::DataMigrator.create_data_schema_table
+        DataMigrate::DataMigrator.assure_data_schema_table
         run_both = ENV["BOTH"] == "true"
         migrations = DataMigrate::DatabaseTasks.pending_migrations.keep_if{|m| m[:version] == version}
 
@@ -52,7 +102,7 @@ namespace :db do
       task :with_data => :environment do
         version = ENV["VERSION"] ? ENV["VERSION"].to_i : nil
         raise "VERSION is required" unless version
-        DataMigrate::DataMigrator.create_data_schema_table
+        DataMigrate::DataMigrator.assure_data_schema_table
         run_both = ENV["BOTH"] == "true"
         migrations = DataMigrate::DatabaseTasks.past_migrations.keep_if{|m| m[:version] == version}
 
@@ -72,7 +122,10 @@ namespace :db do
     namespace :status do
       desc "Display status of data and schema migrations"
       task :with_data => :environment do
-        DataMigrate::Tasks::DataMigrateTasks.status_with_schema
+        ActiveRecord::Base.configurations.configs_for(env_name: ActiveRecord::Tasks::DatabaseTasks.env).each do |db_config|
+          ActiveRecord::Base.establish_connection(db_config)
+          DataMigrate::Tasks::DataMigrateTasks.status_with_schema(db_config)
+        end
       end
     end
   end # END OF MIGRATE NAME SPACE
@@ -81,7 +134,7 @@ namespace :db do
     desc 'Rolls the schema back to the previous version (specify steps w/ STEP=n).'
     task :with_data => :environment do
       step = ENV['STEP'] ? ENV['STEP'].to_i : 1
-      DataMigrate::DataMigrator.create_data_schema_table
+      DataMigrate::DataMigrator.assure_data_schema_table
       DataMigrate::DatabaseTasks.past_migrations[0..(step - 1)].each do | past_migration |
         DataMigrate::DatabaseTasks.run_migration(past_migration, :down)
       end
@@ -94,7 +147,7 @@ namespace :db do
   namespace :forward do
     desc 'Pushes the schema to the next version (specify steps w/ STEP=n).'
     task :with_data => :environment do
-      DataMigrate::DataMigrator.create_data_schema_table
+      DataMigrate::DataMigrator.assure_data_schema_table
       step = ENV['STEP'] ? ENV['STEP'].to_i : 1
       DataMigrate::DatabaseTasks.forward(step)
       Rake::Task["db:_dump"].invoke
@@ -105,7 +158,7 @@ namespace :db do
   namespace :version do
     desc "Retrieves the current schema version numbers for data and schema migrations"
     task :with_data => :environment do
-      DataMigrate::DataMigrator.create_data_schema_table
+      DataMigrate::DataMigrator.assure_data_schema_table
       puts "Current Schema version: #{ActiveRecord::Migrator.current_version}"
       puts "Current Data version: #{DataMigrate::DataMigrator.current_version}"
     end
@@ -146,26 +199,25 @@ namespace :db do
       end
     end
   end
-
-  namespace :prepare do
-    desc "Runs setup if database does not exist, or runs data and schema migrations if it does"
-    task with_data: :environment do
-      DataMigrate::DatabaseTasks.prepare_all_with_data
-    end
-  end
 end
 
 namespace :data do
   desc 'Migrate data migrations (options: VERSION=x, VERBOSE=false)'
   task :migrate => :environment do
-    DataMigrate::Tasks::DataMigrateTasks.migrate
+    original_db_config = ActiveRecord::Base.connection_db_config
+    ActiveRecord::Base.configurations.configs_for(env_name: ActiveRecord::Tasks::DatabaseTasks.env).each do |db_config|
+      ActiveRecord::Base.establish_connection(db_config)
+      DataMigrate::Tasks::DataMigrateTasks.migrate
+    end
     Rake::Task["data:dump"].invoke
+  ensure
+    ActiveRecord::Base.establish_connection(original_db_config)
   end
 
   namespace :migrate do
     desc  'Rollbacks the database one migration and re migrate up (options: STEP=x, VERSION=x).'
     task :redo => :environment do
-      DataMigrate::DataMigrator.create_data_schema_table
+      DataMigrate::DataMigrator.assure_data_schema_table
       if ENV["VERSION"]
         Rake::Task["data:migrate:down"].invoke
         Rake::Task["data:migrate:up"].invoke
@@ -177,7 +229,7 @@ namespace :data do
 
     desc 'Runs the "up" for a given migration VERSION.'
     task :up => :environment do
-      DataMigrate::DataMigrator.create_data_schema_table
+      DataMigrate::DataMigrator.assure_data_schema_table
       version = ENV["VERSION"] ? ENV["VERSION"].to_i : nil
       raise "VERSION is required" unless version
       DataMigrate::DataMigrator.run(:up, DataMigrate::DatabaseTasks.data_migrations_path, version)
@@ -188,20 +240,23 @@ namespace :data do
     task :down => :environment do
       version = ENV["VERSION"] ? ENV["VERSION"].to_i : nil
       raise "VERSION is required" unless version
-      DataMigrate::DataMigrator.create_data_schema_table
+      DataMigrate::DataMigrator.assure_data_schema_table
       DataMigrate::DataMigrator.run(:down, DataMigrate::DatabaseTasks.data_migrations_path, version)
       Rake::Task["data:dump"].invoke
     end
 
     desc "Display status of data migrations"
     task :status => :environment do
-      DataMigrate::Tasks::DataMigrateTasks.status
+      ActiveRecord::Base.configurations.configs_for(env_name: ActiveRecord::Tasks::DatabaseTasks.env).each do |db_config|
+        ActiveRecord::Base.establish_connection(db_config)
+        DataMigrate::Tasks::DataMigrateTasks.status(db_config)
+      end
     end
   end
 
   desc 'Rolls the schema back to the previous version (specify steps w/ STEP=n).'
   task :rollback => :environment do
-    DataMigrate::DataMigrator.create_data_schema_table
+    DataMigrate::DataMigrator.assure_data_schema_table
     step = ENV['STEP'] ? ENV['STEP'].to_i : 1
     DataMigrate::DataMigrator.rollback(DataMigrate::DatabaseTasks.data_migrations_path, step)
     Rake::Task["data:dump"].invoke
@@ -209,7 +264,7 @@ namespace :data do
 
   desc 'Pushes the schema to the next version (specify steps w/ STEP=n).'
   task :forward => :environment do
-    DataMigrate::DataMigrator.create_data_schema_table
+    DataMigrate::DataMigrator.assure_data_schema_table
     step = ENV['STEP'] ? ENV['STEP'].to_i : 1
     # TODO: No worky for .forward
     # DataMigrate::DataMigrator.forward('db/data/', step)
@@ -222,7 +277,7 @@ namespace :data do
 
   desc "Retrieves the current schema version number for data migrations"
   task :version => :environment do
-    DataMigrate::DataMigrator.create_data_schema_table
+    DataMigrate::DataMigrator.assure_data_schema_table
     puts "Current data version: #{DataMigrate::DataMigrator.current_version}"
   end
 
@@ -234,7 +289,10 @@ namespace :data do
 
   desc "Create a db/data_schema.rb file that stores the current data version"
   task dump: :environment do
-    DataMigrate::Tasks::DataMigrateTasks.dump
+    ActiveRecord::Base.configurations.configs_for(env_name: ActiveRecord::Tasks::DatabaseTasks.env).each do |db_config|
+      ActiveRecord::Base.establish_connection(db_config)
+      DataMigrate::Tasks::DataMigrateTasks.dump(db_config)
+    end
 
     # Allow this task to be called as many times as required. An example
     # is the migrate:redo task, which calls other two internally