Unit test fixes for WifiConfigManager/WifiVendorHal am: 28102ffacd am: 8127b1ea3c

am: 052c06786c

Change-Id: I3778b70263fa8d88b3ddb9b9e9bc943ff30cafcd

2 files changed, 2 insertions, 4 deletions

diff --git a/tests/wifitests/src/com/android/server/wifi/WifiConfigManagerTest.java b/tests/wifitests/src/com/android/server/wifi/WifiConfigManagerTest.java
index 20b89dfc3..6b5c79a17 100644
--- a/tests/wifitests/src/com/android/server/wifi/WifiConfigManagerTest.java
+++ b/tests/wifitests/src/com/android/server/wifi/WifiConfigManagerTest.java
@@ -166,7 +166,7 @@ public class WifiConfigManagerTest {
         }).when(mPackageManager).getPackageUidAsUser(anyString(), anyInt(), anyInt());
 
         when(mWifiKeyStore
-                .updateNetworkKeys(any(WifiConfiguration.class), any(WifiConfiguration.class)))
+                .updateNetworkKeys(any(WifiConfiguration.class), any()))
                 .thenReturn(true);
 
         when(mWifiConfigStore.areStoresPresent()).thenReturn(true);
diff --git a/tests/wifitests/src/com/android/server/wifi/WifiVendorHalTest.java b/tests/wifitests/src/com/android/server/wifi/WifiVendorHalTest.java
index 6f1d69c69..36fe7d2df 100644
--- a/tests/wifitests/src/com/android/server/wifi/WifiVendorHalTest.java
+++ b/tests/wifitests/src/com/android/server/wifi/WifiVendorHalTest.java
@@ -56,7 +56,6 @@ import android.net.wifi.WifiScanner;
 import android.net.wifi.WifiSsid;
 import android.net.wifi.WifiWakeReasonAndCounts;
 import android.os.HandlerThread;
-import android.os.Looper;
 import android.os.RemoteException;
 import android.util.Pair;
 
@@ -176,8 +175,7 @@ public class WifiVendorHalTest {
         mWifiVendorHal.initialize();
         ArgumentCaptor<WifiVendorHal.HalDeviceManagerStatusListener> hdmCallbackCaptor =
                 ArgumentCaptor.forClass(WifiVendorHal.HalDeviceManagerStatusListener.class);
-        verify(mHalDeviceManager).registerStatusListener(
-                hdmCallbackCaptor.capture(), any(Looper.class));
+        verify(mHalDeviceManager).registerStatusListener(hdmCallbackCaptor.capture(), any());
         mHalDeviceManagerStatusCallbacks = hdmCallbackCaptor.getValue();
 
     }