use super::kind::PacketType;
use super::PayloadTrait;
#[derive(Debug, Eq, PartialEq, Nom, Serialize)]
#[nom(BigEndian)]
#[cfg_attr(test, derive(test_strategy::Arbitrary))]
pub struct Version {
pub version: u32,
#[serde(serialize_with = "crate::util::vec_u8_as_slice")]
pub extension_data: Vec<u8>
}
impl PayloadTrait for Version {
const Type: PacketType = PacketType::Version;
fn binsize(&self) -> u32 {
4 + self.extension_data.len() as u32
}
}
impl From<Version> for super::Payload {
fn from(p: Version) -> Self {
Self::Version(p)
}
}
#[cfg(test)]
mod tests {
use test_strategy::proptest;
use crate::parser::encode;
use crate::parser::Parser;
use super::*;
#[proptest]
fn roundtrip_whole(input: Version) {
let mut stream = Parser::default();
let packet = input.into_packet();
stream.write(&encode(&packet)).unwrap();
assert_eq!(stream.get_packet(), Ok(Some(packet)));
assert_eq!(stream.get_packet(), Ok(None));
}
}