Merge pull request rails#55077 from JoeDupuis/db-migrate-reset-namespaced

Add namespaced version of the `db:migrate:reset` command.

Author: eileencodes

activerecord/CHANGELOG.md

@@ -1,3 +1,7 @@
+*   Add support for multiple databases to `db:migrate:reset`.
+
+    *Joé Dupuis*
+
 *   Add `affected_rows` to `ActiveRecord::Result`.
 
     *Jenny Shen*

activerecord/lib/active_record/railties/databases.rake

@@ -163,6 +163,18 @@ db_namespace = namespace :db do
     desc "Resets your database using your migrations for the current environment"
     task reset: ["db:drop", "db:create", "db:schema:dump", "db:migrate"]
 
+    namespace :reset do
+      ActiveRecord::Tasks::DatabaseTasks.for_each(databases) do |name|
+        desc "Drop and recreate the #{name} database using migrations"
+        task name => :load_config do
+          db_namespace["drop:#{name}"].invoke
+          db_namespace["create:#{name}"].invoke
+          db_namespace["schema:dump:#{name}"].invoke
+          db_namespace["migrate:#{name}"].invoke
+        end
+      end
+    end
+
     desc 'Run the "up" for a given migration VERSION.'
     task up: :load_config do
       ActiveRecord::Tasks::DatabaseTasks.raise_for_multi_db(command: "db:migrate:up")

railties/test/application/rake/multi_dbs_test.rb

@@ -993,6 +993,38 @@ class TwoMigration < ActiveRecord::Migration::Current
         end
       end
 
+      test "db:migrate:reset:primary regenerates the schema from migrations" do
+        require "#{app_path}/config/environment"
+        Dir.chdir(app_path) do
+          generate_models_for_animals
+          rails "db:migrate"
+          assert_not File.read("db/schema.rb").include?("director")
+
+
+          primary_mtime = File.mtime("db/schema.rb")
+          animals_mtime = File.mtime("db/animals_schema.rb")
+
+          app_file "db/migrate/02_create_movies.rb", <<-MIGRATION
+            class CreateMovies < ActiveRecord::Migration::Current
+              create_table(:movies) { |t| t.string :director }
+            end
+          MIGRATION
+
+          app_file "db/animals_migrate/02_new_animals.rb", <<-MIGRATION
+            class NewAnimals < ActiveRecord::Migration::Current
+              create_table(:cats) {}
+            end
+          MIGRATION
+
+          rails "db:migrate:reset:primary"
+
+          assert File.read("db/schema.rb").include?("director")
+          assert File.mtime("db/schema.rb") > primary_mtime
+          assert_equal animals_mtime, File.mtime("db/animals_schema.rb")
+          assert_not File.read("db/animals_schema.rb").include?("cats")
+        end
+      end
+
       test "db:prepare works on all databases" do
         require "#{app_path}/config/environment"
         db_prepare