如何测试抽象工厂

Question

我正在尝试在 Python 中使用一个抽象工厂，使用以下 3 个文件进行最低限度的复制：

test_factory.py

from factory import Factory

def test_factory():
    factory = Factory.makeFactory()
    product = factory.makeProduct('Hi there')
    print(product)

if __name__ == '__main__':
    test_factory()

factory.py

from abc import ABCMeta, abstractmethod
from product import ConcreteProduct

class Factory(metaclass=ABCMeta):
    @classmethod
    @abstractmethod
    def makeProduct(cls):
        pass

    @classmethod
    def makeFactory(cls):
        return ConcreteFactory()

class ConcreteFactory(Factory):
    @classmethod
    def makeProduct(cls, message):
        return ConcreteProduct(message)

product.py

class ConcreteProduct(object):
    def __init__(self, message):
        self._message = message
    def __str__(self):
        return self._message

我无法弄清楚如何模拟此代码以验证 ConcreteProduct.__init__ 是否使用适当的值调用。由于测试文件永远不会看到product.py，因此我不确定如何完成此操作，或者是否可能。我怀疑我的设计存在更根本的问题。

Answer 1

最简单的方法是patchfactory.ConcreteProduct模块引用。

所以你的测试可以（未测试）：

from factory import Factory
from unittest.mock import *

@patch("factory.ConcreteProduct")
def test_factory(mock_product_factory):
    mock_product = mock_product_factory.return_value
    factory = Factory.makeFactory()
    product = factory.makeProduct('Hi there')
    self.assertIs(product, mock_product)
    mock_product_factory.assert_called_with('Hi there')

if __name__ == '__main__':
    test_factory()

如果（且仅当）factory 模块中的ConcreteProduct 引用在您的测试环境中不存在，您可以使用create=True patch 的属性来注入它。

我想指出factory 模块中的ConcreteProduct 引用已经是一个工厂。 Python 中的类是工厂，但它不是类型语言，工厂概念不像 java 那样死板。我来自 Java 背景，即使在 python 中我仍然使用工厂，但是当您应该操纵输入以创建正确的对象时，它们变得非常有用，如果您的工厂方法只是传递给类引用的参数，请考虑删除中间的人。

Answer 2

到目前为止，我已经得到了@Michele dAmico 的回答所引导的解决方案，并且非常接近他的解决方案。

test_factory.py 变为：

from factory import Factory
from unittest.mock import patch

@patch('product.ConcreteProduct.__init__', return_value=None)
def test_factory(mock_init):
    factory = Factory.makeFactory()
    product = factory.makeProduct('Hi there')
    mock_init.assert_called_with('Hi there')


if __name__ == '__main__':
    test_factory()

请注意，我正在修补 product.，而不是 factory.，所以我基本上是在回避 factory.py 并嘲笑我知道它将导入的内容。我不知道我对以这种方式打破封装的感觉如何，但老实说，这就是我对一般嘲笑的感觉。

我更喜欢这个而不是另一个答案，因为它有点短，而且因为根据the mock docs 它可能很危险：

默认情况下，补丁将无法替换不存在的属性。如果 你传入create=True，属性不存在，patch会 在调用修补函数时为您创建属性，并且 之后再次删除它。这对于编写针对 您的生产代码在运行时创建的属性。关闭 默认，因为它可能很危险。打开它你可以写 针对实际上不存在的 API 通过测试！

我当然会对进一步的讨论感兴趣，因为我仍然有一种感觉，我可以学习一些更好的设计方法来使这个更干净。