Rust+Diesel操作PostgreSQL测试失败，数据库连接异常求助

核心问题 #

你的测试全部失败主要源于两个关键问题：

1. 测试并行执行导致数据交叉污染：Rust默认并行运行所有测试，多个测试同时操作数据库，数据状态互相干扰（比如test_create_task中出现了test_delete_task的测试数据）。
2. 测试环境变量未全局生效：仅在test_establish_connection_success中加载了.env.test，其他测试仍使用生产环境的数据库连接；且establish_connection每次调用都会重新加载.env，覆盖测试环境配置。

分步解决方案 #

1. 禁用测试并行（快速解决数据污染） #

在项目根目录的Cargo.toml中添加测试配置，强制串行执行测试：

[profile.test]
test = false  # 禁用并行测试

或者使用serial_test crate，给每个测试标记串行执行（更灵活）：

# 先在Cargo.toml添加依赖：serial_test = "2.0"
use serial_test::serial;

#[cfg(test)]
pub mod tests {
    use crate::*;
    use serial_test::serial;

    #[test]
    #[serial]
    fn test_create_task() { /* ... */ }

    #[test]
    #[serial]
    fn test_read_tasks() { /* ... */ }

    // 其他测试函数同理添加#[serial]属性
}

2. 全局加载测试环境变量 #

修改测试模块，在所有测试执行前统一加载.env.test，避免establish_connection覆盖配置：

#[cfg(test)]
pub mod tests {
    use crate::*;
    use dotenvy::from_filename;
    use std::sync::Once;

    // 全局初始化逻辑，仅执行一次
    static INIT: Once = Once::new();

    fn setup() {
        INIT.call_once(|| {
            // 优先加载测试环境变量，失败则终止
            from_filename(".env.test").expect("Failed to load .env.test file");
        });
    }

    #[test]
    fn test_establish_connection_success() {
        setup();
        let result = std::panic::catch_unwind(establish_connection);
        assert!(result.is_ok(), "Connection to the database failed");
    }

    #[test]
    #[serial]
    fn test_create_task() {
        setup();
        delete_all();
        let task = NewTodo { title: "Test Task", done: false };
        create_task(task);

        let tasks = read_tasks();
        assert_eq!(tasks.last().unwrap().title, "Test Task");
    }

    // 所有测试函数开头都调用setup()
}

同时修改establish_connection，避免重复加载.env覆盖测试配置：

pub fn establish_connection() -> PgConnection {
    // 仅在未读取到DATABASE_URL时加载.env
    if env::var("DATABASE_URL").is_err() {
        dotenv().ok();
    }

    let database_url = env::var("DATABASE_URL").expect("DATABASE_URL must be set");
    PgConnection::establish(&database_url)
        .unwrap_or_else(|_| panic!("Error connecting to {}", database_url))
}

3. 用事务隔离测试数据（更优雅的长期方案） #

使用Diesel的测试事务机制，每个测试在独立事务中执行，结束后自动回滚，彻底避免数据残留：

#[cfg(test)]
pub mod tests {
    use crate::*;
    use diesel::connection::Connection;

    // 封装测试数据库连接与事务逻辑
    fn with_test_db<F>(test: F)
    where
        F: FnOnce(&mut PgConnection) -> ()
    {
        setup(); // 加载测试环境变量
        let mut conn = establish_connection();
        // 启动测试事务，测试结束自动回滚
        conn.begin_test_transaction().unwrap();
        test(&mut conn);
    }

    // 重构核心数据操作函数，接受连接参数
    pub fn create_task(conn: &mut PgConnection, task: NewTodo) {
        use crate::schema::to_do;
        diesel::insert_into(to_do::table)
            .values(&task)
            .returning(ToDo::as_returning())
            .get_result(conn)
            .expect("Error saving new task");
    }

    // 重构后的测试函数
    #[test]
    fn test_create_task() {
        with_test_db(|conn| {
            let task = NewTodo { title: "Test Task", done: false };
            create_task(conn, task);

            let tasks = read_tasks(conn); // 同步重构read_tasks接受连接参数
            assert_eq!(tasks.last().unwrap().title, "Test Task");
        });
    }
}

这种方式无需手动清理数据，还能安全恢复并行测试。

额外注意事项 #

• 确保.env.test中的DATABASE_URL指向独立的测试数据库，与生产数据库完全隔离。
• 重构业务函数时，尽量让核心数据操作接受连接参数，提升代码的可测试性与灵活性。

内容的提问来源于stack exchange，提问作者Jean Remédios