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