File tree 1 file changed +21
-0
lines changed
Tests/AWSLambdaRuntimeCoreTests
1 file changed +21
-0
lines changed Original file line number Diff line number Diff line change @@ -20,6 +20,27 @@ import NIOPosix
20
20
21
21
@testable import AWSLambdaRuntimeCore
22
22
23
+ func withMockServer< Result> (
24
+ behaviour: some LambdaServerBehavior ,
25
+ port: Int = 0 ,
26
+ keepAlive: Bool = true ,
27
+ _ body: ( _ port: Int ) async throws -> Result
28
+ ) async throws -> Result {
29
+ let eventLoopGroup = NIOSingletons . posixEventLoopGroup
30
+ let server = MockLambdaServer ( behavior: behaviour, port: port, keepAlive: keepAlive)
31
+ let port = try await server. start ( ) . get ( )
32
+
33
+ let result : Swift . Result < Result , any Error >
34
+ do {
35
+ result = . success( try await body ( port) )
36
+ } catch {
37
+ result = . failure( error)
38
+ }
39
+
40
+ try ? await server. stop ( ) . get ( )
41
+ return try result. get ( )
42
+ }
43
+
23
44
final class MockLambdaServer {
24
45
private let logger = Logger ( label: " MockLambdaServer " )
25
46
private let behavior : LambdaServerBehavior
You can’t perform that action at this time.
0 commit comments