activerecord/test/support/config.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

# frozen_string_literal: true
require "yaml"
require "erb"
require "fileutils"
require "pathname"

module ARTest
  class << self
    def config
      @config ||= read_config
    end

    private

    def config_file
      Pathname.new(ENV["ARCONFIG"] || TEST_ROOT + "/config.yml")
    end

    def read_config
      unless config_file.exist?
        FileUtils.cp TEST_ROOT + "/config.example.yml", config_file
      end

      erb = ERB.new(config_file.read)
      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_unittest2"],
               ["arunit_without_prepared_statements", "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