argus-cli 0.1.20

#![feature(rustc_private)]
use std::{
  env, fs,
  path::Path,
  process::{Command, Stdio},
  sync::Once,
};

use anyhow::{Context, Result, ensure};

static SETUP: Once = Once::new();

static DNF_PERF_P: &str = "dnf-perf.csv";

fn run<P: AsRef<Path>>(dir: P, f: impl FnOnce(&mut Command)) -> Result<String> {
  let root = env::temp_dir().join("argus");
  let heredir = Path::new(".").canonicalize()?;

  SETUP.call_once(|| {
    let mut cmd = Command::new("cargo");
    cmd.args(["install", "--path", ".", "--debug", "--locked", "--root"]);
    cmd.arg(&root);
    cmd.current_dir(&heredir);
    let status = cmd.status().unwrap();
    if !status.success() {
      panic!("installing argus failed")
    }

    fs::write(DNF_PERF_P, "Label,N,Time\n").unwrap();
  });

  let mut cmd = Command::new("cargo");
  cmd.arg("argus");
  cmd.arg("bundle");

  let path = format!(
    "{}:{}",
    root.join("bin").display(),
    env::var("PATH").unwrap_or_else(|_| "".into())
  );
  cmd.env("PATH", path);

  let ws = heredir.join("tests").join(dir);
  cmd.current_dir(&ws);

  // NOTE: performance data is written to STDERR, so we capture it and place it in a file.
  let perf_file = fs::OpenOptions::new()
    .create(true)
    .append(true)
    .open(DNF_PERF_P)
    .unwrap();
  cmd.stderr(Stdio::from(perf_file));

  f(&mut cmd);

  let _ = fs::remove_dir_all(ws.join("target"));

  let output = cmd.output().context("Process failed")?;
  ensure!(
    output.status.success(),
    "Process exited with non-zero exit code. Stderr:\n{}",
    String::from_utf8(output.stderr)?
  );

  Ok(String::from_utf8(output.stdout)?)
}

macro_rules! mk_tests_for {
    ($($i:ident),*) => {$(
        #[test]
        fn $i() -> Result<()> {
            _ = run(format!("workspaces/{}", stringify!($i)), |_cmd| {})?;
            Ok(())
        }
    )*}
}

mk_tests_for! {
  axum,
  bevy,
  // chumsky,
  diesel,
  // easy_ml,
  entrait,
  nalgebra,
  uom
}

// TODO: include individual test if we want to see a particular output
//
// We should also specify some type of "blessed output,"
// to make sure that certain errors are present at the right locations.