sqlx数据库创建与迁移：configure_database函数作用及测试问题解析

问题背景 #

在阅读《Zero to Production》74-76页时遇到测试问题：每次执行cargo test，测试启动应用后，插入查询会重复插入相同值，触发数据库唯一键约束报错。书中给出的核心解决方案是为每次测试创建唯一的逻辑数据库，并在该数据库上执行迁移，但最后新增的configure_database函数的具体作用及解决逻辑需要拆解。

configure_database函数的具体作用 #

这个函数封装了测试数据库的创建与初始化全流程，分为两个核心步骤：

1. 创建唯一的测试数据库 #

• 首先通过connection_string_without_db连接到Postgres服务器（而非特定数据库）——因为创建新数据库必须先连接到已存在的系统默认库（如postgres库）。
• 执行CREATE DATABASE语句，用UUID生成的唯一名称创建全新的逻辑数据库，确保每次测试的数据库完全独立。

2. 执行数据库迁移 #

• 用新数据库的连接字符串创建连接池PgPool。
• 调用sqlx::migrate!执行项目中的迁移脚本，将数据库结构初始化到与生产环境一致的状态，保证测试时的数据库结构是正确可用的。

对应的核心代码逻辑：

pub async fn configure_database(config: &DatabaseSettings) -> PgPool {
    // 创建数据库
    let mut connection = PgConnection::connect(&config.connection_string_without_db())
        .await
        .expect("Failed to connect to Postgres");
    connection
        .execute(format!(r#"CREATE DATABASE "{}";"#, config.database_name).as_str())
        .await
        .expect("Failed to create database.");
    // 迁移数据库
    let connection_pool = PgPool::connect(&config.connection_string())
        .await
        .expect("Failed to connect to Postgres.");
    sqlx::migrate!("./migrations")
        .run(&connection_pool)
        .await
        .expect("Failed to migrate the database");
    connection_pool
}

如何解决测试失败的问题 #

之前的测试失败根源是多次测试复用同一个数据库，导致之前测试插入的数据残留，新测试插入相同值触发唯一键约束。而configure_database通过以下逻辑彻底解决这个问题：

• 每次测试生成全新数据库：用UUID作为数据库名称，保证每次运行测试时都是一个完全干净的数据库，没有历史数据残留。
• 独立的数据库环境：每个测试的数据库相互隔离，即使多个测试并行运行，也不会出现数据冲突或约束报错。
• 结构一致性：通过迁移脚本初始化数据库结构，确保测试环境的数据库结构与生产环境一致，测试结果更可靠。

内容的提问来源于stack exchange，提问作者Sarang Dutta