chore(): Updated iOS fetchAuthSession unit test to support Result

aws-amplify · Feb 17, 2025 · 021aa25 · 021aa25
1 parent 962a76b
commit 021aa25
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/packages/amplify_datastore/example/ios/unit_tests/NativeAuthPluginTests.swift b/packages/amplify_datastore/example/ios/unit_tests/NativeAuthPluginTests.swift
@@ -51,10 +51,14 @@ class NativeAuthPluginTests: XCTestCase {
         let binaryMessenger = MockBinaryMessenger(isSignedIn: isSignedIn)
         let nativePlugin = NativeAuthPlugin(binaryMessenger: binaryMessenger)
         let expectation = expectation(description: "fetchAuthSession completes")
-        nativePlugin.fetchAuthSession { session in
-            defer { expectation.fulfill() }
-            XCTAssertNotNil(session)
-            XCTAssertEqual(session.isSignedIn, isSignedIn)
+        nativePlugin.fetchAuthSession { result in
+            switch result {
+            case .success(let session):
+                defer { expectation.fulfill() }
+                assertSession(session, isSignedIn)
+            case .failure(let error):
+                XCTFail("fetchAuthSession failed with error: \(error)")
+            }
         }
         waitForExpectations(timeout: 1)
     }