mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
40 lines
966 B
Ruby
40 lines
966 B
Ruby
|
require 'yaml'
|
||
|
require 'erubis'
|
||
|
require 'fileutils'
|
||
|
|
||
|
module ARTest
|
||
|
class << self
|
||
|
def config
|
||
|
@config ||= read_config
|
||
|
end
|
||
|
|
||
|
private
|
||
|
|
||
|
def read_config
|
||
|
unless File.exist?(TEST_ROOT + '/config.yml')
|
||
|
FileUtils.cp TEST_ROOT + '/config.example.yml', TEST_ROOT + '/config.yml'
|
||
|
end
|
||
|
|
||
|
raw = File.read(TEST_ROOT + '/config.yml')
|
||
|
erb = Erubis::Eruby.new(raw)
|
||
|
expand_config(YAML.parse(erb.result(binding)).transform)
|
||
|
end
|
||
|
|
||
|
def expand_config(config)
|
||
|
config['connections'].each do |adapter, connection|
|
||
|
dbs = [['arunit', 'activerecord_unittest'], ['arunit2', 'activerecord_unittest']]
|
||
|
dbs.each do |name, dbname|
|
||
|
unless connection[name].is_a?(Hash)
|
||
|
connection[name] = { 'database' => connection[name] }
|
||
|
end
|
||
|
|
||
|
connection[name]['database'] ||= dbname
|
||
|
connection[name]['adapter'] ||= adapter
|
||
|
end
|
||
|
end
|
||
|
|
||
|
config
|
||
|
end
|
||
|
end
|
||
|
end
|