update test

open-web3-stack · Oct 25, 2024 · 529f71c · 529f71c
1 parent 8b38266
commit 529f71c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/Networking/Sources/Networking/Connection.swift b/Networking/Sources/Networking/Connection.swift
@@ -50,13 +50,13 @@ public final class Connection<Handler: StreamHandler>: Sendable, ConnectionInfoP
         return UInt32(littleEndian: data.withUnsafeBytes { $0.loadUnaligned(as: UInt32.self) })
     }
 
-    public func request(_ request: Handler.EphemeralHandler.Request) async throws -> [Data] {
+    public func request(_ request: Handler.EphemeralHandler.Request) async throws -> Data {
         let data = try request.encode()
         let kind = request.kind
         let stream = try createStream(kind: kind)
         try stream.send(message: data)
 
-        var reps = [Data()]
+        var reps = Data()
         while let nextData = await stream.receive() {
             if nextData.isEmpty { // fin flag
                 break

diff --git a/Networking/Tests/NetworkingTests/PeerTests.swift b/Networking/Tests/NetworkingTests/PeerTests.swift
@@ -210,7 +210,7 @@ struct PeerTests {
         let dataList1 = try await connection1.request(
             MockRequest(kind: .typeA, data: Data("hello world".utf8))
         )
-        #expect(dataList1.first == Data("hello world".utf8))
+        #expect(dataList1 == Data("hello world".utf8))
 
         let connection2 = try peer2.connect(
             to: NetAddr(ipAddress: "127.0.0.1", port: 8083)!, role: .validator
@@ -220,6 +220,6 @@ struct PeerTests {
         let dataList2 = try await connection2.request(
             MockRequest(kind: .typeB, data: Data("I am jam".utf8))
         )
-        #expect(dataList2.first == Data("I am jam".utf8))
+        #expect(dataList2 == Data("I am jam".utf8))
     }
 }