pytest接口自动化断言封装

在pytest接口自动化中，我们可以通过封装断言函数来实现断言的重用，提高代码的可维护性和可读性。下面是一个示例：

import json

def assert_code(response, expected_code):
    """
    封装断言状态码的函数
    :param response: 接口响应
    :param expected_code: 预期状态码
    """
    assert response.status_code == expected_code, f"响应状态码错误，预期为{expected_code}，实际为{response.status_code}"

def assert_json(response, expected_json):
    """
    封装断言json响应的函数
    :param response: 接口响应
    :param expected_json: 预期json响应
    """
    assert response.headers['Content-Type'] == 'application/json', "响应头Content-Type不是application/json"
    assert response.json() == json.loads(expected_json), f"响应json内容错误，预期为{expected_json}，实际为{response.text}"

在上面的代码中，我们封装了两个断言函数，一个用于断言状态码，一个用于断言json响应。这样，在编写测试用例时，我们只需要调用这些函数即可，如下所示：

def test_login():
    # 发送登录请求
    response = requests.post(url, data=data)
    # 断言状态码
    assert_code(response, 200)
    # 断言json响应
    assert_json(response, '{"code": 0, "msg": "登录成功"}')

这样，我们就可以通过封装断言函数来实现断言的重用，提高代码的可维护性和可读性。