mirror of
https://github.com/swift-server/swift-aws-lambda-runtime.git
synced 2026-06-02 07:27:33 +00:00
Fabian Fett
ab13e51984
Modifications: - Replace `HTTPHandler` with `NIOHTTPClientResponseAggregator`, to fix a CoW issue. This increases performance. - Ensure that we only use one `EventLoop` in tests. - Use `syncOperations` to setup the `HTTPClient`.
96 lines
3.5 KiB
Swift
96 lines
3.5 KiB
Swift
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This source file is part of the SwiftAWSLambdaRuntime open source project
|
|
//
|
|
// Copyright (c) 2017-2021 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 AWSLambdaRuntimeCore
|
|
import Logging
|
|
import NIO
|
|
import XCTest
|
|
|
|
func runLambda(behavior: LambdaServerBehavior, handler: Lambda.Handler) throws {
|
|
try runLambda(behavior: behavior, factory: { $0.eventLoop.makeSucceededFuture(handler) })
|
|
}
|
|
|
|
func runLambda(behavior: LambdaServerBehavior, factory: @escaping Lambda.HandlerFactory) throws {
|
|
let eventLoopGroup = MultiThreadedEventLoopGroup(numberOfThreads: 1)
|
|
defer { XCTAssertNoThrow(try eventLoopGroup.syncShutdownGracefully()) }
|
|
let logger = Logger(label: "TestLogger")
|
|
let configuration = Lambda.Configuration(runtimeEngine: .init(requestTimeout: .milliseconds(100)))
|
|
let runner = Lambda.Runner(eventLoop: eventLoopGroup.next(), configuration: configuration)
|
|
let server = try MockLambdaServer(behavior: behavior).start().wait()
|
|
defer { XCTAssertNoThrow(try server.stop().wait()) }
|
|
try runner.initialize(logger: logger, factory: factory).flatMap { handler in
|
|
runner.run(logger: logger, handler: handler)
|
|
}.wait()
|
|
}
|
|
|
|
struct EchoHandler: LambdaHandler {
|
|
typealias In = String
|
|
typealias Out = String
|
|
|
|
func handle(context: Lambda.Context, event: String, callback: (Result<String, Error>) -> Void) {
|
|
callback(.success(event))
|
|
}
|
|
}
|
|
|
|
struct FailedHandler: LambdaHandler {
|
|
typealias In = String
|
|
typealias Out = Void
|
|
|
|
private let reason: String
|
|
|
|
public init(_ reason: String) {
|
|
self.reason = reason
|
|
}
|
|
|
|
func handle(context: Lambda.Context, event: String, callback: (Result<Void, Error>) -> Void) {
|
|
callback(.failure(TestError(self.reason)))
|
|
}
|
|
}
|
|
|
|
func assertLambdaLifecycleResult(_ result: Result<Int, Error>, shoudHaveRun: Int = 0, shouldFailWithError: Error? = nil, file: StaticString = #file, line: UInt = #line) {
|
|
switch result {
|
|
case .success where shouldFailWithError != nil:
|
|
XCTFail("should fail with \(shouldFailWithError!)", file: file, line: line)
|
|
case .success(let count) where shouldFailWithError == nil:
|
|
XCTAssertEqual(shoudHaveRun, count, "should have run \(shoudHaveRun) times", file: file, line: line)
|
|
case .failure(let error) where shouldFailWithError == nil:
|
|
XCTFail("should succeed, but failed with \(error)", file: file, line: line)
|
|
case .failure(let error) where shouldFailWithError != nil:
|
|
XCTAssertEqual(String(describing: shouldFailWithError!), String(describing: error), "expected error to mactch", file: file, line: line)
|
|
default:
|
|
XCTFail("invalid state")
|
|
}
|
|
}
|
|
|
|
struct TestError: Error, Equatable, CustomStringConvertible {
|
|
let description: String
|
|
|
|
init(_ description: String) {
|
|
self.description = description
|
|
}
|
|
}
|
|
|
|
extension Date {
|
|
internal var millisSinceEpoch: Int64 {
|
|
Int64(self.timeIntervalSince1970 * 1000)
|
|
}
|
|
}
|
|
|
|
extension Lambda.RuntimeError: Equatable {
|
|
public static func == (lhs: Lambda.RuntimeError, rhs: Lambda.RuntimeError) -> Bool {
|
|
// technically incorrect, but good enough for our tests
|
|
String(describing: lhs) == String(describing: rhs)
|
|
}
|
|
}
|