require 'active_record' module TurntableHelper def reload_turntable!(config_file_name = nil) ActiveRecord::Base.send(:include, ActiveRecord::Turntable) ActiveRecord::Base.turntable_config_file = config_file_name ActiveRecord::Turntable::Config.load!(ActiveRecord::Base.turntable_config_file, :test) end def establish_connection_to(env = :test) silence_warnings { Object.const_set('RAILS_ENV', env.to_s) Object.const_set('Rails', Object.new) allow(Rails).to receive(:env) { ActiveSupport::StringInquirer.new(RAILS_ENV) } ActiveRecord::Base.logger = Logger.new("/dev/null") } ActiveRecord::Base.establish_connection(env) end def truncate_shard ActiveRecord::Base.descendants.each do |klass| klass.delete_all end end def migrate(version) ActiveRecord::Migrator.run(:up, MIGRATIONS_ROOT, version) end end