From 13146594026ed3d87356d83a067b6a598425dab4 Mon Sep 17 00:00:00 2001
From: Joshua Goins <josh@redstrate.com>
Date: Tue, 18 Mar 2025 19:53:56 -0400
Subject: [PATCH] Restore the test_packet_sizes test

---
 src/packet/packet.rs | 27 +++++++++++++++++++++++----
 1 file changed, 23 insertions(+), 4 deletions(-)

diff --git a/src/packet/packet.rs b/src/packet/packet.rs
index ae6447a..e66c0a8 100644
--- a/src/packet/packet.rs
+++ b/src/packet/packet.rs
@@ -224,12 +224,31 @@ pub async fn send_keep_alive<T: ReadWriteIpcSegment>(
 
 #[cfg(test)]
 mod tests {
-    // TODO: Restore this test
-    /*
+    use crate::packet::IpcSegment;
+
     use super::*;
+
     /// Ensure that the packet size as reported matches up with what we write
     #[test]
     fn test_packet_sizes() {
+        #[binrw]
+        #[brw(repr = u16)]
+        #[derive(Clone, PartialEq, Debug)]
+        enum ClientLobbyIpcType {
+            Dummy = 0x1,
+        }
+
+        #[binrw]
+        #[br(import(magic: &ClientLobbyIpcType))]
+        #[derive(Debug, Clone)]
+        enum ClientLobbyIpcData {
+            Dummy(),
+        }
+
+        type ClientLobbyIpcSegment = IpcSegment<ClientLobbyIpcType, ClientLobbyIpcData>;
+
+        impl ReadWriteIpcSegment for ClientLobbyIpcSegment {}
+
         let packet_types = [
             SegmentType::InitializeEncryption {
                 phrase: String::new(),
@@ -249,7 +268,7 @@ mod tests {
         for packet in &packet_types {
             let mut cursor = Cursor::new(Vec::new());
 
-            let packet_segment = PacketSegment {
+            let packet_segment: PacketSegment<ClientLobbyIpcSegment> = PacketSegment {
                 source_actor: 0,
                 target_actor: 0,
                 segment_type: packet.clone(),
@@ -260,5 +279,5 @@ mod tests {
 
             assert_eq!(buffer.len(), packet_segment.calc_size() as usize);
         }
-    }*/
+    }
 }