From 2535a8c67c286a973e891f46c2cc1e622ee99f4b Mon Sep 17 00:00:00 2001
From: Di Wu <github@wudi.me>
Date: Thu, 30 Nov 2023 16:00:05 -0800
Subject: [PATCH] update test case to verify the latest synced version is
 applied to api request

---
 .../SyncMutationToCloudOperationTests.swift     | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/AmplifyPlugins/DataStore/Tests/AWSDataStorePluginTests/Sync/MutationQueue/SyncMutationToCloudOperationTests.swift b/AmplifyPlugins/DataStore/Tests/AWSDataStorePluginTests/Sync/MutationQueue/SyncMutationToCloudOperationTests.swift
index 7b66f29c7d..96a0ef2252 100644
--- a/AmplifyPlugins/DataStore/Tests/AWSDataStorePluginTests/Sync/MutationQueue/SyncMutationToCloudOperationTests.swift
+++ b/AmplifyPlugins/DataStore/Tests/AWSDataStorePluginTests/Sync/MutationQueue/SyncMutationToCloudOperationTests.swift
@@ -45,8 +45,10 @@ class SyncMutationToCloudOperationTests: XCTestCase {
         var listenerFromSecondRequestOptional: GraphQLOperation<MutationSync<AnyModel>>.ResultListener?
 
         var numberOfTimesEntered = 0
-        let responder = MutateRequestListenerResponder<MutationSync<AnyModel>> { _, eventListener in
+        let responder = MutateRequestListenerResponder<MutationSync<AnyModel>> { request, eventListener in
             if numberOfTimesEntered == 0 {
+                let requestInputVersion = request.variables.flatMap { $0["input"] as? [String: Any] }.flatMap { $0["_version"] as? Int }
+                XCTAssertEqual(requestInputVersion, 10)
                 listenerFromFirstRequestOptional = eventListener
                 expectFirstCallToAPIMutate.fulfill()
             } else if numberOfTimesEntered == 1 {
@@ -67,9 +69,18 @@ class SyncMutationToCloudOperationTests: XCTestCase {
             expectMutationRequestCompletion.fulfill()
         }
 
+        let model = MockSynced(id: "id-1")
         let operation = await SyncMutationToCloudOperation(
             mutationEvent: mutationEvent,
-            getLatestSyncMetadata: { nil },
+            getLatestSyncMetadata: {
+                MutationSyncMetadata(
+                    modelId: model.id,
+                    modelName: model.modelName,
+                    deleted: false,
+                    lastChangedAt: Date().unixSeconds,
+                    version: 10
+                )
+            },
             api: mockAPIPlugin,
             authModeStrategy: AWSDefaultAuthModeStrategy(),
             networkReachabilityPublisher: publisher,
@@ -93,7 +104,7 @@ class SyncMutationToCloudOperationTests: XCTestCase {
             return
         }
 
-        let model = MockSynced(id: "id-1")
+
         let anyModel = try model.eraseToAnyModel()
         let remoteSyncMetadata = MutationSyncMetadata(modelId: model.id,
                                                       modelName: model.modelName,