Question

我有这样的代码：

def extract(data):
    if len(data) == 3:
       a = 3
    else:
        component = data.split("-")
        if len(component) == 3:
            a,b,c = component
        else:
            raise globals.myException("data1", "Incorrect format", data)

    return a,b,c

这是一个简化的。我想模拟异常类globals.myException。我这样做：

def test_extract_data_throws_exception(self):
        with patch('globals.myException') as mock: 
            mock.__init__("data1", "Incorrect format", "")
            with self.assertRaises(myException):
                self.assertEqual(extract(""), (""))

我总是得到错误：“TypeError：异常必须是旧式类或派生自BaseException，而不是MagicMock”

编辑：正如@Aaron Digulla所说，猴子补丁是正确的解决方案。我发布解决方案以帮助其他人。

def test_extract_data_throws_exception(self):
        #monkey patching
        class ReplaceClass(myException):
            def __init__(self, module, message, detail = u''):
                pass

        globals.myException = ReplaceClass
        with self.assertRaises(myException:
            self.assertEqual(extract(""), (""))

Answer 1

原因是raise检查参数的类型。它必须是字符串（a.k.a“旧样式例外”）或派生自BaseException

由于模拟不是，raise拒绝使用它。

在这种特定情况下，您必须提出异常或使用monkey patching（=在测试中覆盖符号globals.myException并在之后将其恢复）。

如何在python中模拟异常调用？

1 个答案: