require File.expand_path(File.dirname(__FILE__) + '/../test_helper')

class IntegrationMatcherTest < Test::Unit::TestCase
  def setup
    superclass = Class.new
    self.class.const_set('Vehicle', superclass)
    
    @klass = Class.new(superclass)
  end
  
  def test_should_return_nil_if_no_match_found
    assert_nil StateMachine::Integrations.match(@klass)
  end
  
  def test_should_return_integration_class_if_match_found
    integration = Module.new do
      include StateMachine::Integrations::Base
      
      def self.matching_ancestors
        ['IntegrationMatcherTest::Vehicle']
      end
    end
    StateMachine::Integrations.const_set('Custom', integration)
    
    assert_equal integration, StateMachine::Integrations.match(@klass)
  ensure
    StateMachine::Integrations.send(:remove_const, 'Custom')
  end
  
  def test_should_return_nil_if_no_match_found_with_ancestors
    assert_nil StateMachine::Integrations.match_ancestors(['IntegrationMatcherTest::Fake'])
  end
  
  def test_should_return_integration_class_if_match_found_with_ancestors
    integration = Module.new do
      include StateMachine::Integrations::Base
      
      def self.matching_ancestors
        ['IntegrationMatcherTest::Vehicle']
      end
    end
    StateMachine::Integrations.const_set('Custom', integration)
    
    assert_equal integration, StateMachine::Integrations.match_ancestors(['IntegrationMatcherTest::Fake', 'IntegrationMatcherTest::Vehicle'])
  ensure
    StateMachine::Integrations.send(:remove_const, 'Custom')
  end
  
  def teardown
    self.class.send(:remove_const, 'Vehicle')
  end
end

class IntegrationFinderTest < Test::Unit::TestCase
  def test_should_find_base
    assert_equal StateMachine::Integrations::Base, StateMachine::Integrations.find_by_name(:base)
  end
  
  def test_should_find_active_model
    assert_equal StateMachine::Integrations::ActiveModel, StateMachine::Integrations.find_by_name(:active_model)
  end
  
  def test_should_find_active_record
    assert_equal StateMachine::Integrations::ActiveRecord, StateMachine::Integrations.find_by_name(:active_record)
  end
  
  def test_should_find_data_mapper
    assert_equal StateMachine::Integrations::DataMapper, StateMachine::Integrations.find_by_name(:data_mapper)
  end
  
  def test_should_find_mongo_mapper
    assert_equal StateMachine::Integrations::MongoMapper, StateMachine::Integrations.find_by_name(:mongo_mapper)
  end
  
  def test_should_find_sequel
    assert_equal StateMachine::Integrations::Sequel, StateMachine::Integrations.find_by_name(:sequel)
  end
  
  def test_should_raise_an_exception_if_invalid
    exception = assert_raise(StateMachine::IntegrationNotFound) { StateMachine::Integrations.find_by_name(:invalid) }
    assert_equal ':invalid is an invalid integration', exception.message
  end
end