单元测试模拟在 Python 中的 rest API 请求上失败

Question

您好，我是单元测试的新手，我正在研究 mocks 和 pytest。我正在尝试对两个 rest api 请求进行单元测试，其中 GET 请求检查 API 中是否不存在某个项目以及它是否不创建它与 POST 请求并创建一个文件夹，否则如果它存在只是创建一个文件夹。

我尝试使用Mocker()，但当我尝试模拟GET 请求时，我被困在AttributeError: Mocker 上。

这是我要测试的代码：

   client = requests.session()
    # get item by name
    response = client.get(
        f"https://url.com/rest/item/info?name=test_item",
        auth=username, password),
    )

    if (
        response.status_code == 200
        and response.json()["status"]
        == "Item doesn't exist!"
    ):
        logging.info(f"Creating item")
        client.post(
            "https://url.com/rest/item/create?name=test_item",
            auth=username, password),
        )
        # directory creation
        dir_make = "mkdir -p test_item/temperature"
        exec_cmd(dir_make)

    elif response.status_code == 200 and response.json()["status"]=="OK":
        # directory creation
        dir_make = "mkdir -p test_item/temperature"
        exec_cmd(dir_make)

这是以AttributeError 失败的单元测试：

def test_existing_item(requests_mock, monkeypatch):
    with requests_mock.Mocker() as mock:
        mock.get("https://url.com/rest/item/info?name=test_item", text="OK")

        resp = requests.get("https://url.com/rest/item/info?name=test_item")
        assert resp.text == "OK"

编辑：测试未找到项目和 POST 模拟。似乎它没有为else 语句添加覆盖范围。如果该项目存在并且在这种情况下只需要添加文件夹，如何测试？

编辑 2：添加了 elif 语句而不是 else 和 2 个单独的测试，仍然是一个 test_existing_items() 不包括 elif 语句......在这种情况下我做错了什么？

def test_existing_item(monkeypatch):
    with requests_mock.Mocker() as mock_request:
        mock_request.get(requests_mock.ANY, text="success!")
        resp = requests.get(
            "https://url.com/rest/item/info?name=test_item",
            auth=("mock_username", "mock_password"),
        )
        if resp.status_code == 200 and resp.json()["status"] == "OK":
            dir_make = "mkdir -p test_item/temperature"
            exec_cmd(dir_make)


    encoded_auth = b64encode(b"mock_username:mock_password").decode("ascii")

    assert mock_request.last_request.headers["Authorization"] == f"Basic {encoded_auth}"


def test_post_item(monkeypatch):
    with requests_mock.Mocker() as mock_request:
        mock_request.get(requests_mock.ANY, text="success!")
        resp = requests.get(
            "https://url.com/rest/item/info?name=test_item",
            auth=("mock_username", "mock_password"),
        )

        if resp.status_code == 200 and resp.json()["status"] == "ERROR":
            mock_request.get(requests_mock.ANY, text="success!")
            requests.post(
                "https://url.com/rest/item/create?name=test_item",
                auth=("mock_username", "mock_password"),
            )

        dir_make = "mkdir -p test_item/temperature"
        exec_cmd(dir_make)


    encoded_auth = b64encode(b"mock_username:mock_password").decode("ascii")

    assert mock_request.last_request.headers["Authorization"] == f"Basic {encoded_auth}"

我不熟悉单元测试，所以如果能对这段代码进行单元测试，我们将不胜感激。

Answer 1

import requests
import requests_mock


def test_existing_item(monkeypatch):
    with requests_mock.Mocker() as mock:
        mock.get("https://url.com/rest/item/info?name=test_item", text="OK")

        resp = requests.get("https://url.com/rest/item/info?name=test_item")
        assert resp.text == "OK"

不要将requests_mock 作为参数传递，pytest 应该可以正常工作。

编辑：

至于您的编辑：

似乎它没有为 else 语句添加覆盖范围。如果该项目存在并且在这种情况下只需要添加文件夹，如何测试？

那是因为您的if 条件始终为真，因此它永远不会访问else 语句下方的代码。你的第二个问题我不太清楚，但我相信你想写几个测试，一个是你的if 语句，一个是你的else 语句。根据经验，如果您在测试中需要条件逻辑，您就会遇到一个问题：要么每次运行测试时一切都应该按照您的意愿进行，要么您应该按照您的意愿中止并让测试失败代码在每次运行时都具有完全相同的行为 - 以及扩展的测试。