fix failing test

klaviyo · Dec 19, 2024 · b7c0c2a · b7c0c2a
1 parent 8659c9b
commit b7c0c2a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Tests/KlaviyoSwiftTests/KlaviyoSDKTests.swift b/Tests/KlaviyoSwiftTests/KlaviyoSDKTests.swift
@@ -146,6 +146,7 @@ class KlaviyoSDKTests: XCTestCase {
     // MARK: test unhandle push notification
 
     func testUnhandlePushNotification() throws {
+        let expectation = setupActionAssertion(expectedAction: .syncBadgeCount)
         let callback = XCTestExpectation(description: "callback is not made")
         callback.isInverted = true
         let data: [AnyHashable: Any] = [
@@ -161,7 +162,7 @@ class KlaviyoSDKTests: XCTestCase {
             callback.fulfill()
         }
 
-        wait(for: [callback], timeout: 1.0)
+        wait(for: [callback, expectation], timeout: 1.0)
         XCTAssertFalse(handled)
     }