Python模拟异常没有被捕获答案

【问题标题】：Python mocked exception not being caughtPython模拟异常没有被捕获
【发布时间】：2015-09-09 11:08:11
【问题描述】：

努力在标题中简洁地描述这一点......

我有一个要测试的模块：

mod.py：

import subprocess
class MyStuff(object):
    def my_fun(self):
        try:
            print subprocess
            out = subprocess.check_output(["echo", "pirates"])
        except subprocess.CalledProcessError:
            print "caught exception"

还有测试模块test_mod.py：

import unittest
import mock
from mod import MyStuff
import subprocess

class Tests(unittest.TestCase):
    def setUp(self):
        self.patched_subprocess = mock.patch(
            'mod.subprocess', autospec=True)
        self.mock_subprocess = self.patched_subprocess.start()
        self.my_stuff = MyStuff()

    def tearDown(self):
        self.patched_subprocess.stop()

    def test_my_fun(self):
        self.mock_subprocess.check_output = mock.Mock(
            side_effect=subprocess.CalledProcessError(0, "hi", "no"))
        with self.assertRaises(subprocess.CalledProcessError):
            out = self.my_stuff.my_fun()

if __name__ == '__main__':
    unittest.main()

然后我运行python test_mod.py 并看到以下输出：

<NonCallableMagicMock name='subprocess' spec='module' id='140654009377872'>
.
----------------------------------------------------------------------
Ran 1 test in 0.007s

OK

我很高兴 subprocess 对象已被模拟，但为什么 print "caught exception" 语句没有被执行？我猜这是因为真正的异常被抛出test_mod.subprocess.CalledProcessException 而不是subprocess.CalledProcessException，但我不知道如何解决这个问题。有什么建议吗？感谢您的宝贵时间。

【问题讨论】：

标签： python unit-testing mocking

【解决方案1】：

我最终解决了这个问题......

问题是我在模拟整个子进程模块，其中包括 CalledProcessError 异常！这就是为什么它似乎与我在测试模块中提出的异常不匹配，因为它是一个完全不同的对象。

解决方法是只模拟 subprocess.check_output，D'oh！

【讨论】：