如何用pytest进行API测试

使用pytest进行API测试主要涉及以下几个步骤：

1. 安装pytest

首先，确保你已经安装了pytest。如果没有安装，可以使用pip进行安装：

pip install pytest

2. 创建测试文件

创建一个新的Python文件，例如test_api.py，用于存放你的API测试代码。

3. 编写测试用例

在test_api.py文件中编写测试用例。你可以使用requests库来发送HTTP请求。首先，确保你已经安装了requests库：

pip install requests

示例：测试一个GET请求

import requests
import pytest

BASE_URL = "https://api.example.com"

def test_get_endpoint():
    response = requests.get(f"{BASE_URL}/endpoint")
    assert response.status_code == 200
    assert "expected_key" in response.json()

示例：测试一个POST请求

import requests
import pytest

BASE_URL = "https://api.example.com"

def test_post_endpoint():
    payload = {"key": "value"}
    response = requests.post(f"{BASE_URL}/endpoint", json=payload)
    assert response.status_code == 201
    assert response.json()["key"] == "value"

4. 使用pytest的fixture功能

你可以使用pytest的fixture功能来管理测试数据和环境设置。

示例：使用fixture进行环境设置

import requests
import pytest

BASE_URL = "https://api.example.com"

@pytest.fixture(scope="module")
def api_client():
    return requests.Session()

def test_get_endpoint(api_client):
    response = api_client.get(f"{BASE_URL}/endpoint")
    assert response.status_code == 200
    assert "expected_key" in response.json()

5. 运行测试

在命令行中运行pytest来执行你的测试：

pytest test_api.py

6. 添加更多功能

你可以添加更多的功能来增强你的测试，例如：

• 使用pytest.mark.parametrize来参数化测试数据。
• 使用pytest.fixture来管理测试环境。
• 使用pytest.mark.skip和pytest.mark.skipif来跳过某些测试。
• 使用pytest.mark.xfail来标记预期失败的测试。

示例：参数化测试

import requests
import pytest

BASE_URL = "https://api.example.com"

@pytest.mark.parametrize("payload, expected_status", [
    ({"key": "value"}, 201),
    ({"key": "wrong_value"}, 400),
])
def test_post_endpoint(payload, expected_status):
    response = requests.post(f"{BASE_URL}/endpoint", json=payload)
    assert response.status_code == expected_status

通过以上步骤，你可以使用pytest进行API测试，并且可以根据需要添加更多的功能和复杂性。