python - 如何为 Pytest 模拟库类

Question

我是模拟和测试 python 代码(任何代码)的新手。

我正在尝试测试我的函数 main.py

def get_channel_list():
    sc = SlackClient(get_token())
    channels_list_json = sc.api_call("channels.list")
    if channels_list_json['ok'] == True:
        return channels_list_json

这是我要测试的功能

我需要模拟补丁 sc.api_call("channels.list") 来返回 JSON 对象 但我找不到任何这样的例子来帮助我弄清楚如何去做。

我发现的一切都像这个例子Mocking a class method...

我认为它看起来像这样:

@patch.object(SlackClient, 'api_call')
def test_get_channel_list():
    assert get_channel_list() != ""

我不必测试库我需要测试我之前提到的函数中的其余代码。 感谢您的帮助，我真的很满意这个测试。

Answer 1

您需要编写一个单独的模拟函数来返回一个 JSON 对象。

你可以试试这个:

@pytest.fixture
def mock_api_call(monkeypatch):
    monkeypatch.setattr(SlackClient, 'api_call', lambda self, arg: {"ok": True})

def test(mock_api_call):
    sc = SlackClient(get_token())
    channels_list_json = sc.api_call("channels.list")
    assert True == channels_list_json['ok']

def test_get_channel_list(mock_api_call):
    channels_list_json = get_channels_list()
    assert dict == type(channels_list_json)