Skip to content

Commit 7b26da5

Browse files
committed
Test target Swift 6
1 parent 36f3657 commit 7b26da5

File tree

1 file changed

+15
-16
lines changed

1 file changed

+15
-16
lines changed

Tests/AWSLambdaRuntimeCoreTests/MockLambdaServer.swift

Lines changed: 15 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ func withMockServer<Result>(
3737
result = .failure(error)
3838
}
3939

40-
try? await server.stop().get()
40+
try? await server.stop()
4141
return try result.get()
4242
}
4343

@@ -50,7 +50,7 @@ final class MockLambdaServer<Behavior: LambdaServerBehavior> {
5050
private let group: EventLoopGroup
5151

5252
private var channel: Channel?
53-
private var shutdown = false, n,m
53+
private var shutdown = false
5454

5555
init(
5656
behavior: Behavior,
@@ -70,7 +70,7 @@ final class MockLambdaServer<Behavior: LambdaServerBehavior> {
7070
assert(shutdown)
7171
}
7272

73-
func start() async throws -> Int {
73+
fileprivate func start() async throws -> Int {
7474
let logger = self.logger
7575
let keepAlive = self.keepAlive
7676
let behavior = self.behavior
@@ -99,15 +99,12 @@ final class MockLambdaServer<Behavior: LambdaServerBehavior> {
9999
return localAddress.port!
100100
}
101101

102-
func stop() -> EventLoopFuture<Void> {
102+
fileprivate func stop() async throws {
103103
self.logger.info("stopping \(self)")
104-
guard let channel = self.channel else {
105-
return self.group.next().makeFailedFuture(ServerError.notReady)
106-
}
107-
return channel.close().always { _ in
108-
self.shutdown = true
109-
self.logger.info("\(self) stopped")
110-
}
104+
let channel = self.channel!
105+
try? await channel.close().get()
106+
self.shutdown = true
107+
self.logger.info("\(self) stopped")
111108
}
112109
}
113110

@@ -232,28 +229,30 @@ final class HTTPHandler: ChannelInboundHandler {
232229
}
233230
let head = HTTPResponseHead(version: HTTPVersion(major: 1, minor: 1), status: status, headers: headers)
234231

232+
let logger = self.logger
235233
context.write(wrapOutboundOut(.head(head))).whenFailure { error in
236-
self.logger.error("\(self) write error \(error)")
234+
logger.error("write error \(error)")
237235
}
238236

239237
if let b = body {
240238
var buffer = context.channel.allocator.buffer(capacity: b.utf8.count)
241239
buffer.writeString(b)
242240
context.write(wrapOutboundOut(.body(.byteBuffer(buffer)))).whenFailure { error in
243-
self.logger.error("\(self) write error \(error)")
241+
logger.error("write error \(error)")
244242
}
245243
}
246244

247245
let loopBoundContext = NIOLoopBound(context, eventLoop: context.eventLoop)
248246

247+
let keepAlive = self.keepAlive
249248
context.writeAndFlush(wrapOutboundOut(.end(nil))).whenComplete { result in
250249
if case .failure(let error) = result {
251-
self.logger.error("\(self) write error \(error)")
250+
logger.error("write error \(error)")
252251
}
253-
if !self.keepAlive {
252+
if !keepAlive {
254253
let context = loopBoundContext.value
255254
context.close().whenFailure { error in
256-
self.logger.error("\(self) close error \(error)")
255+
logger.error("close error \(error)")
257256
}
258257
}
259258
}

0 commit comments

Comments
 (0)