Rust集成测试中启动的应用进程无法自动终止问题排查

问题分析与解决方案 #

核心错误点 #

• kill_on_drop未生效：调用spawn()后立刻链式执行wait().await，会直接消耗Child进程句柄并等待其退出，kill_on_drop(true)的逻辑完全没机会触发——因为Child在wait执行后就被销毁了，没有保留到测试结束时触发drop操作。
• 后台线程脱离管控：用spawn启动的线程没有被跟踪，测试进程退出后，这个线程可能仍在运行，导致cargo run启动的服务器进程变成孤儿进程，被系统init接管，无法自动终止。
• 不安全的静态变量：unsafe的READY存在并发风险，即使单线程测试也不是合规写法。

修复方案 #

1. 安全持有进程句柄，主动终止服务器

改用线程安全的静态容器存储Child实例，在所有测试结束后主动kill进程：

修改tests/common.rs：

use std::process::{Command, Child};
use std::sync::{Arc, Mutex, OnceCell};
use tokio::time::{sleep, Duration};

// 线程安全的全局进程句柄存储
static SERVER_CHILD: OnceCell<Arc<Mutex<Option<Child>>>> = OnceCell::new();

pub async fn wait_for_server() {
    let child_cell = SERVER_CHILD.get_or_init(|| Arc::new(Mutex::new(None)));
    let mut guard = child_cell.lock().unwrap();
    
    if guard.is_none() {
        let child = Command::new("cargo")
            .env("RUST_LOG", "ERROR")
            .args(&["run", "--", "--ip", "127.0.0.1:3000"])
            .spawn()
            .expect("Could not start server");
        
        *guard = Some(child);
        // 等待服务器启动（可优化为端口轮询）
        sleep(Duration::from_millis(500)).await;
    }
}

// 主动终止服务器的函数
pub fn shutdown_server() {
    if let Some(child_cell) = SERVER_CHILD.get() {
        let mut guard = child_cell.lock().unwrap();
        if let Some(mut child) = guard.take() {
            // 忽略kill和wait的错误，避免测试失败
            let _ = child.kill();
            let _ = child.wait();
        }
    }
}

2. 利用测试钩子自动执行清理

添加ctor依赖到Cargo.toml的dev-dependencies，用#[dtor]钩子在所有测试结束后自动调用清理：

[dev-dependencies]
ctor = "0.2"

修改tests/test_one.rs：

use crate::common::{wait_for_server, shutdown_server};
use ctor::dtor;

// 所有测试完成后自动执行清理
#[dtor]
fn cleanup_server() {
    shutdown_server();
}

#[tokio::test]
async fn my_test_one() {
    wait_for_server().await;
    // 测试逻辑
}

3. 优化：直接启动二进制而非cargo run

cargo run会额外启动一个cargo进程，不如直接编译出二进制再启动，更可控：

// 在wait_for_server中替换Command部分
// 需要在Cargo.toml中指定二进制名称，比如name = "my-server"
let binary_path = env!("CARGO_BIN_EXE_my-server");
let child = Command::new(binary_path)
    .env("RUST_LOG", "ERROR")
    .args(&["--ip", "127.0.0.1:3000"])
    .spawn()
    .expect("Could not start server");

额外优化建议 #

• 替换固定时间等待：改用TCP连接轮询服务器端口，确保服务器真正启动后再执行测试，避免因环境差异导致的等待时间不足。
• 移除unsafe代码：用OnceCell或once_cell库的Lazy来管理全局状态，完全避免unsafe操作。

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