test(swift): add unit test cases for extension functions

packages/amplify_datastore/example/ios/unit_tests/GraphQLRequestExtensionTests.swift

@@ -0,0 +1,61 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+
+import XCTest
+@testable import Amplify
+@testable import amplify_datastore
+
+class GraphQLRequestExtensionTests: XCTestCase {
+    func testToNativeGraphQLRequest_withCorrectData_convertToNativeGraphQLRequestSuccessfully() {
+
+        let graphQLRequest = GraphQLRequest(
+            apiName: UUID().uuidString,
+            document: UUID().uuidString,
+            variables: [UUID().uuidString: UUID().uuidString],
+            responseType: MutationEvent.self
+        )
+
+        let nativeGrqphQLRquest = graphQLRequest.toNativeGraphQLRequest()
+        XCTAssertEqual(graphQLRequest.apiName, nativeGrqphQLRquest.apiName)
+        XCTAssertEqual(graphQLRequest.document, nativeGrqphQLRquest.document)
+        XCTAssertEqual(String(describing: graphQLRequest.responseType), nativeGrqphQLRquest.responseType)
+        let graphQLVariablesJson = """
+        {"\(graphQLRequest.variables!.keys.first!)":"\(graphQLRequest.variables!.values.first!)"}
+        """
+        XCTAssertEqual(graphQLVariablesJson, nativeGrqphQLRquest.variablesJson)
+    }
+
+    func testToNativeGraphQLRequest_withNilVariables_convertToNativeGraphQLRequestWithEmptyVariablesJsonObject() {
+        let graphQLRequest = GraphQLRequest(
+            apiName: UUID().uuidString,
+            document: UUID().uuidString,
+            variables: nil,
+            responseType: MutationEvent.self
+        )
+
+        let nativeGrqphQLRquest = graphQLRequest.toNativeGraphQLRequest()
+        XCTAssertEqual(graphQLRequest.apiName, nativeGrqphQLRquest.apiName)
+        XCTAssertEqual(graphQLRequest.document, nativeGrqphQLRquest.document)
+        XCTAssertEqual(String(describing: graphQLRequest.responseType), nativeGrqphQLRquest.responseType)
+        let graphQLVariablesJson = "{}"
+        XCTAssertEqual(graphQLVariablesJson, nativeGrqphQLRquest.variablesJson)
+    }
+
+    func testToNativeGraphQLRequest_withEmptyVariables_convertToNativeGraphQLRequestWithEmptyVariablesJsonObject() {
+        let graphQLRequest = GraphQLRequest(
+            apiName: UUID().uuidString,
+            document: UUID().uuidString,
+            variables: [:],
+            responseType: MutationEvent.self
+        )
+
+        let nativeGrqphQLRquest = graphQLRequest.toNativeGraphQLRequest()
+        XCTAssertEqual(graphQLRequest.apiName, nativeGrqphQLRquest.apiName)
+        XCTAssertEqual(graphQLRequest.document, nativeGrqphQLRquest.document)
+        XCTAssertEqual(String(describing: graphQLRequest.responseType), nativeGrqphQLRquest.responseType)
+        let graphQLVariablesJson = "{}"
+        XCTAssertEqual(graphQLVariablesJson, nativeGrqphQLRquest.variablesJson)
+    }
+
+}

packages/amplify_datastore/example/ios/unit_tests/PublisherExtensionTests.swift

@@ -0,0 +1,105 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+
+import XCTest
+import Combine
+@testable import amplify_datastore
+
+class PublisherExtensionTests: XCTestCase {
+    func testToAmplifyAsyncThrowingSequence_createAsyncSequenceCorrect() async throws {
+        let subject = PassthroughSubject<Int, TestError>()
+        let (sequence, cancellable) = subject.eraseToAnyPublisher().toAmplifyAsyncThrowingSequence()
+
+        let expectation1 = expectation(description: "element 1 received")
+        let expectation2 = expectation(description: "element 2 received")
+        let expectation3 = expectation(description: "element 3 received")
+        Task {
+            for try await element in sequence {
+                switch element {
+                case 1: expectation1.fulfill()
+                case 2: expectation2.fulfill()
+                case 3: expectation3.fulfill()
+                default: break
+                }
+            }
+        }
+
+        Task {
+            subject.send(1)
+            subject.send(2)
+            subject.send(3)
+            subject.send(completion: .finished)
+        }
+
+        await fulfillment(of: [expectation1, expectation2, expectation3], timeout: 1)
+    }
+
+    func testToAmplifyAsyncThrowingSequence_withThrowingError_createAsyncSequenceCorrect() async throws {
+        let subject = PassthroughSubject<Int, TestError>()
+        let (sequence, cancellable) = subject.eraseToAnyPublisher().toAmplifyAsyncThrowingSequence()
+
+        let expectation1 = expectation(description: "element 1 received")
+        let expectation2 = expectation(description: "element 2 received")
+        let expectation3 = expectation(description: "element 3 received")
+        expectation3.isInverted = true
+        Task {
+            do {
+                for try await element in sequence {
+                    switch element {
+                    case 1: expectation1.fulfill()
+                    case 2: expectation2.fulfill()
+                    case 3: expectation3.fulfill()
+                    default: break
+                    }
+                }
+            } catch {
+                XCTAssertTrue(error is TestError)
+            }
+        }
+
+        subject.send(1)
+        subject.send(2)
+        subject.send(completion: .failure(.error))
+        await fulfillment(of: [expectation1, expectation2, expectation3], timeout: 1)
+    }
+
+    func testToAmplifyAsyncThrowingSequence_withCancelling_createAsyncSequenceCorrect() async throws {
+        let subject = PassthroughSubject<Int, TestError>()
+        let (sequence, cancellable) = subject.eraseToAnyPublisher().toAmplifyAsyncThrowingSequence()
+
+        let expectation1 = expectation(description: "element 1 received")
+        let expectation2 = expectation(description: "element 2 received")
+        let expectation3 = expectation(description: "element 3 received")
+        expectation3.isInverted = true
+        let expectation4 = expectation(description: "element 4 received")
+        expectation4.isInverted = true
+        Task {
+            for try await element in sequence {
+                switch element {
+                case 1: expectation1.fulfill()
+                case 2: expectation2.fulfill()
+                case 3: expectation3.fulfill()
+                case 4: expectation4.fulfill()
+                default: break
+                }
+            }
+        }
+
+        subject.send(1)
+        subject.send(2)
+        cancellable.cancel()
+        subject.send(3)
+        subject.send(4)
+        await fulfillment(of: [
+            expectation1,
+            expectation2,
+            expectation3,
+            expectation4
+        ], timeout: 1)
+    }
+}
+
+fileprivate enum TestError: Error {
+    case error
+}