Używanie makiety Pythona do tymczasowego usunięcia obiektu z dyktafonu

Question

Piszę test dla jakiegoś kodu, który sprawdza wartość w os.environ (Wiem, że to nie jest optymalne, ale muszę iść z tym). Chciałbym usunąć wpis z os.environ na czas trwania testu. Nie jestem pewien, czy mock popiera to. Wiem, że patch.dict można użyć do modyfikacji elementu, ale chcę, aby usunięto parę klucz/wartość. Chciałbym coś w tym stylu:

print os.environ 
{ ... , 'MY_THING': 'foo', ... } 

with mock.patch.dict.delete('os.environ', 'MY_THING'): 
    # run the test 
    # ('MY_THING' in os.environ) should return False 

# everything back to normal now  
print os.environ 
{ ... , 'MY_THING': 'foo', ... } 

Czy istnieje sposób na wykonanie takiego wyczynu?

Answer 1

mock.patch.dict nie bardzo działa jak próbka żądanego kodu. patch.dict to funkcja wymagająca argumentu. Prawdopodobnie chcesz go użyć w następujący sposób:

>>> import os 
>>> import mock 
>>> with mock.patch.dict('os.environ'): 
...  del os.environ['PATH'] 
...  print 'PATH' in os.environ 
... 
False 
>>> print 'PATH' in os.environ 
True 

Answer 2

Do usuwania elementu, można po prostu użyć:

my_thing = os.environ['MY_THING'] # Gotta store it to restore it later 
del os.environ['MY_THING'] 

a następnie przywrócić go z:

os.environ['MY_THING'] = my_thing