aboutsummaryrefslogtreecommitdiffstats
path: root/Library/Homebrew/cask/spec/support/env_helper.rb
blob: 0a302ef45f7c061a6d8cae5b485a8f881b44fe4f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
module EnvHelper
  class << self
    def with_env_var(key, val)
      was_defined = ENV.key? "key"
      old_value = ENV["key"]
      ENV[key] = val
      yield
    ensure
      if was_defined
        ENV[key] = old_value
      else
        ENV.delete(key)
      end
    end
  end
end