Python，模拟：引发异常[关闭]

Question

我在测试中遇到从函数引发异常的问题：

### Implemetation
def MethodToTest():
    myVar = StdObject()
    try:
        myVar.raiseError() # <--- here
        return True
    except Exception as e:
        # ... code to test
        return False

### Test file
@patch('stdLib.StdObject', autospec=True)
def test_MethodeToTest(self, mockedObjectConstructor):
    mockedObj = mockedObjectConstructor.return_value
    mockedObj.raiseError.side_effect = Exception('Test') # <--- do not work
    ret = MethodToTest()
    assert ret is False

我想raiseError() 函数来引发错误。

我在 SO 上找到了几个示例，但没有一个符合我的需要。

Answer 1

我变了

@patch('stdLib.StdObject', autospec=True)

到

@patch('stdLib.StdObject', **{'return_value.raiseError.side_effect': Exception()})

并删除了# <--- do not work 行。

它现在可以工作了。

This 就是一个很好的例子。

编辑：

mockedObj.raiseError.side_effect = Mock(side_effect=Exception('Test'))

也可以。

Answer 2

好的，您提供的答案是有效的，但是您更改了执行方式（这很好。要解决您的原始问题，您需要将函数分配给 side_effect，而不是结果或对象：

def my_side_effect():
    raise Exception("Test")

@patch('stdLib.StdObject', autospec=True)
def test_MethodeToTest(self, mockedObjectConstructor):
    mockedObj = mockedObjectConstructor.return_value
    mockedObj.raiseError.side_effect = my_side_effect # <- note no brackets, 
    ret = MethodToTest()
    assert ret is False

希望对您有所帮助。注意，如果目标方法采用 args，副作用也需要采用 args（我相信）。