From e9cae1a2f2c6fab13a69faa7220a8b3a01e1cb43 Mon Sep 17 00:00:00 2001
From: Chris Roberts <croberts@hashicorp.com>
Date: Fri, 6 Nov 2020 08:01:16 -0800
Subject: [PATCH] Allow `stub_env` to remove environment variable value

---
 test/unit/support/shared/base_context.rb | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/test/unit/support/shared/base_context.rb b/test/unit/support/shared/base_context.rb
index 63d672cc4..7a00e30f9 100644
--- a/test/unit/support/shared/base_context.rb
+++ b/test/unit/support/shared/base_context.rb
@@ -103,9 +103,10 @@ shared_context "unit" do
     allow(ENV).to receive(:[]).and_call_original
 
     hash.each do |key, value|
+      v = value.nil? ? nil : value.to_s
       allow(ENV).to receive(:[])
         .with(key.to_s)
-        .and_return(value.to_s)
+        .and_return(v)
     end
   end