1 files changed, 133 insertions, 0 deletions
diff --git a/activerecord/test/cases/migrator_test.rb b/activerecord/test/cases/migrator_test.rb
new file mode 100644
index 0000000000..143aadd833
--- /dev/null
+++ b/activerecord/test/cases/migrator_test.rb
@@ -0,0 +1,133 @@
+require "cases/helper"
+
+module ActiveRecord
+  class MigratorTest < ActiveRecord::TestCase
+    # Use this class to sense if migrations have gone
+    # up or down.
+    class Sensor < ActiveRecord::Migration
+      attr_reader :went_up, :went_down
+
+      def initialize name, version
+        super
+        @went_up  = false
+        @went_down = false
+      end
+
+      def up; @went_up = true; end
+      def down; @went_down = true; end
+    end
+
+    def setup
+      super
+      ActiveRecord::SchemaMigration.create_table
+      ActiveRecord::SchemaMigration.delete_all rescue nil
+    end
+
+    def teardown
+      super
+      ActiveRecord::SchemaMigration.delete_all rescue nil
+    end
+
+    def test_migrator_with_duplicate_names
+      assert_raises(ActiveRecord::DuplicateMigrationNameError, "Multiple migrations have the name Chunky") do
+        list = [Migration.new('Chunky'), Migration.new('Chunky')]
+        ActiveRecord::Migrator.new(:up, list)
+      end
+    end
+
+    def test_migrator_with_duplicate_versions
+      assert_raises(ActiveRecord::DuplicateMigrationVersionError) do
+        list = [Migration.new('Foo', 1), Migration.new('Bar', 1)]
+        ActiveRecord::Migrator.new(:up, list)
+      end
+    end
+
+    def test_migrator_with_missing_version_numbers
+      assert_raises(ActiveRecord::UnknownMigrationVersionError) do
+        list = [Migration.new('Foo', 1), Migration.new('Bar', 2)]
+        ActiveRecord::Migrator.new(:up, list, 3).run
+      end
+    end
+
+    def test_finds_migrations
+      migrations = ActiveRecord::Migrator.migrations(MIGRATIONS_ROOT + "/valid")
+
+      [[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
+    end
+
+    def test_finds_migrations_in_subdirectories
+      migrations = ActiveRecord::Migrator.migrations(MIGRATIONS_ROOT + "/valid_with_subdirectories")
+
+      [[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
+    end
+
+    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
+
+    def test_deprecated_constructor
+      assert_deprecated do
+        ActiveRecord::Migrator.new(:up, MIGRATIONS_ROOT + "/valid")
+      end
+    end
+
+    def test_relative_migrations
+      list = Dir.chdir(MIGRATIONS_ROOT) do
+        ActiveRecord::Migrator.migrations("valid/")
+      end
+
+      migration_proxy = list.find { |item|
+        item.name == 'ValidPeopleHaveLastNames'
+      }
+      assert migration_proxy, 'should find pending migration'
+    end
+
+    def test_finds_pending_migrations
+      ActiveRecord::SchemaMigration.create!(:version => '1')
+      migration_list = [ Migration.new('foo', 1), Migration.new('bar', 3) ]
+      migrations = ActiveRecord::Migrator.new(:up, migration_list).pending_migrations
+
+      assert_equal 1, migrations.size
+      assert_equal migration_list.last, migrations.first
+    end
+
+    def test_migrator_interleaved_migrations
+      pass_one = [Sensor.new('One', 1)]
+
+      ActiveRecord::Migrator.new(:up, pass_one).migrate
+      assert pass_one.first.went_up
+      refute pass_one.first.went_down
+
+      pass_two = [Sensor.new('One', 1), Sensor.new('Three', 3)]
+      ActiveRecord::Migrator.new(:up, pass_two).migrate
+      refute pass_two[0].went_up
+      assert pass_two[1].went_up
+      assert pass_two.all? { |x| !x.went_down }
+
+      pass_three = [Sensor.new('One', 1),
+                    Sensor.new('Two', 2),
+                    Sensor.new('Three', 3)]
+
+      ActiveRecord::Migrator.new(:down, pass_three).migrate
+      assert pass_three[0].went_down
+      refute pass_three[1].went_down
+      assert pass_three[2].went_down
+    end
+  end
+end