Python接口自动化测试：断言封装详解

概要

在进行API接口测试时，断言起着至关重要的作用。断言是用于验证预期结果与实际结果是否一致的过程。在Python中，我们可以利用一些库来实现断言功能。

1. 安装必要的库

在Python中，我们主要会使用两个库：requests和jsonpath。requests库用于发送HTTP请求，而jsonpath库则用于解析JSON数据。

首先，我们需要安装这两个库。打开命令行，输入以下命令：

pip install requests
pip install jsonpath

2. 封装断言函数

接下来，我们需要编写一个断言函数。该函数负责读取接口返回的信息，并根据预先设定的条件进行断言。以下是一个基本的断言函数示例：

import requests
import jsonpath

def assert_response(response, assertions):
    for assertion in assertions:
        actual_value = jsonpath.jsonpath(response.json(), assertion["jsonpath"])
        expected_value = assertion["expected_value"]
        assert actual_value[0] == expected_value, f"断言失败。期望值为 {expected_value}，实际值为 {actual_value[0]}"

在这个函数中，我们首先使用jsonpath从JSON数据中提取实际值。然后，我们使用assert语句来进行断言，判断实际值是否等于预期值。

3. 使用断言函数

现在，我们可以在接口测试中使用这个断言函数了。

首先，我们需要发送一个HTTP请求，并获取返回的响应。然后，我们可以使用断言函数来检查响应是否符合预期。

以下是一个示例：

# 发送HTTP请求
url = "<http://example.com/api>"
response = requests.get(url)

# 定义断言
assertions = [
    {
        "jsonpath": "$.status",
        "expected_value": "success"
    },
    {
        "jsonpath": "$.data[0].id",
        "expected_value": 1
    }
]

# 使用断言函数
assert_response(response, assertions)

在这个示例中，我们首先发送一个GET请求到http://example.com/api。然后，我们定义了两个断言：一个是检查status是否为success，另一个是检查第一条数据的id是否为1。最后，我们调用assert_response函数进行断言。

4. 高级断言示例

除了基本的断言功能之外，我们还可以进行更多的断言操作。以下是一些高级断言示例：

检查列表长度

有时候，我们需要检查返回结果中某个列表的长度是否符合预期。可以通过以下方式进行断言：

assert len(response.json()["data"]) == expected_length

例如，假设我们期望返回的数据列表长度为10，我们可以使用上述断言来验证。

检查关键字存在

有时候，我们需要检查返回结果中是否包含某个关键字。可以通过以下方式进行断言：

assert keyword in response.json()["data"]

例如，假设我们期望返回的数据中包含关键字"example"，我们可以使用上述断言来验证。

检查响应时间

有时候，我们需要检查接口的响应时间是否在预期范围内。可以通过以下方式进行断言：

assert response.elapsed.total_seconds() <= expected_time

例如，假设我们期望接口的响应时间不超过5秒，我们可以使用上述断言来验证。

检查状态码

在接口测试中，我们经常需要检查返回的状态码是否符合预期。可以通过以下方式进行断言：

assert response.status_code == expected_status_code

例如，假设我们期望返回的状态码为200，我们可以使用上述断言来验证。

通过这些高级断言操作，我们可以更加灵活地进行接口测试和结果验证。

5. 完全封装的断言方法

除了逐个断言检查外，我们还可以将所有断言封装到一个方法中，以便更方便地进行结果验证。以下是一个完全封装的断言方法示例：

import requests
import jsonpath

def assert_responses(responses, assertions):
    for index, response in enumerate(responses):
        for assertion in assertions[index]:
            actual_value = jsonpath.jsonpath(response.json(), assertion["jsonpath"])
            expected_value = assertion["expected_value"]
            assert actual_value[0] == expected_value, f"断言失败。期望值为 {expected_value}，实际值为 {actual_value[0]}"

在这个方法中，我们接受一个包含多个响应的列表和一个包含多个断言的列表。我们使用嵌套循环来遍历每个响应和断言，并进行断言检查。

以下是一个示例：

# 发送HTTP请求
urls = ["<http://example.com/api1>", "<http://example.com/api2>", "<http://example.com/api3>"]
responses = [requests.get(url) for url in urls]

# 定义断言
assertions = [
    [
        {
            "jsonpath": "$.status",
            "expected_value": "success"
        },
        {
            "jsonpath": "$.data[0].id",
            "expected_value": 1
        }
    ],
    [
        {
            "jsonpath": "$.status",
            "expected_value": "success"
        },
        {
            "jsonpath": "$.data[0].name",
            "expected_value": "example"
        }
    ],
    [
        {
            "jsonpath": "$.status",
            "expected_value": "success"
        }
    ]
]

# 使用断言函数
assert_responses(responses, assertions)

在这个示例中，我们发送了三个GET请求到不同的URL，并获得了三个响应。然后，我们定义了三个断言列表，每个列表对应一个响应。我们可以通过调整断言列表来适应不同的测试场景。

通过完全封装的断言方法，我们可以更加灵活地进行多个响应的结果验证。