python unittest setup,关于单元测试：如何在python中使用unittest setUp正确使用mock

淩亂°似流年 2023-01-17 03:17 158阅读 0赞

在我尝试学习TDD时，尝试学习单元测试并使用mock与python。慢慢地掌握它，但不确定我是否正确地这样做。预警：我正在坚持使用python 2.4，因为供应商API是预先编译的2.4 pyc文件，所以我使用模拟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

引出了几个问题：

那看起来不错吗？ :)

这是尝试和模拟在我正在测试的类中实例化的对象的正确方法吗？

可以在setUp上调用@patch装饰器，还是会导致奇怪的副作用？

反正是否有模拟提升ldap.INVALID\_CREDENTIALS异常而不必将异常导入我的testcase文件？

我应该使用patch.object()而不是，如果是这样，怎么样？

谢谢。

1-3)似乎对我很好... 4)import ldap而不是设置side\_effect = ldap.INVALID\_CREDENTIALS？

您可以随时进行相同的测试但是使用自己制作的更简单的对象...

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

@patch('mymodule.SomeClass')

class MyTest(TestCase):

def test\_one(self, MockSomeClass):

self.assertIs(mymodule.SomeClass, MockSomeClass)

见：26.5.3.4。将相同的补丁应用于每种测试方法(也列出替代方案)

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

对于Python3之前的Mock(托管在voidspace.org.uk/python/mock上)，请参阅对每个测试方法应用相同的补丁。

我刚遇到一个问题，我在TestCase类上有一个类级别的模拟，并假设它在setUp()方法中进行调用时已经存在。不是这种情况;类级别的模拟不及时应用于setUp()。我解决了这个问题，相反，创建了一个我在所有测试中使用的辅助方法。不确定这是最好的方法，但它确实有效。

@berto如果你在答案中扩展你的评论，我认为这将有所帮助。这是一个与其他人不同且可能更容易的解决方案。

我将首先回答您的问题，然后我将详细介绍patch()和setUp()如何进行交互。

我不认为它看起来是正确的，请参阅此列表中问题＃3的答案以获取详细信息。

是的，实际调用补丁看起来应该模拟你想要的对象。

不，你几乎不想在setUp()上使用@patch()装饰器。你很幸运，因为该对象是在setUp()中创建的，并且在测试方法期间永远不会被创建。

我不知道如何使模拟对象引发异常而不将该异常导入测试用例文件。

我认为这里不需要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)

我想你想给出你提到的答案＃3的超链接，因为SO根据他们收到的分数对答案进行排序。

我明白你的意思，@ ErdinEray，但我实际上是在讨论OP对问题＃3的回答。

我非常喜欢这种技术，因为它允许您为适用于大多数测试的类级模拟创建默认配置，然后需要模拟以不同方式运行的测试可以覆盖它。真的很不错。

如果要应用许多补丁并且希望它们应用于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()

考虑在tearDown()中使用patch.stop\_all()。

我试过这个 - 似乎还需要启动applied\_pa??tches。考虑一下这样的行：for patch in self.applied\_patches: patch.start()

它是stopall，而不是stop\_all。

公平地说 - 我现在使用"self.addCleanup(patch)"方法。是时候更新这个答案了。

self.addCleanup(patch.stopall)

我想指出一个接受的答案的变体，其中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找到对此的解释：

If patch() is used as a decorator and new is omitted, the created mock is passed in as an extra argument to the decorated function.

上面的答案都省略了新的，但包含它可能很方便。

谢谢你！这有很多帮助，尤其是当有许多模拟的函数类时，不需要特殊的返回值或类似函数。保持测试用例函数定义更清晰。

您可以创建一个修补的内部函数并从setUp调用它。

如果您的原始setUp功能是：

def setUp(self):

some\_work()

然后你可以修改它来修改它：

def setUp(self):

@patch(...)

def mocked\_func():

some\_work()

mocked\_func()

那么为什么不使用patch作为上下文管理器呢？ with patch(...):

python unittest setup,关于单元测试：如何在python中使用unittest setUp正确使用mock

发表评论取消回复

还没有评论，来说两句吧...

相关阅读

相关 Python-unittest单元测试

相关 python unittest setup,关于单元测试：如何在python中使用unittest setUp正确使用mock

相关 python单元测试入门_python 单元测试- unittest

相关 Python单元测试框架-unittest

相关 Python 单元测试库 Mock unittest

相关 python中的单元测试模块unittest

相关 python单元测试框架unittest介绍和使用

相关 Python Flask，单元测试案例，unittest

相关 Python单元测试框架——unittest

相关 Python单元测试框架—Unittest