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