python - 从存储在变量中的类创建的模拟实例

Question

我有以下代码:

class Messenger(object):
    def __init__(self):
        # Class Type of what messages will be created as.
        message_class = Message

    def publish(self, body):
        # Instantiate object of type stored in `message_class`
        message = message_class(body)
        message.publish()

我想断言 Message.publish() 方法被调用。我如何实现这一目标？

我已经尝试过以下方法:

1. 将 message_class 分配给 Mock 或 Mock()。如果我调试 message_class(body) 返回的内容，它是一个模拟，但我似乎无法获取实例并断言它(因为我在测试中分配的模拟不是使用的实例，它是类型)。
2. 使用装饰器修补 Message 类。每当我这样做时，它似乎都没有捕获它。当我调试 message_class(body) 返回其 Message 类型时，而不是 Mock。
3. 尝试模拟 message_class 的 __init__ 方法，希望我可以设置每当代码尝试实例化消息时返回的实例。不起作用，会抛出错误，因为 __init__ 方法不应该有返回值。

Answer 1

如果您要存储实际实例，我想说您可以执行类似 messenger.message.publish.assert_used_once 的操作，但由于 message_class 已被存储，因此让它变得有点棘手。鉴于此，您可以从模拟的 class 中提取 return_value 并以这种方式检查调用。我是这样做的:

信使。请注意将 message_class 分配给 self 的细微修改。我假设你打算这样做，否则如果没有一些全局性的时髦，它就无法工作:

'''messenger.py'''
class Message(object):
    def __init__(self, body):
        self.body = body

    def publish(self):
        print('message published: {}'.format(self.body))


class Messenger(object):
    def __init__(self):
        # Class Type of what messages will be created as.
        self.message_class = Message

    def publish(self, body):
        # Instantiate object of type stored in `message_class`
        message = self.message_class(body)
        message.publish()

测试:

'''test_messenger.py'''    
from unittest import mock, TestCase

from messenger import Messenger

class TestMessenger(TestCase):
    @mock.patch('messenger.Message')
    def test_publish(self, mock_message):
        messenger = Messenger()
        messenger.publish('test body')
        # .return_value gives the mock instance, from there you can make your assertions
        mock_message.return_value.publish.assert_called_once()