ctest: Add generation, compilation, and running of tests for constants.

Author: mbyx

ctest-next/build.rs

@@ -1,20 +1,31 @@
 use std::env;
 
 fn main() {
-    println!(
-        "cargo:rustc-env={}={}",
-        "HOST_PLATFORM",
-        env::var("HOST").unwrap()
-    );
-    println!(
-        "cargo:rustc-env={}={}",
-        "TARGET_PLATFORM",
-        env::var("TARGET").unwrap()
-    );
-    println!(
-        "cargo:rustc-env={}={}",
-        "LINKER",
-        env::var("CC").unwrap_or("".to_string())
-    );
-    println!("cargo:rerun-if-changed-env=TARGET")
+    let host = env::var("HOST").unwrap();
+    let target = env::var("TARGET").unwrap();
+    let target_key = target.replace('-', "_");
+
+    println!("cargo:rustc-env=HOST_PLATFORM={host}");
+    println!("cargo:rustc-env=TARGET_PLATFORM={target}");
+
+    let linker = env::var(format!("CARGO_TARGET_{}_LINKER", target_key.to_uppercase()))
+        .or_else(|_| env::var("CC"))
+        .or_else(|_| env::var(format!("CC_{}", target_key)))
+        .unwrap_or_default();
+
+    let runner =
+        env::var(format!("CARGO_TARGET_{}_RUNNER", target_key.to_uppercase())).unwrap_or_default();
+
+    // As we invoke rustc directly this does not get passed to it, although RUSTFLAGS does.
+    let flags = env::var(format!(
+        "CARGO_TARGET_{}_RUSTFLAGS",
+        target_key.to_uppercase()
+    ))
+    .unwrap_or_default();
+
+    println!("cargo:rustc-env=LINKER={linker}");
+    println!("cargo:rustc-env=RUNNER={runner}");
+    println!("cargo:rustc-env=FLAGS={flags}");
+
+    println!("cargo:rerun-if-changed-env=TARGET");
 }

ctest-next/lib

@@ -48,6 +48,12 @@ pub fn compile_test<P: AsRef<Path>>(
     if !linker.is_empty() {
         cmd.arg(format!("-Clinker={}", linker));
     }
+
+    let flags = env::var("FLAGS").unwrap();
+    if !flags.is_empty() {
+        cmd.args(flags.split_whitespace().collect::<Vec<_>>());
+    }
+
     let output = cmd.output()?;
 
     if !output.status.success() {
@@ -58,8 +64,17 @@ pub fn compile_test<P: AsRef<Path>>(
 }
 
 /// Run the compiled test binary.
-pub fn run_test<P: AsRef<Path>>(test_binary: P) -> Result<String> {
-    let output = Command::new(test_binary.as_ref()).output()?;
+pub fn run_test(test_binary: &str) -> Result<String> {
+    let cmd = env::var("RUNNER").unwrap();
+    let output = if cmd.is_empty() {
+        Command::new(test_binary).output()?
+    } else {
+        let mut cmd = cmd.split_whitespace();
+        Command::new(cmd.next().unwrap())
+            .args(cmd.collect::<Vec<_>>())
+            .arg(test_binary)
+            .output()?
+    };
 
     if !output.status.success() {
         return Err(std::str::from_utf8(&output.stderr)?.into());

ctest-next/src/runner.rs

@@ -34,7 +34,7 @@ fn test_entrypoint_hierarchy() {
     )
     .unwrap();
 
-    assert!(run_test(test_binary).is_ok());
+    assert!(run_test(test_binary.to_str().unwrap()).is_ok());
 }
 
 #[test]
@@ -50,7 +50,7 @@ fn test_entrypoint_simple() {
     let test_binary =
         compile_test(env::temp_dir().to_str().unwrap(), crate_path, "simple_out").unwrap();
 
-    assert!(run_test(test_binary).is_ok());
+    assert!(run_test(test_binary.to_str().unwrap()).is_ok());
 }
 
 #[test]
@@ -66,7 +66,7 @@ fn test_entrypoint_macro() {
     let test_binary =
         compile_test(env::temp_dir().to_str().unwrap(), crate_path, "macro_out").unwrap();
 
-    assert!(run_test(test_binary).is_ok());
+    assert!(run_test(test_binary.to_str().unwrap()).is_ok());
 }
 
 #[test]