PYTEST MOCK:模拟抽象类

Question

我正在尝试使以下代码工作：

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？

Answer 1

而不是尝试修补真实实例，您可以使用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'

这用于创建任何类的模拟实例，而不仅仅是抽象的实例。

Answer 2

使用 pytest-mock或 unittest.mock您可以使用oigher.patch.multiple方法覆盖 __abstractmethods__属性。通过这样做，您将能够创建一个抽象类的实例并测试其非抽象方法。

使用pytest和pytest mock的示例：

def test_something(mocker):
    mocker.patch.multiple(ExampleClass, __abstractmethods__=set())
    instance = ExampleClass()
    ...