77

在我尝试学习 TDD 时,尝试学习单元测试并在 python 中使用 mock。慢慢掌握它,但不确定我是否正确执行此操作。预先警告:我坚持使用 python 2.4,因为供应商 API 是作为预编译的 2.4 pyc 文件提供的,所以我使用的是 mock 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() 吗?如果是,怎么做?

谢谢。

4

6 回答 6

83

您可以patch()用作类装饰器,而不仅仅是函数装饰器。然后,您可以像以前一样传入模拟函数:

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

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

请参阅:对每个测试方法应用相同的补丁(其中还列出了替代方法)

如果您希望对所有测试方法进行修补,那么在 setUp 上以这种方式设置修补程序更有意义。

于 2014-05-12T20:29:04.690 回答
26

patch()我将从回答您的问题开始,然后我将详细举例说明如何setUp()交互。

  1. 我认为它看起来不正确,有关详细信息,请参阅我对此列表中问题 #3 的回答。
  2. 是的,对 patch 的实际调用看起来应该模拟您想要的对象。
  3. 不,你几乎不想@patch()setUp(). 你很幸运,因为对象是在setUp()测试方法中创建的,并且永远不会被创建。
  4. 我不知道有什么方法可以让模拟对象引发异常而不将该异常导入您的测试用例文件。
  5. 我认为这里没有任何必要patch.object()。它只是让您修补对象的属性,而不是将目标指定为字符串。

为了扩展我对问题 #3 的回答,问题是patch()装饰器仅在装饰功能运行时适用。一旦setUp()返回,补丁就会被删除。在您的情况下,这是可行的,但我敢打赌,这会使查看此测试的人感到困惑。如果您真的只希望补丁在 期间发生setUp(),我建议使用该with声明来明确补丁将被删除。

以下示例有两个测试用例。TestPatchAsDecorator表明装饰类将在测试方法期间应用补丁,但不会在setUp(). TestPatchInSetUp展示了如何应用补丁,使其在setUp()测试方法和测试方法中都到位。调用self.addCleanUp()确保补丁将在tearDown().

import unittest
from mock import patch


@patch('__builtin__.sum', return_value=99)
class TestPatchAsDecorator(unittest.TestCase):
    def setUp(self):
        s = sum([1, 2, 3])

        self.assertEqual(6, s)

    def test_sum(self, mock_sum):
        s1 = sum([1, 2, 3])
        mock_sum.return_value = 42
        s2 = sum([1, 2, 3])

        self.assertEqual(99, s1)
        self.assertEqual(42, s2)


class TestPatchInSetUp(unittest.TestCase):
    def setUp(self):
        patcher = patch('__builtin__.sum', return_value=99)
        self.mock_sum = patcher.start()
        self.addCleanup(patcher.stop)

        s = sum([1, 2, 3])

        self.assertEqual(99, s)

    def test_sum(self):
        s1 = sum([1, 2, 3])
        self.mock_sum.return_value = 42
        s2 = sum([1, 2, 3])

        self.assertEqual(99, s1)
        self.assertEqual(42, s2)
于 2016-12-20T23:21:06.870 回答
20

我想指出一个接受的答案的变体,其中一个new参数被传递给patch()装饰器:

from unittest.mock import patch, Mock

MockSomeClass = Mock()

@patch('mymodule.SomeClass', new=MockSomeClass)
class MyTest(TestCase):
    def test_one(self):
        # Do your test here

请注意,在这种情况下,不再需要在MockSomeClass每个测试方法中添加第二个参数 , ,这样可以节省大量代码重复。

可以在https://docs.python.org/3/library/unittest.mock.html#patch找到对此的解释:

如果patch()用作装饰器并且省略new,则创建的模拟将作为额外参数传递给装饰函数。

最重要的答案省略了new,但包含它会很方便。

于 2018-08-24T04:03:57.927 回答
16

如果你有很多补丁要应用,并且你希望它们应用到在 setUp 方法中初始化的东西上,也试试这个:

def setUp(self):
    self.patches = {
        "sut.BaseTestRunner._acquire_slot": mock.Mock(),
        "sut.GetResource": mock.Mock(spec=GetResource),
        "sut.models": mock.Mock(spec=models),
        "sut.DbApi": make_db_api_mock()
    }

    self.applied_patches = [mock.patch(patch, data) for patch, data in self.patches.items()]
    [patch.apply for patch in self.applied_patches]
    .
    . rest of setup
    .


def tearDown(self):
    patch.stopall()
于 2015-03-30T15:20:55.527 回答
0

您可以创建一个修补的内部函数并从setUp.

如果您的原始setUp功能是:

def setUp(self):
    some_work()

然后,您可以通过将其更改为来修补它:

def setUp(self):
    @patch(...)
    def mocked_func():
        some_work()

    mocked_func()
于 2019-02-28T13:51:11.467 回答
0

解决这个问题的另一种方法是在类之外定义补丁,然后重用它来装饰类以及 setUp 和 tearDown 方法,如下所示(重用 Don Kirkby 的示例):

import unittest
from mock import patch

my_patch = patch('__builtin__.sum', return_value=99)

@my_patch
class TestPatchAsDecorator(unittest.TestCase):
    @my_patch
    def setUp(self):
        s = sum([1, 2, 3])

        self.assertEqual(99, s)

    def test_sum(self, mock_sum):
        s1 = sum([1, 2, 3])
        mock_sum.return_value = 42
        s2 = sum([1, 2, 3])

        self.assertEqual(99, s1)
        self.assertEqual(42, s2)
于 2021-11-08T17:34:51.840 回答