forked from M-Labs/nac3
standalone: Add cargo test cases for demos
This commit is contained in:
parent
a19f1065e3
commit
ea121673d3
|
@ -34,6 +34,9 @@ mod basic_symbol_resolver;
|
|||
use basic_symbol_resolver::*;
|
||||
use nac3core::toplevel::composer::ComposerConfig;
|
||||
|
||||
#[cfg(test)]
|
||||
mod test;
|
||||
|
||||
/// Command-line argument parser definition.
|
||||
#[derive(Parser)]
|
||||
#[command(author, version, about, long_about = None)]
|
||||
|
|
|
@ -0,0 +1,148 @@
|
|||
use std::fs;
|
||||
use std::io;
|
||||
use std::path::{Path, PathBuf};
|
||||
use std::process::{Command, Output};
|
||||
use parking_lot::Mutex;
|
||||
|
||||
/// [Mutex] for enforcing only a single `./run_demo.sh` is executing.
|
||||
///
|
||||
/// This is necessary as the command will generate temporary files (specifically `*.o`, `*.bc`, and
|
||||
/// `demo`) in the filesystem.
|
||||
static RUN_EXEC_MTX: Mutex<()> = Mutex::new(());
|
||||
|
||||
/// Dummy struct for automatically removing temporary files after test case execution completes.
|
||||
struct CleanupDemoTempFiles;
|
||||
|
||||
impl Default for CleanupDemoTempFiles {
|
||||
|
||||
fn default() -> Self {
|
||||
CleanupDemoTempFiles {}
|
||||
}
|
||||
}
|
||||
|
||||
impl Drop for CleanupDemoTempFiles {
|
||||
|
||||
fn drop(&mut self) {
|
||||
get_demo_dir().read_dir()
|
||||
.unwrap()
|
||||
.into_iter()
|
||||
.map(|dir_entry| dir_entry.unwrap().path().into_boxed_path())
|
||||
.filter(|p| {
|
||||
let ext = p.extension()
|
||||
.and_then(|ext| ext.to_str())
|
||||
.unwrap_or_default();
|
||||
let file_name = p.file_name()
|
||||
.and_then(|ext| ext.to_str())
|
||||
.unwrap_or_default();
|
||||
|
||||
["bc", "o"].contains(&ext) || file_name == "demo"
|
||||
})
|
||||
.for_each(|p| fs::remove_file(p).unwrap())
|
||||
}
|
||||
}
|
||||
|
||||
/// Returns the absolute [Path] to the `demo` directory.
|
||||
fn get_demo_dir() -> PathBuf {
|
||||
std::env::current_dir()
|
||||
.map(|p| p.join("demo"))
|
||||
.and_then(|p| p.canonicalize())
|
||||
.unwrap()
|
||||
}
|
||||
|
||||
/// Collects all Python tests from `demo/src`.
|
||||
fn collect_demo_tests() -> Vec<PathBuf> {
|
||||
get_demo_dir()
|
||||
.join("src")
|
||||
.read_dir()
|
||||
.unwrap()
|
||||
.into_iter()
|
||||
.filter_map(|dir_entry| {
|
||||
let p = dir_entry.unwrap().path();
|
||||
|
||||
if p.is_file() && p.extension().and_then(|ext| ext.to_str()).unwrap_or_default() == "py" {
|
||||
Some(p)
|
||||
} else {
|
||||
None
|
||||
}
|
||||
})
|
||||
.collect()
|
||||
}
|
||||
|
||||
/// Runs the interpreter on the `file` as if by invoking `interpret_demo.py`.
|
||||
fn run_interpreter(file: &Path) -> io::Result<Output> {
|
||||
Command::new("./interpret_demo.py")
|
||||
.arg(file.to_str().unwrap())
|
||||
.current_dir(get_demo_dir())
|
||||
.output()
|
||||
}
|
||||
|
||||
/// Runs NAC3 on the `file` as if by invoking `run_demo.sh`.
|
||||
fn run_demo(file: &Path, nac3_args: &[&str]) -> io::Result<Output> {
|
||||
let _lk = RUN_EXEC_MTX.lock();
|
||||
|
||||
Command::new("./run_demo.sh")
|
||||
.args(nac3_args)
|
||||
.arg(file.to_str().unwrap())
|
||||
.current_dir(get_demo_dir())
|
||||
.output()
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_demos_optimized() {
|
||||
let _cleanup = CleanupDemoTempFiles::default();
|
||||
|
||||
let tests = collect_demo_tests();
|
||||
|
||||
for test in tests.iter() {
|
||||
let py_interpreted= run_interpreter(test).unwrap();
|
||||
let py_stdout = String::from_utf8(py_interpreted.stdout).unwrap();
|
||||
|
||||
let executed = run_demo(test, &[]).unwrap();
|
||||
let exec_stdout = String::from_utf8(executed.stdout).unwrap();
|
||||
assert_eq!(py_stdout, exec_stdout, "Execution result mismatch for {}", test.file_name().and_then(|p| p.to_str()).unwrap());
|
||||
|
||||
let bc_interpreted = run_demo(test, &["--lli"]).unwrap();
|
||||
let bc_stdout = String::from_utf8(bc_interpreted.stdout).unwrap();
|
||||
assert_eq!(py_stdout, bc_stdout, "Execution result mismatch for {}", test.file_name().and_then(|p| p.to_str()).unwrap());
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_demos_unoptimized() {
|
||||
let _cleanup = CleanupDemoTempFiles::default();
|
||||
|
||||
let tests = collect_demo_tests();
|
||||
|
||||
for test in tests.iter() {
|
||||
let py_interpreted= run_interpreter(test).unwrap();
|
||||
let py_stdout = String::from_utf8(py_interpreted.stdout).unwrap();
|
||||
|
||||
let executed = run_demo(test, &["-O0"]).unwrap();
|
||||
let exec_stdout = String::from_utf8(executed.stdout).unwrap();
|
||||
assert_eq!(py_stdout, exec_stdout, "Execution result differs for {}", test.file_name().and_then(|p| p.to_str()).unwrap());
|
||||
|
||||
let bc_interpreted= run_demo(test, &["--lli", "-O0"]).unwrap();
|
||||
let bc_stdout = String::from_utf8(bc_interpreted.stdout).unwrap();
|
||||
assert_eq!(py_stdout, bc_stdout, "Execution result mismatch for {}", test.file_name().and_then(|p| p.to_str()).unwrap());
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_demos_multithreaded() {
|
||||
let _cleanup = CleanupDemoTempFiles::default();
|
||||
|
||||
let tests = collect_demo_tests();
|
||||
|
||||
for test in tests.iter() {
|
||||
let py_interpreted= run_interpreter(test).unwrap();
|
||||
let py_stdout = String::from_utf8(py_interpreted.stdout).unwrap();
|
||||
|
||||
let executed = run_demo(test, &["-T4"]).unwrap();
|
||||
let exec_stdout = String::from_utf8(executed.stdout).unwrap();
|
||||
assert_eq!(py_stdout, exec_stdout, "Execution result differs for {}", test.file_name().and_then(|p| p.to_str()).unwrap());
|
||||
|
||||
let bc_interpreted= run_demo(test, &["--lli", "-T4"]).unwrap();
|
||||
let bc_stdout = String::from_utf8(bc_interpreted.stdout).unwrap();
|
||||
assert_eq!(py_stdout, bc_stdout, "Execution result mismatch for {}", test.file_name().and_then(|p| p.to_str()).unwrap());
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue