Tests to specify the behaviour of ActiveRecord::Migrator.get_all_versions() [#5066 state:resolved]

Signed-off-by: José Valim <jose.valim@gmail.com>

1 files changed, 14 insertions, 0 deletions

diff --git a/activerecord/test/cases/migration_test.rb b/activerecord/test/cases/migration_test.rb
index 4ce9bdb46d..2c3fc46831 100644
--- a/activerecord/test/cases/migration_test.rb
+++ b/activerecord/test/cases/migration_test.rb
@@ -1358,6 +1358,20 @@ if ActiveRecord::Base.connection.supports_migrations?
       ActiveRecord::Migrator.forward(MIGRATIONS_ROOT + "/valid")
       assert_equal(3, ActiveRecord::Migrator.current_version)
     end
+    
+    def test_get_all_versions
+      ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/valid")
+      assert_equal([1,2,3], ActiveRecord::Migrator.get_all_versions)      
+
+      ActiveRecord::Migrator.rollback(MIGRATIONS_ROOT + "/valid")
+      assert_equal([1,2], ActiveRecord::Migrator.get_all_versions)
+
+      ActiveRecord::Migrator.rollback(MIGRATIONS_ROOT + "/valid")
+      assert_equal([1], ActiveRecord::Migrator.get_all_versions)
+
+      ActiveRecord::Migrator.rollback(MIGRATIONS_ROOT + "/valid")
+      assert_equal([], ActiveRecord::Migrator.get_all_versions)
+    end
 
     def test_schema_migrations_table_name
       ActiveRecord::Base.table_name_prefix = "prefix_"