mirror
/
swift-nio-transport-services
mirror of https://github.com/apple/swift-nio-transport-services.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

PR changes

This commit is contained in:
Gus Cairo 2025-04-15 15:08:35 +01:00
parent d597722cca
commit 988d42102c
3 changed files with 91 additions and 67 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
Tests/NIOTransportServicesTests/NIOTSBootstrapTests.swift
View File

@ -18,7 +18,7 @@ import XCTest
import Network
import NIOCore
import NIOEmbedded
@testable import NIOTransportServices
import NIOTransportServices
import NIOConcurrencyHelpers
import Foundation
@ -384,11 +384,11 @@ final class NIOTSBootstrapTests: XCTestCase {
func channelActive(context: ChannelHandlerContext) {
self.connectionPromise.succeed()
context.fireChannelActive()
}
}
let group = NIOTSEventLoopGroup(loopCount: 1)
try await withEventLoopGroup { group in
let configuratorListenerCounter = NIOLockedValueBox(0)
let configuratorConnectionCounter = NIOLockedValueBox(0)
let waitForConnectionHandler = WaitForConnectionHandler(
@ -422,8 +422,7 @@ final class NIOTSBootstrapTests: XCTestCase {
XCTAssertEqual(2, configuratorListenerCounter.withLockedValue { $0 })
XCTAssertEqual(1, configuratorConnectionCounter.withLockedValue { $0 })
try await group.shutdownGracefully()
}
}
}

7
Tests/NIOTransportServicesTests/NIOTSDatagramConnectionChannelTests.swift
View File

@ -245,11 +245,11 @@ final class NIOTSDatagramConnectionChannelTests: XCTestCase {
func channelActive(context: ChannelHandlerContext) {
self.connectionPromise.succeed()
context.fireChannelActive()
}
}
let group = NIOTSEventLoopGroup(loopCount: 1)
try await withEventLoopGroup { group in
let configuratorListenerCounter = NIOLockedValueBox(0)
let configuratorConnectionCounter = NIOLockedValueBox(0)
let waitForConnectionHandler = WaitForConnectionHandler(
@ -288,8 +288,7 @@ final class NIOTSDatagramConnectionChannelTests: XCTestCase {
XCTAssertEqual(2, configuratorListenerCounter.withLockedValue { $0 })
XCTAssertEqual(1, configuratorConnectionCounter.withLockedValue { $0 })
try await group.shutdownGracefully()
}
}
func testCanExtractTheConnection() throws {

26
Tests/NIOTransportServicesTests/NIOTSTestUtilities.swift Normal file
View File

@ -0,0 +1,26 @@
//===----------------------------------------------------------------------===//
//
// This source file is part of the SwiftNIO open source project
//
// Copyright (c) 2017-2025 Apple Inc. and the SwiftNIO project authors
// Licensed under Apache License v2.0
//
// See LICENSE.txt for license information
// See CONTRIBUTORS.txt for the list of SwiftNIO project authors
//
// SPDX-License-Identifier: Apache-2.0
//
//===----------------------------------------------------------------------===//
import NIOCore
import NIOTransportServices
func withEventLoopGroup(_ test: (EventLoopGroup) async throws -> Void) async rethrows {
let group = NIOTSEventLoopGroup()
do {
try await test(group)
} catch {
try await group.shutdownGracefully()
throw error
}
}