| use flatbuffers_reflection::reflection::{root_as_schema, BaseType, Field}; | |
| use flatbuffers_reflection::{ | |
| get_any_field_float, get_any_field_float_in_struct, get_any_field_integer, | |
| get_any_field_integer_in_struct, get_any_field_string, get_any_field_string_in_struct, | |
| get_any_root, get_field_float, get_field_integer, get_field_string, get_field_struct, | |
| get_field_struct_in_struct, get_field_table, get_field_vector, set_any_field_float, | |
| set_any_field_integer, set_any_field_string, set_field, set_string, FlatbufferError, | |
| }; | |
| use flatbuffers_reflection::{SafeBuffer, Struct}; | |
| use flatbuffers::{FlatBufferBuilder, Table, VerifierOptions}; | |
| use std::error::Error; | |
| use std::fs::File; | |
| use std::io::Read; | |
| use assert_approx_eq::assert_approx_eq; | |
| #[allow(dead_code, unused_imports)] | |
| #[path = "../../monster_test/mod.rs"] | |
| mod monster_test_generated; | |
| pub use monster_test_generated::my_game; | |
| #[test] | |
| fn test_schema_correct_root_table() { | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let root_table = schema.root_table().unwrap(); | |
| assert_eq!(root_table.name(), "MyGame.Example.Monster"); | |
| assert_eq!(root_table.declaration_file().unwrap(), "//monster_test.fbs"); | |
| } | |
| #[test] | |
| fn test_schema_correct_object() { | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| assert_eq!( | |
| schema | |
| .objects() | |
| .lookup_by_key("TableA", |object, key| object.key_compare_with_value(key)) | |
| .unwrap() | |
| .declaration_file() | |
| .unwrap(), | |
| "//include_test/include_test1.fbs" | |
| ); | |
| assert_eq!( | |
| schema | |
| .objects() | |
| .lookup_by_key("MyGame.OtherNameSpace.Unused", |object, key| object | |
| .key_compare_with_value(key)) | |
| .unwrap() | |
| .declaration_file() | |
| .unwrap(), | |
| "//include_test/sub/include_test2.fbs" | |
| ); | |
| } | |
| #[test] | |
| fn test_schema_correct_enum() { | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| assert_eq!( | |
| schema | |
| .enums() | |
| .lookup_by_key("MyGame.OtherNameSpace.FromInclude", |en, key| en | |
| .key_compare_with_value(key)) | |
| .unwrap() | |
| .declaration_file() | |
| .unwrap(), | |
| "//include_test/sub/include_test2.fbs" | |
| ); | |
| } | |
| #[test] | |
| fn test_schema_correct_file() { | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| assert_eq!(schema.fbs_files().unwrap().len(), 3); | |
| let fbs0 = schema.fbs_files().unwrap().get(0); | |
| assert_eq!(fbs0.filename(), "//include_test/include_test1.fbs"); | |
| let fbs0_includes = fbs0.included_filenames().unwrap(); | |
| assert_eq!(fbs0_includes.len(), 2); | |
| assert_eq!(fbs0_includes.get(0), "//include_test/include_test1.fbs"); | |
| assert_eq!(fbs0_includes.get(1), "//include_test/sub/include_test2.fbs"); | |
| let fbs1 = schema.fbs_files().unwrap().get(1); | |
| assert_eq!(fbs1.filename(), "//include_test/sub/include_test2.fbs"); | |
| let fbs1_includes = fbs1.included_filenames().unwrap(); | |
| assert_eq!(fbs1_includes.len(), 2); | |
| assert_eq!(fbs1_includes.get(0), "//include_test/include_test1.fbs"); | |
| assert_eq!(fbs1_includes.get(1), "//include_test/sub/include_test2.fbs"); | |
| let fbs2 = schema.fbs_files().unwrap().get(2); | |
| assert_eq!(fbs2.filename(), "//monster_test.fbs"); | |
| let fbs2_includes = fbs2.included_filenames().unwrap(); | |
| assert_eq!(fbs2_includes.len(), 1); | |
| assert_eq!(fbs2_includes.get(0), "//include_test/include_test1.fbs"); | |
| } | |
| #[test] | |
| fn test_schema_correct_table_field() { | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let root_table = root_as_schema(schema_buffer.as_slice()).unwrap().root_table().unwrap(); | |
| let fields = root_table.fields(); | |
| let hp_field_option = | |
| fields.lookup_by_key("hp", |field, key| field.key_compare_with_value(key)); | |
| assert!(hp_field_option.is_some()); | |
| let hp_field = hp_field_option.unwrap(); | |
| assert_eq!(hp_field.name(), "hp"); | |
| assert_eq!(hp_field.id(), 2); | |
| assert_eq!(hp_field.type_().base_type(), BaseType::Short); | |
| let friendly_field_option = | |
| fields.lookup_by_key("friendly", |field, key| field.key_compare_with_value(key)); | |
| assert!(friendly_field_option.is_some()); | |
| assert!(friendly_field_option.unwrap().attributes().is_some()); | |
| assert!(friendly_field_option | |
| .unwrap() | |
| .attributes() | |
| .unwrap() | |
| .lookup_by_key("priority", |key_value, key| key_value.key_compare_with_value(key)) | |
| .is_some()); | |
| } | |
| #[test] | |
| fn test_schema_correct_table_field_nullability() { | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let root_table = root_as_schema(schema_buffer.as_slice()).unwrap().root_table().unwrap(); | |
| let fields = root_table.fields(); | |
| assert_eq!( | |
| fields | |
| .lookup_by_key("hp", |field, key| field.key_compare_with_value(key)) | |
| .unwrap() | |
| .optional(), | |
| false | |
| ); | |
| assert_eq!( | |
| fields | |
| .lookup_by_key("pos", |field, key| field.key_compare_with_value(key)) | |
| .unwrap() | |
| .optional(), | |
| true | |
| ); | |
| assert_eq!( | |
| fields | |
| .lookup_by_key("name", |field, key| field.key_compare_with_value(key)) | |
| .unwrap() | |
| .optional(), | |
| false | |
| ); | |
| } | |
| #[test] | |
| fn test_schema_correct_child_table_index() { | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let root_table = schema.root_table().unwrap(); | |
| let fields = root_table.fields(); | |
| let pos_field_option = | |
| fields.lookup_by_key("pos", |field, key| field.key_compare_with_value(key)); | |
| assert!(pos_field_option.is_some()); | |
| let pos_field = pos_field_option.unwrap(); | |
| assert_eq!(pos_field.type_().base_type(), BaseType::Obj); | |
| let pos_table = schema.objects().get(pos_field.type_().index() as usize); | |
| assert_eq!(pos_table.name(), "MyGame.Example.Vec3"); | |
| } | |
| #[test] | |
| fn test_buffer_integer_same_type_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let i16_field = get_schema_field(&schema, "hp"); | |
| let value = unsafe { get_field_integer::<i16>(&root_table, &i16_field) }; | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), Some(32767)); | |
| } | |
| #[test] | |
| fn test_buffer_integer_diff_type_same_size_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let i16_field = get_schema_field(&schema, "hp"); | |
| let value = unsafe { get_field_integer::<u16>(&root_table, &i16_field) }; | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), Some(32767)); | |
| } | |
| #[test] | |
| fn test_buffer_integer_diff_size_fails() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let i16_field = get_schema_field(&schema, "hp"); | |
| let value = unsafe { get_field_integer::<i64>(&root_table, &i16_field) }; | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("i64"), String::from("Short")) | |
| ); | |
| } | |
| #[test] | |
| fn test_buffer_float_same_type_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let f32_field = get_schema_field(&schema, "testf"); | |
| let value = unsafe { get_field_float::<f32>(&root_table, &f32_field) }; | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), Some(3.14)); | |
| } | |
| #[test] | |
| fn test_buffer_float_diff_type_fails() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let f32_field = get_schema_field(&schema, "testf"); | |
| let value = unsafe { get_field_float::<f64>(&root_table, &f32_field) }; | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("f64"), String::from("Float")) | |
| ); | |
| } | |
| #[test] | |
| fn test_buffer_string_same_type_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let string_field = get_schema_field(&schema, "name"); | |
| let value = unsafe { get_field_string(&root_table, &string_field) }; | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), Some("MyMonster")); | |
| } | |
| #[test] | |
| fn test_buffer_string_diff_type_fails() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let f32_field = get_schema_field(&schema, "testf"); | |
| let value = unsafe { get_field_string(&root_table, &f32_field) }; | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("String"), String::from("Float")) | |
| ); | |
| } | |
| #[test] | |
| fn test_buffer_struct_same_type_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let struct_field = get_schema_field(&schema, "pos"); | |
| let res = unsafe { get_field_struct(&root_table, &struct_field) }; | |
| assert!(res.is_ok()); | |
| let optional_value = res.unwrap(); | |
| assert!(optional_value.is_some()); | |
| assert_eq!(optional_value.unwrap().buf(), &buffer); | |
| assert!(optional_value.unwrap().loc() > root_table.loc()); | |
| } | |
| #[test] | |
| fn test_buffer_struct_diff_type_fails() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let f32_field = get_schema_field(&schema, "testf"); | |
| let res = unsafe { get_field_struct(&root_table, &f32_field) }; | |
| assert!(res.is_err()); | |
| assert_eq!( | |
| res.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("Obj"), String::from("Float")) | |
| ); | |
| } | |
| #[test] | |
| fn test_buffer_vector_same_type_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let vector_field = get_schema_field(&schema, "inventory"); | |
| let value = unsafe { get_field_vector::<u8>(&root_table, &vector_field) }; | |
| assert!(value.is_ok()); | |
| let optional_vector = value.unwrap(); | |
| assert!(optional_vector.is_some()); | |
| let vector = optional_vector.unwrap(); | |
| assert_eq!(vector.len(), 5); | |
| assert_eq!(vector.get(0), 0); | |
| assert_eq!(vector.get(1), 1); | |
| assert_eq!(vector.get(2), 2); | |
| assert_eq!(vector.get(3), 3); | |
| assert_eq!(vector.get(4), 4); | |
| } | |
| #[test] | |
| fn test_buffer_vector_diff_type_fails() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let f32_field = get_schema_field(&schema, "testf"); | |
| let value = unsafe { get_field_vector::<u8>(&root_table, &f32_field) }; | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("u8"), String::from("Float")) | |
| ); | |
| } | |
| #[test] | |
| fn test_buffer_table_same_type_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let table_field = get_schema_field(&schema, "testempty"); | |
| let value = unsafe { get_field_table(&root_table, &table_field) }; | |
| assert!(value.is_ok()); | |
| let optional_nested_table = value.unwrap(); | |
| assert!(optional_nested_table.is_some()); | |
| let nested_table = optional_nested_table.unwrap(); | |
| let nested_table_fields = &root_as_schema(schema.as_slice()) | |
| .unwrap() | |
| .objects() | |
| .get(table_field.type_().index() as usize) | |
| .fields(); | |
| let nested_table_field = nested_table_fields | |
| .lookup_by_key("id", |field: &Field<'_>, key| field.key_compare_with_value(key)) | |
| .unwrap(); | |
| let nested_table_id = unsafe { get_field_string(&nested_table, &nested_table_field) }; | |
| assert!(nested_table_id.is_ok()); | |
| assert_eq!(nested_table_id.unwrap(), Some("Fred")); | |
| } | |
| #[test] | |
| fn test_buffer_table_diff_type_fails() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let f32_field = get_schema_field(&schema, "testf"); | |
| let value = unsafe { get_field_table(&root_table, &f32_field) }; | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("Obj"), String::from("Float")) | |
| ); | |
| } | |
| #[test] | |
| fn test_buffer_nested_struct_same_type_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let struct_field = get_schema_field(&schema, "pos"); | |
| let struct_value = unsafe { get_field_struct(&root_table, &struct_field).unwrap().unwrap() }; | |
| let struct_schema = root_as_schema(schema.as_slice()) | |
| .unwrap() | |
| .objects() | |
| .get(struct_field.type_().index() as usize); | |
| let nested_struct_field = struct_schema | |
| .fields() | |
| .lookup_by_key("test3", |field, key| field.key_compare_with_value(key)) | |
| .unwrap(); | |
| let res = unsafe { get_field_struct_in_struct(&struct_value, &nested_struct_field) }; | |
| assert!(res.is_ok()); | |
| let value = res.unwrap(); | |
| assert_eq!(value.buf(), &buffer); | |
| assert!(value.loc() > struct_value.loc()); | |
| } | |
| #[test] | |
| fn test_buffer_nested_struct_diff_type_fails() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let struct_field = get_schema_field(&schema, "pos"); | |
| let struct_value = unsafe { get_field_struct(&root_table, &struct_field).unwrap().unwrap() }; | |
| let struct_schema = root_as_schema(schema.as_slice()) | |
| .unwrap() | |
| .objects() | |
| .get(struct_field.type_().index() as usize); | |
| let nested_float_field = struct_schema | |
| .fields() | |
| .lookup_by_key("x", |field, key| field.key_compare_with_value(key)) | |
| .unwrap(); | |
| let res = unsafe { get_field_struct_in_struct(&struct_value, &nested_float_field) }; | |
| assert!(res.is_err()); | |
| assert_eq!( | |
| res.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("Obj"), String::from("Float")) | |
| ); | |
| } | |
| #[test] | |
| fn test_buffer_i16_as_integer_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let i16_field = get_schema_field(&schema, "hp"); | |
| let value = unsafe { get_any_field_integer(&root_table, &i16_field) }; | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), 32767); | |
| } | |
| #[test] | |
| fn test_buffer_f32_as_integer_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let f32_field = get_schema_field(&schema, "testf"); | |
| let value = unsafe { get_any_field_integer(&root_table, &f32_field) }; | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), 3); | |
| } | |
| #[test] | |
| fn test_buffer_inf_as_integer_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let inf_field = get_schema_field(&schema, "inf_default"); | |
| let value = unsafe { get_any_field_integer(&root_table, &inf_field) }; | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), 0); | |
| } | |
| #[test] | |
| fn test_buffer_bool_as_integer_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let bool_field = get_schema_field(&schema, "testbool"); | |
| let value = unsafe { get_any_field_integer(&root_table, &bool_field) }; | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), 0); | |
| } | |
| #[test] | |
| fn test_buffer_nan_as_integer_fails() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let nan_field = get_schema_field(&schema, "nan_default"); | |
| let value = unsafe { get_any_field_integer(&root_table, &nan_field) }; | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("i64"), String::from("Float")) | |
| ); | |
| } | |
| #[test] | |
| fn test_buffer_string_as_integer_fails() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let string_field = get_schema_field(&schema, "name"); | |
| let value = unsafe { get_any_field_integer(&root_table, &string_field) }; | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("i64"), String::from("String")) | |
| ); | |
| } | |
| #[test] | |
| fn test_buffer_i16_as_float_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let i16_field = get_schema_field(&schema, "hp"); | |
| let value = unsafe { get_any_field_float(&root_table, &i16_field) }; | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), 32767f64); | |
| } | |
| #[test] | |
| fn test_buffer_f32_as_float_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let f32_field = get_schema_field(&schema, "testf"); | |
| let value = unsafe { get_any_field_float(&root_table, &f32_field) }; | |
| assert!(value.is_ok()); | |
| assert_approx_eq!(value.unwrap(), 3.14); | |
| } | |
| #[test] | |
| fn test_buffer_string_as_float_fails() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let string_field = get_schema_field(&schema, "name"); | |
| let value = unsafe { get_any_field_float(&root_table, &string_field) }; | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("f64"), String::from("String")) | |
| ); | |
| } | |
| #[test] | |
| fn test_buffer_i16_as_string_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let i16_field = get_schema_field(&schema, "hp"); | |
| let value = unsafe { | |
| get_any_field_string(&root_table, &i16_field, &root_as_schema(schema.as_slice()).unwrap()) | |
| }; | |
| assert_eq!(value, String::from("32767")); | |
| } | |
| #[test] | |
| fn test_buffer_f32_as_string_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let f32_field = get_schema_field(&schema, "testf"); | |
| let mut value = unsafe { | |
| get_any_field_string(&root_table, &f32_field, &root_as_schema(schema.as_slice()).unwrap()) | |
| }; | |
| value.truncate(4); | |
| assert_eq!(value, String::from("3.14")); | |
| } | |
| #[test] | |
| fn test_buffer_string_as_string_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let string_field = get_schema_field(&schema, "name"); | |
| let value = unsafe { | |
| get_any_field_string( | |
| &root_table, | |
| &string_field, | |
| &root_as_schema(schema.as_slice()).unwrap(), | |
| ) | |
| }; | |
| assert_eq!(value, "MyMonster"); | |
| } | |
| #[test] | |
| fn test_buffer_i16_in_struct_as_integer_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let struct_field = get_schema_field(&schema, "pos"); | |
| let struct_value = unsafe { get_field_struct(&root_table, &struct_field).unwrap().unwrap() }; | |
| let struct_schema = root_as_schema(schema.as_slice()) | |
| .unwrap() | |
| .objects() | |
| .get(struct_field.type_().index() as usize); | |
| let nested_struct_field = struct_schema | |
| .fields() | |
| .lookup_by_key("test3", |field, key| field.key_compare_with_value(key)) | |
| .unwrap(); | |
| let nested_struct_value = | |
| unsafe { get_field_struct_in_struct(&struct_value, &nested_struct_field).unwrap() }; | |
| let nested_struct_schema = root_as_schema(schema.as_slice()) | |
| .unwrap() | |
| .objects() | |
| .get(nested_struct_field.type_().index() as usize); | |
| let i16_in_struct = nested_struct_schema | |
| .fields() | |
| .lookup_by_key("a", |field, key| field.key_compare_with_value(key)) | |
| .unwrap(); | |
| let value = unsafe { get_any_field_integer_in_struct(&nested_struct_value, &i16_in_struct) }; | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), 5); | |
| } | |
| #[test] | |
| fn test_buffer_f32_in_struct_as_integer_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let struct_field = get_schema_field(&schema, "pos"); | |
| let struct_value = unsafe { get_field_struct(&root_table, &struct_field).unwrap().unwrap() }; | |
| let struct_schema = root_as_schema(schema.as_slice()) | |
| .unwrap() | |
| .objects() | |
| .get(struct_field.type_().index() as usize); | |
| let f32_in_struct = struct_schema | |
| .fields() | |
| .lookup_by_key("x", |field, key| field.key_compare_with_value(key)) | |
| .unwrap(); | |
| let value = unsafe { get_any_field_integer_in_struct(&struct_value, &f32_in_struct) }; | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), 1); | |
| } | |
| #[test] | |
| fn test_buffer_enum_in_struct_as_integer_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let struct_field = get_schema_field(&schema, "pos"); | |
| let struct_value = unsafe { get_field_struct(&root_table, &struct_field).unwrap().unwrap() }; | |
| let struct_schema = root_as_schema(schema.as_slice()) | |
| .unwrap() | |
| .objects() | |
| .get(struct_field.type_().index() as usize); | |
| let enum_in_struct = struct_schema | |
| .fields() | |
| .lookup_by_key("test2", |field, key| field.key_compare_with_value(key)) | |
| .unwrap(); | |
| let value = unsafe { get_any_field_integer_in_struct(&struct_value, &enum_in_struct) }; | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), 2); | |
| } | |
| #[test] | |
| fn test_buffer_struct_in_struct_as_integer_fails() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let struct_field = get_schema_field(&schema, "pos"); | |
| let struct_value = unsafe { get_field_struct(&root_table, &struct_field).unwrap().unwrap() }; | |
| let struct_schema = root_as_schema(schema.as_slice()) | |
| .unwrap() | |
| .objects() | |
| .get(struct_field.type_().index() as usize); | |
| let struct_in_struct = struct_schema | |
| .fields() | |
| .lookup_by_key("test3", |field, key| field.key_compare_with_value(key)) | |
| .unwrap(); | |
| let value = unsafe { get_any_field_integer_in_struct(&struct_value, &struct_in_struct) }; | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("i64"), String::from("Obj")) | |
| ); | |
| } | |
| #[test] | |
| fn test_buffer_i16_in_struct_as_float_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let struct_field = get_schema_field(&schema, "pos"); | |
| let struct_value = unsafe { get_field_struct(&root_table, &struct_field).unwrap().unwrap() }; | |
| let struct_schema = root_as_schema(schema.as_slice()) | |
| .unwrap() | |
| .objects() | |
| .get(struct_field.type_().index() as usize); | |
| let nested_struct_field = struct_schema | |
| .fields() | |
| .lookup_by_key("test3", |field, key| field.key_compare_with_value(key)) | |
| .unwrap(); | |
| let nested_struct_value = | |
| unsafe { get_field_struct_in_struct(&struct_value, &nested_struct_field).unwrap() }; | |
| let nested_struct_schema = root_as_schema(schema.as_slice()) | |
| .unwrap() | |
| .objects() | |
| .get(nested_struct_field.type_().index() as usize); | |
| let i16_in_struct = nested_struct_schema | |
| .fields() | |
| .lookup_by_key("a", |field, key| field.key_compare_with_value(key)) | |
| .unwrap(); | |
| let value = unsafe { get_any_field_float_in_struct(&nested_struct_value, &i16_in_struct) }; | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), 5f64); | |
| } | |
| #[test] | |
| fn test_buffer_f32_in_struct_as_float_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let struct_field = get_schema_field(&schema, "pos"); | |
| let struct_value = unsafe { get_field_struct(&root_table, &struct_field).unwrap().unwrap() }; | |
| let struct_schema = root_as_schema(schema.as_slice()) | |
| .unwrap() | |
| .objects() | |
| .get(struct_field.type_().index() as usize); | |
| let f32_in_struct = struct_schema | |
| .fields() | |
| .lookup_by_key("x", |field, key| field.key_compare_with_value(key)) | |
| .unwrap(); | |
| let value = unsafe { get_any_field_float_in_struct(&struct_value, &f32_in_struct) }; | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), 1f64); | |
| } | |
| #[test] | |
| fn test_buffer_enum_in_struct_as_float_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let struct_field = get_schema_field(&schema, "pos"); | |
| let struct_value = unsafe { get_field_struct(&root_table, &struct_field).unwrap().unwrap() }; | |
| let struct_schema = root_as_schema(schema.as_slice()) | |
| .unwrap() | |
| .objects() | |
| .get(struct_field.type_().index() as usize); | |
| let enum_in_struct = struct_schema | |
| .fields() | |
| .lookup_by_key("test2", |field, key| field.key_compare_with_value(key)) | |
| .unwrap(); | |
| let value = unsafe { get_any_field_float_in_struct(&struct_value, &enum_in_struct) }; | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), 2f64); | |
| } | |
| #[test] | |
| fn test_buffer_struct_in_struct_as_float_fails() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let struct_field = get_schema_field(&schema, "pos"); | |
| let struct_value = unsafe { get_field_struct(&root_table, &struct_field).unwrap().unwrap() }; | |
| let struct_schema = root_as_schema(schema.as_slice()) | |
| .unwrap() | |
| .objects() | |
| .get(struct_field.type_().index() as usize); | |
| let struct_in_struct = struct_schema | |
| .fields() | |
| .lookup_by_key("test3", |field, key| field.key_compare_with_value(key)) | |
| .unwrap(); | |
| let value = unsafe { get_any_field_float_in_struct(&struct_value, &struct_in_struct) }; | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("f64"), String::from("Obj")) | |
| ); | |
| } | |
| #[test] | |
| fn test_buffer_i16_in_struct_as_string_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let struct_field = get_schema_field(&schema, "pos"); | |
| let struct_value = unsafe { get_field_struct(&root_table, &struct_field).unwrap().unwrap() }; | |
| let struct_schema = root_as_schema(schema.as_slice()) | |
| .unwrap() | |
| .objects() | |
| .get(struct_field.type_().index() as usize); | |
| let nested_struct_field = struct_schema | |
| .fields() | |
| .lookup_by_key("test3", |field, key| field.key_compare_with_value(key)) | |
| .unwrap(); | |
| let nested_struct_value = | |
| unsafe { get_field_struct_in_struct(&struct_value, &nested_struct_field).unwrap() }; | |
| let nested_struct_schema = root_as_schema(schema.as_slice()) | |
| .unwrap() | |
| .objects() | |
| .get(nested_struct_field.type_().index() as usize); | |
| let i16_in_struct = nested_struct_schema | |
| .fields() | |
| .lookup_by_key("a", |field, key| field.key_compare_with_value(key)) | |
| .unwrap(); | |
| let value = unsafe { | |
| get_any_field_string_in_struct( | |
| &nested_struct_value, | |
| &i16_in_struct, | |
| &root_as_schema(schema.as_slice()).unwrap(), | |
| ) | |
| }; | |
| assert_eq!(value, String::from("5")); | |
| } | |
| #[test] | |
| fn test_buffer_f32_in_struct_as_string_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let struct_field = get_schema_field(&schema, "pos"); | |
| let struct_value = unsafe { get_field_struct(&root_table, &struct_field).unwrap().unwrap() }; | |
| let struct_schema = root_as_schema(schema.as_slice()) | |
| .unwrap() | |
| .objects() | |
| .get(struct_field.type_().index() as usize); | |
| let f32_in_struct = struct_schema | |
| .fields() | |
| .lookup_by_key("x", |field, key| field.key_compare_with_value(key)) | |
| .unwrap(); | |
| let value = unsafe { | |
| get_any_field_string_in_struct( | |
| &struct_value, | |
| &f32_in_struct, | |
| &root_as_schema(schema.as_slice()).unwrap(), | |
| ) | |
| }; | |
| assert_eq!(value, String::from("1")); | |
| } | |
| #[test] | |
| fn test_buffer_enum_in_struct_as_string_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let struct_field = get_schema_field(&schema, "pos"); | |
| let struct_value = unsafe { get_field_struct(&root_table, &struct_field).unwrap().unwrap() }; | |
| let struct_schema = root_as_schema(schema.as_slice()) | |
| .unwrap() | |
| .objects() | |
| .get(struct_field.type_().index() as usize); | |
| let enum_in_struct = struct_schema | |
| .fields() | |
| .lookup_by_key("test2", |field, key| field.key_compare_with_value(key)) | |
| .unwrap(); | |
| let value = unsafe { | |
| get_any_field_string_in_struct( | |
| &struct_value, | |
| &enum_in_struct, | |
| &root_as_schema(schema.as_slice()).unwrap(), | |
| ) | |
| }; | |
| assert_eq!(value, String::from("2")); | |
| } | |
| #[test] | |
| fn test_buffer_struct_in_struct_as_string_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let root_table = unsafe { get_any_root(&buffer) }; | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let struct_field = get_schema_field(&schema, "pos"); | |
| let struct_value = unsafe { get_field_struct(&root_table, &struct_field).unwrap().unwrap() }; | |
| let struct_schema = root_as_schema(schema.as_slice()) | |
| .unwrap() | |
| .objects() | |
| .get(struct_field.type_().index() as usize); | |
| let struct_in_struct = struct_schema | |
| .fields() | |
| .lookup_by_key("test3", |field, key| field.key_compare_with_value(key)) | |
| .unwrap(); | |
| let value = unsafe { | |
| get_any_field_string_in_struct( | |
| &struct_value, | |
| &struct_in_struct, | |
| &root_as_schema(schema.as_slice()).unwrap(), | |
| ) | |
| }; | |
| assert_eq!(value, String::from("MyGame.Example.Test { a: 5, b: 6, }")); | |
| } | |
| #[test] | |
| fn test_buffer_set_valid_int_to_i16_succeeds() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let i16_field = get_schema_field(&schema, "hp"); | |
| let res = unsafe { set_any_field_integer(&mut buffer, table_loc, &i16_field, 111) }; | |
| assert!(res.is_ok()); | |
| let updated_table = unsafe { get_any_root(&buffer) }; | |
| let updated_value = unsafe { get_field_integer::<i16>(&updated_table, &i16_field) }; | |
| assert!(updated_value.is_ok()); | |
| assert_eq!(updated_value.unwrap(), Some(111)); | |
| } | |
| #[test] | |
| fn test_buffer_set_integer_to_f32_succeeds() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let f32_field = get_schema_field(&schema, "testf"); | |
| let res = unsafe { set_any_field_integer(&mut buffer, table_loc, &f32_field, 111) }; | |
| assert!(res.is_ok()); | |
| let updated_table = unsafe { get_any_root(&buffer) }; | |
| let updated_value = unsafe { get_field_float::<f32>(&updated_table, &f32_field) }; | |
| assert!(updated_value.is_ok()); | |
| assert_approx_eq!(updated_value.unwrap().unwrap(), 111f32); | |
| } | |
| #[test] | |
| fn test_buffer_set_overflow_to_i16_fails() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let i16_field = get_schema_field(&schema, "hp"); | |
| let res = unsafe { set_any_field_integer(&mut buffer, table_loc, &i16_field, 32768) }; | |
| assert!(res.is_err()); | |
| assert_eq!( | |
| res.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("i64"), String::from("Short")) | |
| ); | |
| } | |
| #[test] | |
| fn test_buffer_set_integer_to_string_fails() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let string_field = get_schema_field(&schema, "name"); | |
| let res = unsafe { set_any_field_integer(&mut buffer, table_loc, &string_field, 1) }; | |
| assert!(res.is_err()); | |
| assert_eq!(res.unwrap_err(), FlatbufferError::SetValueNotSupported); | |
| } | |
| #[test] | |
| fn test_buffer_set_integer_to_unset_fails() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let unset_field = get_schema_field(&schema, "testf3"); | |
| let res = unsafe { set_any_field_integer(&mut buffer, table_loc, &unset_field, 1) }; | |
| assert!(res.is_err()); | |
| assert_eq!(res.unwrap_err(), FlatbufferError::SetValueNotSupported); | |
| } | |
| #[test] | |
| fn test_buffer_set_valid_float_to_f32_succeeds() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let f32_field = get_schema_field(&schema, "testf"); | |
| let res = unsafe { set_any_field_float(&mut buffer, table_loc, &f32_field, 111.11) }; | |
| assert!(res.is_ok()); | |
| let updated_table = unsafe { get_any_root(&buffer) }; | |
| let updated_value = unsafe { get_field_float::<f32>(&updated_table, &f32_field) }; | |
| assert!(updated_value.is_ok()); | |
| assert_approx_eq!(updated_value.unwrap().unwrap(), 111.11); | |
| } | |
| #[test] | |
| fn test_buffer_set_float_to_i16_succeeds() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let i16_field = get_schema_field(&schema, "hp"); | |
| let res = unsafe { set_any_field_float(&mut buffer, table_loc, &i16_field, 111.11) }; | |
| assert!(res.is_ok()); | |
| let updated_table = unsafe { get_any_root(&buffer) }; | |
| let updated_value = unsafe { get_field_integer::<i16>(&updated_table, &i16_field) }; | |
| assert!(updated_value.is_ok()); | |
| assert_eq!(updated_value.unwrap(), Some(111)); | |
| } | |
| #[test] | |
| fn test_buffer_set_overflow_to_f32_fails() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let f32_field = get_schema_field(&schema, "testf"); | |
| let res = unsafe { set_any_field_float(&mut buffer, table_loc, &f32_field, f64::MAX) }; | |
| assert!(res.is_err()); | |
| assert_eq!( | |
| res.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("f64"), String::from("Float")) | |
| ); | |
| } | |
| #[test] | |
| fn test_buffer_set_float_to_string_fails() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let string_field = get_schema_field(&schema, "name"); | |
| let res = unsafe { set_any_field_float(&mut buffer, table_loc, &string_field, 1.1) }; | |
| assert!(res.is_err()); | |
| assert_eq!(res.unwrap_err(), FlatbufferError::SetValueNotSupported); | |
| } | |
| #[test] | |
| fn test_buffer_set_float_to_unset_fails() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let unset_field = get_schema_field(&schema, "testf3"); | |
| let res = unsafe { set_any_field_float(&mut buffer, table_loc, &unset_field, 1.1) }; | |
| assert!(res.is_err()); | |
| assert_eq!(res.unwrap_err(), FlatbufferError::SetValueNotSupported); | |
| } | |
| #[test] | |
| fn test_buffer_set_float_str_to_f32_succeeds() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let f32_field = get_schema_field(&schema, "testf"); | |
| let res = unsafe { set_any_field_string(&mut buffer, table_loc, &f32_field, "111.11") }; | |
| assert!(res.is_ok()); | |
| let updated_table = unsafe { get_any_root(&buffer) }; | |
| let updated_value = unsafe { get_field_float::<f32>(&updated_table, &f32_field) }; | |
| assert!(updated_value.is_ok()); | |
| assert_approx_eq!(updated_value.unwrap().unwrap(), 111.11); | |
| } | |
| #[test] | |
| fn test_buffer_set_int_str_to_i16_succeeds() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let i16_field = get_schema_field(&schema, "hp"); | |
| let res = unsafe { set_any_field_string(&mut buffer, table_loc, &i16_field, "111") }; | |
| assert!(res.is_ok()); | |
| let updated_table = unsafe { get_any_root(&buffer) }; | |
| let updated_value = unsafe { get_field_integer::<i16>(&updated_table, &i16_field) }; | |
| assert!(updated_value.is_ok()); | |
| assert_eq!(updated_value.unwrap(), Some(111)); | |
| } | |
| #[test] | |
| fn test_buffer_set_non_num_str_to_f32_fails() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let f32_field = get_schema_field(&schema, "testf"); | |
| let res = unsafe { set_any_field_string(&mut buffer, table_loc, &f32_field, "any") }; | |
| assert!(res.is_err()); | |
| assert_eq!(res.unwrap_err().to_string(), "invalid float literal"); | |
| } | |
| #[test] | |
| fn test_buffer_set_int_str_to_string_fails() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let string_field = get_schema_field(&schema, "name"); | |
| let res = unsafe { set_any_field_string(&mut buffer, table_loc, &string_field, "1") }; | |
| assert!(res.is_err()); | |
| assert_eq!(res.unwrap_err(), FlatbufferError::SetValueNotSupported); | |
| } | |
| #[test] | |
| fn test_buffer_set_int_str_to_unset_fails() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let unset_field = get_schema_field(&schema, "testf3"); | |
| let res = unsafe { set_any_field_string(&mut buffer, table_loc, &unset_field, "1") }; | |
| assert!(res.is_err()); | |
| assert_eq!(res.unwrap_err(), FlatbufferError::SetValueNotSupported); | |
| } | |
| #[test] | |
| fn test_buffer_set_i16_to_i16_succeeds() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let i16_field = get_schema_field(&schema, "hp"); | |
| let res = unsafe { set_field::<i16>(&mut buffer, table_loc, &i16_field, 111) }; | |
| assert!(res.is_ok()); | |
| let updated_table = unsafe { get_any_root(&buffer) }; | |
| let updated_value = unsafe { get_field_integer::<i16>(&updated_table, &i16_field) }; | |
| assert!(updated_value.is_ok()); | |
| assert_eq!(updated_value.unwrap(), Some(111)); | |
| } | |
| #[test] | |
| fn test_buffer_set_i32_to_i16_fails() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let i16_field = get_schema_field(&schema, "hp"); | |
| let res = unsafe { set_field::<i32>(&mut buffer, table_loc, &i16_field, 111) }; | |
| assert!(res.is_err()); | |
| assert_eq!( | |
| res.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("i32"), String::from("Short")) | |
| ); | |
| } | |
| #[test] | |
| fn test_buffer_set_f32_to_f32_succeeds() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let f32_field = get_schema_field(&schema, "testf"); | |
| let res = unsafe { set_field::<f32>(&mut buffer, table_loc, &f32_field, 111.11) }; | |
| assert!(res.is_ok()); | |
| let updated_table = unsafe { get_any_root(&buffer) }; | |
| let updated_value = unsafe { get_field_float::<f32>(&updated_table, &f32_field) }; | |
| assert!(updated_value.is_ok()); | |
| assert_approx_eq!(updated_value.unwrap().unwrap(), 111.11); | |
| } | |
| #[test] | |
| fn test_buffer_set_f64_to_f32_fails() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let f32_field = get_schema_field(&schema, "testf"); | |
| let res = unsafe { set_field::<f64>(&mut buffer, table_loc, &f32_field, 111.11) }; | |
| assert!(res.is_err()); | |
| assert_eq!( | |
| res.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("f64"), String::from("Float")) | |
| ); | |
| } | |
| #[test] | |
| fn test_buffer_set_f32_to_f32_unset_fails() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let unset_field = get_schema_field(&schema, "testf3"); | |
| let res = unsafe { set_field::<f32>(&mut buffer, table_loc, &unset_field, 111.11) }; | |
| assert!(res.is_err()); | |
| assert_eq!(res.unwrap_err(), FlatbufferError::SetValueNotSupported); | |
| } | |
| #[test] | |
| fn test_buffer_set_string_same_str_succeeds() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let string_field = get_schema_field(&schema, "name"); | |
| let res = unsafe { | |
| set_string( | |
| &mut buffer, | |
| table_loc, | |
| &string_field, | |
| "MyMonster", | |
| &root_as_schema(schema.as_slice()).unwrap(), | |
| ) | |
| }; | |
| assert!(res.is_ok()); | |
| let updated_table = unsafe { get_any_root(&buffer) }; | |
| let updated_value = unsafe { get_field_string(&updated_table, &string_field) }; | |
| assert!(updated_value.is_ok()); | |
| assert_eq!(updated_value.unwrap(), Some("MyMonster")); | |
| } | |
| #[test] | |
| fn test_buffer_set_string_same_size_succeeds() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let string_field = get_schema_field(&schema, "name"); | |
| let res = unsafe { | |
| set_string( | |
| &mut buffer, | |
| table_loc, | |
| &string_field, | |
| "YoMonster", | |
| &root_as_schema(schema.as_slice()).unwrap(), | |
| ) | |
| }; | |
| assert!(res.is_ok()); | |
| let updated_table = unsafe { get_any_root(&buffer) }; | |
| let updated_value = unsafe { get_field_string(&updated_table, &string_field) }; | |
| assert!(updated_value.is_ok()); | |
| assert_eq!(updated_value.unwrap(), Some("YoMonster")); | |
| } | |
| #[test] | |
| fn test_buffer_set_string_bigger_size_succeeds() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let string_field = get_schema_field(&schema, "name"); | |
| let res = unsafe { | |
| set_string( | |
| &mut buffer, | |
| table_loc, | |
| &string_field, | |
| "AStringWithSlightlyBiggerSize", | |
| &root_as_schema(schema.as_slice()).unwrap(), | |
| ) | |
| }; | |
| assert!(res.is_ok()); | |
| let updated_table = unsafe { get_any_root(&buffer) }; | |
| let updated_value = unsafe { get_field_string(&updated_table, &string_field) }; | |
| assert!(updated_value.is_ok()); | |
| assert_eq!(updated_value.unwrap(), Some("AStringWithSlightlyBiggerSize")); | |
| } | |
| #[test] | |
| fn test_buffer_set_string_smaller_size_succeeds() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let string_field = get_schema_field(&schema, "name"); | |
| let res = unsafe { | |
| set_string( | |
| &mut buffer, | |
| table_loc, | |
| &string_field, | |
| "s", | |
| &root_as_schema(schema.as_slice()).unwrap(), | |
| ) | |
| }; | |
| assert!(res.is_ok()); | |
| let updated_table = unsafe { get_any_root(&buffer) }; | |
| let updated_value = unsafe { get_field_string(&updated_table, &string_field) }; | |
| assert!(updated_value.is_ok()); | |
| assert_eq!(updated_value.unwrap(), Some("s")); | |
| } | |
| #[test] | |
| fn test_buffer_set_string_diff_type_fails() { | |
| let mut buffer = create_test_buffer(); | |
| let table_loc = unsafe { get_any_root(&buffer) }.loc(); | |
| let schema = load_file_as_buffer("../monster_test.bfbs"); | |
| let f32_field = get_schema_field(&schema, "testf"); | |
| let res = unsafe { | |
| set_string( | |
| &mut buffer, | |
| table_loc, | |
| &f32_field, | |
| "any", | |
| &root_as_schema(schema.as_slice()).unwrap(), | |
| ) | |
| }; | |
| assert!(res.is_err()); | |
| assert_eq!( | |
| res.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("String"), String::from("Float")) | |
| ); | |
| } | |
| #[test] | |
| fn test_create_safe_buffer_default_options_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema); | |
| assert!(safe_buffer.is_ok()); | |
| } | |
| #[test] | |
| fn test_create_safe_buffer_limit_max_depth_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let verify_options = VerifierOptions { max_depth: 1, ..Default::default() }; | |
| let safe_buffer = SafeBuffer::new_with_options(&buffer, &schema, &verify_options); | |
| assert!(safe_buffer.is_err()); | |
| assert!(format!("{:#?}", safe_buffer.err().unwrap()).contains("DepthLimitReached")); | |
| } | |
| #[test] | |
| fn test_create_safe_buffer_limit_max_table_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let verify_options = VerifierOptions { max_tables: 1, ..Default::default() }; | |
| let safe_buffer = SafeBuffer::new_with_options(&buffer, &schema, &verify_options); | |
| assert!(safe_buffer.is_err()); | |
| assert!(format!("{:#?}", safe_buffer.err().unwrap()).contains("TooManyTables")); | |
| } | |
| #[test] | |
| fn test_create_safe_buffer_limit_max_size_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let verify_options = VerifierOptions { max_apparent_size: 1 << 6, ..Default::default() }; | |
| let safe_buffer = SafeBuffer::new_with_options(&buffer, &schema, &verify_options); | |
| assert!(safe_buffer.is_err()); | |
| assert!(format!("{:#?}", safe_buffer.err().unwrap()).contains("ApparentSizeTooLarge")); | |
| } | |
| #[test] | |
| fn test_safe_buffer_integer_same_type_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_field_integer::<i16>("hp"); | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), Some(32767)); | |
| } | |
| #[test] | |
| fn test_safe_buffer_integer_invalid_field_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_field_integer::<i16>("nonexistent"); | |
| assert!(value.is_err()); | |
| assert_eq!(value.unwrap_err(), FlatbufferError::FieldNotFound); | |
| } | |
| #[test] | |
| fn test_safe_buffer_integer_diff_size_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_field_integer::<i64>("hp"); | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("i64"), String::from("Short")) | |
| ); | |
| } | |
| #[test] | |
| fn test_safe_buffer_float_same_type_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_field_float::<f32>("testf"); | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), Some(3.14)); | |
| } | |
| #[test] | |
| fn test_safe_buffer_float_invalid_field_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_field_float::<f32>("non_existent"); | |
| assert!(value.is_err()); | |
| assert_eq!(value.unwrap_err(), FlatbufferError::FieldNotFound); | |
| } | |
| #[test] | |
| fn test_safe_buffer_float_diff_type_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_field_float::<f64>("testf"); | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("f64"), String::from("Float")) | |
| ); | |
| } | |
| #[test] | |
| fn test_safe_buffer_string_same_type_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_field_string("name"); | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), Some("MyMonster")); | |
| } | |
| #[test] | |
| fn test_safe_buffer_string_invalid_field_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_field_string("non_existent"); | |
| assert!(value.is_err()); | |
| assert_eq!(value.unwrap_err(), FlatbufferError::FieldNotFound); | |
| } | |
| #[test] | |
| fn test_safe_buffer_string_diff_type_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_field_string("testf"); | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("String"), String::from("Float")) | |
| ); | |
| } | |
| #[test] | |
| fn test_safe_buffer_struct_same_type_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_field_struct("pos"); | |
| assert!(value.is_ok()); | |
| let optional_value = value.unwrap(); | |
| assert!(optional_value.is_some()); | |
| let safe_struct = optional_value.unwrap(); | |
| assert!(safe_struct.get_field_struct("test3").is_ok()); | |
| } | |
| #[test] | |
| fn test_safe_buffer_struct_invalid_field_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_field_struct("non_existent"); | |
| assert!(value.is_err()); | |
| assert_eq!(value.unwrap_err(), FlatbufferError::FieldNotFound); | |
| } | |
| #[test] | |
| fn test_safe_buffer_struct_diff_type_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_field_struct("testf"); | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("Obj"), String::from("Float")) | |
| ); | |
| } | |
| #[test] | |
| fn test_safe_buffer_vector_same_type_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_field_vector::<u8>("inventory"); | |
| assert!(value.is_ok()); | |
| let optional_vector = value.unwrap(); | |
| assert!(optional_vector.is_some()); | |
| let vector = optional_vector.unwrap(); | |
| assert_eq!(vector.len(), 5); | |
| assert_eq!(vector.get(0), 0); | |
| assert_eq!(vector.get(1), 1); | |
| assert_eq!(vector.get(2), 2); | |
| assert_eq!(vector.get(3), 3); | |
| assert_eq!(vector.get(4), 4); | |
| } | |
| #[test] | |
| fn test_safe_buffer_vector_invalid_field_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_field_vector::<u8>("non_existent"); | |
| assert!(value.is_err()); | |
| assert_eq!(value.unwrap_err(), FlatbufferError::FieldNotFound); | |
| } | |
| #[test] | |
| fn test_safe_buffer_vector_diff_type_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_field_vector::<u8>("testf"); | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("u8"), String::from("Float")) | |
| ); | |
| } | |
| #[test] | |
| fn test_safe_buffer_table_same_type_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_field_table("testempty"); | |
| assert!(value.is_ok()); | |
| let optional_nested_table = value.unwrap(); | |
| assert!(optional_nested_table.is_some()); | |
| let nested_table = optional_nested_table.unwrap(); | |
| let nested_field_value = nested_table.get_field_string("id"); | |
| assert!(nested_field_value.is_ok()); | |
| assert_eq!(nested_field_value.unwrap(), Some("Fred")); | |
| } | |
| #[test] | |
| fn test_safe_buffer_table_invalid_field_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_field_table("non_existent"); | |
| assert!(value.is_err()); | |
| assert_eq!(value.unwrap_err(), FlatbufferError::FieldNotFound); | |
| } | |
| #[test] | |
| fn test_safe_buffer_table_diff_type_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_field_table("testf"); | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("Obj"), String::from("Float")) | |
| ); | |
| } | |
| #[test] | |
| fn test_safe_buffer_i16_as_integer_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_any_field_integer("hp"); | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), 32767); | |
| } | |
| #[test] | |
| fn test_safe_buffer_i16_as_integer_invalid_field_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_any_field_integer("non_existent"); | |
| assert!(value.is_err()); | |
| assert_eq!(value.unwrap_err(), FlatbufferError::FieldNotFound); | |
| } | |
| #[test] | |
| fn test_safe_buffer_string_as_integer_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_any_field_integer("name"); | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("i64"), String::from("String")) | |
| ); | |
| } | |
| #[test] | |
| fn test_safe_buffer_i16_as_float_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_any_field_float("hp"); | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), 32767f64); | |
| } | |
| #[test] | |
| fn test_safe_buffer_i16_as_float_invalid_field_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_any_field_float("non_existent"); | |
| assert!(value.is_err()); | |
| assert_eq!(value.unwrap_err(), FlatbufferError::FieldNotFound); | |
| } | |
| #[test] | |
| fn test_safe_buffer_string_as_float_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_any_field_float("name"); | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("f64"), String::from("String")) | |
| ); | |
| } | |
| #[test] | |
| fn test_safe_buffer_string_as_string_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_any_field_string("name"); | |
| assert!(value.is_ok()); | |
| assert_eq!(value.unwrap(), "MyMonster"); | |
| } | |
| #[test] | |
| fn test_safe_buffer_i16_as_string_invalid_field_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let value = root_table.get_any_field_string("non_existent"); | |
| assert!(value.is_err()); | |
| assert_eq!(value.unwrap_err(), FlatbufferError::FieldNotFound); | |
| } | |
| #[test] | |
| fn test_safe_buffer_nested_struct_same_type_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let struct_value = root_table.get_field_struct("pos").unwrap().unwrap(); | |
| let nested_struct_value = struct_value.get_field_struct("test3"); | |
| assert!(nested_struct_value.is_ok()); | |
| let value = nested_struct_value.unwrap(); | |
| assert!(value.get_any_field_integer("a").is_ok()); | |
| } | |
| #[test] | |
| fn test_safe_buffer_nested_struct_invalid_field_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let struct_value = root_table.get_field_struct("pos").unwrap().unwrap(); | |
| let value = struct_value.get_field_struct("non_existent"); | |
| assert!(value.is_err()); | |
| assert_eq!(value.unwrap_err(), FlatbufferError::FieldNotFound); | |
| } | |
| #[test] | |
| fn test_safe_buffer_nested_struct_diff_type_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let struct_value = root_table.get_field_struct("pos").unwrap().unwrap(); | |
| let nested_struct_value = struct_value.get_field_struct("x"); | |
| assert!(nested_struct_value.is_err()); | |
| assert_eq!( | |
| nested_struct_value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("Obj"), String::from("Float")) | |
| ); | |
| } | |
| #[test] | |
| fn test_safe_buffer_enum_in_struct_as_integer_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let struct_value = root_table.get_field_struct("pos").unwrap().unwrap(); | |
| let nested_enum_value = struct_value.get_any_field_integer("test2"); | |
| assert!(nested_enum_value.is_ok()); | |
| assert_eq!(nested_enum_value.unwrap(), 2); | |
| } | |
| #[test] | |
| fn test_safe_buffer_enum_in_struct_as_integer_invalid_field_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let struct_value = root_table.get_field_struct("pos").unwrap().unwrap(); | |
| let value = struct_value.get_any_field_integer("non_existent"); | |
| assert!(value.is_err()); | |
| assert_eq!(value.unwrap_err(), FlatbufferError::FieldNotFound); | |
| } | |
| #[test] | |
| fn test_safe_buffer_struct_in_struct_as_integer_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let struct_value = root_table.get_field_struct("pos").unwrap().unwrap(); | |
| let nested_struct_value = struct_value.get_any_field_integer("test3"); | |
| assert!(nested_struct_value.is_err()); | |
| assert_eq!( | |
| nested_struct_value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("i64"), String::from("Obj")) | |
| ); | |
| } | |
| #[test] | |
| fn test_safe_buffer_i16_in_struct_as_float_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let struct_value = root_table.get_field_struct("pos").unwrap().unwrap(); | |
| let nested_struct_value = struct_value.get_field_struct("test3").unwrap(); | |
| let nested_i16_value = nested_struct_value.get_any_field_float("a"); | |
| assert!(nested_i16_value.is_ok()); | |
| assert_eq!(nested_i16_value.unwrap(), 5f64); | |
| } | |
| #[test] | |
| fn test_safe_buffer_child_in_struct_as_float_invalid_field_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let struct_value = root_table.get_field_struct("pos").unwrap().unwrap(); | |
| let value = struct_value.get_any_field_float("non_existent"); | |
| assert!(value.is_err()); | |
| assert_eq!(value.unwrap_err(), FlatbufferError::FieldNotFound); | |
| } | |
| #[test] | |
| fn test_safe_buffer_struct_in_struct_as_float_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let struct_value = root_table.get_field_struct("pos").unwrap().unwrap(); | |
| let value = struct_value.get_any_field_float("test3"); | |
| assert!(value.is_err()); | |
| assert_eq!( | |
| value.unwrap_err(), | |
| FlatbufferError::FieldTypeMismatch(String::from("f64"), String::from("Obj")) | |
| ); | |
| } | |
| #[test] | |
| fn test_safe_buffer_f32_in_struct_as_string_succeeds() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let struct_value = root_table.get_field_struct("pos").unwrap().unwrap(); | |
| let nested_f32_value = struct_value.get_any_field_string("x"); | |
| assert!(nested_f32_value.is_ok()); | |
| assert_eq!(nested_f32_value.unwrap(), String::from("1")); | |
| } | |
| #[test] | |
| fn test_safe_buffer_child_in_struct_as_string_invalid_field_fails() { | |
| let buffer = create_test_buffer(); | |
| let schema_buffer = load_file_as_buffer("../monster_test.bfbs"); | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let safe_buffer = SafeBuffer::new(&buffer, &schema).unwrap(); | |
| let root_table = safe_buffer.get_root(); | |
| let struct_value = root_table.get_field_struct("pos").unwrap().unwrap(); | |
| let value = struct_value.get_any_field_string("non_existent"); | |
| assert!(value.is_err()); | |
| assert_eq!(value.unwrap_err(), FlatbufferError::FieldNotFound); | |
| } | |
| fn load_file_as_buffer(path: &str) -> Vec<u8> { | |
| std::fs::read(path).unwrap() | |
| } | |
| fn get_schema_field<'a>(schema_buffer: &'a Vec<u8>, field_name: &'a str) -> Field<'a> { | |
| let schema = root_as_schema(schema_buffer.as_slice()).unwrap(); | |
| let root_table = schema.root_table().unwrap(); | |
| let fields = root_table.fields(); | |
| fields.lookup_by_key(field_name, |field, key| field.key_compare_with_value(key)).unwrap() | |
| } | |
| fn create_test_buffer() -> Vec<u8> { | |
| let mut builder = FlatBufferBuilder::new(); | |
| let mon = { | |
| let s0 = builder.create_string("test1"); | |
| let s1 = builder.create_string("test2"); | |
| let fred_name = builder.create_string("Fred"); | |
| // can't inline creation of this Vec3 because we refer to it by reference, so it must live | |
| // long enough to be used by MonsterArgs. | |
| let pos = my_game::example::Vec3::new( | |
| 1.0, | |
| 2.0, | |
| 3.0, | |
| 3.0, | |
| my_game::example::Color::Green, | |
| &my_game::example::Test::new(5i16, 6i8), | |
| ); | |
| let args = my_game::example::MonsterArgs { | |
| hp: 32767, | |
| testf: 3.14, | |
| mana: 150, | |
| name: Some(builder.create_string("MyMonster")), | |
| pos: Some(&pos), | |
| test_type: my_game::example::Any::Monster, | |
| testempty: Some(my_game::example::Stat::create( | |
| &mut builder, | |
| &my_game::example::StatArgs { id: Some(fred_name), ..Default::default() }, | |
| )), | |
| test: Some( | |
| my_game::example::Monster::create( | |
| &mut builder, | |
| &my_game::example::MonsterArgs { name: Some(fred_name), ..Default::default() }, | |
| ) | |
| .as_union_value(), | |
| ), | |
| inventory: Some(builder.create_vector(&[0u8, 1, 2, 3, 4])), | |
| test4: Some(builder.create_vector(&[ | |
| my_game::example::Test::new(10, 20), | |
| my_game::example::Test::new(30, 40), | |
| ])), | |
| testarrayofstring: Some(builder.create_vector(&[s0, s1])), | |
| ..Default::default() | |
| }; | |
| my_game::example::Monster::create(&mut builder, &args) | |
| }; | |
| my_game::example::finish_monster_buffer(&mut builder, mon); | |
| let (flatbuf, start_loc) = builder.mut_finished_buffer(); | |
| flatbuf[start_loc..].to_vec() | |
| } |