use super::*;
impl<N: Network> FromBytes for Authorization<N> {
fn read_le<R: Read>(mut reader: R) -> IoResult<Self> {
let version = u8::read_le(&mut reader)?;
if version != 1 {
return Err(error("Invalid authorization version"));
}
let num_requests = u8::read_le(&mut reader)?;
if num_requests == 0 {
return Err(error("Authorization (from 'read_le') has no requests"));
}
let requests = (0..num_requests).map(|_| Request::read_le(&mut reader)).collect::<IoResult<Vec<_>>>()?;
let num_transitions = u8::read_le(&mut reader)?;
if num_transitions == 0 {
return Err(error("Authorization (from 'read_le') has no transitions"));
}
let transitions =
(0..num_transitions).map(|_| Transition::read_le(&mut reader)).collect::<IoResult<Vec<_>>>()?;
Self::try_from((requests, transitions)).map_err(error)
}
}
impl<N: Network> ToBytes for Authorization<N> {
fn write_le<W: Write>(&self, mut writer: W) -> IoResult<()> {
let requests = self.requests.read();
let transitions = self.transitions.read();
1u8.write_le(&mut writer)?;
u8::try_from(requests.len()).map_err(error)?.write_le(&mut writer)?;
requests.iter().try_for_each(|request| request.write_le(&mut writer))?;
u8::try_from(transitions.len()).map_err(error)?.write_le(&mut writer)?;
transitions.values().try_for_each(|transition| transition.write_le(&mut writer))
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_bytes() -> Result<()> {
let rng = &mut TestRng::default();
let expected = crate::stack::authorization::test_helpers::sample_authorization(rng);
let expected_bytes = expected.to_bytes_le()?;
assert_eq!(expected, Authorization::read_le(&expected_bytes[..])?);
Ok(())
}
}