77 lines
1.8 KiB
Rust
77 lines
1.8 KiB
Rust
use prometeu_compiler::ir_core::*;
|
|
use serde_json;
|
|
|
|
#[test]
|
|
fn test_ir_core_manual_construction() {
|
|
let mut const_pool = ConstPool::new();
|
|
const_pool.insert(ConstantValue::String("hello".to_string()));
|
|
|
|
let program = Program {
|
|
const_pool,
|
|
modules: vec![Module {
|
|
name: "main".to_string(),
|
|
functions: vec![Function {
|
|
id: FunctionId(10),
|
|
name: "entry".to_string(),
|
|
blocks: vec![Block {
|
|
id: 0,
|
|
instrs: vec![
|
|
Instr::PushConst(ConstId(0)),
|
|
Instr::Call(FunctionId(11), 0),
|
|
],
|
|
terminator: Terminator::Return,
|
|
}],
|
|
}],
|
|
}],
|
|
};
|
|
|
|
let json = serde_json::to_string_pretty(&program).unwrap();
|
|
|
|
// Snapshot check for deterministic shape
|
|
let expected = r#"{
|
|
"const_pool": {
|
|
"constants": [
|
|
{
|
|
"String": "hello"
|
|
}
|
|
]
|
|
},
|
|
"modules": [
|
|
{
|
|
"name": "main",
|
|
"functions": [
|
|
{
|
|
"id": 10,
|
|
"name": "entry",
|
|
"blocks": [
|
|
{
|
|
"id": 0,
|
|
"instrs": [
|
|
{
|
|
"PushConst": 0
|
|
},
|
|
{
|
|
"Call": [
|
|
11,
|
|
0
|
|
]
|
|
}
|
|
],
|
|
"terminator": "Return"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}"#;
|
|
assert_eq!(json, expected);
|
|
}
|
|
|
|
#[test]
|
|
fn test_ir_core_ids() {
|
|
assert_eq!(serde_json::to_string(&FunctionId(1)).unwrap(), "1");
|
|
assert_eq!(serde_json::to_string(&ConstId(2)).unwrap(), "2");
|
|
assert_eq!(serde_json::to_string(&TypeId(3)).unwrap(), "3");
|
|
}
|