use crate::{
metadata::tables::{
methodptr::MethodPtrRaw,
types::{RowWritable, TableId, TableInfoRef},
},
utils::write_le_at_dyn,
Result,
};
impl RowWritable for MethodPtrRaw {
fn row_write(
&self,
data: &mut [u8],
offset: &mut usize,
_rid: u32,
sizes: &TableInfoRef,
) -> Result<()> {
write_le_at_dyn(
data,
offset,
self.method,
sizes.is_large(TableId::MethodDef),
)?;
Ok(())
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::metadata::{
tables::types::{RowReadable, TableId, TableInfo, TableRow},
token::Token,
};
#[test]
fn test_round_trip_serialization_short() {
let original_row = MethodPtrRaw {
rid: 1,
token: Token::new(0x05000001),
offset: 0,
method: 0x0101,
};
let table_info = std::sync::Arc::new(TableInfo::new_test(
&[(TableId::MethodDef, 1)],
false,
false,
false,
));
let row_size = <MethodPtrRaw as TableRow>::row_size(&table_info) as usize;
let mut buffer = vec![0u8; row_size];
let mut offset = 0;
original_row
.row_write(&mut buffer, &mut offset, 1, &table_info)
.expect("Serialization should succeed");
let mut read_offset = 0;
let deserialized_row = MethodPtrRaw::row_read(&buffer, &mut read_offset, 1, &table_info)
.expect("Deserialization should succeed");
assert_eq!(original_row.method, deserialized_row.method);
assert_eq!(offset, row_size, "Offset should match expected row size");
}
#[test]
fn test_round_trip_serialization_long() {
let original_row = MethodPtrRaw {
rid: 1,
token: Token::new(0x05000001),
offset: 0,
method: 0x01010101,
};
let table_info = std::sync::Arc::new(TableInfo::new_test(
&[(TableId::MethodDef, u16::MAX as u32 + 3)],
true,
true,
true,
));
let row_size = <MethodPtrRaw as TableRow>::row_size(&table_info) as usize;
let mut buffer = vec![0u8; row_size];
let mut offset = 0;
original_row
.row_write(&mut buffer, &mut offset, 1, &table_info)
.expect("Serialization should succeed");
let mut read_offset = 0;
let deserialized_row = MethodPtrRaw::row_read(&buffer, &mut read_offset, 1, &table_info)
.expect("Deserialization should succeed");
assert_eq!(original_row.method, deserialized_row.method);
assert_eq!(offset, row_size, "Offset should match expected row size");
}
#[test]
fn test_known_binary_format_short() {
let expected_data = vec![
0x01, 0x01, ];
let row = MethodPtrRaw {
rid: 1,
token: Token::new(0x05000001),
offset: 0,
method: 0x0101,
};
let table_info = std::sync::Arc::new(TableInfo::new_test(
&[(TableId::MethodDef, 1)],
false,
false,
false,
));
let row_size = <MethodPtrRaw as TableRow>::row_size(&table_info) as usize;
let mut buffer = vec![0u8; row_size];
let mut offset = 0;
row.row_write(&mut buffer, &mut offset, 1, &table_info)
.expect("Serialization should succeed");
assert_eq!(
buffer, expected_data,
"Generated binary should match expected format"
);
assert_eq!(
offset,
expected_data.len(),
"Offset should match data length"
);
}
#[test]
fn test_known_binary_format_long() {
let expected_data = vec![
0x01, 0x01, 0x01, 0x01, ];
let row = MethodPtrRaw {
rid: 1,
token: Token::new(0x05000001),
offset: 0,
method: 0x01010101,
};
let table_info = std::sync::Arc::new(TableInfo::new_test(
&[(TableId::MethodDef, u16::MAX as u32 + 3)],
true,
true,
true,
));
let row_size = <MethodPtrRaw as TableRow>::row_size(&table_info) as usize;
let mut buffer = vec![0u8; row_size];
let mut offset = 0;
row.row_write(&mut buffer, &mut offset, 1, &table_info)
.expect("Serialization should succeed");
assert_eq!(
buffer, expected_data,
"Generated binary should match expected format"
);
assert_eq!(
offset,
expected_data.len(),
"Offset should match data length"
);
}
}