vendor/rails/actionpack/test/active_record_unit.rb

   1 require 'abstract_unit'
   2
   3 # Define the essentials
   4 class ActiveRecordTestConnector
   5   cattr_accessor :able_to_connect
   6   cattr_accessor :connected
   7
   8   # Set our defaults
   9   self.connected = false
  10   self.able_to_connect = true
  11 end
  12
  13 # Try to grab AR
  14 if defined?(ActiveRecord) && defined?(Fixtures)
  15   $stderr.puts 'Active Record is already loaded, running tests'
  16 else
  17   $stderr.print 'Attempting to load Active Record... '
  18   begin
  19     PATH_TO_AR = "#{File.dirname(__FILE__)}/../../activerecord/lib"
  20     raise LoadError, "#{PATH_TO_AR} doesn't exist" unless File.directory?(PATH_TO_AR)
  21     $LOAD_PATH.unshift PATH_TO_AR
  22     require 'active_record'
  23     require 'active_record/fixtures'
  24     $stderr.puts 'success'
  25   rescue LoadError => e
  26     $stderr.print "failed. Skipping Active Record assertion tests: #{e}"
  27     ActiveRecordTestConnector.able_to_connect = false
  28   end
  29 end
  30 $stderr.flush
  31
  32
  33 # Define the rest of the connector
  34 class ActiveRecordTestConnector
  35   class << self
  36     def setup
  37       unless self.connected || !self.able_to_connect
  38         setup_connection
  39         load_schema
  40         require_fixture_models
  41         self.connected = true
  42       end
  43     rescue Exception => e  # errors from ActiveRecord setup
  44       $stderr.puts "\nSkipping ActiveRecord assertion tests: #{e}"
  45       #$stderr.puts "  #{e.backtrace.join("\n  ")}\n"
  46       self.able_to_connect = false
  47     end
  48
  49     private
  50       def setup_connection
  51         if Object.const_defined?(:ActiveRecord)
  52           defaults = { :database => ':memory:' }
  53           begin
  54             options = defaults.merge :adapter => 'sqlite3', :timeout => 500
  55             ActiveRecord::Base.establish_connection(options)
  56             ActiveRecord::Base.configurations = { 'sqlite3_ar_integration' => options }
  57             ActiveRecord::Base.connection
  58           rescue Exception  # errors from establishing a connection
  59             $stderr.puts 'SQLite 3 unavailable; trying SQLite 2.'
  60             options = defaults.merge :adapter => 'sqlite'
  61             ActiveRecord::Base.establish_connection(options)
  62             ActiveRecord::Base.configurations = { 'sqlite2_ar_integration' => options }
  63             ActiveRecord::Base.connection
  64           end
  65
  66           Object.send(:const_set, :QUOTED_TYPE, ActiveRecord::Base.connection.quote_column_name('type')) unless Object.const_defined?(:QUOTED_TYPE)
  67         else
  68           raise "Can't setup connection since ActiveRecord isn't loaded."
  69         end
  70       end
  71
  72       # Load actionpack sqlite tables
  73       def load_schema
  74         File.read(File.dirname(__FILE__) + "/fixtures/db_definitions/sqlite.sql").split(';').each do |sql|
  75           ActiveRecord::Base.connection.execute(sql) unless sql.blank?
  76         end
  77       end
  78
  79       def require_fixture_models
  80         Dir.glob(File.dirname(__FILE__) + "/fixtures/*.rb").each {|f| require f}
  81       end
  82   end
  83 end
  84
  85 class ActiveRecordTestCase < ActiveSupport::TestCase
  86   # Set our fixture path
  87   if ActiveRecordTestConnector.able_to_connect
  88     self.fixture_path = [FIXTURE_LOAD_PATH]
  89     self.use_transactional_fixtures = false
  90   end
  91
  92   def self.fixtures(*args)
  93     super if ActiveRecordTestConnector.connected
  94   end
  95
  96   def run(*args)
  97     super if ActiveRecordTestConnector.connected
  98   end
  99
 100   def default_test; end
 101 end
 102
 103 ActiveRecordTestConnector.setup