Fix build_two

mozilla · Sep 25, 2024 · 2b0103a · 2b0103a
1 parent 08c49e6
commit 2b0103a
Showing 1 changed file with 14 additions and 15 deletions.
diff --git 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]