diff options
Diffstat (limited to 'test/rake/in_environment.rb')
-rw-r--r-- | test/rake/in_environment.rb | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/test/rake/in_environment.rb b/test/rake/in_environment.rb new file mode 100644 index 0000000000..f0a340fac9 --- /dev/null +++ b/test/rake/in_environment.rb @@ -0,0 +1,30 @@ +module InEnvironment + private + + # Create an environment for a test. At the completion of the yielded + # block, the environment is restored to its original conditions. + def in_environment(settings) + original_settings = set_env(settings) + yield + ensure + set_env(original_settings) if original_settings + end + + # Set the environment according to the settings hash. + def set_env(settings) # :nodoc: + result = {} + settings.each do |k, v| + result[k] = ENV[k] + if k == 'PWD' + result[k] = Dir.pwd + Dir.chdir(v) + elsif v.nil? + ENV.delete(k) + else + ENV[k] = v + end + end + result + end + +end |