tests: fix a flaky test in NotificationGeenerationProcessorTests

Author: jinliu9508

OneSignalSDK/onesignal/notifications/src/test/java/com/onesignal/notifications/internal/generation/NotificationGenerationProcessorTests.kt

@@ -5,6 +5,8 @@ import com.onesignal.common.threading.suspendifyOnIO
 import com.onesignal.debug.LogLevel
 import com.onesignal.debug.internal.logging.Logging
 import com.onesignal.mocks.AndroidMockHelper
+import com.onesignal.mocks.IOMockHelper
+import com.onesignal.mocks.IOMockHelper.awaitIO
 import com.onesignal.mocks.MockHelper
 import com.onesignal.notifications.INotificationReceivedEvent
 import com.onesignal.notifications.INotificationWillDisplayEvent
@@ -96,6 +98,9 @@ private class Mocks {
 }
 
 class NotificationGenerationProcessorTests : FunSpec({
+
+    listener(IOMockHelper)
+
     beforeAny {
         Logging.logLevel = LogLevel.NONE
 
@@ -276,16 +281,17 @@ class NotificationGenerationProcessorTests : FunSpec({
         coEvery { mocks.notificationLifecycleService.externalNotificationWillShowInForeground(any()) } coAnswers {
             val willDisplayEvent = firstArg<INotificationWillDisplayEvent>()
             willDisplayEvent.preventDefault(false)
+            // Call preventDefault(true) synchronously to wake the waiter before it times out
+            willDisplayEvent.preventDefault(true)
+            // Then call display() asynchronously to ensure it happens after waitForWake completes
             suspendifyOnIO {
-                delay(100)
-                willDisplayEvent.preventDefault(true)
-                delay(100)
                 willDisplayEvent.notification.display()
             }
         }
 
         // When
         mocks.notificationGenerationProcessor.processNotificationData(mocks.context, 1, mocks.notificationPayload, false, 1111)
+        awaitIO()
 
         // Then
         coVerify(exactly = 0) {
@@ -301,15 +307,14 @@ class NotificationGenerationProcessorTests : FunSpec({
             val receivedEvent = firstArg<INotificationReceivedEvent>()
             receivedEvent.preventDefault(false)
             suspendifyOnIO {
-                delay(100)
                 receivedEvent.preventDefault(true)
-                delay(100)
                 receivedEvent.notification.display()
             }
         }
 
         // When
         mocks.notificationGenerationProcessor.processNotificationData(mocks.context, 1, mocks.notificationPayload, true, 1111)
+        awaitIO()
 
         // Then
         coVerify(exactly = 0) {