如何在带有unittest setUp的python中正确使用模拟

Question

在尝试学习TDD的过程中，尝试学习单元测试并在python中使用模拟。慢慢掌握它，但不确定我是否正确执行此操作。预先警告：由于供应商API是预编译的2.4
pyc文件，因此我正在使用python 2.4进行卡住，因此我正在使用模拟0.8.0和unittest（不是unittest2）

在“ mymodule.py”中给出此示例代码

import ldap

class MyCustomException(Exception):
    pass

class MyClass:
    def __init__(self, server, user, passwd):
        self.ldap = ldap.initialize(server)
        self.user = user
        self.passwd = passwd

    def connect(self):
        try:
            self.ldap.simple_bind_s(self.user, self.passwd)
        except ldap.INVALID_CREDENTIALS:
            # do some stuff
            raise MyCustomException

现在在我的测试用例文件“
test_myclass.py”中，我要模拟ldap对象。ldap.initialize返回ldap.ldapobject.SimpleLDAPObject，因此我认为这是我必须模拟的方法。

import unittest
from ldap import INVALID_CREDENTIALS
from mock import patch, MagicMock
from mymodule import MyClass

class LDAPConnTests(unittest.TestCase):
    @patch('ldap.initialize')
    def setUp(self, mock_obj):
        self.ldapserver = MyClass('myserver','myuser','mypass')
        self.mocked_inst = mock_obj.return_value

    def testRaisesMyCustomException(self):
        self.mocked_inst.simple_bind_s = MagicMock()
        # set our side effect to the ldap exception to raise
        self.mocked_inst.simple_bind_s.side_effect = INVALID_CREDENTIALS
        self.assertRaises(mymodule.MyCustomException, self.ldapserver.connect)

    def testMyNextTestCase(self):
        # blah blah

带我几个问题：

1. 看起来正确吗？:)
2. 这是尝试模拟在我正在测试的类中实例化的对象的正确方法吗？
3. 可以在setUp上调用@patch装饰器，还是会引起怪异的副作用？
4. 无论如何，有没有要引发ldap.INVALID_CREDENTIALS异常的模拟而不必将异常导入到我的测试用例文件中？
5. 我应该改用patch.object（），如果是，怎么办？

谢谢。

Answer 1

您可以patch()用作类装饰器，而不仅可以用作函数装饰器。然后，您可以像以前一样传递模拟函数：

@patch('mymodule.SomeClass')
class MyTest(TestCase):

    def test_one(self, MockSomeClass):
        self.assertIs(mymodule.SomeClass, MockSomeClass)

请参阅：将相同的补丁程序应用于每种测试方法（还列出了替代方法）

如果要对所有测试方法进行修补，则在setUp上以这种方式设置修补程序更有意义。