#![cfg(not(any(
feature = "parser_tests",
feature = "analyzer_tests",
feature = "codegen_tests",
feature = "interpreter_tests",
feature = "conformance_tests",
feature = "integration_tests",
)))]
use std::fs;
use std::process::Command;
#[test]
fn test_option_field_unwrap_on_borrowed_param() {
let source = r#"
struct Node {
pub value: int,
pub children: Option<Vec<int>>
}
fn get_sum(node: Node) -> int {
if node.children.is_some() {
let children = node.children.unwrap()
children[0] + children[1]
} else {
0
}
}
fn main() {
let n = Node { value: 1, children: Some(vec![2, 3]) }
let sum = get_sum(n)
assert_eq(sum, 5)
}
"#;
let _tmp = tempfile::tempdir().unwrap();
let temp_dir = _tmp.path();
let test_id = format!(
"wj_test_{}",
std::time::SystemTime::now()
.duration_since(std::time::UNIX_EPOCH)
.unwrap()
.as_nanos()
);
let test_dir = temp_dir.join(&test_id);
fs::create_dir_all(&test_dir).unwrap();
let wj_file = test_dir.join("test.wj");
fs::write(&wj_file, source).unwrap();
let out_dir = test_dir.join("out");
let wj_binary = env!("CARGO_BIN_EXE_wj");
let _output = Command::new(wj_binary)
.arg("build")
.arg(&wj_file)
.arg("--target")
.arg("rust")
.arg("--output")
.arg(&out_dir)
.arg("--no-cargo")
.output()
.expect("Failed to run wj compiler");
let rust_file = out_dir.join("test.rs");
let generated = fs::read_to_string(&rust_file).expect("Failed to read generated Rust file");
println!("Generated code:\n{}", generated);
let rustc_output = Command::new("rustc")
.arg(&rust_file)
.arg("--crate-type")
.arg("bin")
.arg("--edition")
.arg("2021")
.arg("-o")
.arg(test_dir.join("test_bin"))
.output()
.expect("Failed to run rustc");
if !rustc_output.status.success() {
let stderr = String::from_utf8_lossy(&rustc_output.stderr);
panic!(
"Compilation failed:\n{}\n\nGenerated code:\n{}",
stderr, generated
);
}
assert!(
generated.contains(".as_ref().unwrap()") || generated.contains(".clone().unwrap()"),
"Option::unwrap on borrowed field should use .as_ref() or .clone()"
);
}