use std::io::{Read, Seek, Write};
use byteorder::{LittleEndian, ReadBytesExt, WriteBytesExt};
use crate::header::mode::Mode;
use crate::header::Header;
use crate::RefPackResult;
pub struct Reference {
_private: (),
}
impl Mode for Reference {
const LENGTH: usize = 4;
fn read<R: Read + Seek>(reader: &mut R) -> RefPackResult<Header> {
let decompressed_length = reader.read_u32::<LittleEndian>()?;
Ok(Header {
decompressed_length,
compressed_length: None,
})
}
fn write<W: Write + Seek>(header: Header, writer: &mut W) -> RefPackResult<()> {
writer.write_u32::<LittleEndian>(header.decompressed_length)?;
Ok(())
}
}
#[cfg(test)]
mod test {
use std::io::Cursor;
use proptest::prop_assert_eq;
use test_strategy::proptest;
use super::*;
use crate::header::Header;
#[proptest]
fn symmetrical_read_write(header: Header) {
let expected = Header {
decompressed_length: header.decompressed_length,
compressed_length: None,
};
let mut write_buf = vec![];
let mut write_cur = Cursor::new(&mut write_buf);
header.write::<Reference>(&mut write_cur).unwrap();
let mut read_cur = Cursor::new(&mut write_buf);
let got = Header::read::<Reference>(&mut read_cur).unwrap();
prop_assert_eq!(expected, got);
}
}