update peer test

Author: MacOMNI

Networking/Sources/Networking/Connection.swift

@@ -12,6 +12,10 @@ public protocol ConnectionInfoProtocol {
     var remoteAddress: NetAddr { get }
 }
 
+enum ConnectionError: Error {
+    case receiveFailed
+}
+
 public final class Connection<Handler: StreamHandler>: Sendable, ConnectionInfoProtocol {
     let connection: QuicConnection
     let impl: PeerImpl<Handler>
@@ -49,7 +53,15 @@ public final class Connection<Handler: StreamHandler>: Sendable, ConnectionInfoP
             response.append(nextData)
             break
         }
-        return response
+        guard response.count >= 4 else {
+            stream.close(abort: true)
+            throw ConnectionError.receiveFailed
+        }
+        let lengthData = response.prefix(4)
+        let length = UInt32(
+            littleEndian: lengthData.withUnsafeBytes { $0.loadUnaligned(as: UInt32.self) }
+        )
+        return response.dropFirst(4).prefix(Int(length))
     }
 
     @discardableResult

Networking/Tests/NetworkingTests/PeerTests.swift

@@ -211,7 +211,7 @@ struct PeerTests {
         let data1 = try await connection1.request(
             MockRequest(kind: .typeA, data: Data("hello world".utf8))
         )
-        #expect(data1.suffix(from: 4) == Data("hello world".utf8))
+        #expect(data1 == Data("hello world".utf8))
 
         let connection2 = try peer2.connect(
             to: NetAddr(ipAddress: "127.0.0.1", port: 8083)!, role: .validator
@@ -221,6 +221,6 @@ struct PeerTests {
         let data2 = try await connection2.request(
             MockRequest(kind: .typeB, data: Data("I am jam".utf8))
         )
-        #expect(data2.suffix(from: 4) == Data("I am jam".utf8))
+        #expect(data2 == Data("I am jam".utf8))
     }
 }