#![cfg(feature = "serde")] use rhai::{ serde::{from_dynamic, to_dynamic}, Dynamic, Engine, EvalAltResult, ImmutableString, INT, }; use serde::{Deserialize, Serialize}; #[cfg(not(feature = "no_index"))] use rhai::Array; #[cfg(not(feature = "no_object"))] use rhai::Map; #[cfg(not(feature = "no_float"))] use rhai::FLOAT; #[cfg(not(feature = "no_float"))] #[cfg(feature = "decimal")] use rust_decimal::Decimal; #[test] fn test_serde_ser_primary_types() -> Result<(), Box> { assert!(to_dynamic(42_u64)?.is::()); assert!(to_dynamic(u64::MAX)?.is::()); assert!(to_dynamic(42 as INT)?.is::()); assert!(to_dynamic(true)?.is::()); assert!(to_dynamic(())?.is::<()>()); #[cfg(not(feature = "no_float"))] { assert!(to_dynamic(123.456_f64)?.is::()); assert!(to_dynamic(123.456_f32)?.is::()); } #[cfg(feature = "no_float")] #[cfg(feature = "decimal")] { assert!(to_dynamic(123.456_f64)?.is::()); assert!(to_dynamic(123.456_f32)?.is::()); } assert!(to_dynamic("hello".to_string())?.is::()); Ok(()) } #[test] fn test_serde_ser_integer_types() -> Result<(), Box> { assert!(to_dynamic(42_i8)?.is::()); assert!(to_dynamic(42_i16)?.is::()); assert!(to_dynamic(42_i32)?.is::()); assert!(to_dynamic(42_i64)?.is::()); assert!(to_dynamic(42_u8)?.is::()); assert!(to_dynamic(42_u16)?.is::()); assert!(to_dynamic(42_u32)?.is::()); assert!(to_dynamic(42_u64)?.is::()); Ok(()) } #[test] #[cfg(not(feature = "no_index"))] fn test_serde_ser_array() -> Result<(), Box> { let arr: Vec = vec![123, 456, 42, 999]; let d = to_dynamic(arr)?; assert!(d.is::()); assert_eq!(4, d.cast::().len()); Ok(()) } #[test] #[cfg(not(feature = "no_index"))] #[cfg(not(feature = "no_object"))] fn test_serde_ser_struct() -> Result<(), Box> { #[derive(Debug, Serialize, PartialEq)] struct Hello { a: INT, b: bool, } #[derive(Debug, Serialize, PartialEq)] struct Test { int: u32, seq: Vec, obj: Hello, } let x = Test { int: 42, seq: vec!["hello".into(), "kitty".into(), "world".into()], obj: Hello { a: 123, b: true }, }; let d = to_dynamic(x)?; assert!(d.is::()); let mut map = d.cast::(); let obj = map.remove("obj").unwrap().cast::(); let mut seq = map.remove("seq").unwrap().cast::(); assert_eq!(Ok(123), obj["a"].as_int()); assert!(obj["b"].as_bool().unwrap()); assert_eq!(Ok(42), map["int"].as_int()); assert_eq!(seq.len(), 3); assert_eq!("kitty", seq.remove(1).into_string().unwrap()); Ok(()) } #[test] fn test_serde_ser_unit_enum() -> Result<(), Box> { #[derive(Serialize)] enum MyEnum { VariantFoo, VariantBar, } let d = to_dynamic(MyEnum::VariantFoo)?; assert_eq!("VariantFoo", d.into_string().unwrap()); let d = to_dynamic(MyEnum::VariantBar)?; assert_eq!("VariantBar", d.into_string().unwrap()); Ok(()) } #[test] #[cfg(not(feature = "no_object"))] fn test_serde_ser_externally_tagged_enum() -> Result<(), Box> { #[derive(Serialize)] enum MyEnum { VariantUnit, #[cfg(not(feature = "no_index"))] VariantUnitTuple(), VariantNewtype(i32), #[cfg(not(feature = "no_index"))] VariantTuple(i32, i32), VariantEmptyStruct {}, VariantStruct { a: i32, }, } { assert_eq!( "VariantUnit", to_dynamic(MyEnum::VariantUnit)? .into_immutable_string() .unwrap() .as_str() ); } #[cfg(not(feature = "no_index"))] { let mut map = to_dynamic(MyEnum::VariantUnitTuple())?.cast::(); let content = map.remove("VariantUnitTuple").unwrap().cast::(); assert!(map.is_empty()); assert!(content.is_empty()); } let mut map = to_dynamic(MyEnum::VariantNewtype(123))?.cast::(); let content = map.remove("VariantNewtype").unwrap(); assert!(map.is_empty()); assert_eq!(Ok(123), content.as_int()); #[cfg(not(feature = "no_index"))] { let mut map = to_dynamic(MyEnum::VariantTuple(123, 456))?.cast::(); let content = map.remove("VariantTuple").unwrap().cast::(); assert!(map.is_empty()); assert_eq!(2, content.len()); assert_eq!(Ok(123), content[0].as_int()); assert_eq!(Ok(456), content[1].as_int()); } let mut map = to_dynamic(MyEnum::VariantEmptyStruct {})?.cast::(); let map_inner = map.remove("VariantEmptyStruct").unwrap().cast::(); assert!(map.is_empty()); assert!(map_inner.is_empty()); let mut map = to_dynamic(MyEnum::VariantStruct { a: 123 })?.cast::(); let mut map_inner = map.remove("VariantStruct").unwrap().cast::(); assert!(map.is_empty()); assert_eq!(Ok(123), map_inner.remove("a").unwrap().as_int()); assert!(map_inner.is_empty()); Ok(()) } #[test] #[cfg(not(feature = "no_object"))] fn test_serde_ser_internally_tagged_enum() -> Result<(), Box> { #[derive(Serialize)] #[serde(tag = "tag")] enum MyEnum { VariantEmptyStruct {}, VariantStruct { a: i32 }, } let mut map = to_dynamic(MyEnum::VariantEmptyStruct {})?.cast::(); assert_eq!( "VariantEmptyStruct", map.remove("tag") .unwrap() .into_immutable_string() .unwrap() .as_str() ); assert!(map.is_empty()); let mut map = to_dynamic(MyEnum::VariantStruct { a: 123 })?.cast::(); assert_eq!( "VariantStruct", map.remove("tag") .unwrap() .into_immutable_string() .unwrap() .as_str() ); assert_eq!(Ok(123), map.remove("a").unwrap().as_int()); assert!(map.is_empty()); Ok(()) } #[test] #[cfg(not(feature = "no_object"))] fn test_serde_ser_adjacently_tagged_enum() -> Result<(), Box> { #[derive(Serialize)] #[serde(tag = "tag", content = "content")] enum MyEnum { VariantUnit, #[cfg(not(feature = "no_index"))] VariantUnitTuple(), VariantNewtype(i32), #[cfg(not(feature = "no_index"))] VariantTuple(i32, i32), VariantEmptyStruct {}, VariantStruct { a: i32, }, } let mut map = to_dynamic(MyEnum::VariantUnit)?.cast::(); assert_eq!( "VariantUnit", map.remove("tag") .unwrap() .into_immutable_string() .unwrap() .as_str() ); assert!(map.is_empty()); #[cfg(not(feature = "no_index"))] { let mut map = to_dynamic(MyEnum::VariantUnitTuple())?.cast::(); assert_eq!( "VariantUnitTuple", map.remove("tag") .unwrap() .into_immutable_string() .unwrap() .as_str() ); let content = map.remove("content").unwrap().cast::(); assert!(map.is_empty()); assert!(content.is_empty()); } let mut map = to_dynamic(MyEnum::VariantNewtype(123))?.cast::(); assert_eq!( "VariantNewtype", map.remove("tag") .unwrap() .into_immutable_string() .unwrap() .as_str() ); let content = map.remove("content").unwrap(); assert!(map.is_empty()); assert_eq!(Ok(123), content.as_int()); #[cfg(not(feature = "no_index"))] { let mut map = to_dynamic(MyEnum::VariantTuple(123, 456))?.cast::(); assert_eq!( "VariantTuple", map.remove("tag") .unwrap() .into_immutable_string() .unwrap() .as_str() ); let content = map.remove("content").unwrap().cast::(); assert!(map.is_empty()); assert_eq!(2, content.len()); assert_eq!(Ok(123), content[0].as_int()); assert_eq!(Ok(456), content[1].as_int()); } let mut map = to_dynamic(MyEnum::VariantEmptyStruct {})?.cast::(); assert_eq!( "VariantEmptyStruct", map.remove("tag") .unwrap() .into_immutable_string() .unwrap() .as_str() ); let map_inner = map.remove("content").unwrap().cast::(); assert!(map.is_empty()); assert!(map_inner.is_empty()); let mut map = to_dynamic(MyEnum::VariantStruct { a: 123 })?.cast::(); assert_eq!( "VariantStruct", map.remove("tag").unwrap().into_string().unwrap() ); let mut map_inner = map.remove("content").unwrap().cast::(); assert!(map.is_empty()); assert_eq!(Ok(123), map_inner.remove("a").unwrap().as_int()); assert!(map_inner.is_empty()); Ok(()) } #[test] #[cfg(not(feature = "no_object"))] fn test_serde_ser_untagged_enum() -> Result<(), Box> { #[derive(Serialize)] #[serde(untagged)] enum MyEnum { VariantEmptyStruct {}, VariantStruct1 { a: i32 }, VariantStruct2 { b: i32 }, } let map = to_dynamic(MyEnum::VariantEmptyStruct {})?.cast::(); assert!(map.is_empty()); let mut map = to_dynamic(MyEnum::VariantStruct1 { a: 123 })?.cast::(); assert_eq!(Ok(123), map.remove("a").unwrap().as_int()); assert!(map.is_empty()); let mut map = to_dynamic(MyEnum::VariantStruct2 { b: 123 })?.cast::(); assert_eq!(Ok(123), map.remove("b").unwrap().as_int()); assert!(map.is_empty()); Ok(()) } #[test] fn test_serde_de_primary_types() -> Result<(), Box> { assert_eq!(42, from_dynamic::(&Dynamic::from(42_u16))?); assert_eq!(42, from_dynamic::(&(42 as INT).into())?); assert_eq!(true, from_dynamic::(&true.into())?); assert_eq!((), from_dynamic::<()>(&().into())?); #[cfg(not(feature = "no_float"))] { assert_eq!(123.456, from_dynamic::(&123.456.into())?); assert_eq!(123.456, from_dynamic::(&Dynamic::from(123.456_f32))?); } #[cfg(feature = "no_float")] #[cfg(feature = "decimal")] { let d: Dynamic = Decimal::from_str("123.456").unwrap().into(); assert_eq!(123.456, from_dynamic::(&d)?); assert_eq!(123.456, from_dynamic::(&d)?); } assert_eq!( "hello", from_dynamic::(&"hello".to_string().into())? ); Ok(()) } #[test] fn test_serde_de_integer_types() -> Result<(), Box> { assert_eq!(42, from_dynamic::(&Dynamic::from(42 as INT))?); assert_eq!(42, from_dynamic::(&Dynamic::from(42 as INT))?); assert_eq!(42, from_dynamic::(&Dynamic::from(42 as INT))?); assert_eq!(42, from_dynamic::(&Dynamic::from(42 as INT))?); assert_eq!(42, from_dynamic::(&Dynamic::from(42 as INT))?); assert_eq!(42, from_dynamic::(&Dynamic::from(42 as INT))?); assert_eq!(42, from_dynamic::(&Dynamic::from(42 as INT))?); assert_eq!(42, from_dynamic::(&Dynamic::from(42 as INT))?); Ok(()) } #[test] #[cfg(not(feature = "no_index"))] fn test_serde_de_array() -> Result<(), Box> { let arr: Vec = vec![123, 456, 42, 999]; assert_eq!(arr, from_dynamic::>(&arr.clone().into())?); Ok(()) } #[test] #[cfg(not(feature = "no_index"))] #[cfg(not(feature = "no_object"))] fn test_serde_de_struct() -> Result<(), Box> { #[derive(Debug, Deserialize, PartialEq)] struct Hello { a: INT, b: bool, } #[derive(Debug, Deserialize, PartialEq)] struct Test { int: u32, seq: Vec, obj: Hello, } let mut map = Map::new(); map.insert("int".into(), Dynamic::from(42_u32)); let mut map2 = Map::new(); map2.insert("a".into(), (123 as INT).into()); map2.insert("b".into(), true.into()); map.insert("obj".into(), map2.into()); let arr: Array = vec!["hello".into(), "kitty".into(), "world".into()]; map.insert("seq".into(), arr.into()); let expected = Test { int: 42, seq: vec!["hello".into(), "kitty".into(), "world".into()], obj: Hello { a: 123, b: true }, }; assert_eq!(expected, from_dynamic(&map.into())?); Ok(()) } #[test] #[cfg(not(feature = "no_index"))] #[cfg(not(feature = "no_object"))] #[cfg(not(feature = "no_float"))] fn test_serde_de_script() -> Result<(), Box> { #[derive(Debug, Deserialize)] struct Point { x: FLOAT, y: FLOAT, } #[derive(Debug, Deserialize)] struct MyStruct { a: i64, b: Vec, c: bool, d: Point, } let engine = Engine::new(); let result: Dynamic = engine.eval( r#" #{ a: 42, b: [ "hello", "world" ], c: true, d: #{ x: 123.456, y: 999.0 } } "#, )?; // Convert the 'Dynamic' object map into 'MyStruct' let _: MyStruct = from_dynamic(&result)?; Ok(()) } #[test] fn test_serde_de_unit_enum() -> Result<(), Box> { #[derive(Debug, PartialEq, Deserialize)] enum MyEnum { VariantFoo, VariantBar, } let d = Dynamic::from("VariantFoo".to_string()); assert_eq!(MyEnum::VariantFoo, from_dynamic(&d)?); let d = Dynamic::from("VariantBar".to_string()); assert_eq!(MyEnum::VariantBar, from_dynamic(&d)?); Ok(()) } #[test] #[cfg(not(feature = "no_object"))] fn test_serde_de_externally_tagged_enum() -> Result<(), Box> { #[derive(Debug, PartialEq, Deserialize)] #[serde(deny_unknown_fields)] enum MyEnum { VariantUnit, #[cfg(not(feature = "no_index"))] VariantUnitTuple(), VariantNewtype(i32), #[cfg(not(feature = "no_index"))] VariantTuple(i32, i32), VariantEmptyStruct {}, VariantStruct { a: i32, }, } let d = Dynamic::from("VariantUnit".to_string()); assert_eq!(MyEnum::VariantUnit, from_dynamic(&d).unwrap()); #[cfg(not(feature = "no_index"))] { let array: Array = vec![]; let mut map_outer = Map::new(); map_outer.insert("VariantUnitTuple".into(), array.into()); assert_eq!( MyEnum::VariantUnitTuple(), from_dynamic(&map_outer.into()).unwrap() ); } let mut map_outer = Map::new(); map_outer.insert("VariantNewtype".into(), (123 as INT).into()); assert_eq!( MyEnum::VariantNewtype(123), from_dynamic(&map_outer.into()).unwrap() ); #[cfg(not(feature = "no_index"))] { let array: Array = vec![(123 as INT).into(), (456 as INT).into()]; let mut map_outer = Map::new(); map_outer.insert("VariantTuple".into(), array.into()); assert_eq!( MyEnum::VariantTuple(123, 456), from_dynamic(&map_outer.into()).unwrap() ); } let map_inner = Map::new(); let mut map_outer = Map::new(); map_outer.insert("VariantEmptyStruct".into(), map_inner.into()); assert_eq!( MyEnum::VariantEmptyStruct {}, from_dynamic(&map_outer.into()).unwrap() ); let mut map_inner = Map::new(); map_inner.insert("a".into(), (123 as INT).into()); let mut map_outer = Map::new(); map_outer.insert("VariantStruct".into(), map_inner.into()); assert_eq!( MyEnum::VariantStruct { a: 123 }, from_dynamic(&map_outer.into()).unwrap() ); Ok(()) } #[test] #[cfg(not(feature = "no_object"))] fn test_serde_de_internally_tagged_enum() -> Result<(), Box> { #[derive(Debug, PartialEq, Deserialize)] #[serde(tag = "tag", deny_unknown_fields)] enum MyEnum { VariantEmptyStruct {}, VariantStruct { a: i32 }, } let mut map = Map::new(); map.insert("tag".into(), "VariantStruct".into()); map.insert("a".into(), (123 as INT).into()); assert_eq!( MyEnum::VariantStruct { a: 123 }, from_dynamic(&map.into()).unwrap() ); let mut map = Map::new(); map.insert("tag".into(), "VariantEmptyStruct".into()); assert_eq!( MyEnum::VariantEmptyStruct {}, from_dynamic(&map.into()).unwrap() ); Ok(()) } #[test] #[cfg(not(feature = "no_object"))] fn test_serde_de_adjacently_tagged_enum() -> Result<(), Box> { #[derive(Debug, PartialEq, Deserialize)] #[serde(tag = "tag", content = "content", deny_unknown_fields)] enum MyEnum { VariantUnit, #[cfg(not(feature = "no_index"))] VariantUnitTuple(), VariantNewtype(i32), #[cfg(not(feature = "no_index"))] VariantTuple(i32, i32), VariantEmptyStruct {}, VariantStruct { a: i32, }, } let mut map_outer = Map::new(); map_outer.insert("tag".into(), "VariantUnit".into()); assert_eq!( MyEnum::VariantUnit, from_dynamic(&map_outer.into()).unwrap() ); #[cfg(not(feature = "no_index"))] { let array: Array = vec![]; let mut map_outer = Map::new(); map_outer.insert("tag".into(), "VariantUnitTuple".into()); map_outer.insert("content".into(), array.into()); assert_eq!( MyEnum::VariantUnitTuple(), from_dynamic(&map_outer.into()).unwrap() ); } let mut map_outer = Map::new(); map_outer.insert("tag".into(), "VariantNewtype".into()); map_outer.insert("content".into(), (123 as INT).into()); assert_eq!( MyEnum::VariantNewtype(123), from_dynamic(&map_outer.into()).unwrap() ); #[cfg(not(feature = "no_index"))] { let array: Array = vec![(123 as INT).into(), (456 as INT).into()]; let mut map_outer = Map::new(); map_outer.insert("tag".into(), "VariantTuple".into()); map_outer.insert("content".into(), array.into()); assert_eq!( MyEnum::VariantTuple(123, 456), from_dynamic(&map_outer.into()).unwrap() ); } let map_inner = Map::new(); let mut map_outer = Map::new(); map_outer.insert("tag".into(), "VariantEmptyStruct".into()); map_outer.insert("content".into(), map_inner.into()); assert_eq!( MyEnum::VariantEmptyStruct {}, from_dynamic(&map_outer.into()).unwrap() ); let mut map_inner = Map::new(); map_inner.insert("a".into(), (123 as INT).into()); let mut map_outer = Map::new(); map_outer.insert("tag".into(), "VariantStruct".into()); map_outer.insert("content".into(), map_inner.into()); assert_eq!( MyEnum::VariantStruct { a: 123 }, from_dynamic(&map_outer.into()).unwrap() ); Ok(()) } #[test] #[cfg(not(feature = "no_object"))] fn test_serde_de_untagged_enum() -> Result<(), Box> { #[derive(Debug, PartialEq, Deserialize)] #[serde(untagged, deny_unknown_fields)] enum MyEnum { VariantEmptyStruct {}, VariantStruct1 { a: i32 }, VariantStruct2 { b: i32 }, } let map = Map::new(); assert_eq!( MyEnum::VariantEmptyStruct {}, from_dynamic(&map.into()).unwrap() ); let mut map = Map::new(); map.insert("a".into(), (123 as INT).into()); assert_eq!( MyEnum::VariantStruct1 { a: 123 }, from_dynamic(&map.into()).unwrap() ); let mut map = Map::new(); map.insert("b".into(), (123 as INT).into()); assert_eq!( MyEnum::VariantStruct2 { b: 123 }, from_dynamic(&map.into()).unwrap() ); Ok(()) } #[test] #[cfg(feature = "metadata")] #[cfg(not(feature = "no_object"))] #[cfg(not(feature = "no_index"))] fn test_serde_json() -> serde_json::Result<()> { let s: ImmutableString = "hello".into(); assert_eq!(serde_json::to_string(&s)?, r#""hello""#); let mut map = Map::new(); map.insert("a".into(), (123 as INT).into()); let arr: Array = vec![(1 as INT).into(), (2 as INT).into(), (3 as INT).into()]; map.insert("b".into(), arr.into()); map.insert("c".into(), true.into()); let d: Dynamic = map.into(); let json = serde_json::to_string(&d)?; assert!(json.contains("\"a\":123")); assert!(json.contains("\"b\":[1,2,3]")); assert!(json.contains("\"c\":true")); let d2: Dynamic = serde_json::from_str(&json)?; assert!(d2.is::()); let mut m = d2.cast::(); assert_eq!(m["a"].as_int().unwrap(), 123); assert!(m["c"].as_bool().unwrap()); let a = m.remove("b").unwrap().cast::(); assert_eq!(a.len(), 3); assert_eq!(format!("{:?}", a), "[1, 2, 3]"); Ok(()) } #[test] #[cfg(not(feature = "no_object"))] fn test_serde_optional() -> Result<(), Box> { #[derive(Debug, Clone, PartialEq, Eq, Deserialize)] struct TestStruct { foo: Option, } let mut engine = Engine::new(); engine.register_type_with_name::("TestStruct"); let r = engine.eval::("#{ foo: 'a' }")?; assert_eq!( from_dynamic::(&r)?, TestStruct { foo: Some('a') } ); let r = engine.eval::("#{ foo: () }")?; assert_eq!(from_dynamic::(&r)?, TestStruct { foo: None }); let r = engine.eval::("#{ }")?; assert_eq!(from_dynamic::(&r)?, TestStruct { foo: None }); Ok(()) } #[test] #[cfg(not(feature = "no_index"))] fn test_serde_blob() -> Result<(), Box> { let engine = Engine::new(); let r = engine.eval::( " let x = blob(10); for i in 0..10 { x[i] = i; } x ", )?; let r = from_dynamic::(&r)?; assert_eq!(r.to_vec(), vec![0_u8, 1, 2, 3, 4, 5, 6, 7, 8, 9]); Ok(()) }