Chris@909: require File.dirname(__FILE__) + '/../test_helper'
Chris@909: require 'rails_generator'
Chris@909: require 'rails_generator/scripts/generate'
Chris@909: 
Chris@909: class MigrationsTest < Test::Unit::TestCase
Chris@909:   
Chris@909:   @@migration_dir = "#{RAILS_ROOT}/db/migrate"
Chris@909: 
Chris@909:   def setup
Chris@909:     ActiveRecord::Migration.verbose = false
Chris@909:     Engines.plugins[:test_migration].migrate(0)
Chris@909:   end
Chris@909:   
Chris@909:   def teardown
Chris@909:     FileUtils.rm_r(@@migration_dir) if File.exist?(@@migration_dir)
Chris@909:   end
Chris@909:   
Chris@909:   def test_engine_migrations_can_run_down
Chris@909:     assert !table_exists?('tests'), ActiveRecord::Base.connection.tables.inspect
Chris@909:     assert !table_exists?('others'), ActiveRecord::Base.connection.tables.inspect
Chris@909:     assert !table_exists?('extras'), ActiveRecord::Base.connection.tables.inspect
Chris@909:   end
Chris@909:     
Chris@909:   def test_engine_migrations_can_run_up
Chris@909:     Engines.plugins[:test_migration].migrate(3)
Chris@909:     assert table_exists?('tests')
Chris@909:     assert table_exists?('others')
Chris@909:     assert table_exists?('extras')
Chris@909:   end
Chris@909:   
Chris@909:   def test_engine_migrations_can_upgrade_incrementally
Chris@909:     Engines.plugins[:test_migration].migrate(1)
Chris@909:     assert table_exists?('tests')
Chris@909:     assert !table_exists?('others')
Chris@909:     assert !table_exists?('extras')
Chris@909:     assert_equal 1, Engines::Plugin::Migrator.current_version(Engines.plugins[:test_migration])
Chris@909:     
Chris@909:     
Chris@909:     Engines.plugins[:test_migration].migrate(2)
Chris@909:     assert table_exists?('others')
Chris@909:     assert_equal 2, Engines::Plugin::Migrator.current_version(Engines.plugins[:test_migration])
Chris@909:     
Chris@909:     
Chris@909:     Engines.plugins[:test_migration].migrate(3)
Chris@909:     assert table_exists?('extras')
Chris@909:     assert_equal 3, Engines::Plugin::Migrator.current_version(Engines.plugins[:test_migration])
Chris@909:   end
Chris@909:     
Chris@909:   def test_generator_creates_plugin_migration_file
Chris@909:     Rails::Generator::Scripts::Generate.new.run(['plugin_migration', 'test_migration'], :quiet => true)
Chris@909:     assert migration_file, "migration file is missing"
Chris@909:   end
Chris@909:   
Chris@909:   private
Chris@909:   
Chris@909:   def table_exists?(table)
Chris@909:     ActiveRecord::Base.connection.tables.include?(table)
Chris@909:   end
Chris@909:   
Chris@909:   def migration_file
Chris@909:     Dir["#{@@migration_dir}/*test_migration_to_version_3.rb"][0]
Chris@909:   end
Chris@909: end