Throw exception после первого вызова

Question

У меня есть цикл, где я обрабатываю добавление записей в zip-файл. Я издевался над своим объектом zipfile, и я хочу поднять исключение, чтобы убедиться, что моя логика для обработки больших zip-файлов будет работать правильно.

Есть ли способ с помощью MagicMocks или простых макетов принять первый вызов, но вызвать исключение на втором?

Answer 1

с генератором и Mock side_effect

from unittest.mock import Mock 

def sample_generator(): 
    yield 1 
    yield 2 
    raise Exception() 

gen = sample_generator() 

def sideeffect(): 
    global gen 
    for x in gen: 
     return x 


m = Mock(side_effect=sideeffect) 
m() #1 
m() #2 
m() #3 

вы можете получить:

File "test.py", line 22, in <module> 
    m() #3 
    .... 
    raise Exception() 
Exception 

Я уверен, что вы можете сделать его немного чище с некоторым усилием, но это должно решить основную проблему

Answer 2

Самый простой способ - использовать side_effect, которые принимают либо итерируемый, вызываемый, либо исключение (класс или экземпляр)

В качестве альтернативы side_effect может быть классом или экземпляром исключения. В этом случае исключение будет возникать при вызове mock.

Как показано в Quick Guide вы можете использовать side_effect, чтобы вызвать исключение, просто

>>> mock = Mock(side_effect=KeyError('foo')) 
>>> mock() 
Traceback (most recent call last): 
... 
KeyError: 'foo' 

Более того, вы можете использовать список и исключения вместе в side_effect задания. Поэтому самый простой способ сделать то, что вам нужно, это что-то вроде этого:

>>> m = Mock(side_effect=[1, KeyError("bar"), 3]) 
>>> m("a") 
1 
>>> m("b") 
... 
KeyError: 'bar' 
>>> m("c") 
3 

Другого способа сделать это может быть использовать вызываемый писать логику как ваш макет должен реагировать. В этом случае вы можете свободно выбирать, основываясь на аргументах или состоянии вашего теста.