如何mock.patch一个生成器/有产量的类方法-解网

问：

如何修补生成器类方法。例如，在下面的课堂上，我们将如何修补get_changed_diff_patch方法？

class PassiveJsonMixin(JsonMixin):
    """
    passive items that do not have id/key
    """

    def __init__(self, *args, **kwargs):
        JsonMixin.__init__(self, *args, **kwargs)
        # self.build_json()
        pass
    def get_changed_diff_patch(self, parent_hidden=False):
        print ("From PassiveJsonMixin: get_changed_diff_patch")
        return
        yield

通常的方法

patch.object(mymod.PassiveJsonMixin, 'get_changed_diff_patch',wrapper(mymod.PassiveJsonMixin.get_changed_diff_patch) )

似乎不起作用。

生成器补丁产量 python-unittest.mock

from unittest import mock

class MyClass:
    def some_generator(self):
        for i in range(5):
            yield i

    def method_using_generator(self):
        values = []
        for value in self.some_generator():
            values.append(value)
        return values


with mock.patch.object(MyClass, 'some_generator', return_value=[10, 20, 30, 40, 50]):
    my_instance = MyClass()
    result = my_instance.method_using_generator()
    print (result)

上一个：如何将 Python 生成器转换为异步生成器？

下一个：什么决定了发电机的尺寸？

如何mock.patch一个生成器/有产量的类方法

How to mock.patch a class method which is a generator/has yield

评论