Skip to content

Commit 2c31f54

Browse files
committed
Update unit tests
1 parent c1ab258 commit 2c31f54

1 file changed

Lines changed: 11 additions & 9 deletions

File tree

Tests/SocketTests/SocketTests.swift

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -248,7 +248,8 @@ struct SocketTests {
248248
#expect(boundUdpAddress.address.rawValue == "127.0.0.1", "Bound UDP socket should be on loopback address")
249249
}
250250

251-
func testTCPNoDelay() async throws {
251+
@Test
252+
func tcpNoDelay() async throws {
252253
// Create a TCP socket
253254
let socket = try await Socket(IPv4Protocol.tcp)
254255
defer { Task { await socket.close() } }
@@ -259,24 +260,25 @@ struct SocketTests {
259260

260261
// Test getting TCP_NODELAY option
261262
let retrievedOption = try socket[TCPSocketOption.NoDelay.self]
262-
XCTAssertEqual(retrievedOption.boolValue, true)
263+
#expect(retrievedOption.boolValue == true)
263264

264265
// Test disabling TCP_NODELAY
265266
let disableNodelay = TCPSocketOption.NoDelay(false)
266267
try socket.setOption(disableNodelay)
267268

268269
let retrievedDisabled = try socket[TCPSocketOption.NoDelay.self]
269-
XCTAssertEqual(retrievedDisabled.boolValue, false)
270+
#expect(retrievedDisabled.boolValue == false)
270271

271272
// Test with boolean literal
272273
let literalOption: TCPSocketOption.NoDelay = true
273274
try socket.setOption(literalOption)
274275

275276
let retrievedLiteral = try socket[TCPSocketOption.NoDelay.self]
276-
XCTAssertEqual(retrievedLiteral.boolValue, true)
277+
#expect(retrievedLiteral.boolValue == true)
277278
}
278279

279-
func testTCPNoDelayBehavior() async throws {
280+
@Test
281+
func tcpNoDelayBehavior() async throws {
280282
// This test verifies that TCP_NODELAY option works by testing with a connected socket pair
281283
let port = UInt16.random(in: 8080 ..< .max)
282284
print("Testing TCP_NODELAY behavior on port \(port)")
@@ -296,7 +298,7 @@ struct SocketTests {
296298
// Test that we can set TCP_NODELAY before connecting
297299
try client.setOption(TCPSocketOption.NoDelay(true))
298300
let nodeDelayBeforeConnect = try client[TCPSocketOption.NoDelay.self]
299-
XCTAssertEqual(nodeDelayBeforeConnect.boolValue, true)
301+
#expect(nodeDelayBeforeConnect.boolValue == true)
300302

301303
// Connect to server
302304
do { try await client.connect(to: address) }
@@ -308,20 +310,20 @@ struct SocketTests {
308310

309311
// Verify TCP_NODELAY is still set after connection
310312
let nodeDelayAfterConnect = try client[TCPSocketOption.NoDelay.self]
311-
XCTAssertEqual(nodeDelayAfterConnect.boolValue, true)
313+
#expect(nodeDelayAfterConnect.boolValue == true)
312314

313315
// Test setting TCP_NODELAY on the server's accepted connection
314316
try serverConnection.setOption(TCPSocketOption.NoDelay(false))
315317
let serverNodeDelay = try serverConnection[TCPSocketOption.NoDelay.self]
316-
XCTAssertEqual(serverNodeDelay.boolValue, false)
318+
#expect(serverNodeDelay.boolValue == false)
317319

318320
// Demonstrate that small writes work with TCP_NODELAY
319321
// (The actual latency difference is hard to measure reliably on localhost)
320322
let testData = Data("Hello".utf8)
321323
try await client.write(testData)
322324

323325
let receivedData = try await serverConnection.read(testData.count)
324-
XCTAssertEqual(testData, receivedData)
326+
#expect(testData == receivedData)
325327

326328
print("Successfully tested TCP_NODELAY option setting and data transmission")
327329
}

0 commit comments

Comments
 (0)