Process database.yml configuration via ERB

Author: orien

script/jack_hammer

@@ -15,6 +15,8 @@ require 'optparse'
 require 'bundler/setup'
 require 'active_record'
 require 'database_cleaner'
+require 'erb'
+require 'yaml'
 
 support = File.expand_path("../../spec/support/", __FILE__)
 
@@ -25,7 +27,11 @@ if db_engine == 'sqlite'
   exit 0
 end
 
-ActiveRecord::Base.establish_connection YAML.load_file(File.join(support, "database.yml"))[db_engine]
+database_config_file = File.join(__dir__, '..', 'spec', 'support', 'database.yml')
+database_config_raw = File.read(database_config_file)
+database_config_yaml = ERB.new(database_config_raw).result
+database_config = YAML.load(database_config_yaml)
+ActiveRecord::Base.establish_connection(database_config[db_engine])
 require "#{support}/schema"
 
 lib = File.expand_path('../../lib', __FILE__)

spec/support/database.rb

@@ -1,18 +1,23 @@
 require 'active_record'
 require 'database_cleaner'
+require 'erb'
+require 'yaml'
 
 FileUtils.mkdir_p 'tmp'
 
 db_engine = ENV['DB'] || 'mysql'
-database_config_file = File.expand_path('../database.yml', __FILE__)
+database_config_file = File.join(__dir__, 'database.yml')
 
 raise <<-MSG.strip_heredoc unless File.exist?(database_config_file)
   Please configure your spec/support/database.yml file.
   See spec/support/database.example.yml'
 MSG
 
 ActiveRecord::Base.belongs_to_required_by_default = true if ActiveRecord.version.version >= '5'
-ActiveRecord::Base.establish_connection(YAML.load_file(database_config_file)[db_engine])
+database_config_raw = File.read(database_config_file)
+database_config_yaml = ERB.new(database_config_raw).result
+database_config = YAML.load(database_config_yaml)
+ActiveRecord::Base.establish_connection(database_config[db_engine])
 
 RSpec.configure do |config|
   config.before(:suite) do