Refactor with_env method

Largo · Aug 4, 2024 · 6d7710f · 6d7710f
1 parent cfb6d22
commit 6d7710f
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/test/test_ocra.rb b/test/test_ocra.rb
@@ -80,17 +80,12 @@ def with_fixture(name, target_path = nil)
   # block. When the block exits, the environment variables are set
   # back to their original values.
   def with_env(hash)
-    old = {}
-    hash.each do |k,v|
-      old[k] = ENV[k]
-      ENV[k] = v
-    end
+    old = ENV.except(hash.keys)
+    ENV.update(hash)
     begin
       yield
     ensure
-      hash.each do |k,v|
-        ENV[k] = old[k]
-      end
+      ENV.update(old)
     end
   end