feat: Test code-generated Waiters components

codegen/Package.swift

@@ -59,6 +59,8 @@ appendLibTarget(name: "aws_restjson", path: "\(baseDirLocal)/aws-restjson")
 appendTstTarget(name: "aws_restjsonTests", path: "\(baseDirLocal)/aws-restjson", dependency: "aws_restjson")
 appendLibTarget(name: "rest_json_extras", path: "\(baseDirLocal)/rest_json_extras")
 appendTstTarget(name: "rest_json_extrasTests", path: "\(baseDirLocal)/rest_json_extras", dependency: "rest_json_extras")
+appendLibTarget(name: "Waiters", path: "\(baseDirLocal)/Waiters")
+appendTstTarget(name: "WaitersTests", path: "./protocol-test-codegen-local/Tests", dependency: "Waiters")
 
 func appendLibTarget(name: String, path: String) {
     package.targets.append(
@@ -103,7 +105,7 @@ if let smithySwiftDir = ProcessInfo.processInfo.environment["SMITHY_SWIFT_CI_DIR
     ]
 } else {
     package.dependencies += [
-        .package(path: "~/Projects/Amplify/SwiftSDK/smithy-swift"),
-        .package(path: "~/Projects/Amplify/SwiftSDK/aws-sdk-swift"),
+        .package(path: "../../smithy-swift"),
+        .package(path: "../../aws-sdk-swift"),
     ]
 }

codegen/protocol-test-codegen-local/Tests/swift-codegen/WaitersTests/ErrorTypeMatcherTests.swift

@@ -0,0 +1,66 @@
+//
+// Copyright Amazon.com Inc. or its affiliates.
+// All Rights Reserved.
+//
+// SPDX-License-Identifier: Apache-2.0
+//
+
+import XCTest
+@testable import Waiters
+@testable import ClientRuntime
+
+#if swift(>=5.7)
+
+class ErrorTypeMatcherTests: XCTestCase {
+
+    // expected errorType for these tests: "MyError"
+
+    // MARK: - errorType matcher
+
+    func test_errorType_matchesWhenErrorTypeMatchesAndErrorIsAWaiterTypedError() async throws {
+        let error = WaiterTypedErrorImpl()
+        let subject = try WaitersClient.errorTypeMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .failure(error))
+        XCTAssertEqual(match, .success(.failure(error)))
+    }
+
+    func test_errorType_doesNotMatchWhenErrorTypeDoesNotMatchAndErrorIsAWaiterTypedError() async throws {
+        let error = WaiterTypedErrorNoMatchImpl()
+        let subject = try WaitersClient.errorTypeMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .failure(error))
+        XCTAssertNil(match)
+    }
+
+    func test_errorType_doesNotMatchWhenErrorTypeMatchesButErrorIsNotAWaiterTypedError() async throws {
+        let error = WaiterTypedErrorNoMatchImpl()
+        let subject = try WaitersClient.errorTypeMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .failure(error))
+        XCTAssertNil(match)
+    }
+
+    func test_errorType_doesNotMatchWhenResultIsSuccess() async throws {
+        let response = GetWidgetOutputResponse()
+        let subject = try WaitersClient.errorTypeMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(response))
+        XCTAssertNil(match)
+    }
+}
+
+// Error types used in tests above
+
+private struct WaiterTypedErrorImpl: WaiterTypedError, Equatable {
+
+    var waiterErrorType: String? { "MyError" }
+}
+
+private struct WaiterTypedErrorNoMatchImpl: WaiterTypedError, Equatable {
+
+    var waiterErrorType: String? { "OtherError" }
+}
+
+private struct NotAWaiterTypedErrorImpl: Error, Equatable {  // An error but not a WaiterTypedError
+
+    var waiterErrorType: String? { "MyError" }
+}
+
+#endif

codegen/protocol-test-codegen-local/Tests/swift-codegen/WaitersTests/InputOutputMatcherTests.swift

@@ -0,0 +1,40 @@
+//
+// Copyright Amazon.com Inc. or its affiliates.
+// All Rights Reserved.
+//
+// SPDX-License-Identifier: Apache-2.0
+//
+
+import XCTest
+@testable import Waiters
+@testable import ClientRuntime
+
+#if swift(>=5.7)
+class InputOutputMatcherTests: XCTestCase {
+
+    // JMESPath expression: input.stringProperty == output.stringProperty
+    // JMESPath comparator: booleanEquals
+    // JMESPath expected value: true
+
+    // inputOutput tests are just on the input & output properties, because all the other logic
+    // in them is shared with output matchers, which are tested more comprehensively.
+
+    func test_inputOutput_acceptorMatchesWhenInputAndOutputPropertiesMatch() async throws {
+        let value = UUID().uuidString
+        let input = GetWidgetInput(stringProperty: value)
+        let output = GetWidgetOutputResponse(stringProperty: value)
+        let subject = try WaitersClient.inputOutputPropertyMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: input, result: .success(output))
+        XCTAssertEqual(match, .success(.success(output)))
+    }
+
+    func test_inputOutput_acceptorFailsToMatchWhenInputAndOutputPropertiesDontMatch() async throws {
+        let value = UUID().uuidString
+        let input = GetWidgetInput(stringProperty: value)
+        let output = GetWidgetOutputResponse(stringProperty: value + "xxx")
+        let subject = try WaitersClient.inputOutputPropertyMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: input, result: .success(output))
+        XCTAssertNil(match)
+    }
+}
+#endif

codegen/protocol-test-codegen-local/Tests/swift-codegen/WaitersTests/OutputMatcherTests.swift

@@ -0,0 +1,236 @@
+//
+// Copyright Amazon.com Inc. or its affiliates.
+// All Rights Reserved.
+//
+// SPDX-License-Identifier: Apache-2.0
+//
+
+import XCTest
+@testable import Waiters
+@testable import ClientRuntime
+
+#if swift(>=5.7)
+class OutputMatcherTests: XCTestCase {
+
+    // MARK: properties & stringEquals comparator
+
+    // JMESPath expression: stringProperty
+    // JMESPath comparator: stringEquals
+    // JMESPath expected value: payload property contents
+
+    func test_outputStringProperty_acceptorMatchesOnPropertyMatch() async throws {
+        let output = GetWidgetOutputResponse(stringProperty: "payload property contents")
+        let subject = try WaitersClient.outputStringPropertyMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(output))
+        XCTAssertEqual(match, .success(.success(output)))
+    }
+
+    func test_outputStringProperty_acceptorFailsToMatchOnPropertyMismatch() async throws {
+        let output = GetWidgetOutputResponse(stringProperty: "not the payload property contents")
+        let subject = try WaitersClient.outputStringPropertyMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(output))
+        XCTAssertNil(match)
+    }
+
+    func test_outputStringProperty_acceptorFailsToMatchOnNullProperty() async throws {
+        let output = GetWidgetOutputResponse(stringProperty: nil)
+        let subject = try WaitersClient.outputStringPropertyMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(output))
+        XCTAssertNil(match)
+    }
+
+    // MARK: properties & booleanEquals comparator
+
+    // JMESPath expression: booleanProperty
+    // JMESPath comparator: booleanEquals
+    // JMESPath expected value: false
+
+    func test_outputBooleanProperty_acceptorMatchesOnPropertyMatch() async throws {
+        let output = GetWidgetOutputResponse(booleanProperty: false)
+        let subject = try WaitersClient.outputBooleanPropertyMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(output))
+        XCTAssertEqual(match, .success(.success(output)))
+    }
+
+    func test_outputBooleanProperty_acceptorFailsToMatchOnPropertyMismatch() async throws {
+        let output = GetWidgetOutputResponse(booleanProperty: true)
+        let subject = try WaitersClient.outputBooleanPropertyMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(output))
+        XCTAssertNil(match)
+    }
+
+    func test_outputBooleanProperty_acceptorFailsToMatchOnNullProperty() async throws {
+        let output = GetWidgetOutputResponse(booleanProperty: nil)
+        let subject = try WaitersClient.outputBooleanPropertyMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(output))
+        XCTAssertNil(match)
+    }
+
+    // MARK: properties & allStringEquals comparator
+
+    // JMESPath expression: stringArrayProperty
+    // JMESPath comparator: allStringEquals
+    // JMESPath expected value: payload property contents
+
+    func test_arrayPropertyAll_acceptorMatchesWhenArrayElementsAllMatch() async throws {
+        let expected = "payload property contents"
+        let output = GetWidgetOutputResponse(stringArrayProperty: [expected, expected, expected])
+        let subject = try WaitersClient.outputStringArrayAllPropertyMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(output))
+        XCTAssertEqual(match, .success(.success(output)))
+    }
+
+    func test_arrayPropertyAll_acceptorFailsToMatchWhenArrayElementsDontMatch() async throws {
+        let expected = "payload property contents"
+        let output = GetWidgetOutputResponse(stringArrayProperty: [expected, expected, "unexpected"])
+        let subject = try WaitersClient.outputStringArrayAllPropertyMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(output))
+        XCTAssertNil(match)
+    }
+
+    func test_arrayPropertyAll_acceptorFailsToMatchWhenArrayIsEmpty() async throws {
+        let output = GetWidgetOutputResponse(stringArrayProperty: [])
+        let subject = try WaitersClient.outputStringArrayAllPropertyMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(output))
+        XCTAssertNil(match)
+    }
+
+    func test_arrayPropertyAll_acceptorFailsToMatchWhenArrayIsNull() async throws {
+        let output = GetWidgetOutputResponse(stringArrayProperty: nil)
+        let subject = try WaitersClient.outputStringArrayAllPropertyMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(output))
+        XCTAssertNil(match)
+    }
+
+    // MARK: properties & anyStringEquals comparator
+
+    // JMESPath expression: stringArrayProperty
+    // JMESPath comparator: anyStringEquals
+    // JMESPath expected value: payload property contents
+
+    func test_arrayPropertyAny_acceptorMatchesWhenArrayElementsAllMatch() async throws {
+        let expected = "payload property contents"
+        let output = GetWidgetOutputResponse(stringArrayProperty: [expected, expected, expected])
+        let subject = try WaitersClient.outputStringArrayAnyPropertyMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(output))
+        XCTAssertEqual(match, .success(.success(output)))
+    }
+
+    func test_arrayPropertyAny_acceptorMatchesWhenAllButOneElementMismatches() async throws {
+        let expected = "payload property contents"
+        let output = GetWidgetOutputResponse(stringArrayProperty: [expected, expected, "unexpected"])
+        let subject = try WaitersClient.outputStringArrayAnyPropertyMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(output))
+        XCTAssertEqual(match, .success(.success(output)))
+    }
+
+    func test_arrayPropertyAny_acceptorFailsToMatchWhenAllElementsMismatch() async throws {
+        let unexpected = "unexpected"
+        let output = GetWidgetOutputResponse(stringArrayProperty: [unexpected, unexpected, unexpected])
+        let subject = try WaitersClient.outputStringArrayAnyPropertyMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(output))
+        XCTAssertNil(match)
+    }
+
+    func test_arrayPropertyAny_acceptorFailsToMatchWhenArrayIsEmpty() async throws {
+        let output = GetWidgetOutputResponse(stringArrayProperty: [])
+        let subject = try WaitersClient.outputStringArrayAnyPropertyMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(output))
+        XCTAssertNil(match)
+    }
+
+    func test_arrayPropertyAny_acceptorFailsToMatchWhenArrayIsNull() async throws {
+        let output = GetWidgetOutputResponse(stringArrayProperty: nil)
+        let subject = try WaitersClient.outputStringArrayAnyPropertyMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(output))
+        XCTAssertNil(match)
+    }
+
+    // MARK: - Flatten operator
+
+    // JMESPath expression: children[].grandchildren[].name
+    // JMESPath comparator: anyStringEquals
+    // JMESPath expected value: expected name
+
+    func test_flatten_acceptorMatchesWhenFlattenedValueMatches() async throws {
+        let expected = "expected name"
+        let output = outputTree(embeddedName: expected)
+        let subject = try WaitersClient.flattenMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(output))
+        XCTAssertEqual(match, .success(.success(output)))
+    }
+
+    func test_flatten_acceptorDoesNotMatchWhenNoFlattenedValueMatches() async throws {
+        let unexpected = "not the expected name"
+        let output = outputTree(embeddedName: unexpected)
+        let subject = try WaitersClient.flattenMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(output))
+        XCTAssertNil(match)
+    }
+
+    // JMESPath expression: length(children[].grandchildren[]) == `6`
+    // JMESPath comparator: booleanEquals
+    // JMESPath expected value: true
+
+    func test_flattenLength_acceptorMatchesWhenFlattenedValueMatchesCount() async throws {
+        let unexpected = "not the expected name"
+        let output = outputTree(embeddedName: unexpected)
+        let subject = try WaitersClient.flattenLengthMatcherWaiterConfig().acceptors[0]
+        let match = subject.evaluate(input: anInput, result: .success(output))
+        XCTAssertEqual(match, .success(.success(output)))
+    }
+
+    // MARK: - Filter
+
+    // JMESPath expression: length((children[].grandchildren[])[?number > `4`]) == `3`
+    // JMESPath comparator: booleanEquals
+    // JMESPath expected value: true
+
+    func test_filter_acceptorMatchesWhenFilterMatches() async throws {
+        let output1 = outputTree()
+        let subject = try WaitersClient.filterMatcherWaiterConfig().acceptors[0]
+        let match1 = subject.evaluate(input: anInput, result: .success(output1))
+        XCTAssertNil(match1)
+        let output2 = outputTree(appendBonusKid: true)
+        let match2 = subject.evaluate(input: anInput, result: .success(output2))
+        XCTAssertEqual(match2, .success(.success(output2)))
+    }
+
+    // MARK: - Projections
+
+    // JMESPath expression: length(children[?length(grandchildren) == `3`]) == `1`
+    // JMESPath comparator: booleanEquals
+    // JMESPath expected value: true
+
+    func test_projectedLength_acceptorMatchesWhenFlattenedValueMatchesCount() async throws {
+        let output1 = outputTree()
+        let subject = try WaitersClient.projectedLengthMatcherWaiterConfig().acceptors[0]
+        let match1 = subject.evaluate(input: anInput, result: .success(output1))
+        XCTAssertNil(match1)
+        let output2 = outputTree(appendBonusKid: true)
+        let match2 = subject.evaluate(input: anInput, result: .success(output2))
+        XCTAssertEqual(match2, .success(.success(output2)))
+    }
+
+    // MARK: - Helper methods
+
+    private func outputTree(globalName: String? = nil, embeddedName: String? = "c", appendBonusKid: Bool = false) -> GetWidgetOutputResponse {
+        var grandchildren2: [WaitersClientTypes.Grandchild] = [
+            .init(name: embeddedName ?? globalName, number: 1),
+            .init(name: globalName ?? "d", number: 2)
+        ]
+        if appendBonusKid { grandchildren2.append(.init(name: "bonus kid", number: 7))}
+        return GetWidgetOutputResponse(children: [
+            .init(grandchildren: [
+                .init(name: globalName ?? "a", number: 3),
+                .init(name: globalName ?? "b", number: 4)
+            ]),
+            .init(grandchildren: grandchildren2),
+            .init(grandchildren: [
+                .init(name: globalName ?? "e", number: 5),
+                .init(name: globalName ?? "f", number: 6)
+            ])
+        ])
+    }
+}
+#endif