如何为HTTP服务器事件循环编写不修改公共接口的集成测试

完全不需要为测试修改run函数的公共接口，核心思路是将阻塞的服务逻辑放到独立后台线程运行，通过端口探活验证启动状态即可，所有适配逻辑都写在测试代码侧。

具体实现步骤 #

• 用标准库的线程能力把server::run()放到后台线程执行，避免阻塞测试主线程，这一步不需要对生产代码做任何改动。
• 增加带超时的端口轮询检测：服务启动绑定端口需要几毫秒时间，不要拉起线程就直接判定启动成功，每隔100毫秒尝试连接目标端口，能正常连通就说明服务已经完成绑定、正常启动。
• 测试完成后回收后台线程资源，避免端口残留占用影响其他测试用例。

测试代码示例 #

// tests/server.rs
use std::net::TcpStream;
use std::thread;
use std::time::{Duration, Instant};

// 注意如果server模块在crate内部，测试里要正确引入，比如use your_crate_name::server;
#[test]
fn verify_server_can_boot_normally() {
    // 拉起后台线程跑阻塞的服务逻辑
    let server_thread = thread::spawn(|| {
        server::run();
    });

    // 最多等待2秒检测服务是否启动
    let check_timeout = Duration::from_secs(2);
    let start_time = Instant::now();
    let mut is_server_up = false;

    while start_time.elapsed() < check_timeout {
        if TcpStream::connect("127.0.0.1:7878").is_ok() {
            is_server_up = true;
            break;
        }
        thread::sleep(Duration::from_millis(100));
    }

    // 断言服务启动成功
    assert!(is_server_up, "server failed to bind port and complete startup within 2 seconds");

    // 回收线程资源，测试结束自动释放端口
    server_thread.join().ok();
}

补充说明 #

这个方案没有侵入生产代码的公共接口，不需要给run函数加关闭信号、返回Result类型之类为测试做的妥协，完全满足需求。
如果遇到测试环境7878端口被随机占用的问题，可以考虑把端口改成动态分配的空闲端口，但这属于生产代码的逻辑优化，不属于为测试做的接口改造范畴。如果严格要求完全不动生产代码，固定端口加超时探活的方案已经足够稳定。

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