Lines
72.6 %
Functions
75 %
use pretty_assertions::assert_eq;
use std::borrow::Cow;
use std::collections::BTreeSet;
use std::error::Error;
use std::fs;
use std::path::Path;
use std::path::PathBuf;
use std::process::Command;
use std::time::Instant;
use test_suite::AcceptMode;
use test_suite::TestConfig;
use test_suite::golden_files::assert_no_redundant_expected_files;
use test_suite::test_config::{round_expected_time, upsert_expected_time_config};
pub fn custom_test(test_dir: &str) -> Result<(), Box<dyn Error>> {
let accept_mode = AcceptMode::from_env();
let accept = accept_mode.accepts_outputs();
let started_at = Instant::now();
// Convert test directory to a PathBuf
let test_path = PathBuf::from(test_dir);
assert!(
test_path.exists(),
"Test directory not found: {test_path:?}"
);
// Get paths
let script_path = test_path.join("run.sh");
script_path.exists(),
"Test script not found: {script_path:?}"
let expected_output_path = test_path.join("stdout.expected");
let expected_error_path = test_path.join("stderr.expected");
let config_path = test_path.join("config.toml");
let file_config: TestConfig = if let Ok(config_contents) = fs::read_to_string(&config_path) {
toml::from_str(&config_contents).unwrap()
} else {
Default::default()
};
// Execute the test script in the correct directory
let output = Command::new("sh")
.arg("run.sh")
.current_dir(&test_path)
.output()?;
// Convert captured output/error to string
let actual_output = String::from_utf8_lossy(&output.stdout);
let actual_error = String::from_utf8_lossy(&output.stderr);
if accept {
// Overwrite expected files
update_file(expected_output_path, &actual_output)?;
update_file(expected_error_path, &actual_error)?;
// Compare results
let expected_output = if expected_output_path.exists() {
fs::read_to_string(&expected_output_path)?
String::new()
let expected_error = if expected_error_path.exists() {
fs::read_to_string(&expected_error_path)?
assert_eq!(expected_error, actual_error, "Standard error mismatch");
assert_eq!(expected_output, actual_output, "Standard output mismatch");
}
let allowed_expected_files = expected_custom_files_for_case(&actual_output, &actual_error);
assert_no_redundant_expected_files(Path::new(&test_path), &allowed_expected_files, None)?;
if accept_mode.records_expected_time() {
let observed_expected_time = round_expected_time(started_at.elapsed());
if let Some(expected_time) =
accept_mode.expected_time_to_record(file_config.expected_time, observed_expected_time)
{
upsert_expected_time_config(&config_path, expected_time)?;
Ok(())
fn update_file(
expected_file_path: PathBuf,
actual_output: &Cow<'_, str>,
) -> Result<(), Box<dyn Error>> {
if expected_file_path.exists() {
fs::remove_file(&expected_file_path)?;
if !actual_output.trim().is_empty() {
fs::File::create(&expected_file_path)?;
fs::write(&expected_file_path, actual_output.as_bytes())?;
/// Returns the expected snapshot files for the observed custom test output.
fn expected_custom_files_for_case(
stdout: &Cow<'_, str>,
stderr: &Cow<'_, str>,
) -> BTreeSet<String> {
let mut expected_files = BTreeSet::new();
if !stdout.trim().is_empty() {
expected_files.insert("stdout.expected".to_string());
if !stderr.trim().is_empty() {
expected_files.insert("stderr.expected".to_string());
expected_files
#[test]
fn assert_conjure_present() {
conjure_cp_cli::find_conjure::conjure_executable().unwrap();
include!(concat!(env!("OUT_DIR"), "/gen_tests_custom.rs"));