rails--rails/railties/test/env_helpers.rb

28 lines
562 B
Ruby

# frozen_string_literal: true
require "rails"
module EnvHelpers
private
def with_rails_env(env, &block)
Rails.instance_variable_set :@_env, nil
switch_env "RAILS_ENV", env do
switch_env "RACK_ENV", nil, &block
end
end
def with_rack_env(env, &block)
Rails.instance_variable_set :@_env, nil
switch_env "RACK_ENV", env do
switch_env "RAILS_ENV", nil, &block
end
end
def switch_env(key, value)
old, ENV[key] = ENV[key], value
yield
ensure
ENV[key] = old
end
end