Использование python mock для временного удаления объекта из dict

Question

Я пишу тест для некоторого кода, который проверяет значение в os.environ (я знаю, что это не оптимально, но я должен пойти с ним). Я хотел бы удалить запись из os.environ в течение всего теста. Я не уверен, что макет поддерживает это. Я знаю, что patch.dict может использоваться для изменения элемента, но я хочу, чтобы пара ключ/значение удалялась. Мне хотелось бы что-то в этом направлении:

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', ... } 

Есть ли способ совершить такой подвиг?

Answer 1

mock.patch.dict не совсем работает, как ваш образец желаемого кода. patch.dict - это функция, которая требует аргумента. Вероятно, вы хотите использовать его так:

>>> 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

Для удаления элемента, вы можете просто использовать:

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

А затем восстановить его:

os.environ['MY_THING'] = my_thing