require 'yaml' require 'erb' class SpecClient def self.load_config(file) YAML.load(ERB.new(File.read(file)).result) end def self.mysql(name) file = File.join(File.dirname(__FILE__), '..', 'config', 'connections', 'mysql', "#{name}.yml") config = self.load_config(file) db = config[:database] config.delete(:database) connection = ::Mysql2::Client.new(config) begin connection.query("use `#{db}`") rescue Exception => e puts "#{e} => will create new databse #{db}" end connection end def self.elasticsearch(name) file = File.join(File.dirname(__FILE__), '..', 'config', 'connections', 'elasticsearch', "#{name}.yml") config = self.load_config(file) ::Elasticsearch::Client.new(config) end end