use super::*;
impl<N: Network> FromBytes for FinalizeOperation<N> {
fn read_le<R: Read>(mut reader: R) -> IoResult<Self> {
let variant = u8::read_le(&mut reader)?;
match variant {
0 => {
let mapping_id = Field::read_le(&mut reader)?;
Ok(Self::InitializeMapping(mapping_id))
}
1 => {
let mapping_id = Field::read_le(&mut reader)?;
let key_id = Field::read_le(&mut reader)?;
let value_id = Field::read_le(&mut reader)?;
Ok(Self::InsertKeyValue(mapping_id, key_id, value_id))
}
2 => {
let mapping_id = Field::read_le(&mut reader)?;
let key_id = Field::read_le(&mut reader)?;
let value_id = Field::read_le(&mut reader)?;
Ok(Self::UpdateKeyValue(mapping_id, key_id, value_id))
}
3 => {
let mapping_id = Field::read_le(&mut reader)?;
let key_id = Field::read_le(&mut reader)?;
Ok(Self::RemoveKeyValue(mapping_id, key_id))
}
4 => {
let mapping_id = Field::read_le(&mut reader)?;
Ok(Self::ReplaceMapping(mapping_id))
}
5 => {
let mapping_id = Field::read_le(&mut reader)?;
Ok(Self::RemoveMapping(mapping_id))
}
6.. => Err(error(format!("Failed to decode finalize operation variant {variant}"))),
}
}
}
impl<N: Network> ToBytes for FinalizeOperation<N> {
fn write_le<W: Write>(&self, mut writer: W) -> IoResult<()> {
match self {
Self::InitializeMapping(mapping_id) => {
0u8.write_le(&mut writer)?;
mapping_id.write_le(&mut writer)?;
}
Self::InsertKeyValue(mapping_id, key_id, value_id) => {
1u8.write_le(&mut writer)?;
mapping_id.write_le(&mut writer)?;
key_id.write_le(&mut writer)?;
value_id.write_le(&mut writer)?;
}
Self::UpdateKeyValue(mapping_id, key_id, value_id) => {
2u8.write_le(&mut writer)?;
mapping_id.write_le(&mut writer)?;
key_id.write_le(&mut writer)?;
value_id.write_le(&mut writer)?;
}
Self::RemoveKeyValue(mapping_id, key_id) => {
3u8.write_le(&mut writer)?;
mapping_id.write_le(&mut writer)?;
key_id.write_le(&mut writer)?;
}
Self::ReplaceMapping(mapping_id) => {
4u8.write_le(&mut writer)?;
mapping_id.write_le(&mut writer)?;
}
Self::RemoveMapping(mapping_id) => {
5u8.write_le(&mut writer)?;
mapping_id.write_le(&mut writer)?;
}
}
Ok(())
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_bytes() {
for expected in crate::logic::finalize_operation::test_helpers::sample_finalize_operations() {
let expected_bytes = expected.to_bytes_le().unwrap();
assert_eq!(expected, FinalizeOperation::read_le(&expected_bytes[..]).unwrap());
}
}
}