| 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() |
| } |