fixes testRemove unit test

Iterable · Jul 7, 2023 · fa4fc2d · fa4fc2d
1 parent 12b52dc
commit fa4fc2d
Showing 1 changed file with 9 additions and 10 deletions.
diff --git a/tests/unit-tests/InboxTests.swift b/tests/unit-tests/InboxTests.swift
@@ -194,22 +194,21 @@ class InboxTests: XCTestCase {
         mockInAppFetcher.mockInAppPayloadFromServer(internalApi: internalAPI, payload).onSuccess { _ in
             let messages = internalAPI.inAppManager.getInboxMessages()
             XCTAssertEqual(messages.count, 2)
-
+            
             let messageToRemove = messages[0]
             internalAPI.inAppManager.remove(
                 message: messageToRemove,
                 location: .inbox,
                 source: .inboxSwipe,
-                successHandler: { _ in
-                    // Success handler code
-                    expectation1.fulfill()
-                },
-                failureHandler: { _, _ in
-                    // Failure handler code
-                    XCTFail("Failed to remove message")
-                    expectation1.fulfill()
-                }
+                successHandler: { _ in },
+                failureHandler: { _, _ in }
             )
+
+            DispatchQueue.main.asyncAfter(deadline: .now() + 0.05) {
+                let newMessages = internalAPI.inAppManager.getInboxMessages()
+                XCTAssertEqual(newMessages.count, 1)
+                expectation1.fulfill()
+            }
         }
 
         wait(for: [expectation1], timeout: testExpectationTimeout)