FastAPI+SQLAlchemy测试环境中Base.metadata.create_all无法创建测试数据库表问题

阿华AIGC实验室

2026-4-27

问题诊断与解决方案

我之前也碰到过几乎一模一样的问题，核心原因其实很明确：你调用Base.metadata.create_all()的时候，SQLAlchemy的元数据里根本没包含URL表的定义，所以测试数据库自然不会生成对应的表，最终触发表不存在的报错。

为什么会出现这种情况？因为SQLAlchemy的declarative_base()生成的Base类，只有在你导入并加载了继承它的模型（比如你的URL类）之后，才会把表结构注册到Base.metadata中。如果测试代码里先执行了create_all()，却没加载模型类，那元数据是空的，自然不会创建任何表。

下面给你几个具体的解决步骤：

1. 先加载模型类，再创建表

在测试代码的Base.metadata.create_all()调用之前，显式导入你的URL模型（以及其他所有继承自Base的模型）。比如：

# 替换成你实际存放URL模型的模块路径
from your_app.models import URL

# 确保模型加载后，再执行表创建
Base.metadata.create_all(bind=engine)

这样Base.metadata里就会包含urls表的定义，create_all才能在测试数据库中生成对应的表结构。

2. 确认测试用的是正确的Base实例

绝对不要在测试代码里重新创建一个新的declarative_base()，必须直接从生产代码模块导入已经关联了模型的Base实例。比如：

# 从你的数据库配置模块导入Base，而不是自己新建
from your_app.database import Base

如果测试代码用了新的Base实例，它和生产代码里关联URL模型的Base完全是两个独立的对象，根本不知道有urls表的存在。

3. 用pytest Fixture优化测试数据库流程（推荐）

手动处理表的创建和清理容易出错，建议用pytest的fixture封装整个测试数据库的生命周期，做到测试前自动建表、测试后自动清理：

import pytest
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from your_app.database import Base, get_db
from your_app.main import app
from fastapi.testclient import TestClient

@pytest.fixture(scope="module")
def test_engine():
    return create_engine(
        "mysql+pymysql://{user}:{password}@{ip}:{port}/testurldb".format(
            user=user, password=password, ip=ip, port=port
        )
    )

@pytest.fixture(scope="module")
def test_session(test_engine):
    # 预先创建所有表
    Base.metadata.create_all(bind=test_engine)
    Session = sessionmaker(bind=test_engine)
    session = Session()
    yield session
    # 测试结束后清理数据并删除表
    session.rollback()
    Base.metadata.drop_all(bind=test_engine)
    session.close()

@pytest.fixture(scope="module")
def client(test_session):
    def override_get_db():
        try:
            yield test_session
        finally:
            test_session.close()
    app.dependency_overrides[get_db] = override_get_db
    yield TestClient(app)
    # 测试完成后恢复原依赖，避免影响其他测试
    del app.dependency_overrides[get_db]

这种方式不仅能确保表被正确创建，还能避免测试数据残留，让测试更可靠。