无法捕获Python模拟异常

Question

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

我有一个要测试的模块：

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

我很高兴子流程对象已被模拟，但是为什么不执行print "caught exception"语句？ 我猜这是因为真正的例外，在获得罚球test_mod.subprocess.CalledProcessException而不是subprocess.CalledProcessException因为我打算，但我不知道如何解决这个问题。 有什么建议吗？ 谢谢你的时间。

Answer 1

我终于解决了这个...

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

解决方法是模拟subprocess.check_output ，哦！