zluda_inject/tests/inject.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

use std::{env, io, path::PathBuf, process::Command};

#[test]
fn direct_cuinit() -> io::Result<()> {
    let zluda_with_exe = PathBuf::from(env!("CARGO_BIN_EXE_zluda_with"));
    let mut zluda_redirect_dll = zluda_with_exe.parent().unwrap().to_path_buf();
    zluda_redirect_dll.push("zluda_redirect.dll");
    let helpers_dir = env!("HELPERS_OUT_DIR");
    let exe_under_test = format!(
        "{}{}direct_cuinit.exe",
        helpers_dir,
        std::path::MAIN_SEPARATOR
    );
    let mut test_cmd = Command::new(&zluda_with_exe);
    test_cmd
        .arg(&zluda_redirect_dll)
        .arg("--")
        .arg(&exe_under_test);
    let test_output = test_cmd.output()?;
    let stderr_text = String::from_utf8(test_output.stderr).unwrap();
    assert!(stderr_text.contains("ZLUDA_DUMP"));
    Ok(())
}