From 2b0103aff4065c241914746002008eeeb9c880a9 Mon Sep 17 00:00:00 2001 From: Max Inden Date: Wed, 25 Sep 2024 20:18:15 +0200 Subject: [PATCH] Fix build_two --- neqo-transport/src/packet/mod.rs | 29 ++++++++++++++--------------- 1 file changed, 14 insertions(+), 15 deletions(-) diff --git a/neqo-transport/src/packet/mod.rs b/neqo-transport/src/packet/mod.rs index aded640c2..e28a9f164 100644 --- a/neqo-transport/src/packet/mod.rs +++ b/neqo-transport/src/packet/mod.rs @@ -1181,21 +1181,20 @@ mod tests { builder.encode(&[0; 3]); let encoder = builder.build(&mut prot).expect("build"); assert_eq!(encoder.len(), 45); - // TODO - // let first = encoder.clone(); - - // // TODO: 0 ideal here? - // let mut builder = - // PacketBuilder::short(encoder, false, Some(ConnectionId::from(SERVER_CID)), 0); - // builder.pn(1, 3); - // builder.encode(&[0]); // Minimal size (packet number is big enough). - // let encoder = builder.build(&mut prot).expect("build"); - // assert_eq!( - // first.as_ref(), - // &encoder.as_ref()[..first.len()], - // "the first packet should be a prefix" - // ); - // assert_eq!(encoder.len(), 45 + 29); + let first = encoder.to_vec(); + + // TODO: 0 ideal here? + let mut builder = + PacketBuilder::short(encoder, false, Some(ConnectionId::from(SERVER_CID)), 0); + builder.pn(1, 3); + builder.encode(&[0]); // Minimal size (packet number is big enough). + let encoder = builder.build(&mut prot).expect("build"); + assert_eq!( + first.as_slice(), + &encoder.as_ref()[..first.len()], + "the first packet should be a prefix" + ); + assert_eq!(encoder.len(), 45 + 29); } #[test]