robustify monkeypatch

1 files changed, 14 insertions, 0 deletions

diff --git a/testing/test_monkeypatch.py b/testing/test_monkeypatch.py
index 973811025..bc093ef5e 100644
--- a/testing/test_monkeypatch.py
+++ b/testing/test_monkeypatch.py
@@ -59,6 +59,20 @@ def test_setitem():
     monkeypatch.undo()
     assert d['x'] == 5
 
+def test_setitem_deleted_meanwhile():
+    d = {}
+    monkeypatch = MonkeyPatch()
+    monkeypatch.setitem(d, 'x', 2)
+    del d['x']
+    monkeypatch.undo()
+    assert not d
+
+def test_setenv_deleted_meanwhile():
+    monkeypatch = MonkeyPatch()
+    monkeypatch.setenv('XYZ123', 'hello')
+    monkeypatch.undo()
+    assert 'XYZ123' not in os.environ
+
 def test_delitem():
     d = {'x': 1}
     monkeypatch = MonkeyPatch()