Fix CTS test

NotificationAssistantServiceTest#testOnNotificationVisibilityChanged

Bug: 186781761
Test: NotificationAssistantServiceTest
Change-Id: I03b3e6c6bd789257f4866f986eb070c622229267

tests/tests/notificationlegacy/notificationlegacy29/src/android/app/notification/legacy29/cts/NotificationAssistantServiceTest.java

diff --git a/tests/tests/notificationlegacy/notificationlegacy29/src/android/app/notification/legacy29/cts/NotificationAssistantServiceTest.java b/tests/tests/notificationlegacy/notificationlegacy29/src/android/app/notification/legacy29/cts/NotificationAssistantServiceTest.java
index af70d69..4e61610 100644
--- a/tests/tests/notificationlegacy/notificationlegacy29/src/android/app/notification/legacy29/cts/NotificationAssistantServiceTest.java
+++ b/tests/tests/notificationlegacy/notificationlegacy29/src/android/app/notification/legacy29/cts/NotificationAssistantServiceTest.java
@@ -566,14 +566,13 @@
         turnScreenOn();
         mUi.adoptShellPermissionIdentity("android.permission.EXPAND_STATUS_BAR");
 
-        mNotificationAssistantService.resetNotificationVisibilityCounts();
+        sendNotification(1, ICON_ID);
+        Thread.sleep(SLEEP_TIME * 2);
 
         // Initialize as closed
         mStatusBarManager.collapsePanels();
-
-        sendNotification(1, ICON_ID);
-        assertEquals(0, mNotificationAssistantService.notificationVisibleCount);
-        assertEquals(0, mNotificationAssistantService.notificationHiddenCount);
+        Thread.sleep(SLEEP_TIME * 2);
+        mNotificationAssistantService.resetNotificationVisibilityCounts();
 
         mStatusBarManager.expandNotificationsPanel();
         Thread.sleep(SLEEP_TIME * 2);