update tests

Author: sebsto

Tests/AWSLambdaRuntimeTests/ControlPlaneRequestEncoderTests.swift

@@ -25,10 +25,10 @@ import FoundationEssentials
 import Foundation
 #endif
 
-@available(LambdaSwift 2.0, *)
 struct ControlPlaneRequestEncoderTests {
     let host = "192.168.0.1"
 
+    @available(LambdaSwift 2.0, *)
     func createChannels() -> (client: EmbeddedChannel, server: EmbeddedChannel) {
         let client = EmbeddedChannel(handler: ControlPlaneRequestEncoderHandler(host: self.host))
         let server = EmbeddedChannel(handlers: [
@@ -39,6 +39,7 @@ struct ControlPlaneRequestEncoderTests {
     }
 
     @Test
+    @available(LambdaSwift 2.0, *)
     func testNextRequest() throws {
         let (client, server) = createChannels()
         defer {
@@ -59,6 +60,7 @@ struct ControlPlaneRequestEncoderTests {
     }
 
     @Test
+    @available(LambdaSwift 2.0, *)
     func testPostInvocationSuccessWithoutBody() throws {
         let (client, server) = createChannels()
         defer {
@@ -81,6 +83,7 @@ struct ControlPlaneRequestEncoderTests {
     }
 
     @Test
+    @available(LambdaSwift 2.0, *)
     func testPostInvocationSuccessWithBody() throws {
         let (client, server) = createChannels()
         defer {
@@ -106,6 +109,7 @@ struct ControlPlaneRequestEncoderTests {
     }
 
     @Test
+    @available(LambdaSwift 2.0, *)
     func testPostInvocationErrorWithBody() throws {
         let (client, server) = createChannels()
         defer {
@@ -134,6 +138,7 @@ struct ControlPlaneRequestEncoderTests {
     }
 
     @Test
+    @available(LambdaSwift 2.0, *)
     func testPostStartupError() throws {
         let (client, server) = createChannels()
         defer {
@@ -160,6 +165,7 @@ struct ControlPlaneRequestEncoderTests {
     }
 
     @Test
+    @available(LambdaSwift 2.0, *)
     func testMultipleNextAndResponseSuccessRequests() throws {
         let (client, server) = createChannels()
         defer {
@@ -184,6 +190,7 @@ struct ControlPlaneRequestEncoderTests {
         }
     }
 
+    @available(LambdaSwift 2.0, *)
     func sendRequest(
         _ request: ControlPlaneRequest,
         client: EmbeddedChannel,

Tests/AWSLambdaRuntimeTests/InvocationTests.swift

@@ -24,9 +24,9 @@ import Foundation
 #endif
 
 @Suite
-@available(LambdaSwift 2.0, *)
 struct InvocationTest {
     @Test
+    @available(LambdaSwift 2.0, *)
     func testInvocationTraceID() throws {
         let headers = HTTPHeaders([
             (AmazonHeaders.requestID, "test"),

Tests/AWSLambdaRuntimeTests/Lambda+CodableTests.swift

@@ -24,7 +24,6 @@ import Foundation
 #endif
 
 @Suite
-@available(LambdaSwift 2.0, *)
 struct JSONTests {
 
     let logger = Logger(label: "JSONTests")
@@ -47,6 +46,7 @@ struct JSONTests {
     }
 
     @Test
+    @available(LambdaSwift 2.0, *)
     func testJSONHandlerWithOutput() async {
         let jsonEncoder = JSONEncoder()
         let jsonDecoder = JSONDecoder()

Tests/AWSLambdaRuntimeTests/LambdaClockTests.swift

@@ -23,10 +23,10 @@ import Foundation
 #endif
 
 @Suite("LambdaClock Tests")
-@available(LambdaSwift 2.0, *)
 struct LambdaClockTests {
 
     @Test("Clock provides current time")
+    @available(LambdaSwift 2.0, *)
     func clockProvidesCurrentTime() {
         let clock = LambdaClock()
         let now = clock.now
@@ -37,6 +37,7 @@ struct LambdaClockTests {
     }
 
     @Test("Instant can be advanced by duration")
+    @available(LambdaSwift 2.0, *)
     func instantCanBeAdvancedByDuration() {
         let clock = LambdaClock()
         let start = clock.now
@@ -46,6 +47,7 @@ struct LambdaClockTests {
     }
 
     @Test("Duration calculation between instants")
+    @available(LambdaSwift 2.0, *)
     func durationCalculationBetweenInstants() {
         let clock = LambdaClock()
         let start = clock.now
@@ -56,6 +58,7 @@ struct LambdaClockTests {
     }
 
     @Test("Instant comparison works correctly")
+    @available(LambdaSwift 2.0, *)
     func instantComparisonWorksCorrectly() {
         let clock = LambdaClock()
         let earlier = clock.now
@@ -66,12 +69,14 @@ struct LambdaClockTests {
     }
 
     @Test("Clock minimum resolution is milliseconds")
+    @available(LambdaSwift 2.0, *)
     func clockMinimumResolutionIsMilliseconds() {
         let clock = LambdaClock()
         #expect(clock.minimumResolution == .milliseconds(1))
     }
 
     @Test("Sleep until deadline works")
+    @available(LambdaSwift 2.0, *)
     func sleepUntilDeadlineWorks() async throws {
         let clock = LambdaClock()
         let start = clock.now
@@ -88,6 +93,7 @@ struct LambdaClockTests {
     }
 
     @Test("Sleep with past deadline returns immediately")
+    @available(LambdaSwift 2.0, *)
     func sleepWithPastDeadlineReturnsImmediately() async throws {
         let clock = LambdaClock()
         let now = clock.now
@@ -103,6 +109,7 @@ struct LambdaClockTests {
     }
 
     @Test("Duration to future instant returns negative duration")
+    @available(LambdaSwift 2.0, *)
     func durationToFutureInstantReturnsNegativeDuration() {
         let clock = LambdaClock()
         let futureDeadline = clock.now.advanced(by: .seconds(30))
@@ -117,6 +124,7 @@ struct LambdaClockTests {
     }
 
     @Test("LambdaClock now matches Foundation Date within tolerance")
+    @available(LambdaSwift 2.0, *)
     func lambdaClockNowMatchesFoundationDate() {
 
         let clock = LambdaClock()
@@ -138,6 +146,7 @@ struct LambdaClockTests {
         )
     }
     @Test("Instant renders as string with an epoch number")
+    @available(LambdaSwift 2.0, *)
     func instantRendersAsStringWithEpochNumber() {
         let clock = LambdaClock()
         let instant = clock.now

Tests/AWSLambdaRuntimeTests/LambdaContextTests.swift

@@ -19,7 +19,6 @@ import Testing
 @testable import AWSLambdaRuntime
 
 @Suite("LambdaContext ClientContext Tests")
-@available(LambdaSwift 2.0, *)
 struct LambdaContextTests {
 
     @Test("ClientContext with full data resolves correctly")
@@ -115,6 +114,7 @@ struct LambdaContextTests {
     }
 
     @Test("getRemainingTime returns positive duration for future deadline")
+    @available(LambdaSwift 2.0, *)
     func getRemainingTimeReturnsPositiveDurationForFutureDeadline() {
 
         // Create context with deadline 30 seconds in the future

Tests/AWSLambdaRuntimeTests/LambdaResponseStreamWriter+HeadersTests.swift

@@ -24,10 +24,10 @@ import Foundation
 #endif
 
 @Suite("LambdaResponseStreamWriter+Headers Tests")
-@available(LambdaSwift 2.0, *)
 struct LambdaResponseStreamWriterHeadersTests {
 
     @Test("Write status and headers with minimal response (status code only)")
+    @available(LambdaSwift 2.0, *)
     func testWriteStatusAndHeadersMinimal() async throws {
         let writer = MockLambdaResponseStreamWriter()
         let response = StreamingLambdaStatusAndHeadersResponse(statusCode: 200)
@@ -44,6 +44,7 @@ struct LambdaResponseStreamWriterHeadersTests {
     }
 
     @Test("Write status and headers with full response (all fields populated)")
+    @available(LambdaSwift 2.0, *)
     func testWriteStatusAndHeadersFull() async throws {
         let writer = MockLambdaResponseStreamWriter()
         let response = StreamingLambdaStatusAndHeadersResponse(
@@ -80,6 +81,7 @@ struct LambdaResponseStreamWriterHeadersTests {
     }
 
     @Test("Write status and headers with custom encoder")
+    @available(LambdaSwift 2.0, *)
     func testWriteStatusAndHeadersWithCustomEncoder() async throws {
         let writer = MockLambdaResponseStreamWriter()
         let response = StreamingLambdaStatusAndHeadersResponse(
@@ -107,6 +109,7 @@ struct LambdaResponseStreamWriterHeadersTests {
     }
 
     @Test("Write status and headers with only headers (no multiValueHeaders)")
+    @available(LambdaSwift 2.0, *)
     func testWriteStatusAndHeadersOnlyHeaders() async throws {
         let writer = MockLambdaResponseStreamWriter()
         let response = StreamingLambdaStatusAndHeadersResponse(
@@ -132,6 +135,7 @@ struct LambdaResponseStreamWriterHeadersTests {
     }
 
     @Test("Write status and headers with only multiValueHeaders (no headers)")
+    @available(LambdaSwift 2.0, *)
     func testWriteStatusAndHeadersOnlyMultiValueHeaders() async throws {
         let writer = MockLambdaResponseStreamWriter()
         let response = StreamingLambdaStatusAndHeadersResponse(
@@ -162,6 +166,7 @@ struct LambdaResponseStreamWriterHeadersTests {
     }
 
     @Test("Verify JSON serialization format matches expected structure")
+    @available(LambdaSwift 2.0, *)
     func testJSONSerializationFormat() async throws {
         let writer = MockLambdaResponseStreamWriter()
         let response = StreamingLambdaStatusAndHeadersResponse(
@@ -199,6 +204,7 @@ struct LambdaResponseStreamWriterHeadersTests {
     }
 
     @Test("Verify buffer contains both JSON and null byte separator")
+    @available(LambdaSwift 2.0, *)
     func testBufferContainsJsonAndSeparator() async throws {
         let writer = MockLambdaResponseStreamWriter()
         let response = StreamingLambdaStatusAndHeadersResponse(statusCode: 200)
@@ -219,6 +225,7 @@ struct LambdaResponseStreamWriterHeadersTests {
     // MARK: - Error Handling Tests
 
     @Test("JSON serialization error propagation")
+    @available(LambdaSwift 2.0, *)
     func testJSONSerializationErrorPropagation() async throws {
         let writer = MockLambdaResponseStreamWriter()
         let response = StreamingLambdaStatusAndHeadersResponse(statusCode: 200)
@@ -236,6 +243,7 @@ struct LambdaResponseStreamWriterHeadersTests {
     }
 
     @Test("Write method error propagation")
+    @available(LambdaSwift 2.0, *)
     func testWriteMethodErrorPropagation() async throws {
         let writer = FailingMockLambdaResponseStreamWriter(failOnWriteCall: 1)  // Fail on first write
         let response = StreamingLambdaStatusAndHeadersResponse(statusCode: 200)
@@ -252,6 +260,7 @@ struct LambdaResponseStreamWriterHeadersTests {
     // This test is no longer needed since we only have one write operation now
 
     @Test("Error types and messages are properly handled")
+    @available(LambdaSwift 2.0, *)
     func testErrorTypesAndMessages() async throws {
         let writer = MockLambdaResponseStreamWriter()
         let response = StreamingLambdaStatusAndHeadersResponse(statusCode: 200)
@@ -272,6 +281,7 @@ struct LambdaResponseStreamWriterHeadersTests {
     }
 
     @Test("JSONEncoder error propagation with invalid data")
+    @available(LambdaSwift 2.0, *)
     func testJSONEncoderErrorPropagation() async throws {
         let writer = MockLambdaResponseStreamWriter()
 
@@ -293,6 +303,7 @@ struct LambdaResponseStreamWriterHeadersTests {
     // MARK: - Integration Tests
 
     @Test("Integration: writeStatusAndHeaders with existing streaming methods")
+    @available(LambdaSwift 2.0, *)
     func testIntegrationWithExistingStreamingMethods() async throws {
         let writer = MockLambdaResponseStreamWriter()
         let response = StreamingLambdaStatusAndHeadersResponse(
@@ -337,6 +348,7 @@ struct LambdaResponseStreamWriterHeadersTests {
     }
 
     @Test("Integration: multiple header writes work correctly")
+    @available(LambdaSwift 2.0, *)
     func testMultipleHeaderWrites() async throws {
         let writer = MockLambdaResponseStreamWriter()
 
@@ -371,6 +383,7 @@ struct LambdaResponseStreamWriterHeadersTests {
     }
 
     @Test("Integration: header write followed by body streaming compatibility")
+    @available(LambdaSwift 2.0, *)
     func testHeaderWriteFollowedByBodyStreaming() async throws {
         let writer = MockLambdaResponseStreamWriter()
 
@@ -419,6 +432,7 @@ struct LambdaResponseStreamWriterHeadersTests {
     }
 
     @Test("Integration: verify method works with different LambdaResponseStreamWriter implementations")
+    @available(LambdaSwift 2.0, *)
     func testWithDifferentWriterImplementations() async throws {
         // Test with basic mock implementation
         let basicWriter = MockLambdaResponseStreamWriter()
@@ -442,6 +456,7 @@ struct LambdaResponseStreamWriterHeadersTests {
     }
 
     @Test("Integration: complex scenario with headers, streaming, and finish")
+    @available(LambdaSwift 2.0, *)
     func testComplexIntegrationScenario() async throws {
         let writer = MockLambdaResponseStreamWriter()
 
@@ -494,6 +509,7 @@ struct LambdaResponseStreamWriterHeadersTests {
     }
 
     @Test("Integration: verify compatibility with protocol requirements")
+    @available(LambdaSwift 2.0, *)
     func testProtocolCompatibility() async throws {
         let writer = MockLambdaResponseStreamWriter()
         let response = StreamingLambdaStatusAndHeadersResponse(statusCode: 200)