针对SQLModel与SQLite应用的测试策略：使用临时数据库的实践指南

import sqlite3
# ...
con = sqlite3.connect(f"sqlite:///{tmp_path}/db.db") # 错误：包含sqlite:///前缀

import sqlite3
import pytest
from pathlib import Path

def test_sqlite3_connection(tmp_path: Path):
    temp_db_file = tmp_path / "test.db"
    # 正确：直接传递文件路径
    con = sqlite3.connect(str(temp_db_file))
    cur = con.cursor()
    cur.execute("CREATE TABLE test_table (id INTEGER PRIMARY KEY, name TEXT)")
    con.close()
    assert temp_db_file.exists()

# db.py
from sqlmodel import create_engine
sqlite_url = "sqlite:///database.db" # 默认生产数据库
engine = create_engine(sqlite_url) # 在导入时创建，指向database.db

# test_app.py
import db # 此时db.engine已经指向database.db

def temp_db(path):
    db.sqlite_url = f"sqlite:///{path}/db.db" # 尝试修改url
    # 但db.engine仍然是旧的，指向database.db

# db.py (稍作调整，使engine可被外部替换)
from sqlmodel import SQLModel, create_engine, Session

sqlite_url = "sqlite:///database.db"
engine = create_engine(sqlite_url) # 初始引擎

def create_db_and_tables():
    """在当前配置的引擎上创建所有表结构。"""
    SQLModel.metadata.create_all(engine)

def get_session():
    """获取数据库会话。"""
    with Session(engine) as session:
        yield session

# test_app.py (使用pytest fixture来管理临时数据库)
import sqlite3
import pytest
from typer.testing import CliRunner
from pathlib import Path

from projects import db # 导入db模块
from projects.app_typer import app # 导入Typer应用

@pytest.fixture(name="temp_db_file")
def temp_db_fixture(tmp_path: Path):
    """
    为每个测试提供一个临时的SQLite数据库文件路径，
    并配置SQLModel引擎指向该路径，同时创建表结构。
    """
    temp_db_path = tmp_path / "test.db"

    # 关键：重新创建并赋值给db.engine，确保SQLModel使用临时数据库
    db.engine = db.create_engine(f"sqlite:///{temp_db_path}")

    # 在临时数据库上创建表结构，确保应用逻辑能找到表
    db.create_db_and_tables() 

    yield temp_db_path # 将临时数据库文件路径传递给测试函数

    # 清理工作：tmp_path由pytest自动管理，无需手动删除文件

runner = CliRunner()

def test_add_item_to_db(temp_db_file: Path):
    """
    测试向CLI应用添加项目的功能，并使用sqlite3直接验证临时数据库。
    """
    # 调用CLI命令。Typer应用会在其主回调中调用db.create_db_and_tables()
    # 但由于fixture已提前设置好引擎并创建表，这里是安全的。
    # 确保CLI应用的add命令最终会通过db.engine进行数据操作。
    result = runner.invoke(app, ["add", "public", "-n", "ProjectName", "-p", "00-00"])

    assert result.exit_code == 0
    # 根据实际应用输出调整断言
    assert "Project added successfully" in result.stdout or "Success" in result.stdout 

    # 使用sqlite3直接连接临时数据库进行验证
    # 注意：sqlite3.connect()需要直接文件路径
    con = sqlite3.connect(str(temp_db_file)) # 将Path对象转换为字符串
    cur = con.cursor()

    # 查询并验证数据是否正确插入
    db_entry = cur.execute("SELECT * FROM project WHERE name = 'ProjectName'").fetchone()
    con.close() # 关闭数据库连接

    assert db_entry is not None
    assert db_entry[1] == "ProjectName" # 假设name是表的第二个字段
    assert db_entry[2] == "00-00" # 假设project_number是表的第三个字段

import typer
from projects.db import create_db_and_tables # 导入修改后的函数名

app = typer.Typer(add_completion=False)

@app.callback(invoke_without_command=True, no_args_is_help=True)
def main():
    """
    Typer应用的主回调，确保在任何命令执行前数据库表结构已存在。
    在测试环境中，db.engine已被fixture替换为指向临时数据库。
    """
    create_db_and_tables()

# 假设您的add命令是这样定义的
# from projects import add_command_module
# app.add_typer(add_command_module.app, name="add", help="Add a project to the DB.")