176 lines
5.9 KiB
Rust
176 lines
5.9 KiB
Rust
#[cfg(all(test, feature = "cranelift-jit", not(feature = "jit-direct-only")))]
|
||
mod tests {
|
||
use std::collections::HashMap;
|
||
|
||
use crate::box_factory::RuntimeError;
|
||
use crate::box_trait::NyashBox;
|
||
use crate::mir::{
|
||
BasicBlockId, ConstValue, EffectMask, FunctionSignature, MirFunction, MirInstruction,
|
||
MirModule, MirType,
|
||
};
|
||
|
||
// Minimal Person factory: creates InstanceBox with fields [name, age]
|
||
#[allow(dead_code)]
|
||
struct PersonFactory;
|
||
impl crate::box_factory::BoxFactory for PersonFactory {
|
||
fn create_box(
|
||
&self,
|
||
name: &str,
|
||
_args: &[Box<dyn NyashBox>],
|
||
) -> Result<Box<dyn NyashBox>, RuntimeError> {
|
||
if name != "Person" {
|
||
return Err(RuntimeError::InvalidOperation {
|
||
message: format!("Unknown Box type: {}", name),
|
||
});
|
||
}
|
||
let fields = vec!["name".to_string(), "age".to_string()];
|
||
let methods: HashMap<String, crate::ast::ASTNode> = HashMap::new();
|
||
let inst = crate::instance_v2::InstanceBox::from_declaration(
|
||
"Person".to_string(),
|
||
fields,
|
||
methods,
|
||
);
|
||
Ok(Box::new(inst))
|
||
}
|
||
fn box_types(&self) -> Vec<&str> {
|
||
vec!["Person"]
|
||
}
|
||
fn is_builtin_factory(&self) -> bool {
|
||
true
|
||
}
|
||
}
|
||
|
||
#[allow(dead_code)]
|
||
fn build_person_module() -> MirModule {
|
||
let mut module = MirModule::new("identical_person".to_string());
|
||
let sig = FunctionSignature {
|
||
name: "main".into(),
|
||
params: vec![],
|
||
return_type: MirType::String,
|
||
effects: EffectMask::PURE,
|
||
};
|
||
let mut f = MirFunction::new(sig, BasicBlockId::new(0));
|
||
let bb = f.entry_block;
|
||
|
||
let person = f.next_value_id();
|
||
f.get_block_mut(bb)
|
||
.unwrap()
|
||
.add_instruction(MirInstruction::NewBox {
|
||
dst: person,
|
||
box_type: "Person".into(),
|
||
args: vec![],
|
||
});
|
||
|
||
// person.setField("name", "Alice")
|
||
let k_name = f.next_value_id();
|
||
f.get_block_mut(bb)
|
||
.unwrap()
|
||
.add_instruction(MirInstruction::Const {
|
||
dst: k_name,
|
||
value: ConstValue::String("name".into()),
|
||
});
|
||
let v_alice = f.next_value_id();
|
||
f.get_block_mut(bb)
|
||
.unwrap()
|
||
.add_instruction(MirInstruction::Const {
|
||
dst: v_alice,
|
||
value: ConstValue::String("Alice".into()),
|
||
});
|
||
f.get_block_mut(bb)
|
||
.unwrap()
|
||
.add_instruction(MirInstruction::BoxCall {
|
||
dst: None,
|
||
box_val: person,
|
||
method: "setField".into(),
|
||
args: vec![k_name, v_alice],
|
||
method_id: None,
|
||
effects: EffectMask::PURE,
|
||
});
|
||
|
||
// person.setField("age", 25)
|
||
let k_age = f.next_value_id();
|
||
f.get_block_mut(bb)
|
||
.unwrap()
|
||
.add_instruction(MirInstruction::Const {
|
||
dst: k_age,
|
||
value: ConstValue::String("age".into()),
|
||
});
|
||
let v_25 = f.next_value_id();
|
||
f.get_block_mut(bb)
|
||
.unwrap()
|
||
.add_instruction(MirInstruction::Const {
|
||
dst: v_25,
|
||
value: ConstValue::Integer(25),
|
||
});
|
||
f.get_block_mut(bb)
|
||
.unwrap()
|
||
.add_instruction(MirInstruction::BoxCall {
|
||
dst: None,
|
||
box_val: person,
|
||
method: "setField".into(),
|
||
args: vec![k_age, v_25],
|
||
method_id: None,
|
||
effects: EffectMask::PURE,
|
||
});
|
||
|
||
// name = person.getField("name"); return name
|
||
let k_name2 = f.next_value_id();
|
||
f.get_block_mut(bb)
|
||
.unwrap()
|
||
.add_instruction(MirInstruction::Const {
|
||
dst: k_name2,
|
||
value: ConstValue::String("name".into()),
|
||
});
|
||
let out_name = f.next_value_id();
|
||
f.get_block_mut(bb)
|
||
.unwrap()
|
||
.add_instruction(MirInstruction::BoxCall {
|
||
dst: Some(out_name),
|
||
box_val: person,
|
||
method: "getField".into(),
|
||
args: vec![k_name2],
|
||
method_id: None,
|
||
effects: EffectMask::PURE,
|
||
});
|
||
f.get_block_mut(bb)
|
||
.unwrap()
|
||
.add_instruction(MirInstruction::Return {
|
||
value: Some(out_name),
|
||
});
|
||
|
||
module.add_function(f);
|
||
module
|
||
}
|
||
|
||
#[cfg(feature = "cranelift-jit")]
|
||
#[test]
|
||
#[ignore = "ABI_STRICT vtable path diverges; JIT host-bridge parity pending"]
|
||
fn identical_vm_and_jit_person_get_set_slots() {
|
||
// Build runtime with Person factory
|
||
let mut rt_builder = crate::runtime::NyashRuntimeBuilder::new();
|
||
rt_builder = rt_builder.with_factory(Arc::new(PersonFactory));
|
||
let runtime = rt_builder.build();
|
||
|
||
// Also register factory globally for JIT path (host-bridge creates via global registry)
|
||
crate::runtime::register_user_defined_factory(Arc::new(PersonFactory));
|
||
|
||
// Build module
|
||
let module = build_person_module();
|
||
|
||
// VM (VTABLE on)
|
||
std::env::set_var("NYASH_ABI_VTABLE", "1");
|
||
let mut vm = VM::new();
|
||
let vm_out = vm.execute_module(&module).expect("VM exec");
|
||
let vm_s = vm_out.to_string_box().value;
|
||
|
||
// JIT(host-bridge on)
|
||
std::env::set_var("NYASH_JIT_HOST_BRIDGE", "1");
|
||
let jit_out = crate::backend::cranelift_compile_and_execute(&module, "identical_person")
|
||
.expect("JIT exec");
|
||
let jit_s = jit_out.to_string_box().value;
|
||
|
||
assert_eq!(vm_s, jit_s);
|
||
assert_eq!(vm_s, "Alice");
|
||
}
|
||
}
|