mirror of
https://github.com/swift-server/swift-aws-lambda-runtime.git
synced 2026-06-02 07:27:33 +00:00
Fabian Fettandtomer doron
19fc295aa0
- Add an `AsyncLambdaHandler`. Will be renamed to `LambdaHandler` as soon as we drop the current callback based `LambdaHandler`. - The default way to use an `AsyncLambdaHandler` is to use `@main` to execute it. Don't use `Lambda.run` for it. We wan't to remove `Lambda.run` for 1.0. Co-authored-by: tomer doron <tomerd@apple.com>
93 lines
3.3 KiB
Swift
93 lines
3.3 KiB
Swift
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This source file is part of the SwiftAWSLambdaRuntime open source project
|
|
//
|
|
// Copyright (c) 2017-2018 Apple Inc. and the SwiftAWSLambdaRuntime project authors
|
|
// Licensed under Apache License v2.0
|
|
//
|
|
// See LICENSE.txt for license information
|
|
// See CONTRIBUTORS.txt for the list of SwiftAWSLambdaRuntime project authors
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
@testable import AWSLambdaRuntime
|
|
@testable import AWSLambdaRuntimeCore
|
|
import Logging
|
|
import NIO
|
|
import NIOFoundationCompat
|
|
import XCTest
|
|
|
|
class CodableLambdaTest: XCTestCase {
|
|
var eventLoopGroup: EventLoopGroup!
|
|
let allocator = ByteBufferAllocator()
|
|
|
|
override func setUp() {
|
|
self.eventLoopGroup = MultiThreadedEventLoopGroup(numberOfThreads: 1)
|
|
}
|
|
|
|
override func tearDown() {
|
|
try! self.eventLoopGroup.syncShutdownGracefully()
|
|
}
|
|
|
|
func testCodableVoidClosureWrapper() {
|
|
let request = Request(requestId: UUID().uuidString)
|
|
var inputBuffer: ByteBuffer?
|
|
var outputBuffer: ByteBuffer?
|
|
|
|
let closureWrapper = CodableVoidClosureWrapper { (_, _: Request, completion) in
|
|
XCTAssertEqual(request, request)
|
|
completion(.success(()))
|
|
}
|
|
|
|
XCTAssertNoThrow(inputBuffer = try JSONEncoder().encode(request, using: self.allocator))
|
|
XCTAssertNoThrow(outputBuffer = try closureWrapper.handle(context: self.newContext(), event: XCTUnwrap(inputBuffer)).wait())
|
|
XCTAssertNil(outputBuffer)
|
|
}
|
|
|
|
func testCodableClosureWrapper() {
|
|
let request = Request(requestId: UUID().uuidString)
|
|
var inputBuffer: ByteBuffer?
|
|
var outputBuffer: ByteBuffer?
|
|
var response: Response?
|
|
|
|
let closureWrapper = CodableClosureWrapper { (_, req: Request, completion: (Result<Response, Error>) -> Void) in
|
|
XCTAssertEqual(request, req)
|
|
completion(.success(Response(requestId: req.requestId)))
|
|
}
|
|
|
|
XCTAssertNoThrow(inputBuffer = try JSONEncoder().encode(request, using: self.allocator))
|
|
XCTAssertNoThrow(outputBuffer = try closureWrapper.handle(context: self.newContext(), event: XCTUnwrap(inputBuffer)).wait())
|
|
XCTAssertNoThrow(response = try JSONDecoder().decode(Response.self, from: XCTUnwrap(outputBuffer)))
|
|
XCTAssertEqual(response?.requestId, request.requestId)
|
|
}
|
|
|
|
// convencience method
|
|
func newContext() -> Lambda.Context {
|
|
Lambda.Context(requestID: UUID().uuidString,
|
|
traceID: "abc123",
|
|
invokedFunctionARN: "aws:arn:",
|
|
deadline: .now() + .seconds(3),
|
|
cognitoIdentity: nil,
|
|
clientContext: nil,
|
|
logger: Logger(label: "test"),
|
|
eventLoop: self.eventLoopGroup.next(),
|
|
allocator: ByteBufferAllocator())
|
|
}
|
|
}
|
|
|
|
private struct Request: Codable, Equatable {
|
|
let requestId: String
|
|
init(requestId: String) {
|
|
self.requestId = requestId
|
|
}
|
|
}
|
|
|
|
private struct Response: Codable, Equatable {
|
|
let requestId: String
|
|
init(requestId: String) {
|
|
self.requestId = requestId
|
|
}
|
|
}
|