def test_nonexistent_key_not_changed(self):
env = {"a": 1}
state = plugins._EnvironKeyState(env, "key")
state.restore()
self.assertEqual({"a": 1}, env)
def test_deleted_key_restored(self):
env = {"a": 1, "key": "value"}
state = plugins._EnvironKeyState(env, "key")
del env["key"]
state.restore()
self.assertEqual({"a": 1, "key": "value"}, env)
def test_overwritten_key(self):
env = {"a": 1, "key": "value"}
state = plugins._EnvironKeyState(env, "key")
env["key"] = "another value"
state.restore()
self.assertEqual({"a": 1, "key": "value"}, env)
def test_nonexistent_key(self):
env = {"a": 1}
state = plugins._EnvironKeyState(env, "key")
env["key"] = "create"
state.restore()
self.assertEqual({"a": 1}, env)
