WificondPnoScannerTest.java revision a8367288377cbaed6371256ca837b7aa22280706
162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius/* 262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * Copyright (C) 2016 The Android Open Source Project 362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * 462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * Licensed under the Apache License, Version 2.0 (the "License"); 562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * you may not use this file except in compliance with the License. 662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * You may obtain a copy of the License at 762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * 862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * http://www.apache.org/licenses/LICENSE-2.0 962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * 1062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * Unless required by applicable law or agreed to in writing, software 1162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * distributed under the License is distributed on an "AS IS" BASIS, 1262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 1362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * See the License for the specific language governing permissions and 14a8367288377cbaed6371256ca837b7aa22280706Mitchell Wills * limitations under the License. 1562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius */ 1662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 17a8367288377cbaed6371256ca837b7aa22280706Mitchell Willspackage com.android.server.wifi.scanner; 1862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 19e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Piusimport static com.android.server.wifi.ScanTestUtil.NativeScanSettingsBuilder; 20e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Piusimport static com.android.server.wifi.ScanTestUtil.assertScanDataEquals; 21e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius 22e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Piusimport static org.junit.Assert.*; 23e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Piusimport static org.mockito.Mockito.*; 24e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius 2562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport android.content.Context; 2662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport android.net.wifi.WifiConfiguration; 2762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport android.net.wifi.WifiScanner; 2862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport android.test.suitebuilder.annotation.SmallTest; 2962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 3062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport com.android.internal.R; 31a8367288377cbaed6371256ca837b7aa22280706Mitchell Willsimport com.android.server.wifi.MockAlarmManager; 32a8367288377cbaed6371256ca837b7aa22280706Mitchell Willsimport com.android.server.wifi.MockLooper; 33a8367288377cbaed6371256ca837b7aa22280706Mitchell Willsimport com.android.server.wifi.MockResources; 34a8367288377cbaed6371256ca837b7aa22280706Mitchell Willsimport com.android.server.wifi.MockWifiMonitor; 35a8367288377cbaed6371256ca837b7aa22280706Mitchell Willsimport com.android.server.wifi.ScanResults; 36a8367288377cbaed6371256ca837b7aa22280706Mitchell Willsimport com.android.server.wifi.WifiMonitor; 37a8367288377cbaed6371256ca837b7aa22280706Mitchell Willsimport com.android.server.wifi.WifiNative; 3862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport com.android.server.wifi.scanner.ChannelHelper.ChannelCollection; 3962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 4062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport org.junit.Before; 4162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport org.junit.Test; 4262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport org.mockito.InOrder; 4362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport org.mockito.Mock; 4462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport org.mockito.MockitoAnnotations; 4562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 4662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport java.util.HashSet; 4762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport java.util.Set; 4862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 4962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 5062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius/** 51a8367288377cbaed6371256ca837b7aa22280706Mitchell Wills * Unit tests for {@link com.android.server.wifi.scanner.SupplicantWifiScannerImpl.setPnoList}. 5262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius */ 5362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius@SmallTest 5462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piuspublic class SupplicantPnoScannerTest { 5562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 5662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius @Mock Context mContext; 5762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius MockAlarmManager mAlarmManager; 5862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius MockWifiMonitor mWifiMonitor; 5962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius MockLooper mLooper; 6062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius @Mock WifiNative mWifiNative; 6162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius MockResources mResources; 6262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius SupplicantWifiScannerImpl mScanner; 6362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 6462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius @Before 6562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius public void setup() throws Exception { 6662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius MockitoAnnotations.initMocks(this); 6762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 6862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mLooper = new MockLooper(); 6962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mAlarmManager = new MockAlarmManager(); 7062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mWifiMonitor = new MockWifiMonitor(); 7162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mResources = new MockResources(); 7262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 7362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mWifiNative.getInterfaceName()).thenReturn("a_test_interface_name"); 7462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mContext.getSystemService(Context.ALARM_SERVICE)) 7562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius .thenReturn(mAlarmManager.getAlarmManager()); 7662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mContext.getResources()).thenReturn(mResources); 7762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 7862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 7962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius /** 806c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius * Verify that the HW disconnected PNO scan triggers a supplicant PNO scan and invokes the 816c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius * OnPnoNetworkFound callback when the scan results are received. 8262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius */ 8362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius @Test 84dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius public void startHwDisconnectedPnoScan() { 8562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius createScannerWithHwPnoScanSupport(); 8662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 8762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.PnoEventHandler pnoEventHandler = mock(WifiNative.PnoEventHandler.class); 88dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius WifiNative.PnoSettings pnoSettings = createDummyPnoSettings(false); 8962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius ScanResults scanResults = createDummyScanResults(); 9062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 9162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius InOrder order = inOrder(pnoEventHandler, mWifiNative); 9262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Start PNO scan 9362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius startSuccessfulPnoScan(null, pnoSettings, null, pnoEventHandler); 94dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius expectSuccessfulHwDisconnectedPnoScan(order, pnoSettings, pnoEventHandler, scanResults); 9562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius verifyNoMoreInteractions(pnoEventHandler); 9662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 9762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 9862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius /** 996c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius * Verify that we pause & resume HW PNO scan when a single scan is scheduled and invokes the 1006c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius * OnPnoNetworkFound callback when the scan results are received. 10162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius */ 10262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius @Test 103dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius public void pauseResumeHwDisconnectedPnoScanForSingleScan() { 10462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius createScannerWithHwPnoScanSupport(); 10562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 10662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.PnoEventHandler pnoEventHandler = mock(WifiNative.PnoEventHandler.class); 107dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius WifiNative.PnoSettings pnoSettings = createDummyPnoSettings(false); 10862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.ScanEventHandler eventHandler = mock(WifiNative.ScanEventHandler.class); 10962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.ScanSettings settings = createDummyScanSettings(); 11062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius ScanResults scanResults = createDummyScanResults(); 11162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 11262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius InOrder order = inOrder(eventHandler, mWifiNative); 11362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Start PNO scan 11462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius startSuccessfulPnoScan(null, pnoSettings, null, pnoEventHandler); 11562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Start single scan 11662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius assertTrue(mScanner.startSingleScan(settings, eventHandler)); 11762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Verify that the PNO scan was paused and single scan runs successfully 11862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius expectSuccessfulSingleScanWithHwPnoEnabled(order, eventHandler, 11962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius expectedBandScanFreqs(WifiScanner.WIFI_BAND_24_GHZ), new HashSet<Integer>(), 12062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius scanResults); 12162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius verifyNoMoreInteractions(eventHandler); 12262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 12362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius order = inOrder(pnoEventHandler, mWifiNative); 124e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius // Resume PNO scan after the single scan results are received and PNO monitor debounce 125e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius // alarm fires. 126e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius assertTrue("dispatch pno monitor alarm", 127e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius mAlarmManager.dispatch( 128e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius SupplicantWifiScannerImpl.HwPnoDebouncer.PNO_DEBOUNCER_ALARM_TAG)); 129e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius assertEquals("dispatch message after alarm", 1, mLooper.dispatchAll()); 13062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Now verify that PNO scan is resumed successfully 131dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius expectSuccessfulHwDisconnectedPnoScan(order, pnoSettings, pnoEventHandler, scanResults); 13262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius verifyNoMoreInteractions(pnoEventHandler); 13362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 13462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 13562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius /** 1366c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius * Verify that the SW disconnected PNO scan triggers a background scan and invokes the 1376c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius * background scan callbacks when scan results are received. 13862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius */ 13962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius @Test 140dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius public void startSwDisconnectedPnoScan() { 14162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius createScannerWithSwPnoScanSupport(); 1426c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius doSuccessfulSwPnoScanTest(false); 143dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius } 144dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius 145dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius /** 1466c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius * Verify that the HW connected PNO scan triggers a background scan and invokes the 1476c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius * background scan callbacks when scan results are received. 148dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius */ 149dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius @Test 150dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius public void startHwConnectedPnoScan() { 151dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius createScannerWithHwPnoScanSupport(); 1526c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius doSuccessfulSwPnoScanTest(true); 153dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius } 154dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius 155dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius /** 1566c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius * Verify that the SW connected PNO scan triggers a background scan and invokes the 1576c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius * background scan callbacks when scan results are received. 158dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius */ 159dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius @Test 160dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius public void startSwConnectedPnoScan() { 161dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius createScannerWithSwPnoScanSupport(); 1626c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius doSuccessfulSwPnoScanTest(true); 1636c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius } 164dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius 165e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius /** 166e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius * Verify that the HW PNO delayed failure cleans up the scan settings cleanly. 167e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius * 1. Start Hw PNO. 168e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius * 2. Start Single Scan which should pause PNO scan. 169e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius * 3. Fail the PNO scan resume and verify that the OnPnoScanFailed callback is invoked. 170e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius * 4. Now restart a new PNO scan to ensure that the failure was cleanly handled. 171e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius */ 172e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius @Test 173e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius public void delayedHwDisconnectedPnoScanFailure() { 174e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius createScannerWithHwPnoScanSupport(); 175e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius 176e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius WifiNative.PnoEventHandler pnoEventHandler = mock(WifiNative.PnoEventHandler.class); 177e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius WifiNative.PnoSettings pnoSettings = createDummyPnoSettings(false); 178e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius WifiNative.ScanEventHandler eventHandler = mock(WifiNative.ScanEventHandler.class); 179e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius WifiNative.ScanSettings settings = createDummyScanSettings(); 180e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius ScanResults scanResults = createDummyScanResults(); 181e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius 182e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius InOrder order = inOrder(eventHandler, mWifiNative); 183e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius // Start PNO scan 184e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius startSuccessfulPnoScan(null, pnoSettings, null, pnoEventHandler); 185e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius // Start single scan 186e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius assertTrue(mScanner.startSingleScan(settings, eventHandler)); 187e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius // Verify that the PNO scan was paused and single scan runs successfully 188e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius expectSuccessfulSingleScanWithHwPnoEnabled(order, eventHandler, 189e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius expectedBandScanFreqs(WifiScanner.WIFI_BAND_24_GHZ), new HashSet<Integer>(), 190e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius scanResults); 191e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius verifyNoMoreInteractions(eventHandler); 192e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius 193e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius // Fail the PNO resume and check that the OnPnoScanFailed callback is invoked. 194e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius order = inOrder(pnoEventHandler, mWifiNative); 195e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius when(mWifiNative.setPnoScan(true)).thenReturn(false); 196e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius assertTrue("dispatch pno monitor alarm", 197e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius mAlarmManager.dispatch( 198e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius SupplicantWifiScannerImpl.HwPnoDebouncer.PNO_DEBOUNCER_ALARM_TAG)); 199e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius assertEquals("dispatch message after alarm", 1, mLooper.dispatchAll()); 200e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius order.verify(pnoEventHandler).onPnoScanFailed(); 201e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius verifyNoMoreInteractions(pnoEventHandler); 202e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius 203e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius // Add a new PNO scan request 204e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius startSuccessfulPnoScan(null, pnoSettings, null, pnoEventHandler); 205e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius assertTrue("dispatch pno monitor alarm", 206e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius mAlarmManager.dispatch( 207e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius SupplicantWifiScannerImpl.HwPnoDebouncer.PNO_DEBOUNCER_ALARM_TAG)); 208e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius assertEquals("dispatch message after alarm", 1, mLooper.dispatchAll()); 209e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius expectSuccessfulHwDisconnectedPnoScan(order, pnoSettings, pnoEventHandler, scanResults); 210e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius verifyNoMoreInteractions(pnoEventHandler); 211e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius } 212e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius 2136c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius private void doSuccessfulSwPnoScanTest(boolean isConnectedPno) { 214dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius WifiNative.PnoEventHandler pnoEventHandler = mock(WifiNative.PnoEventHandler.class); 2156c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius WifiNative.PnoSettings pnoSettings = createDummyPnoSettings(isConnectedPno); 21662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.ScanEventHandler scanEventHandler = mock(WifiNative.ScanEventHandler.class); 21762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.ScanSettings scanSettings = createDummyScanSettings(); 21862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius ScanResults scanResults = createDummyScanResults(); 21962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 2206c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius InOrder order = inOrder(scanEventHandler, mWifiNative); 22162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 22262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Start PNO scan 22362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius startSuccessfulPnoScan(scanSettings, pnoSettings, scanEventHandler, pnoEventHandler); 22462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 2256c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius expectSuccessfulSwPnoScan(order, scanEventHandler, scanResults); 22662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 22762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius verifyNoMoreInteractions(pnoEventHandler); 22862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 22962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 23062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius private void createScannerWithHwPnoScanSupport() { 23162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mResources.setBoolean(R.bool.config_wifi_background_scan_support, true); 23262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mScanner = new SupplicantWifiScannerImpl(mContext, mWifiNative, mLooper.getLooper()); 23362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 23462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 23562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius private void createScannerWithSwPnoScanSupport() { 23662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mResources.setBoolean(R.bool.config_wifi_background_scan_support, false); 23762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mScanner = new SupplicantWifiScannerImpl(mContext, mWifiNative, mLooper.getLooper()); 23862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 23962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 240dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius private WifiNative.PnoSettings createDummyPnoSettings(boolean isConnected) { 24162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.PnoSettings pnoSettings = new WifiNative.PnoSettings(); 242dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius pnoSettings.isConnected = isConnected; 24362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius pnoSettings.networkList = new WifiNative.PnoNetwork[2]; 24462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius pnoSettings.networkList[0] = new WifiNative.PnoNetwork(); 24562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius pnoSettings.networkList[0].ssid = "ssid_pno_1"; 24662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius pnoSettings.networkList[0].networkId = 1; 24762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius pnoSettings.networkList[0].priority = 1; 24862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius pnoSettings.networkList[1] = new WifiNative.PnoNetwork(); 24962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius pnoSettings.networkList[1].ssid = "ssid_pno_2"; 25062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius pnoSettings.networkList[1].networkId = 2; 25162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius pnoSettings.networkList[1].priority = 2; 25262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius return pnoSettings; 25362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 25462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 25562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius private WifiNative.ScanSettings createDummyScanSettings() { 25662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.ScanSettings settings = new NativeScanSettingsBuilder() 25762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius .withBasePeriod(10000) 25862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius .withMaxApPerScan(10) 25962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius .addBucketWithBand(10000, WifiScanner.REPORT_EVENT_AFTER_EACH_SCAN, 26062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiScanner.WIFI_BAND_24_GHZ) 26162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius .build(); 26262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius return settings; 26362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 26462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 26562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius private ScanResults createDummyScanResults() { 26662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius return ScanResults.create(0, 2400, 2450, 2450, 2400, 2450, 2450, 2400, 2450, 2450); 26762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 26862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 26962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius private void startSuccessfulPnoScan(WifiNative.ScanSettings scanSettings, 27062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.PnoSettings pnoSettings, WifiNative.ScanEventHandler scanEventHandler, 27162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.PnoEventHandler pnoEventHandler) { 27262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mWifiNative.setNetworkVariable(anyInt(), anyString(), anyString())).thenReturn(true); 27362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mWifiNative.enableNetworkWithoutConnect(anyInt())).thenReturn(true); 27462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Scans succeed 27562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mWifiNative.scan(any(Set.class), any(Set.class))).thenReturn(true); 276e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius when(mWifiNative.setPnoScan(anyBoolean())).thenReturn(true); 2776c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius 278e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius if (mScanner.isHwPnoSupported(pnoSettings.isConnected)) { 279e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius // This should happen only for HW PNO scan 280e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius assertTrue(mScanner.setHwPnoList(pnoSettings, pnoEventHandler)); 281e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius } else { 282e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius // This should happen only for SW PNO scan 28362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius assertTrue(mScanner.startBatchedScan(scanSettings, scanEventHandler)); 284e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius 28562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 28662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 28762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 28862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius private Set<Integer> expectedBandScanFreqs(int band) { 28962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius ChannelCollection collection = mScanner.getChannelHelper().createChannelCollection(); 29062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius collection.addBand(band); 29162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius return collection.getSupplicantScanFreqs(); 29262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 29362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 29462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius /** 29562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * Verify that the PNO scan was successfully started. 29662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius */ 297dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius private void expectSuccessfulHwDisconnectedPnoScan(InOrder order, 298dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius WifiNative.PnoSettings pnoSettings, WifiNative.PnoEventHandler eventHandler, 299dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius ScanResults scanResults) { 30062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius for (int i = 0; i < pnoSettings.networkList.length; i++) { 30162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.PnoNetwork network = pnoSettings.networkList[i]; 30262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius order.verify(mWifiNative).setNetworkVariable(network.networkId, 30362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiConfiguration.priorityVarName, Integer.toString(network.priority)); 30462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius order.verify(mWifiNative).enableNetworkWithoutConnect(network.networkId); 30562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 30662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Verify HW PNO scan started 307e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius order.verify(mWifiNative).setPnoScan(true); 30862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 30962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Setup scan results 31062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mWifiNative.getScanResults()).thenReturn(scanResults.getScanDetailArrayList()); 31162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 31262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Notify scan has finished 31362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mWifiMonitor.sendMessage(mWifiNative.getInterfaceName(), WifiMonitor.SCAN_RESULTS_EVENT); 31462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius assertEquals("dispatch message after results event", 1, mLooper.dispatchAll()); 31562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 31662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius order.verify(eventHandler).onPnoNetworkFound(scanResults.getRawScanResults()); 31762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 31862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 31962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius /** 32062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * Verify that the single scan results were delivered and that the PNO scan was paused and 32162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * resumed either side of it. 32262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius */ 32362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius private void expectSuccessfulSingleScanWithHwPnoEnabled(InOrder order, 32462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.ScanEventHandler eventHandler, Set<Integer> expectedScanFreqs, 32562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius Set<Integer> expectedHiddenNetIds, ScanResults scanResults) { 32662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Pause PNO scan first 327e7ba2963bedf426a1d8ba09ab535260ef364512bRoshan Pius order.verify(mWifiNative).setPnoScan(false); 32862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 32962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius order.verify(mWifiNative).scan(eq(expectedScanFreqs), eq(expectedHiddenNetIds)); 33062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 33162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mWifiNative.getScanResults()).thenReturn(scanResults.getScanDetailArrayList()); 33262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 33362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Notify scan has finished 33462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mWifiMonitor.sendMessage(mWifiNative.getInterfaceName(), WifiMonitor.SCAN_RESULTS_EVENT); 33562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius assertEquals("dispatch message after results event", 1, mLooper.dispatchAll()); 33662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 33762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius order.verify(eventHandler).onScanStatus(WifiNative.WIFI_SCAN_RESULTS_AVAILABLE); 33862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius assertScanDataEquals(scanResults.getScanData(), mScanner.getLatestSingleScanResults()); 33962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 34062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 34162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius /** 342dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius * Verify that the SW PNO scan was successfully started. This could either be disconnected 343dcd877d6c143db557884993ea437e2a432cb0ba3Roshan Pius * or connected PNO. 3446c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius * This is basically ensuring that the background scan runs successfully and returns the 3456c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius * expected result. 34662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius */ 34762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius private void expectSuccessfulSwPnoScan(InOrder order, 3486c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius WifiNative.ScanEventHandler eventHandler, ScanResults scanResults) { 34962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 35062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Verify scan started 35162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius order.verify(mWifiNative).scan(any(Set.class), any(Set.class)); 35262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 35362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Make sure that HW PNO scan was not started 3548ba794562167643688ee38352f98345403fa22c8Roshan Pius verify(mWifiNative, never()).setPnoScan(anyBoolean()); 35562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 35662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Setup scan results 35762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mWifiNative.getScanResults()).thenReturn(scanResults.getScanDetailArrayList()); 35862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 35962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Notify scan has finished 36062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mWifiMonitor.sendMessage(mWifiNative.getInterfaceName(), WifiMonitor.SCAN_RESULTS_EVENT); 36162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius assertEquals("dispatch message after results event", 1, mLooper.dispatchAll()); 36262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 3636c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius // Verify background scan results delivered 3646c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius order.verify(eventHandler).onScanStatus(WifiNative.WIFI_SCAN_RESULTS_AVAILABLE); 3656c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius WifiScanner.ScanData[] scanData = mScanner.getLatestBatchedScanResults(true); 3666c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius WifiScanner.ScanData lastScanData = scanData[scanData.length -1]; 3676c5018cef1eb7acbcfa7fc6c9b7c018bab7ba7baRoshan Pius assertScanDataEquals(scanResults.getScanData(), lastScanData); 36862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 36962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius} 370