[英]pytest-mock: Mock an abstract class
我正在尝试使以下代码起作用:
from pytest_mock import mocker
class TestClass(mocker):
def setup_method(self):
self.some_mock = SomeAbstractClass()
self.testsubject = ClassThatIsBeingTested(self.some_mock)
def test_1(self):
mocker.patch(self.some_mock, 'some_function', return_value=5)
assert 5 == self.testsubject.function_that_internally_uses_the_mock()
但是我因为尝试实例化一个抽象类而得到一个TypeError
。
我如何模拟SomeAbstractClass
?
您可以使用unittest.mock
模块(也作为 Python 2.7 的mock
包提供)来创建模拟实例,而不是尝试修补真实实例:
import sys
if sys.version_info[0] == 2:
from mock import Mock
else:
from unittest.mock import Mock
# ...
some_mock = Mock(spec=SomeAbstractClass)
some_mock.some_function.return_value = 5
some_mock.some_function()
# => 5
some_mock.non_existent_function()
# => AttributeError: Mock object has no attribute 'non_existent_function'
这适用于创建任何类的模拟实例——而不仅仅是抽象类。
使用pytest-mock
或unittest.mock
你可以使用mocker.patch.multiple方法来覆盖__abstractmethods__
属性。 通过这样做,您将能够创建抽象类的实例并测试它的非抽象方法。
使用 pytest 和 pytest-mock 的示例:
def test_something(mocker):
mocker.patch.multiple(ExampleClass, __abstractmethods__=set())
instance = ExampleClass()
...
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.