Fix ENV.delete with block

krekoten · Apr 5, 2012 · 8141ff4 · 8141ff4
1 parent aebaa23
commit 8141ff4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/kernel/common/env.rb b/kernel/common/env.rb
@@ -57,7 +57,11 @@ def each
 
     def delete(key)
       existing_value = self[key]
-      self[key] = nil if existing_value
+      if existing_value
+        self[key] = nil
+      elsif block_given?
+        yield key
+      end
       existing_value
     end