Rust集成测试：如何获取线程/Worker ID实现独立测试数据库？

在Rust测试里，要解决并行测试的数据库冲突问题，有几种实用方案，和Node里的Worker ID思路异曲同工：

1. 利用线程ID生成唯一数据库名 #

Rust的标准库可以获取当前测试线程的ID，把它转成字符串后用来命名数据库，确保每个线程用独立库：

use std::thread;

fn get_test_db_name() -> String {
    let thread_id = thread::current().id();
    // 把ThreadId的调试输出转换成简洁的字符串
    let id_str = format!("{:?}", thread_id)
        .trim_start_matches("ThreadId(")
        .trim_end_matches(")")
        .to_string();
    format!("test_db_{}", id_str)
}

这个方案简单直接，测试线程的ID在并行测试中是唯一的，足够应对大部分场景。

2. 为每个测试生成唯一标识（UUID） #

如果需要更彻底的隔离（比如同一线程内的多个测试也用独立库），可以用UUID生成唯一名称，配合参数化测试库rstest更方便：

use rstest::rstest;
use uuid::Uuid;

#[rstest]
#[case(Uuid::new_v4())]
fn test_user_creation(#[case] test_id: Uuid) {
    let db_name = format!("test_db_{}", test_id);
    // 初始化数据库
    init_database(&db_name);
    // 执行测试逻辑
    assert!(create_user(&db_name, "test_user").is_ok());
    // 测试后清理
    drop_database(&db_name);
}

UUID的唯一性保证了不管测试怎么并行跑，数据库都不会冲突。

3. 自定义环境变量传递Worker ID #

和Node的JEST_WORKER_ID思路一致，你可以手动在测试启动时设置环境变量，然后在代码里读取：
首先在Cargo.toml里加自定义测试脚本（控制线程数和环境变量）：

[package]
# ...其他配置

[scripts]
test_parallel = "for worker_id in {0..3}; do TEST_WORKER_ID=$worker_id cargo test -- --test-threads=4 & done; wait"

然后在测试代码里读取这个变量：

use std::env;

fn get_test_db_name() -> String {
    let worker_id = env::var("TEST_WORKER_ID").unwrap_or("0".to_string());
    format!("test_db_{}", worker_id)
}

这个方案适合需要和现有Node测试流程对齐的场景，完全手动控制Worker ID的分配。

4. 异步测试的临时数据库方案 #

如果是用tokio这类异步框架做数据库测试，推荐用sqlx这类库的临时数据库功能，自动创建和销毁隔离的测试库：

use sqlx::postgres::PgPoolOptions;
use uuid::Uuid;

async fn get_test_pool() -> sqlx::PgPool {
    let test_id = Uuid::new_v4();
    let db_name = format!("test_db_{}", test_id);

    // 先连接到管理员数据库创建新库
    let admin_pool = PgPoolOptions::new()
        .connect("postgres://user:password@localhost/postgres")
        .await
        .expect("Failed to connect to admin DB");
    
    sqlx::query(&format!("CREATE DATABASE \"{}\"", db_name))
        .execute(&admin_pool)
        .await
        .expect("Failed to create test DB");

    // 连接到新创建的测试库
    PgPoolOptions::new()
        .connect(&format!("postgres://user:password@localhost/{}", db_name))
        .await
        .expect("Failed to connect to test DB")
}

#[tokio::test]
async fn test_async_query() {
    let pool = get_test_pool().await;
    // 执行异步测试逻辑
    let count = sqlx::query_scalar!("SELECT COUNT(*) FROM users")
        .fetch_one(&pool)
        .await
        .unwrap();
    assert_eq!(count, 0);

    // 测试结束后删除数据库
    let admin_pool = PgPoolOptions::new()
        .connect("postgres://user:password@localhost/postgres")
        .await
        .unwrap();
    sqlx::query(&format!("DROP DATABASE \"{}\"", pool.database_name().unwrap()))
        .execute(&admin_pool)
        .await
        .unwrap();
}

这种方式完全自动化，不用手动管理数据库名称，异步场景下也能保证隔离。

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