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

adjust tests

Browse source
This commit is contained in:
Robin Appelman 2021-07-22 14:58:43 +02:00
commit 00596f946f
8 changed files with 2544 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

27
tests/message_reencode.rs
View file

@ -2,30 +2,36 @@ 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};
use tf_demo_parser::demo::packet::Packet;
use tf_demo_parser::demo::parser::{DemoHandler, Encode, NullHandler};
use tf_demo_parser::{MessageType, Parse};
#[test_case("test_data/messages/game_event_list.bin", MessageType::GameEventList; "game_event_list")]
// #[test_case("test_data/messages/packet_entities.bin", MessageType::PacketEntities; "packet_entities")]
fn message_reencode(input_file: &str, ty: MessageType) {
let data = fs::read(input_file).unwrap();
let state = ParserState::new(|_| false, true);
let mut handler = DemoHandler::parse_all_with_analyser(NullHandler);
setup(&mut handler, "test_data/messages/setup_data_tables.bin");
setup(&mut handler, "test_data/messages/setup_string_tables.bin");
let state = &handler.state_handler;
let mut stream = BitReadStream::new(BitReadBuffer::new(&data, LittleEndian));
let parsed = Message::from_type(ty, &mut stream, &state).unwrap();
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();
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();
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!(written_bits, stream.pos());
assert_eq!(data.len(), out.len());
if data.len() > 16 {
@ -35,3 +41,10 @@ fn message_reencode(input_file: &str, ty: MessageType) {
pretty_assertions::assert_eq!(data, out);
}
}
fn setup(handler: &mut DemoHandler<NullHandler>, input: &str) {
let data = fs::read(input).unwrap();
let mut stream = BitReadStream::new(BitReadBuffer::new_owned(data, LittleEndian));
let packet = Packet::parse(&mut stream, &handler.state_handler).unwrap();
handler.handle_packet(packet).unwrap();
}