Chris@909: require 'test/unit'
Chris@909: 
Chris@909: unless defined?(ActiveRecord)
Chris@909:   plugin_root = File.join(File.dirname(__FILE__), '..')
Chris@909: 
Chris@909:   # first look for a symlink to a copy of the framework
Chris@909:   if framework_root = ["#{plugin_root}/rails", "#{plugin_root}/../../rails"].find { |p| File.directory? p }
Chris@909:     puts "found framework root: #{framework_root}"
Chris@909:     # this allows for a plugin to be tested outside an app
Chris@909:     $:.unshift "#{framework_root}/activesupport/lib", "#{framework_root}/activerecord/lib", "#{framework_root}/actionpack/lib"
Chris@909:   else
Chris@909:     # is the plugin installed in an application?
Chris@909:     app_root = plugin_root + '/../../..'
Chris@909: 
Chris@909:     if File.directory? app_root + '/config'
Chris@909:       puts 'using config/boot.rb'
Chris@909:       ENV['RAILS_ENV'] = 'test'
Chris@909:       require File.expand_path(app_root + '/config/boot')
Chris@909:     else
Chris@909:       # simply use installed gems if available
Chris@909:       puts 'using rubygems'
Chris@909:       require 'rubygems'
Chris@909:       gem 'actionpack'; gem 'activerecord'
Chris@909:     end
Chris@909:   end
Chris@909: 
Chris@909:   %w(action_pack active_record action_controller active_record/fixtures action_controller/test_process).each {|f| require f}
Chris@909: 
Chris@909:   Dependencies.load_paths.unshift "#{plugin_root}/lib"
Chris@909: end
Chris@909: 
Chris@909: # Define the connector
Chris@909: class ActiveRecordTestConnector
Chris@909:   cattr_accessor :able_to_connect
Chris@909:   cattr_accessor :connected
Chris@909: 
Chris@909:   # Set our defaults
Chris@909:   self.connected = false
Chris@909:   self.able_to_connect = true
Chris@909: 
Chris@909:   class << self
Chris@909:     def setup
Chris@909:       unless self.connected || !self.able_to_connect
Chris@909:         setup_connection
Chris@909:         load_schema
Chris@909:         require_fixture_models
Chris@909:         self.connected = true
Chris@909:       end
Chris@909:     rescue Exception => e  # errors from ActiveRecord setup
Chris@909:       $stderr.puts "\nSkipping ActiveRecord assertion tests: #{e}"
Chris@909:       #$stderr.puts "  #{e.backtrace.join("\n  ")}\n"
Chris@909:       self.able_to_connect = false
Chris@909:     end
Chris@909: 
Chris@909:     private
Chris@909: 
Chris@909:     def setup_connection
Chris@909:       if Object.const_defined?(:ActiveRecord)
Chris@909:         defaults = { :database => ':memory:' }
Chris@909:         begin
Chris@909:           options = defaults.merge :adapter => 'sqlite3', :timeout => 500
Chris@909:           ActiveRecord::Base.establish_connection(options)
Chris@909:           ActiveRecord::Base.configurations = { 'sqlite3_ar_integration' => options }
Chris@909:           ActiveRecord::Base.connection
Chris@909:         rescue Exception  # errors from establishing a connection
Chris@909:           $stderr.puts 'SQLite 3 unavailable; trying SQLite 2.'
Chris@909:           options = defaults.merge :adapter => 'sqlite'
Chris@909:           ActiveRecord::Base.establish_connection(options)
Chris@909:           ActiveRecord::Base.configurations = { 'sqlite2_ar_integration' => options }
Chris@909:           ActiveRecord::Base.connection
Chris@909:         end
Chris@909: 
Chris@909:         Object.send(:const_set, :QUOTED_TYPE, ActiveRecord::Base.connection.quote_column_name('type')) unless Object.const_defined?(:QUOTED_TYPE)
Chris@909:       else
Chris@909:         raise "Can't setup connection since ActiveRecord isn't loaded."
Chris@909:       end
Chris@909:     end
Chris@909: 
Chris@909:     # Load actionpack sqlite tables
Chris@909:     def load_schema
Chris@909:       File.read(File.dirname(__FILE__) + "/fixtures/schema.sql").split(';').each do |sql|
Chris@909:         ActiveRecord::Base.connection.execute(sql) unless sql.blank?
Chris@909:       end
Chris@909:     end
Chris@909: 
Chris@909:     def require_fixture_models
Chris@909:       Dir.glob(File.dirname(__FILE__) + "/fixtures/*.rb").each {|f| require f}
Chris@909:     end
Chris@909:   end
Chris@909: end
Chris@909: 
Chris@909: # Test case for inheritance
Chris@909: class ActiveRecordTestCase < Test::Unit::TestCase
Chris@909:   # Set our fixture path
Chris@909:   if ActiveRecordTestConnector.able_to_connect
Chris@909:     self.fixture_path = "#{File.dirname(__FILE__)}/fixtures/"
Chris@909:     self.use_transactional_fixtures = false
Chris@909:   end
Chris@909: 
Chris@909:   def self.fixtures(*args)
Chris@909:     super if ActiveRecordTestConnector.connected
Chris@909:   end
Chris@909: 
Chris@909:   def run(*args)
Chris@909:     super if ActiveRecordTestConnector.connected
Chris@909:   end
Chris@909: 
Chris@909:   # Default so Test::Unit::TestCase doesn't complain
Chris@909:   def test_truth
Chris@909:   end
Chris@909: end
Chris@909: 
Chris@909: ActiveRecordTestConnector.setup
Chris@909: ActionController::Routing::Routes.reload rescue nil
Chris@909: ActionController::Routing::Routes.draw do |map|
Chris@909:   map.connect ':controller/:action/:id'
Chris@909: end