require 'rubygems'
require 'mocha'
require 'active_support'
require 'active_support/test_case'
require 'activerecord'

ActiveRecord.load_all!

module Mysqlplus
  class Test
    
    CONNECTION_SPEC = { :adapter  => 'mysqlplus',
                        :username => 'root',
                        :database => 'mysql',
                        :pool => 5,
                        :warmup => true }
    
    MODELS_DIR = "#{File.dirname(__FILE__)}/models".freeze
    
    class << self
      
      def prepare!
        require 'test/unit'
        connect!
        require_models()
      end
            
      def setup!
        setup_constants!
        setup_config!
      end
        
      def mysqlplus_connection
        "#{File.dirname(__FILE__)}/connections/mysqlplus"
      end

      def active_record_test_files
        returning([]) do |files|
          files << glob( "#{AR_TEST_SUITE}/cases/**/*_test{,_mysqlplus}.rb" )
        end.sort
      end

      def test_files
        glob( "#{File.dirname(__FILE__)}/**/*_test.rb" )
      end

      private

      def connect!
        ::ActiveRecord::Base.establish_connection( CONNECTION_SPEC )
      end

      def require_models
        Dir.entries( MODELS_DIR ).grep(/.rb/).each do |model|
          require_model( model )
        end  
      end  

      def require_model( model )
        require "#{MODELS_DIR}/#{model}"
      end

      def setup_constants!
        set_constant( 'MYSQL_DB_USER' ){ 'rails' }
        set_constant( 'AR_TEST_SUITE' ) do 
          ENV['AR_TEST_SUITE'] || find_active_record_test_suite()
        end
      end
      
      def setup_config!
        unless Object.const_defined?( 'MIGRATIONS_ROOT' )
          require "#{::AR_TEST_SUITE}/config"
        end
      end

      def set_constant( constant )
        Object.const_set(constant, yield ) unless Object.const_defined?( constant )
      end
 
      def find_active_record_test_suite
        returning( ($:).grep( /activerecord/ ).last.split('/') ) do |ar_ts|
          ar_ts.pop
          ar_ts << 'test'
        end.join('/') 
      end

      def glob( pattern )
        Dir.glob( pattern )
      end        
            
    end
  end
end

Mysqlplus::Test.setup!