在Python中使用输入参数模拟嵌套函数

Question

我的文件夹结构

main.py
    class Class1
        generate_columns()
        column_api()
utilities.py
    class Class2
        get_response()

我想在column_api()中模拟class Class1函数。

功能签名如下：

class Class1:
    def generate_columns():
        calls `get_response()` from `class Class2` and returns response

    def column_api(data=List[Dict]]):
        resp = self.generate_columns(data)
        for item in resp:
            if data["name"] == item["name"]:
                ret.append(item)
    return ret

我想嘲笑column_api。这是我到目前为止的测试：

def test_column_api():
    testClass = Class1()
    mock_response = [{"id":"1", "name":"test1"}]
    load_data = [{"id":"1", "name":"test1"}, {"id":"2", "name":"test2"}]
    with patch("package.main.Class1.generate_columns") as api_mock:
        api_mock.return_value = mock_response

        result = testClass.column_api(load_data)
        assert len(result) == 1

它在这里失败了AssertionError。无论我尝试多少，返回的列表总是空的。但理想情况下，根据我的函数定义，返回的列表应该有一个字典，其名称与响应对象中的名称匹配。

我对Python中的Mock和patch概念很新。在阅读了关于SO的各种帖子和Mock的实际文档之后，我可以到此为止。我认为我的api_mock没有被column_api功能使用。我认为这就是问题所在，但我不知道如何让column_api认出api_mock。对此有任何帮助非常感谢！希望我的功能定义在这里很清楚，但我很乐意提供我可能错过的任何其他信息。

非常感谢你提前。我已经打破了这个问题很长一段时间了。希望能在这里找到答案。

Answer 1

我的一位好同事帮助我解决了这个问题，我将其发布在下面。这个解决方案适用于我的目的。

@pytest.fixture
def mocked_gen_col(monkeypatch):

    def mock_gen_technical(*args, **kwargs):
        return [
            {"name": "a"},
            {"name": "b"}
        ]
    monkeypatch.setattr(Class1, "generate_columns", mock_gen_technical)

def test_technical_column_api(mocked_gen_col):
    data = [{
        "id": "1",
        "name": "a"
    }]

    list = Class1().column_api(data=data)
    assert list[0]["name"] == "a"
    assert len(list) == 1