函数级测试夹具自增主键未重置导致Pytest测试失败

1. 移除硬编码主键，改用动态获取 #

测试用例中依赖1、2这类硬编码主键值，是导致测试受自增ID变化影响的核心原因。优化方式是让数据填充夹具返回创建的节点实例，测试时直接引用实例的ID。

修改公共conftest.py中的_populate_nodes夹具：

@fixture(scope="function", name="_populate_nodes")
def fixture_populate_nodes(_db_session):
    """Populate the nodes table with sample data and return the instances."""
    sample_data = [
        {
            "title": "Node 1",
            "logo": "logo1.jpg",
            "seo_description": "SEO description for Node 1",
        },
        {
            "title": "Node 2",
            "logo": "logo2.jpg",
            "seo_description": "SEO description for Node 2",
        },
    ]
    nodes = []
    for data in sample_data:
        node = Nodes(**data)
        _db_session.add(node)
    # 用flush替代commit，既生成ID又不提交事务，避免影响后续回滚
    _db_session.flush()
    return nodes

对应修改测试用例，比如test_get_node：

def test_get_node(test_client, _populate_nodes):
    """Test get_node endpoint."""
    target_node = _populate_nodes[0]
    response = test_client.get(f"/nodes/{target_node.id}")
    assert response.status_code == 200

2. 拆分测试依赖，减少不必要的前置数据 #

像test_create_node这类测试不需要预先存在的节点数据，完全可以独立执行，无需依赖_populate_nodes，既加快测试速度，又避免无关数据干扰：

def test_create_node(test_client):
    """Test create_node endpoint."""
    node_data = {
        "title": "Test Node 1",
        "logo": None,
        "seo_description": None,
    }
    response = test_client.post("/nodes/", json=node_data)
    assert response.status_code == 200
    # 额外验证返回数据的正确性，提升测试健壮性
    assert response.json()["title"] == node_data["title"]

3. 优化事务管理，避免不必要的提交 #

当前_populate_nodes中的commit会触发事务提交，虽然最终会回滚，但会导致Postgres自增序列无法重置（这也是ID持续递增的核心原因之一）。改用flush可在不提交事务的前提下生成主键ID，所有操作都在同一个事务中执行，最后统一回滚，序列不会被持久化更新。

同时确认_db_session夹具的事务逻辑：保证所有测试操作都在同一个事务中完成，避免中途提交导致的状态不一致。

4. 引入参数化测试，减少重复代码 #

对于批量操作的测试（如test_get_nodes、test_update_nodes、test_delete_nodes），用Pytest的@pytest.mark.parametrize复用测试逻辑，覆盖更多场景：

import pytest

@pytest.mark.parametrize("node_id_sets, expected_status", [
    ([], 200),
    ([1], 200),
    ([1,2], 200),
    ([999], 404),
])
def test_get_nodes(test_client, _populate_nodes, node_id_sets, expected_status):
    """Test get_nodes endpoint with various ID sets."""
    existing_ids = [node.id for node in _populate_nodes]
    # 替换参数中的无效ID，保留有效ID用于测试
    request_ids = [id if id in existing_ids else id for id in node_id_sets]
    query_params = "&".join([f"node_ids={id}" for id in request_ids])
    response = test_client.get(f"/nodes/?{query_params}")
    assert response.status_code == expected_status

5. 清晰划分夹具职责与作用域 #

• 公共conftest.py仅保留全局通用夹具（数据库连接、表创建销毁、基础数据填充）
• 服务级conftest.py专注于FastAPI客户端创建与依赖注入覆盖，避免职责混淆
• 特定业务的数据填充夹具放到对应测试模块的conftest.py中，不全部堆积在公共层

6. 增强测试断言的完整性 #

当前测试仅验证状态码，建议补充对返回数据结构、内容的断言，提升测试可信度：

def test_update_node(test_client, _populate_nodes):
    """Test update_node endpoint."""
    target_node = _populate_nodes[0]
    node_data = {"title": "Updated Node"}
    response = test_client.patch(f"/nodes/{target_node.id}/", json=node_data)
    assert response.status_code == 200
    updated_node = response.json()
    assert updated_node["title"] == node_data["title"]
    assert updated_node["id"] == target_node.id

内容的提问来源于stack exchange，提问作者Asad Amir Khwaja