rails--rails/railties/test/application/rake/migrations_test.rb

require "isolation/abstract_unit"

module ApplicationTests
  module RakeTests
    class RakeMigrationsTest < ActiveSupport::TestCase
      def setup
        build_app
        boot_rails
        FileUtils.rm_rf("#{app_path}/config/environments")
      end

      def teardown
        teardown_app
      end

      test 'running migrations with given scope' do
        Dir.chdir(app_path) do
          `rails generate model user username:string password:string`
        end
        app_file "db/migrate/01_a_migration.bukkits.rb", <<-MIGRATION
          class AMigration < ActiveRecord::Migration
          end
        MIGRATION

        output = Dir.chdir(app_path) { `rake db:migrate SCOPE=bukkits` }
        assert_no_match(/create_table\(:users\)/, output)
        assert_no_match(/CreateUsers/, output)
        assert_no_match(/add_column\(:users, :email, :string\)/, output)

        assert_match(/AMigration: migrated/, output)

        output = Dir.chdir(app_path) { `rake db:migrate SCOPE=bukkits VERSION=0` }
        assert_no_match(/drop_table\(:users\)/, output)
        assert_no_match(/CreateUsers/, output)
        assert_no_match(/remove_column\(:users, :email\)/, output)

        assert_match(/AMigration: reverted/, output)
      end

      test 'model and migration generator with change syntax' do
        Dir.chdir(app_path) do
          `rails generate model user username:string password:string`
          `rails generate migration add_email_to_users email:string`
        end

        output = Dir.chdir(app_path){ `rake db:migrate` }
        assert_match(/create_table\(:users\)/, output)
        assert_match(/CreateUsers: migrated/, output)
        assert_match(/add_column\(:users, :email, :string\)/, output)
        assert_match(/AddEmailToUsers: migrated/, output)

        output = Dir.chdir(app_path){ `rake db:rollback STEP=2` }
        assert_match(/drop_table\("users"\)/, output)
        assert_match(/CreateUsers: reverted/, output)
        assert_match(/remove_column\("users", :email\)/, output)
        assert_match(/AddEmailToUsers: reverted/, output)
      end

      test 'migration status when schema migrations table is not present' do
        output = Dir.chdir(app_path){ `rake db:migrate:status` }
        assert_equal "Schema migrations table does not exist yet.\n", output
      end

      test 'test migration status' do
        Dir.chdir(app_path) do
          `rails generate model user username:string password:string`
          `rails generate migration add_email_to_users email:string`
        end

        Dir.chdir(app_path) { `rake db:migrate` }
        output = Dir.chdir(app_path) { `rake db:migrate:status` }

        assert_match(/up\s+\d{14}\s+Create users/, output)
        assert_match(/up\s+\d{14}\s+Add email to users/, output)

        Dir.chdir(app_path) { `rake db:rollback STEP=1` }
        output = Dir.chdir(app_path) { `rake db:migrate:status` }

        assert_match(/up\s+\d{14}\s+Create users/, output)
        assert_match(/down\s+\d{14}\s+Add email to users/, output)
      end

      test 'migration status without timestamps' do
        add_to_config('config.active_record.timestamped_migrations = false')

        Dir.chdir(app_path) do
          `rails generate model user username:string password:string`
          `rails generate migration add_email_to_users email:string`
        end

        Dir.chdir(app_path) { `rake db:migrate` }
        output = Dir.chdir(app_path) { `rake db:migrate:status` }

        assert_match(/up\s+\d{3,}\s+Create users/, output)
        assert_match(/up\s+\d{3,}\s+Add email to users/, output)

        Dir.chdir(app_path) { `rake db:rollback STEP=1` }
        output = Dir.chdir(app_path) { `rake db:migrate:status` }

        assert_match(/up\s+\d{3,}\s+Create users/, output)
        assert_match(/down\s+\d{3,}\s+Add email to users/, output)
      end

      test 'test migration status after rollback and redo' do
        Dir.chdir(app_path) do
          `rails generate model user username:string password:string`
          `rails generate migration add_email_to_users email:string`
        end

        Dir.chdir(app_path) { `rake db:migrate` }
        output = Dir.chdir(app_path) { `rake db:migrate:status` }

        assert_match(/up\s+\d{14}\s+Create users/, output)
        assert_match(/up\s+\d{14}\s+Add email to users/, output)

        Dir.chdir(app_path) { `rake db:rollback STEP=2` }
        output = Dir.chdir(app_path) { `rake db:migrate:status` }

        assert_match(/down\s+\d{14}\s+Create users/, output)
        assert_match(/down\s+\d{14}\s+Add email to users/, output)

        Dir.chdir(app_path) { `rake db:migrate:redo` }
        output = Dir.chdir(app_path) { `rake db:migrate:status` }

        assert_match(/up\s+\d{14}\s+Create users/, output)
        assert_match(/up\s+\d{14}\s+Add email to users/, output)
      end

      test 'migration status after rollback and redo without timestamps' do
        add_to_config('config.active_record.timestamped_migrations = false')

        Dir.chdir(app_path) do
          `rails generate model user username:string password:string`
          `rails generate migration add_email_to_users email:string`
        end

        Dir.chdir(app_path) { `rake db:migrate` }
        output = Dir.chdir(app_path) { `rake db:migrate:status` }

        assert_match(/up\s+\d{3,}\s+Create users/, output)
        assert_match(/up\s+\d{3,}\s+Add email to users/, output)

        Dir.chdir(app_path) { `rake db:rollback STEP=2` }
        output = Dir.chdir(app_path) { `rake db:migrate:status` }

        assert_match(/down\s+\d{3,}\s+Create users/, output)
        assert_match(/down\s+\d{3,}\s+Add email to users/, output)

        Dir.chdir(app_path) { `rake db:migrate:redo` }
        output = Dir.chdir(app_path) { `rake db:migrate:status` }

        assert_match(/up\s+\d{3,}\s+Create users/, output)
        assert_match(/up\s+\d{3,}\s+Add email to users/, output)
      end
    end
  end
end