diff options
author | Marc-Andre Lafortune <github@marc-andre.ca> | 2012-11-11 14:43:57 -0500 |
---|---|---|
committer | Marc-Andre Lafortune <github@marc-andre.ca> | 2012-12-21 13:54:51 -0500 |
commit | 99770e4c659013461fb308040e3d594f2038ed24 (patch) | |
tree | 9d929f687f5ff003d0d23f962be7e196c2196f41 /activerecord/test/cases | |
parent | 65e154f33b54acf40b51082fc5b681ba605015d9 (diff) | |
download | rails-99770e4c659013461fb308040e3d594f2038ed24.tar.gz rails-99770e4c659013461fb308040e3d594f2038ed24.tar.bz2 rails-99770e4c659013461fb308040e3d594f2038ed24.zip |
Add Migration#reversible for reversible data operations [#8267]
Diffstat (limited to 'activerecord/test/cases')
-rw-r--r-- | activerecord/test/cases/invertible_migration_test.rb | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/activerecord/test/cases/invertible_migration_test.rb b/activerecord/test/cases/invertible_migration_test.rb index 6b81e600af..484b4c0fea 100644 --- a/activerecord/test/cases/invertible_migration_test.rb +++ b/activerecord/test/cases/invertible_migration_test.rb @@ -28,6 +28,26 @@ module ActiveRecord end end + class InvertibleByPartsMigration < SilentMigration + attr_writer :test + def change + create_table("new_horses") do |t| + t.column :breed, :string + end + reversible do |dir| + @test.yield :both + dir.up { @test.yield :up } + dir.down { @test.yield :down } + end + revert do + create_table("horses") do |t| + t.column :content, :text + t.column :remind_at, :datetime + end + end + end + end + class NonInvertibleMigration < SilentMigration def change create_table("horses") do |t| @@ -107,6 +127,25 @@ module ActiveRecord assert !migration.connection.table_exists?("horses") end + def test_migrate_revert_by_part + InvertibleMigration.new.migrate :up + received = [] + migration = InvertibleByPartsMigration.new + migration.test = ->(dir){ + assert migration.connection.table_exists?("horses") + assert migration.connection.table_exists?("new_horses") + received << dir + } + migration.migrate :up + assert_equal [:both, :up], received + assert !migration.connection.table_exists?("horses") + assert migration.connection.table_exists?("new_horses") + migration.migrate :down + assert_equal [:both, :up, :both, :down], received + assert migration.connection.table_exists?("horses") + assert !migration.connection.table_exists?("new_horses") + end + def test_migrate_revert_whole_migration migration = InvertibleMigration.new [LegacyMigration, InvertibleMigration].each do |klass| |