1
0
Fork 0
mirror of https://codeberg.org/demostf/parser.git synced 2026-06-03 10:14:06 +02:00

gameeventlist reencode tests

This commit is contained in:
Robin Appelman 2021-07-22 13:27:20 +02:00
commit 07d675a52d
2 changed files with 37 additions and 0 deletions

Binary file not shown.

View file

@ -0,0 +1,37 @@
use bitbuffer::{BitReadBuffer, BitReadStream, BitWriteStream, LittleEndian};
use std::fs;
use test_case::test_case;
use tf_demo_parser::demo::message::Message;
use tf_demo_parser::demo::parser::Encode;
use tf_demo_parser::{MessageType, ParserState};
#[test_case("test_data/messages/game_event_list.bin", MessageType::GameEventList; "game_event_list")]
fn message_reencode(input_file: &str, ty: MessageType) {
let data = fs::read(input_file).unwrap();
let state = ParserState::new(|_| false, true);
let mut stream = BitReadStream::new(BitReadBuffer::new(&data, LittleEndian));
let parsed = Message::from_type(ty, &mut stream, &state).unwrap();
let mut out = Vec::with_capacity(data.len());
let written_bits = {
let mut write = BitWriteStream::new(&mut out, LittleEndian);
parsed.encode(&mut write, &state).unwrap();
write.bit_len()
};
let mut re_stream = BitReadStream::new(BitReadBuffer::new(&out, LittleEndian));
let re_parsed = Message::from_type(ty, &mut re_stream, &state).unwrap();
assert_eq!(written_bits, stream.pos());
assert_eq!(parsed, re_parsed);
assert_eq!(data.len(), out.len());
if data.len() > 16 {
pretty_assertions::assert_eq!(&data[0..data.len() - 8], &out[0..out.len() - 8]);
pretty_assertions::assert_eq!(&data[data.len() - 8..], &out[out.len() - 8..]);
} else {
pretty_assertions::assert_eq!(data, out);
}
}