Rust异步函数中write!宏在Tokio测试用例输出异常求助

问题原因 #

1. 测试未检查函数返回错误：测试代码调用show_status时未处理其返回的Result。若walk_components在测试环境中返回错误（例如main_repo_dir路径无效、目录结构不符合预期），函数会通过await?提前返回错误，后续的writeln!("ids: ...")和flush不会执行，导致output仅保留第一行内容。
2. 原函数未处理IO操作错误：writeln!和flush均会返回std::io::Result，但原函数未用?传播这些错误，若IO操作出现异常（虽Vec<u8>基本不会出错，但规范上需处理），会静默失败导致输出不完整。

解决方案 #

1. 测试中强制检查函数执行结果

修改测试代码，确保show_status执行成功，否则直接终止测试并暴露错误：

#[tokio::test]
async fn test_show_status() {
    let mut output = Vec::new();
    // 若函数返回错误，直接 panic 并输出提示信息
    show_status(&main_repo_dir, &mut output).await.expect("show_status failed");

    assert_eq!(std::str::from_utf8(output.as_slice()).unwrap(), "status:\nids: [0, 1]\n");
}

这样如果show_status因walk_components出错返回错误，测试会立即终止并展示具体错误信息，便于排查测试环境中的路径或逻辑问题。

2. 完善原函数的错误传播逻辑

在原函数中，为所有IO操作的返回值添加?，确保任何错误都能被正确向上传播：

async fn show_status(root_dir: &PathBuf, mut output: impl std::io::Write) -> Result<(), Error> {
    writeln!(output, "status:")?;

    let mut commponent_ids = walk_components(&StatusVisitor::new(), &root_dir, None).await?;
    writeln!(output, "ids: {:?}", commponent_ids)?;
    output.flush()?;

    Ok(())
}

3. 验证测试环境路径有效性

确认测试用main_repo_dir指向的目录结构符合walk_components的预期，确保其能正常返回[0, 1]。可在测试中临时打印main_repo_dir的绝对路径，或添加调试日志确认目录内容。

补充说明 #

你提到在main函数中调用正常，是因为main环境下的main_repo_dir路径有效，walk_components执行成功，所有IO操作都完成；而测试环境中可能路径配置有误，导致walk_components出错，函数提前返回，仅输出第一行内容。

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