Contents

require 'erb'
ActiveRecord::Migration.verbose = false unless ENV.has_key?('DEBUG')

module ScopedSearch::RSpec::Database

  def self.establish_connection
    if ENV['DATABASE']
      self.establish_named_connection(ENV['DATABASE'])
    else
      self.establish_default_connection
    end
  end

  def self.test_databases_configuration
    file = File.expand_path("../database.yml", File.dirname(__FILE__))
    @database_connections ||= YAML.load(ERB.new(File.read(file)).result)
  end
  
  def self.test_databases
    test_databases_configuration.keys
  end

  def self.establish_named_connection(name)
    raise "#{name} database not configured" if test_databases_configuration[name.to_s].nil?
    ActiveRecord::Base.establish_connection(test_databases_configuration[name.to_s])
  end

  def self.establish_default_connection
    ActiveRecord::Base.establish_connection(:adapter => 'sqlite3', :database => ':memory:')
  end

  def self.close_connection
    ActiveRecord::Base.remove_connection
  end

  def self.create_model(fields)
    table_name = "model_#{rand}".gsub(/\./, '')
    ActiveRecord::Migration.create_table(table_name) do |t|
      fields.each do |name, field_type|
        t.send(field_type.to_s.gsub(/^unindexed_/, '').to_sym, name)
      end
    end

    klass = Class.new(ActiveRecord::Base)
    klass.set_table_name(table_name)
    yield(klass) if block_given?
    return klass
  end

  def self.drop_model(klass)
    ActiveRecord::Migration.drop_table(klass.table_name)
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
scoped_search-2.3.0	spec/lib/database.rb
scoped_search-2.2.1	spec/lib/database.rb