1mod common;23use std::path::PathBuf;45use gcmodule::Cc;6use jrsonnet_evaluator::{7 error::Result,8 function::{builtin, builtin::Builtin, CallLocation, FuncVal},9 gc::TraceBox,10 tb,11 typed::Typed,12 State, Val,13};1415#[builtin]16fn a() -> Result<u32> {17 Ok(1)18}1920#[test]21fn basic_function() -> Result<()> {22 let s = State::default();23 let a: a = a {};24 let v = u32::from_untyped(25 a.call(26 s.clone(),27 s.create_default_context(),28 CallLocation::native(),29 &(),30 )?,31 s.clone(),32 )?;3334 ensure_eq!(v, 1);35 Ok(())36}3738#[builtin]39fn native_add(a: u32, b: u32) -> Result<u32> {40 Ok(a + b)41}4243#[test]44fn call_from_code() -> Result<()> {45 let s = State::default();46 s.with_stdlib();47 s.settings_mut().globals.insert(48 "nativeAdd".into(),49 Val::Func(FuncVal::StaticBuiltin(native_add::INST)),50 );5152 let v = s.evaluate_snippet_raw(53 PathBuf::new().into(),54 "55 assert nativeAdd(1, 2) == 3;56 assert nativeAdd(100, 200) == 300;57 null58 "59 .into(),60 )?;61 ensure_val_eq!(s.clone(), v, Val::Null);62 Ok(())63}6465#[builtin(fields(66 a: u3267))]68fn curried_add(this: &curried_add, b: u32) -> Result<u32> {69 Ok(this.a + b)70}7172#[builtin]73fn curry_add(a: u32) -> Result<FuncVal> {74 Ok(FuncVal::Builtin(Cc::new(tb!(curried_add { a }))))75}7677#[test]78fn nonstatic_builtin() -> Result<()> {79 let s = State::default();80 s.with_stdlib();81 s.settings_mut().globals.insert(82 "curryAdd".into(),83 Val::Func(FuncVal::StaticBuiltin(curry_add::INST)),84 );8586 let v = s.evaluate_snippet_raw(87 PathBuf::new().into(),88 "89 local a = curryAdd(1);90 local b = curryAdd(4);9192 assert a(2) == 3;93 assert a(200) == 201;9495 assert b(2) == 6;96 assert b(200) == 204;97 null98 "99 .into(),100 )?;101 ensure_val_eq!(s.clone(), v, Val::Null);102 Ok(())103}