Add AsyncLambdaHandler tests

Author: fabianfett

Sources/AWSLambdaRuntimeCore/LambdaHandler.swift

@@ -108,7 +108,7 @@ extension AsyncLambdaHandler {
 
     @asyncHandler private func _run(context: Lambda.Context, event: In, promise: EventLoopPromise<Out>) {
         do {
-            let result = try await handle(context: context, event: event)
+            let result = try await self.handle(context: context, event: event)
             promise.succeed(result)
         } catch {
             promise.fail(error)

Tests/AWSLambdaRuntimeCoreTests/Lambda+StringTest.swift renamed to Tests/AWSLambdaRuntimeCoreTests/LambdaHandlerTest.swift

@@ -16,7 +16,9 @@
 import NIO
 import XCTest
 
-class StringLambdaTest: XCTestCase {
+class LambdaHandlerTest: XCTestCase {
+    // MARK: Callback
+
     func testCallbackSuccess() {
         let server = MockLambdaServer(behavior: Behavior())
         XCTAssertNoThrow(try server.start().wait())
@@ -77,6 +79,71 @@ class StringLambdaTest: XCTestCase {
         assertLambdaLifecycleResult(result, shoudHaveRun: maxTimes)
     }
 
+    #if compiler(>=5.5) && $AsyncAwait
+
+    // MARK: AsyncLambdaHandler
+
+    func testAsyncHandlerSuccess() {
+        let server = MockLambdaServer(behavior: Behavior())
+        XCTAssertNoThrow(try server.start().wait())
+        defer { XCTAssertNoThrow(try server.stop().wait()) }
+
+        struct Handler: AsyncLambdaHandler {
+            typealias In = String
+            typealias Out = String
+
+            func handle(context: Lambda.Context, event: String) async throws -> String {
+                event
+            }
+        }
+
+        let maxTimes = Int.random(in: 1 ... 10)
+        let configuration = Lambda.Configuration(lifecycle: .init(maxTimes: maxTimes))
+        let result = Lambda.run(configuration: configuration, handler: Handler())
+        assertLambdaLifecycleResult(result, shoudHaveRun: maxTimes)
+    }
+
+    func testVoidAsyncHandlerSuccess() {
+        let server = MockLambdaServer(behavior: Behavior(result: .success(nil)))
+        XCTAssertNoThrow(try server.start().wait())
+        defer { XCTAssertNoThrow(try server.stop().wait()) }
+
+        struct Handler: AsyncLambdaHandler {
+            typealias In = String
+            typealias Out = Void
+
+            func handle(context: Lambda.Context, event: String) async throws {}
+        }
+
+        let maxTimes = Int.random(in: 1 ... 10)
+        let configuration = Lambda.Configuration(lifecycle: .init(maxTimes: maxTimes))
+        let result = Lambda.run(configuration: configuration, handler: Handler())
+        assertLambdaLifecycleResult(result, shoudHaveRun: maxTimes)
+    }
+
+    func testAsyncHandlerFailure() {
+        let server = MockLambdaServer(behavior: Behavior(result: .failure(TestError("boom"))))
+        XCTAssertNoThrow(try server.start().wait())
+        defer { XCTAssertNoThrow(try server.stop().wait()) }
+
+        struct Handler: AsyncLambdaHandler {
+            typealias In = String
+            typealias Out = String
+
+            func handle(context: Lambda.Context, event: String) async throws -> String {
+                throw TestError("boom")
+            }
+        }
+
+        let maxTimes = Int.random(in: 1 ... 10)
+        let configuration = Lambda.Configuration(lifecycle: .init(maxTimes: maxTimes))
+        let result = Lambda.run(configuration: configuration, handler: Handler())
+        assertLambdaLifecycleResult(result, shoudHaveRun: maxTimes)
+    }
+    #endif
+
+    // MARK: EventLoop
+
     func testEventLoopSuccess() {
         let server = MockLambdaServer(behavior: Behavior())
         XCTAssertNoThrow(try server.start().wait())
@@ -137,6 +204,8 @@ class StringLambdaTest: XCTestCase {
         assertLambdaLifecycleResult(result, shoudHaveRun: maxTimes)
     }
 
+    // MARK: Closure
+
     func testClosureSuccess() {
         let server = MockLambdaServer(behavior: Behavior())
         XCTAssertNoThrow(try server.start().wait())

docker/docker-compose.al2.main.yaml

@@ -10,9 +10,14 @@ services:
 
   test:
     image: swift-aws-lambda:al2-main
+    command: /bin/bash -cl "swift test --enable-test-discovery -Xswiftc -warnings-as-errors $${SANITIZER_ARG-} -Xswiftc -Xfrontend -Xswiftc -enable-experimental-concurrency"
 
   test-samples:
     image: swift-aws-lambda:al2-main
+    command: >-
+      /bin/bash -clx "
+      swift build -Xswiftc -Xfrontend -Xswiftc -enable-experimental-concurrency --package-path Examples/LambdaFunctions &&
+      swift build -Xswiftc -Xfrontend -Xswiftc -enable-experimental-concurrency --package-path Examples/LocalDebugging/MyLambda"
 
   shell:
     image: swift-aws-lambda:al2-main