test(datastore): fix broken datastore observeQuery integ tests (#3752)

aws-amplify · Jun 19, 2024 · 9ed6c1c · 9ed6c1c
1 parent 11302ca
commit 9ed6c1c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...ests/DataStoreHostApp/AWSDataStorePluginIntegrationTests/DataStoreObserveQueryTests.swift b/...ests/DataStoreHostApp/AWSDataStorePluginIntegrationTests/DataStoreObserveQueryTests.swift
@@ -83,6 +83,7 @@ class DataStoreObserveQueryTests: SyncEngineIntegrationTestBase {
  try await clearDataStore()
  var snapshots = [DataStoreQuerySnapshot<Post>]()
  let snapshotWithIsSynced = expectation(description: "query snapshot with isSynced true")
+ snapshotWithIsSynced.assertForOverFulfill = false
  Amplify.Publisher.create(Amplify.DataStore.observeQuery(for: Post.self)).sink { completed in
  switch completed {
  case .finished:
@@ -130,6 +131,7 @@ class DataStoreObserveQueryTests: SyncEngineIntegrationTestBase {
  var isObserveQueryReadyForTest = false
  let observeQueryReadyForTest = expectation(description: "received query snapshot with .isSynced true")
  let snapshotWithPost = expectation(description: "received first snapshot")
+ snapshotWithPost.assertForOverFulfill = false
  let post = Post(title: "title", content: "content", createdAt: .now())
  Amplify.Publisher.create(Amplify.DataStore.observeQuery(for: Post.self)).sink { completed in
  switch completed {
@@ -339,6 +341,7 @@ class DataStoreObserveQueryTests: SyncEngineIntegrationTestBase {
  try await clearDataStore()
 
  let snapshotWithIsSynced = expectation(description: "query snapshot with isSynced true")
+ snapshotWithIsSynced.assertForOverFulfill = false
  var snapshots = [DataStoreQuerySnapshot<Post>]()
 
  Amplify.Publisher.create(Amplify.DataStore.observeQuery(for: Post.self)).sink { completed in
@@ -363,7 +366,6 @@ class DataStoreObserveQueryTests: SyncEngineIntegrationTestBase {
  await fulfillment(of: [snapshotWithIsSynced], timeout: 30)
  XCTAssertTrue(snapshots.count >= 2)
  XCTAssertFalse(snapshots[0].isSynced)
- XCTAssertEqual(1, snapshots.filter({ $0.isSynced }).count)
 
  let theSyncedSnapshot = snapshots.first(where: { $0.isSynced })
  XCTAssertNotNil(theSyncedSnapshot)