swift-aws-lambda-runtime/Tests/AWSLambdaRuntimeTests/InvocationTests.swift

//===----------------------------------------------------------------------===//
//
// This source file is part of the SwiftAWSLambdaRuntime open source project
//
// Copyright SwiftAWSLambdaRuntime project authors
// Copyright (c) Amazon.com, Inc. or its affiliates.
// 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
//
//===----------------------------------------------------------------------===//

import NIOHTTP1
import Testing

@testable import AWSLambdaRuntime

#if canImport(FoundationEssentials)
import FoundationEssentials
#else
import Foundation
#endif

@Suite
struct InvocationTest {
    @Test
    @available(LambdaSwift 2.0, *)
    func testInvocationTraceID() throws {
        let headers = HTTPHeaders([
            (AmazonHeaders.requestID, "test"),
            (AmazonHeaders.deadline, String(Date(timeIntervalSinceNow: 60).millisSinceEpoch)),
            (AmazonHeaders.invokedFunctionARN, "arn:aws:lambda:us-east-1:123456789012:function:custom-runtime"),
        ])

        var maybeInvocation: InvocationMetadata?

        #expect(throws: Never.self) { maybeInvocation = try InvocationMetadata(headers: headers) }
        let invocation = try #require(maybeInvocation)
        #expect(!invocation.traceID.isEmpty)
    }

    @Test
    @available(LambdaSwift 2.0, *)
    func testInvocationTenantID() throws {
        let tenantID = "123"
        let headers = HTTPHeaders([
            (AmazonHeaders.requestID, "test"),
            (AmazonHeaders.deadline, String(Date(timeIntervalSinceNow: 60).millisSinceEpoch)),
            (AmazonHeaders.invokedFunctionARN, "arn:aws:lambda:us-east-1:123456789012:function:custom-runtime"),
            (AmazonHeaders.tenantID, tenantID),
        ])

        var maybeInvocation: InvocationMetadata?

        #expect(throws: Never.self) { maybeInvocation = try InvocationMetadata(headers: headers) }
        let invocation = try #require(maybeInvocation)
        #expect(invocation.tenantID == tenantID)
    }

}