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