use super::*;
impl<N: Network> FromBytes for Origin<N> {
fn read_le<R: Read>(mut reader: R) -> IoResult<Self> {
let variant = Variant::read_le(&mut reader)?;
let origin = match variant {
0 => {
let commitment: Field<N> = FromBytes::read_le(&mut reader)?;
Self::Commitment(commitment)
}
1 => {
let state_root: N::StateRoot = FromBytes::read_le(&mut reader)?;
Self::StateRoot(state_root)
}
2.. => return Err(error(format!("Failed to decode transition origin variant {variant}"))),
};
Ok(origin)
}
}
impl<N: Network> ToBytes for Origin<N> {
fn write_le<W: Write>(&self, mut writer: W) -> IoResult<()> {
match self {
Self::Commitment(commitment) => {
(0 as Variant).write_le(&mut writer)?;
commitment.write_le(&mut writer)
}
Self::StateRoot(state_root) => {
(1 as Variant).write_le(&mut writer)?;
state_root.write_le(&mut writer)
}
}
}
}
#[cfg(test)]
mod tests {
use super::*;
use console::network::Testnet3;
type CurrentNetwork = Testnet3;
const ITERATIONS: u32 = 1000;
fn check_bytes(expected: Origin<CurrentNetwork>) -> Result<()> {
let expected_bytes = expected.to_bytes_le()?;
assert_eq!(expected, Origin::read_le(&expected_bytes[..])?);
Ok(())
}
#[test]
fn test_bytes() -> Result<()> {
let rng = &mut test_rng();
for _ in 0..ITERATIONS {
check_bytes(Origin::<CurrentNetwork>::Commitment(Uniform::rand(rng)))?;
check_bytes(Origin::<CurrentNetwork>::StateRoot(Uniform::rand(rng)))?;
}
Ok(())
}
}