From ab00a8e4c8fa59f91135faa7570008fb7401e695 Mon Sep 17 00:00:00 2001
From: Ying Xu <yinxu@google.com>
Date: Mon, 25 Mar 2024 11:43:45 -0700
Subject: [PATCH] Fix the test flakiness.

Bug: 326387894
Test: atest ConnectivityCoverageTests:android.net.connectivity.com.android.server.ConnectivityServiceTest
Change-Id: I2fdb0913dcb2f110a40c9bc7307e9fea19fcce4f
---
 .../unit/java/com/android/server/ConnectivityServiceTest.java  | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/unit/java/com/android/server/ConnectivityServiceTest.java b/tests/unit/java/com/android/server/ConnectivityServiceTest.java
index f41d7b288a..1f8a743390 100755
--- a/tests/unit/java/com/android/server/ConnectivityServiceTest.java
+++ b/tests/unit/java/com/android/server/ConnectivityServiceTest.java
@@ -17434,11 +17434,12 @@ public class ConnectivityServiceTest {
         }
 
         mWiFiAgent.disconnect();
-        waitForIdle();
 
         if (expectUnavailable) {
+            testFactory.expectRequestRemove();
             testFactory.assertRequestCountEquals(0);
         } else {
+            testFactory.expectRequestAdd();
             testFactory.assertRequestCountEquals(1);
         }