含#[ctor]属性的函数中使用println!致Rust集成测试崩溃

问题原因 #

ctor crate的#[ctor]标注的初始化函数会在进程启动后、Rust测试框架完成线程初始化之前执行。而println!会触发Rust标准库的线程本地存储（TLS）初始化逻辑，这和测试框架后续的thread::set_current操作冲突，导致出现thread::set_current should only be called once per thread的致命运行时错误，最终进程异常终止。

解决方案 #

方案1：改用测试安全的延迟初始化（推荐） #

放弃ctor的全局初始化，改用once_cell（或lazy_static）实现只执行一次的初始化，确保初始化逻辑在测试线程完全初始化后执行：

首先在Cargo.toml添加依赖：

[dependencies]
once_cell = "1.18.0"

修改测试文件tests/int_test.rs：

use once_cell::sync::OnceCell;
use int_test_demo::add;

// 全局初始化标记，确保init只执行一次
static INITIALIZED: OnceCell<()> = OnceCell::new();

fn init() {
    println!("init");
}

#[test]
fn it_works() {
    // 首次调用时执行init，后续测试不会重复执行
    INITIALIZED.get_or_init(init);
    let result = add(2, 2);
    assert_eq!(result, 4);
}

方案2：限制ctor初始化逻辑的执行场景 #

如果必须保留ctor，可以通过编译条件判断，在测试环境跳过依赖TLS的操作（比如println!）：

修改测试文件tests/int_test.rs：

use ctor::{ctor, dtor};
use int_test_demo::add;

#[test]
fn it_works() {
    let result = add(2, 2);
    assert_eq!(result, 4);
}

#[ctor]
fn init() {
    // 仅在非测试环境执行println!
    if !cfg!(test) {
        println!("init");
    }
}

#[dtor]
fn term() {
    if !cfg!(test) {
        println!("term");
    }
}

方案3：使用测试专用的初始化工具 #

可以改用test-ctor crate，它专门针对测试场景设计，避免了和标准测试框架的冲突：

添加依赖：

[dev-dependencies]
test-ctor = "0.1.0"

修改测试文件：

use test_ctor::test_ctor;
use int_test_demo::add;

#[test]
fn it_works() {
    let result = add(2, 2);
    assert_eq!(result, 4);
}

#[test_ctor]
fn init() {
    println!("init");
}

验证 #

执行cargo test --nocapture，可以看到测试正常运行，同时初始化信息也能正常打印，不会再出现进程崩溃的问题。

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