Add a #populate method to migrations (#31082)

* Add a #populate method to migrations

* Address rubocop issues

* Rename to #up_only and use #execute in the examples intead of the model

* Update CHANGELOG

[Rich Daley & Rafael Mendonça França]

1 files changed, 25 insertions, 0 deletions

diff --git a/activerecord/test/cases/invertible_migration_test.rb b/activerecord/test/cases/invertible_migration_test.rb
index 60c628511f..e0f40710eb 100644
--- a/activerecord/test/cases/invertible_migration_test.rb
+++ b/activerecord/test/cases/invertible_migration_test.rb
@@ -161,6 +161,13 @@ module ActiveRecord
       end
     end
 
+    class UpOnlyMigration < SilentMigration
+      def change
+        add_column :horses, :oldie, :boolean, default: false
+        up_only { execute "update horses set oldie = 'true'" }
+      end
+    end
+
     setup do
       @verbose_was, ActiveRecord::Migration.verbose = ActiveRecord::Migration.verbose, false
     end
@@ -378,5 +385,23 @@ module ActiveRecord
               "horses_index_named index should not exist"
       end
     end
+
+    def test_up_only
+      InvertibleMigration.new.migrate(:up)
+      horse1 = Horse.create
+      # populates existing horses with oldie=true but new ones have default false
+      UpOnlyMigration.new.migrate(:up)
+      Horse.reset_column_information
+      horse1.reload
+      horse2 = Horse.create
+
+      assert horse1.oldie? # created before migration
+      assert !horse2.oldie? # created after migration
+
+      UpOnlyMigration.new.migrate(:down) # should be no error
+      connection = ActiveRecord::Base.connection
+      assert !connection.column_exists?(:horses, :oldie)
+      Horse.reset_column_information
+    end
   end
 end