如何使用 side_effect 修补对象的属性

Question

所以我有一个文件some_class.py ：

class SomeReader:
    def read_path(self, url):
        return "read_path"


class SomeClass:
    def __init__(self, url):
        self.reader = SomeReader(url)
        print(self.reader.read_path(""))

还有一个测试文件some_class_test.py ：

from some_class import SomeClass, SomeReader

@patch("some_class.SomeReader")
def test_some_class(mock_some_reader):
    def mock_read_path(url):
        return "mock_read_path"

    mock_some_reader.read_path.side_effect = mock_read_path

    SomeClass("")

我期望当我运行这个测试时，它会打印mock_read_path而是打印<MagicMock name='SomeReader().read_path()' id='140701381288480'> 。 我该如何解决？ 我想模拟 SomeReader 的SomeReader初始化，因此我使用@patch("some_class.SomeReader") 。 但我也想模拟read_path的 read_path SomeReader ，因此我有mock_some_reader.read_path.side_effect = mock_read_path但这似乎不起作用。

Answer 1

您正在做的是模拟 class 本身，而不是类的实例。 模拟替换调用以实例化SomeReader(url) （基本上替换类的__init__方法）。

然后你要做的是模拟SomeReader(url)创建的假实例的返回值

@patch("some_class.SomeReader")
def test_some_class(mock_some_reader):
    def mock_read_path(url):
        return "mock_read_path"

    mock_some_reader.return_value.read_path.side_effect = mock_read_path

    SomeClass("")