1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00
rails--rails/activerecord/test/cases/migrator_test.rb

389 lines
11 KiB
Ruby
Raw Normal View History

require "cases/helper"
require "cases/migration/helper"
2014-08-28 17:31:31 -04:00
class MigratorTest < ActiveRecord::TestCase
self.use_transactional_fixtures = false
# Use this class to sense if migrations have gone
# up or down.
class Sensor < ActiveRecord::Migration
attr_reader :went_up, :went_down
2014-08-28 17:31:31 -04:00
def initialize name = self.class.name, version = nil
super
2014-08-28 17:31:31 -04:00
@went_up = false
@went_down = false
end
2014-08-28 17:31:31 -04:00
def up; @went_up = true; end
def down; @went_down = true; end
end
2014-08-28 17:31:31 -04:00
def setup
super
ActiveRecord::SchemaMigration.create_table
ActiveRecord::SchemaMigration.delete_all rescue nil
@verbose_was = ActiveRecord::Migration.verbose
ActiveRecord::Migration.message_count = 0
ActiveRecord::Migration.class_eval do
undef :puts
def puts(*)
ActiveRecord::Migration.message_count += 1
end
end
2014-08-28 17:31:31 -04:00
end
2014-08-28 17:31:31 -04:00
teardown do
ActiveRecord::SchemaMigration.delete_all rescue nil
ActiveRecord::Migration.verbose = @verbose_was
ActiveRecord::Migration.class_eval do
undef :puts
def puts(*)
super
end
end
2014-08-28 17:31:31 -04:00
end
2014-08-28 17:31:31 -04:00
def test_migrator_with_duplicate_names
assert_raises(ActiveRecord::DuplicateMigrationNameError, "Multiple migrations have the name Chunky") do
list = [ActiveRecord::Migration.new('Chunky'), ActiveRecord::Migration.new('Chunky')]
ActiveRecord::Migrator.new(:up, list)
end
2014-08-28 17:31:31 -04:00
end
2014-08-28 17:31:31 -04:00
def test_migrator_with_duplicate_versions
assert_raises(ActiveRecord::DuplicateMigrationVersionError) do
list = [ActiveRecord::Migration.new('Foo', 1), ActiveRecord::Migration.new('Bar', 1)]
ActiveRecord::Migrator.new(:up, list)
end
end
2014-08-28 17:31:31 -04:00
def test_migrator_with_missing_version_numbers
assert_raises(ActiveRecord::UnknownMigrationVersionError) do
list = [ActiveRecord::Migration.new('Foo', 1), ActiveRecord::Migration.new('Bar', 2)]
ActiveRecord::Migrator.new(:up, list, 3).run
end
2014-08-28 17:31:31 -04:00
end
2014-08-28 17:31:31 -04:00
def test_finds_migrations
migrations = ActiveRecord::Migrator.migrations(MIGRATIONS_ROOT + "/valid")
2014-08-28 17:31:31 -04:00
[[1, 'ValidPeopleHaveLastNames'], [2, 'WeNeedReminders'], [3, 'InnocentJointable']].each_with_index do |pair, i|
assert_equal migrations[i].version, pair.first
assert_equal migrations[i].name, pair.last
end
2014-08-28 17:31:31 -04:00
end
2014-08-28 17:31:31 -04:00
def test_finds_migrations_in_subdirectories
migrations = ActiveRecord::Migrator.migrations(MIGRATIONS_ROOT + "/valid_with_subdirectories")
2014-08-28 17:31:31 -04:00
[[1, 'ValidPeopleHaveLastNames'], [2, 'WeNeedReminders'], [3, 'InnocentJointable']].each_with_index do |pair, i|
assert_equal migrations[i].version, pair.first
assert_equal migrations[i].name, pair.last
end
2014-08-28 17:31:31 -04:00
end
2014-08-28 17:31:31 -04:00
def test_finds_migrations_from_two_directories
directories = [MIGRATIONS_ROOT + '/valid_with_timestamps', MIGRATIONS_ROOT + '/to_copy_with_timestamps']
migrations = ActiveRecord::Migrator.migrations directories
[[20090101010101, "PeopleHaveHobbies"],
[20090101010202, "PeopleHaveDescriptions"],
[20100101010101, "ValidWithTimestampsPeopleHaveLastNames"],
[20100201010101, "ValidWithTimestampsWeNeedReminders"],
[20100301010101, "ValidWithTimestampsInnocentJointable"]].each_with_index do |pair, i|
assert_equal pair.first, migrations[i].version
assert_equal pair.last, migrations[i].name
end
end
2014-08-28 17:31:31 -04:00
def test_finds_migrations_in_numbered_directory
migrations = ActiveRecord::Migrator.migrations [MIGRATIONS_ROOT + '/10_urban']
assert_equal 9, migrations[0].version
assert_equal 'AddExpressions', migrations[0].name
end
def test_relative_migrations
list = Dir.chdir(MIGRATIONS_ROOT) do
ActiveRecord::Migrator.migrations("valid")
end
2014-08-28 17:31:31 -04:00
migration_proxy = list.find { |item|
item.name == 'ValidPeopleHaveLastNames'
}
assert migration_proxy, 'should find pending migration'
end
2014-08-28 17:31:31 -04:00
def test_finds_pending_migrations
ActiveRecord::SchemaMigration.create!(:version => '1')
migration_list = [ActiveRecord::Migration.new('foo', 1), ActiveRecord::Migration.new('bar', 3)]
migrations = ActiveRecord::Migrator.new(:up, migration_list).pending_migrations
2014-08-28 17:31:31 -04:00
assert_equal 1, migrations.size
assert_equal migration_list.last, migrations.first
end
2014-08-28 17:31:31 -04:00
def test_migrator_interleaved_migrations
pass_one = [Sensor.new('One', 1)]
2014-08-28 17:31:31 -04:00
ActiveRecord::Migrator.new(:up, pass_one).migrate
assert pass_one.first.went_up
assert_not pass_one.first.went_down
2014-08-28 17:31:31 -04:00
pass_two = [Sensor.new('One', 1), Sensor.new('Three', 3)]
ActiveRecord::Migrator.new(:up, pass_two).migrate
assert_not pass_two[0].went_up
assert pass_two[1].went_up
assert pass_two.all? { |x| !x.went_down }
2014-08-28 17:31:31 -04:00
pass_three = [Sensor.new('One', 1),
Sensor.new('Two', 2),
Sensor.new('Three', 3)]
2012-01-16 13:33:39 -05:00
2014-08-28 17:31:31 -04:00
ActiveRecord::Migrator.new(:down, pass_three).migrate
assert pass_three[0].went_down
assert_not pass_three[1].went_down
assert pass_three[2].went_down
end
2012-01-16 13:33:39 -05:00
2014-08-28 17:31:31 -04:00
def test_up_calls_up
migrations = [Sensor.new(nil, 0), Sensor.new(nil, 1), Sensor.new(nil, 2)]
ActiveRecord::Migrator.new(:up, migrations).migrate
assert migrations.all? { |m| m.went_up }
assert migrations.all? { |m| !m.went_down }
assert_equal 2, ActiveRecord::Migrator.current_version
end
2012-01-16 13:33:39 -05:00
2014-08-28 17:31:31 -04:00
def test_down_calls_down
test_up_calls_up
2012-01-16 13:33:39 -05:00
2014-08-28 17:31:31 -04:00
migrations = [Sensor.new(nil, 0), Sensor.new(nil, 1), Sensor.new(nil, 2)]
ActiveRecord::Migrator.new(:down, migrations).migrate
assert migrations.all? { |m| !m.went_up }
assert migrations.all? { |m| m.went_down }
assert_equal 0, ActiveRecord::Migrator.current_version
end
2014-08-28 17:31:31 -04:00
def test_current_version
ActiveRecord::SchemaMigration.create!(:version => '1000')
assert_equal 1000, ActiveRecord::Migrator.current_version
end
2014-08-28 17:31:31 -04:00
def test_migrator_one_up
calls, migrations = sensors(3)
2014-08-28 17:31:31 -04:00
ActiveRecord::Migrator.new(:up, migrations, 1).migrate
assert_equal [[:up, 1]], calls
calls.clear
2014-08-28 17:31:31 -04:00
ActiveRecord::Migrator.new(:up, migrations, 2).migrate
assert_equal [[:up, 2]], calls
end
2014-08-28 17:31:31 -04:00
def test_migrator_one_down
calls, migrations = sensors(3)
2014-08-28 17:31:31 -04:00
ActiveRecord::Migrator.new(:up, migrations).migrate
assert_equal [[:up, 1], [:up, 2], [:up, 3]], calls
calls.clear
2014-08-28 17:31:31 -04:00
ActiveRecord::Migrator.new(:down, migrations, 1).migrate
2014-08-28 17:31:31 -04:00
assert_equal [[:down, 3], [:down, 2]], calls
end
2014-08-28 17:31:31 -04:00
def test_migrator_one_up_one_down
calls, migrations = sensors(3)
2014-08-28 17:31:31 -04:00
ActiveRecord::Migrator.new(:up, migrations, 1).migrate
assert_equal [[:up, 1]], calls
calls.clear
2014-08-28 17:31:31 -04:00
ActiveRecord::Migrator.new(:down, migrations, 0).migrate
assert_equal [[:down, 1]], calls
end
2014-08-28 17:31:31 -04:00
def test_migrator_double_up
calls, migrations = sensors(3)
assert_equal(0, ActiveRecord::Migrator.current_version)
2014-08-28 17:31:31 -04:00
ActiveRecord::Migrator.new(:up, migrations, 1).migrate
assert_equal [[:up, 1]], calls
calls.clear
2014-08-28 17:31:31 -04:00
ActiveRecord::Migrator.new(:up, migrations, 1).migrate
assert_equal [], calls
end
2014-08-28 17:31:31 -04:00
def test_migrator_double_down
calls, migrations = sensors(3)
2014-08-28 17:31:31 -04:00
assert_equal(0, ActiveRecord::Migrator.current_version)
2014-08-28 17:31:31 -04:00
ActiveRecord::Migrator.new(:up, migrations, 1).run
assert_equal [[:up, 1]], calls
calls.clear
2014-08-28 17:31:31 -04:00
ActiveRecord::Migrator.new(:down, migrations, 1).run
assert_equal [[:down, 1]], calls
calls.clear
2014-08-28 17:31:31 -04:00
ActiveRecord::Migrator.new(:down, migrations, 1).run
assert_equal [], calls
2014-08-28 17:31:31 -04:00
assert_equal(0, ActiveRecord::Migrator.current_version)
end
2014-08-28 17:31:31 -04:00
def test_migrator_verbosity
_, migrations = sensors(3)
2014-08-28 17:31:31 -04:00
ActiveRecord::Migrator.new(:up, migrations, 1).migrate
assert_not_equal 0, ActiveRecord::Migration.message_count
2014-08-28 17:31:31 -04:00
ActiveRecord::Migration.message_count = 0
2014-08-28 17:31:31 -04:00
ActiveRecord::Migrator.new(:down, migrations, 0).migrate
assert_not_equal 0, ActiveRecord::Migration.message_count
end
2014-08-28 17:31:31 -04:00
def test_migrator_verbosity_off
_, migrations = sensors(3)
2014-08-28 17:31:31 -04:00
ActiveRecord::Migration.message_count = 0
ActiveRecord::Migration.verbose = false
ActiveRecord::Migrator.new(:up, migrations, 1).migrate
assert_equal 0, ActiveRecord::Migration.message_count
ActiveRecord::Migrator.new(:down, migrations, 0).migrate
assert_equal 0, ActiveRecord::Migration.message_count
end
2012-01-16 16:58:32 -05:00
2014-08-28 17:31:31 -04:00
def test_target_version_zero_should_run_only_once
calls, migrations = sensors(3)
2012-01-16 16:58:32 -05:00
2014-08-28 17:31:31 -04:00
# migrate up to 1
ActiveRecord::Migrator.new(:up, migrations, 1).migrate
assert_equal [[:up, 1]], calls
calls.clear
2012-01-16 16:58:32 -05:00
2014-08-28 17:31:31 -04:00
# migrate down to 0
ActiveRecord::Migrator.new(:down, migrations, 0).migrate
assert_equal [[:down, 1]], calls
calls.clear
2012-01-16 16:58:32 -05:00
2014-08-28 17:31:31 -04:00
# migrate down to 0 again
ActiveRecord::Migrator.new(:down, migrations, 0).migrate
assert_equal [], calls
end
2014-08-28 17:31:31 -04:00
def test_migrator_going_down_due_to_version_target
calls, migrator = migrator_class(3)
2014-08-28 17:31:31 -04:00
migrator.up("valid", 1)
assert_equal [[:up, 1]], calls
calls.clear
2014-08-28 17:31:31 -04:00
migrator.migrate("valid", 0)
assert_equal [[:down, 1]], calls
calls.clear
2014-08-28 17:31:31 -04:00
migrator.migrate("valid")
assert_equal [[:up, 1], [:up, 2], [:up, 3]], calls
end
2014-08-28 17:31:31 -04:00
def test_migrator_rollback
_, migrator = migrator_class(3)
2014-08-28 17:31:31 -04:00
migrator.migrate("valid")
assert_equal(3, ActiveRecord::Migrator.current_version)
2014-08-28 17:31:31 -04:00
migrator.rollback("valid")
assert_equal(2, ActiveRecord::Migrator.current_version)
2014-08-28 17:31:31 -04:00
migrator.rollback("valid")
assert_equal(1, ActiveRecord::Migrator.current_version)
2014-08-28 17:31:31 -04:00
migrator.rollback("valid")
assert_equal(0, ActiveRecord::Migrator.current_version)
2014-08-28 17:31:31 -04:00
migrator.rollback("valid")
assert_equal(0, ActiveRecord::Migrator.current_version)
end
2014-08-28 17:31:31 -04:00
def test_migrator_db_has_no_schema_migrations_table
_, migrator = migrator_class(3)
2014-08-28 17:31:31 -04:00
ActiveRecord::Base.connection.execute("DROP TABLE schema_migrations")
assert_not ActiveRecord::Base.connection.table_exists?('schema_migrations')
migrator.migrate("valid", 1)
assert ActiveRecord::Base.connection.table_exists?('schema_migrations')
end
2014-08-28 17:31:31 -04:00
def test_migrator_forward
_, migrator = migrator_class(3)
migrator.migrate("/valid", 1)
assert_equal(1, ActiveRecord::Migrator.current_version)
2014-08-28 17:31:31 -04:00
migrator.forward("/valid", 2)
assert_equal(3, ActiveRecord::Migrator.current_version)
2014-08-28 17:31:31 -04:00
migrator.forward("/valid")
assert_equal(3, ActiveRecord::Migrator.current_version)
end
2014-08-28 17:31:31 -04:00
def test_only_loads_pending_migrations
# migrate up to 1
ActiveRecord::SchemaMigration.create!(:version => '1')
2014-08-28 17:31:31 -04:00
calls, migrator = migrator_class(3)
migrator.migrate("valid", nil)
2014-08-28 17:31:31 -04:00
assert_equal [[:up, 2], [:up, 3]], calls
end
2012-01-16 18:24:27 -05:00
2014-08-28 17:31:31 -04:00
def test_get_all_versions
_, migrator = migrator_class(3)
2012-01-16 18:24:27 -05:00
2014-08-28 17:31:31 -04:00
migrator.migrate("valid")
assert_equal([1,2,3], ActiveRecord::Migrator.get_all_versions)
2012-01-16 18:24:27 -05:00
2014-08-28 17:31:31 -04:00
migrator.rollback("valid")
assert_equal([1,2], ActiveRecord::Migrator.get_all_versions)
2012-01-16 18:24:27 -05:00
2014-08-28 17:31:31 -04:00
migrator.rollback("valid")
assert_equal([1], ActiveRecord::Migrator.get_all_versions)
2012-01-16 18:24:27 -05:00
2014-08-28 17:31:31 -04:00
migrator.rollback("valid")
assert_equal([], ActiveRecord::Migrator.get_all_versions)
end
private
def m(name, version, &block)
x = Sensor.new name, version
x.extend(Module.new {
define_method(:up) { block.call(:up, x); super() }
define_method(:down) { block.call(:down, x); super() }
}) if block_given?
end
2014-08-28 17:31:31 -04:00
def sensors(count)
calls = []
migrations = count.times.map { |i|
m(nil, i + 1) { |c,migration|
calls << [c, migration.version]
}
2014-08-28 17:31:31 -04:00
}
[calls, migrations]
end
2014-08-28 17:31:31 -04:00
def migrator_class(count)
calls, migrations = sensors(count)
2014-08-28 17:31:31 -04:00
migrator = Class.new(ActiveRecord::Migrator).extend(Module.new {
define_method(:migrations) { |paths|
migrations
}
})
[calls, migrator]
end
end