From fa4fc2d73a6a98cdaff14b583875aa806aa84d20 Mon Sep 17 00:00:00 2001
From: "evan.greer@iterable.com" <evan.greer@evan.greer>
Date: Thu, 6 Jul 2023 18:05:32 -0600
Subject: [PATCH] fixes testRemove unit test

---
 tests/unit-tests/InboxTests.swift | 19 +++++++++----------
 1 file changed, 9 insertions(+), 10 deletions(-)

diff --git a/tests/unit-tests/InboxTests.swift b/tests/unit-tests/InboxTests.swift
index e7612bc30..79f466015 100644
--- 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)