環境変数をモックする処理をstubを使って書き換える

fjordllc · Nov 7, 2023 · f546eb4 · f546eb4
1 parent 5ba8640
commit f546eb4
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/test/supports/mock_env_helper.rb b/test/supports/mock_env_helper.rb
@@ -1,13 +1,8 @@
 # frozen_string_literal: true
 
 module MockEnvHelper
-  def mock_env(partial_env_hash)
-    old = ENV.to_hash
-    ENV.update partial_env_hash
-    begin
-      yield
-    ensure
-      ENV.replace old
-    end
+  def mock_env(hash_for_mock, &block)
+    env_hash = ENV.to_hash.merge(hash_for_mock)
+    ENV.stub(:[], ->(key) { env_hash[key] }, &block)
   end
 end