Contents

require 'database_cleaner/generic/base'
require 'active_record'

module DatabaseCleaner
  module ActiveRecord

    def self.available_strategies
      %w[truncation transaction deletion]
    end

    def self.config_file_location
      "#{DatabaseCleaner.app_root}/config/database.yml"
    end

    module Base
      include ::DatabaseCleaner::Generic::Base

      attr_accessor :connection_hash

      def db=(desired_db)
        @db = desired_db
        load_config
      end

      def db
        @db || super
      end

      def load_config
        connection_details = YAML::load(ERB.new(IO.read(ActiveRecord.config_file_location)).result)
        self.connection_hash = connection_details[self.db.to_s]
      end

      def create_connection_klass
        Class.new(::ActiveRecord::Base)
      end

      def connection_klass
        return ::ActiveRecord::Base if connection_hash.nil?
        klass = create_connection_klass
        klass.send :establish_connection, connection_hash
        klass
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
database_cleaner-0.6.0	lib/database_cleaner/active_record/base.rb