Merge "Remove use of MockLooper AutoDispatch from non-blocking calls" into mm-wireless-dev

am: 2150117638

* commit '21501176384e4312dce3a51b8ae8ca132f5d9ab4':
  Remove use of MockLooper AutoDispatch from non-blocking calls

1 files changed, 2 insertions, 4 deletions

diff --git a/tests/wifitests/src/com/android/server/wifi/WifiStateMachineTest.java b/tests/wifitests/src/com/android/server/wifi/WifiStateMachineTest.java
index 8f5f7880e..2cdbb0b7d 100644
--- a/tests/wifitests/src/com/android/server/wifi/WifiStateMachineTest.java
+++ b/tests/wifitests/src/com/android/server/wifi/WifiStateMachineTest.java
@@ -765,9 +765,8 @@ public class WifiStateMachineTest {
         dhcpResults.addDns("8.8.8.8");
         dhcpResults.setLeaseDuration(3600);
 
-        mLooper.startAutoDispatch();
         mTestIpManager.injectDhcpSuccess(dhcpResults);
-        mLooper.stopAutoDispatch();
+        mLooper.dispatchAll();
 
         assertEquals("ConnectedState", getCurrentState().getName());
     }
@@ -794,9 +793,8 @@ public class WifiStateMachineTest {
 
         assertEquals("ObtainingIpState", getCurrentState().getName());
 
-        mLooper.startAutoDispatch();
         mTestIpManager.injectDhcpFailure();
-        mLooper.stopAutoDispatch();
+        mLooper.dispatchAll();
 
         assertEquals("DisconnectingState", getCurrentState().getName());
     }