17 lines
285 B
Ruby
17 lines
285 B
Ruby
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
|