Fix flake - not waiting for notification

LOW_COVERAGE_REASON=TEST_ONLY
Bug: 351017432
Test: atest SafetyCenterHostSideTestCases
Relnote: N/A test flake fix
Flag: EXEMPT test flake fix
Change-Id: Ibada243967cca83f040ef5c83b24b73eaf263936

1 files changed, 7 insertions, 1 deletions

diff --git a/tests/hostside/safetycenter/helper-app/src/android/safetycenter/hostside/device/SafetyCenterNotificationLoggingHelperTests.kt b/tests/hostside/safetycenter/helper-app/src/android/safetycenter/hostside/device/SafetyCenterNotificationLoggingHelperTests.kt
index 60e6e41ec..c56c913b6 100644
--- a/tests/hostside/safetycenter/helper-app/src/android/safetycenter/hostside/device/SafetyCenterNotificationLoggingHelperTests.kt
+++ b/tests/hostside/safetycenter/helper-app/src/android/safetycenter/hostside/device/SafetyCenterNotificationLoggingHelperTests.kt
@@ -69,6 +69,12 @@ class SafetyCenterNotificationLoggingHelperTests {
     @Test
     fun sendNotification() {
         safetyCenterTestHelper.setData(SINGLE_SOURCE_ID, newTestDataWithNotifiableIssue())
+        TestNotificationListener.waitForSingleNotificationMatching(
+            NotificationCharacteristics(
+                actions = listOf("See issue"),
+                safetySourceId = SINGLE_SOURCE_ID,
+            )
+        )
     }
 
     @Test
@@ -104,7 +110,7 @@ class SafetyCenterNotificationLoggingHelperTests {
                 statusBarNotificationWithChannel.statusBarNotification.notification.contentIntent
             SafetyCenterActivityLauncher.executeBlockAndExit(
                 launchActivity = { PendingIntentSender.send(contentIntent) },
-                block = {} // No action required
+                block = {}, // No action required
             )
         }
     }