Sha256: c02cad0a0cc79be2819e12a1b83ea3ba96bcba1f8c78dcff7071a7e0b694385f
Contents?: true
Size: 1.23 KB
Versions: 10
Compression:
Stored size: 1.23 KB
Contents
require 'yaml'
class DBConnectionHelper
DATABASE_CONFIG_FILE = "database.yml"
class << self
def db_config
@db_config ||= YAML.load_file(File.expand_path(File.dirname(__FILE__)) + "/../#{DATABASE_CONFIG_FILE}")
end
def new_mysql_connection(role = :master, with_data = false, toxic = false)
key = role.to_s + toxic_postfix(toxic)
conn = ActiveRecord::Base.establish_connection(
:host => '127.0.0.1',
:adapter => DATABASE.adapter,
:username => db_config[key]['user'],
:password => db_config[key]['password'],
:database => test_db_name,
:port => db_config[key]['port']
)
conn = conn.connection
init_with_dummy_data(conn) if with_data
conn
end
def toxic_postfix(toxic)
toxic ? "_toxic" : ""
end
def test_db_name
@test_db_name ||= "test"
end
def test_table_name
@test_table_name ||= "test"
end
def init_with_dummy_data(conn)
conn.execute("DROP TABLE IF EXISTS #{test_table_name} ")
conn.execute("CREATE TABLE #{test_table_name} (id int)")
1.upto(9) do |i|
query = "INSERT INTO #{test_table_name} (id) VALUE (#{i})"
conn.execute(query)
end
end
end
end
Version data entries
10 entries across 10 versions & 1 rubygems