考虑以下代码:
import unittest
from unittest.mock import patch
class Foo(object):
def __init__(self, bar=None):
self.bar = bar
def methodA(self):
print("In methodA")
def methodB(self):
print("In methodB")
def my_func(bar):
foo = Foo(bar)
if foo.bar:
foo.methodA()
foo.methodB()
class MyTestCase(unittest.TestCase):
def test_my_func(self):
bar = None
with patch("__main__.Foo", autospec=True) as foo:
my_func(bar)
foo.methodB.assert_called_once_with()
if __name__ == '__main__':
unittest.main()
这个想法很简单。我有一个函数,其行为切换实例属性的存在与否。我正在尝试编写一个单元测试来验证只执行某些Foo
方法,具体取决于属性。
基于模拟库的patch和autospeccing文档,我认为在autospec=True
上下文管理器中设置patch()
就足够了。它没有。生成的Mock()
正确包含methodA
和methodB
的模拟,但测试失败并出现此错误:
======================================================================
ERROR: test_my_func (__main__.MyTestCase)
----------------------------------------------------------------------
Traceback (most recent call last):
File "so.py", line 28, in test_my_func
my_func(bar)
File "trash.py", line 18, in my_func
if foo.bar:
File "/.../python3.3/unittest/mock.py", line 549, in __getattr__
raise AttributeError("Mock object has no attribute %r" % name)
AttributeError: Mock object has no attribute 'bar'
我确定我错过了一些明显的东西,但我似乎无法弄清楚是什么。我如何进行单元测试my_func()
?
答案 0 :(得分:1)
我的部分问题是对补丁行为的误解。我设置的上下文管理器返回__main__.Foo
模拟的实例,而不是my_func()
中使用的相同实例。换句话说,即使我能够正确地模拟Foo
,没有自动规范,我也无法对其任何方法执行assert_called_once_with()
:它不是同一个对象。
一种解决方案是模拟方法本身。这有效:
def test_my_func(self):
bar = None
with patch('__main__.Foo.methodB') as mock_methodB:
my_func(bar)
mock_methodB.assert_called_once_with()
另一种方法是修改my_func()
以返回foo:
def my_func(bar):
foo = Foo(bar)
if foo.bar:
foo.methodA()
foo.methodB()
return foo
因为函数返回被测试的模拟,所以应该有以下工作:
def test_my_func(self):
bar = None
with patch('__main__.Foo', spec=True, bar=None):
foo = my_func(bar)
assert foo.methodB.called
assert not foo.methodA.called