module StubENV
  def stub_env(key, value)
    allow(ENV).to receive(:[]).and_call_original unless @env_already_stubbed
    @env_already_stubbed ||= true
    allow(ENV).to receive(:[]).with(key).and_return(value)
  end
end

# It's possible that the state of the class variables are not reset across
# test runs.
RSpec.configure do |config|
  config.after(:each) do
    @env_already_stubbed = nil
  end
end