### 如何在 Pytest 中对 HTTP 状态码或返回码进行断言 为了实现对 HTTP 请求的测试并验证其状态码，在 Python 的 `pytest` 框架中通常会结合第三方库如 `requests` 来发送 HTTP 请求，并利用 `assert` 关键字来进行断言操作。 对于简单的 GET 或 POST 请求及其状态码的检验，可以直接使用 `requests.get()` 或者 `requests.post()` 方法发起请求并将响应对象传递给后续处理逻辑。之后可以通过访问 `.status_code` 属性来获取服务器端返回的状态码并与期望值做比较[^2]。 #### 使用 requests 库配合 pytest 进行基本的 HTTP 测试 ```python import pytest import requests def test_http_status(): response = requests.get('https://www.baidu.com') # 对单个状态码进行断言 assert response.status_code == 200, f"Expected status code to be 200 but got {response.status_code}" ``` 当涉及到更复杂的场景比如需要支持多个可能的成功状态码时，则可以在断言条件里加入更多的灵活性： ```python def test_multiple_success_codes(): success_codes = (200, 201) response = requests.get('https://example.com/api/resource') # 支持多状态码断言 assert response.status_code in success_codes, \ f"Status code should be one of {success_codes}, actual was {response.status_code}" ``` 如果要模拟 YAML 文件中的配置项用于定义哪些状态码视为成功的情况，也可以这样做： ```python # 假设这是从某个地方读取到的配置数据结构 config_data = { 'expected_status_codes': ['200', '201'] } def test_with_configured_status_codes(): expected_codes = list(map(int, config_data['expected_status_codes'])) response = requests.get('http://someapi.example.org/endpoint') # 利用配置的数据源来做断言 assert response.status_code in expected_codes, \ f"Response did not match any configured successful codes ({expected_codes})" ``` 上述例子展示了如何基于不同的需求灵活调整断言的方式，无论是针对单一还是多种成功的 HTTP 状态码都能很好地覆盖测试案例的需求[^1]。 另外值得注意的是除了直接对比整数形式的状态码外还可以考虑其他类型的断言例如检查 JSON 响应体内的特定字段是否存在或是具有预期的内容等[^3]。