SupplicantPnoScannerTest.java revision 62bc101940ae1f5e60c4d8861a149b900dbf5e5c
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 1462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * limitations under the License 1562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius */ 1662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 1762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piuspackage com.android.server.wifi; 1862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 1962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport android.content.Context; 2062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport android.net.wifi.ScanResult; 2162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport android.net.wifi.WifiConfiguration; 2262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport android.net.wifi.WifiScanner; 2362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport android.test.suitebuilder.annotation.SmallTest; 2462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 2562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport com.android.internal.R; 2662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport com.android.server.wifi.scanner.ChannelHelper.ChannelCollection; 2762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 2862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport org.junit.Before; 2962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport org.junit.Test; 3062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport org.mockito.InOrder; 3162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport org.mockito.Mock; 3262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport org.mockito.MockitoAnnotations; 3362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 3462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport java.util.Arrays; 3562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport java.util.Collections; 3662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport java.util.HashSet; 3762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport java.util.List; 3862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport java.util.Set; 3962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 4062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport static com.android.server.wifi.ScanTestUtil.NativeScanSettingsBuilder; 4162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport static com.android.server.wifi.ScanTestUtil.assertScanDataEquals; 4262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport static org.junit.Assert.assertEquals; 4362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport static org.junit.Assert.assertTrue; 4462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport static org.mockito.Mockito.any; 4562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport static org.mockito.Mockito.anyBoolean; 4662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport static org.mockito.Mockito.anyInt; 4762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport static org.mockito.Mockito.anyObject; 4862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport static org.mockito.Mockito.anyString; 4962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport static org.mockito.Mockito.eq; 5062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport static org.mockito.Mockito.inOrder; 5162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport static org.mockito.Mockito.mock; 5262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport static org.mockito.Mockito.never; 5362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport static org.mockito.Mockito.verify; 5462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport static org.mockito.Mockito.verifyNoMoreInteractions; 5562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piusimport static org.mockito.Mockito.when; 5662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 5762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius/** 5862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * Unit tests for {@link com.android.server.wifi.SupplicantWifiScannerImpl.setPnoList}. 5962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius */ 6062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius@SmallTest 6162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Piuspublic class SupplicantPnoScannerTest { 6262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 6362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius @Mock Context mContext; 6462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius MockAlarmManager mAlarmManager; 6562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius MockWifiMonitor mWifiMonitor; 6662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius MockLooper mLooper; 6762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius @Mock WifiNative mWifiNative; 6862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius MockResources mResources; 6962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius SupplicantWifiScannerImpl mScanner; 7062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 7162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius @Before 7262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius public void setup() throws Exception { 7362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius MockitoAnnotations.initMocks(this); 7462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 7562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mLooper = new MockLooper(); 7662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mAlarmManager = new MockAlarmManager(); 7762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mWifiMonitor = new MockWifiMonitor(); 7862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mResources = new MockResources(); 7962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 8062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mWifiNative.getInterfaceName()).thenReturn("a_test_interface_name"); 8162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mContext.getSystemService(Context.ALARM_SERVICE)) 8262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius .thenReturn(mAlarmManager.getAlarmManager()); 8362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mContext.getResources()).thenReturn(mResources); 8462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 8562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 8662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius /** 8762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * Verify that the HW PNO scan triggers a supplicant PNO scan and invokes the OnPnoNetworkFound 8862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * callback when the scan results are received. 8962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius */ 9062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius @Test 9162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius public void startHwPnoScan() { 9262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius createScannerWithHwPnoScanSupport(); 9362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 9462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.PnoEventHandler pnoEventHandler = mock(WifiNative.PnoEventHandler.class); 9562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.PnoSettings pnoSettings = createDummyPnoSettings(); 9662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius ScanResults scanResults = createDummyScanResults(); 9762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 9862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius InOrder order = inOrder(pnoEventHandler, mWifiNative); 9962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 10062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Start PNO scan 10162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius startSuccessfulPnoScan(null, pnoSettings, null, pnoEventHandler); 10262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 10362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius expectSuccessfulHwPnoScan(order, pnoSettings, pnoEventHandler, scanResults); 10462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 10562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius verifyNoMoreInteractions(pnoEventHandler); 10662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 10762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 10862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius /** 10962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * Verify that we pause & resume HW PNO scan when a single scan is scheduled. 11062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius */ 11162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius @Test 11262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius public void pauseResumeHwPnoScanForSingleScan() { 11362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius createScannerWithHwPnoScanSupport(); 11462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 11562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.PnoEventHandler pnoEventHandler = mock(WifiNative.PnoEventHandler.class); 11662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.PnoSettings pnoSettings = createDummyPnoSettings(); 11762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.ScanEventHandler eventHandler = mock(WifiNative.ScanEventHandler.class); 11862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.ScanSettings settings = createDummyScanSettings(); 11962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius ScanResults scanResults = createDummyScanResults(); 12062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 12162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius InOrder order = inOrder(eventHandler, mWifiNative); 12262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 12362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Start PNO scan 12462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius startSuccessfulPnoScan(null, pnoSettings, null, pnoEventHandler); 12562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 12662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Start single scan 12762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius assertTrue(mScanner.startSingleScan(settings, eventHandler)); 12862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Verify that the PNO scan was paused and single scan runs successfully 12962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius expectSuccessfulSingleScanWithHwPnoEnabled(order, eventHandler, 13062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius expectedBandScanFreqs(WifiScanner.WIFI_BAND_24_GHZ), new HashSet<Integer>(), 13162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius scanResults); 13262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius verifyNoMoreInteractions(eventHandler); 13362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 13462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius order = inOrder(pnoEventHandler, mWifiNative); 13562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 13662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Now verify that PNO scan is resumed successfully 13762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius expectSuccessfulHwPnoScan(order, pnoSettings, pnoEventHandler, scanResults); 13862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius verifyNoMoreInteractions(pnoEventHandler); 13962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 14062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 14162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius /** 14262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * Verify that the SW PNO scan triggers a background scan and invokes the OnPnoNetworkFound 14362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * callback when the scan results are received. 14462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius */ 14562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius @Test 14662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius public void startSwPnoScan() { 14762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius createScannerWithSwPnoScanSupport(); 14862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 14962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.PnoEventHandler pnoEventHandler = mock(WifiNative.PnoEventHandler.class); 15062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.PnoSettings pnoSettings = createDummyPnoSettings(); 15162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.ScanEventHandler scanEventHandler = mock(WifiNative.ScanEventHandler.class); 15262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.ScanSettings scanSettings = createDummyScanSettings(); 15362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius ScanResults scanResults = createDummyScanResults(); 15462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 15562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius InOrder order = inOrder(pnoEventHandler, mWifiNative); 15662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 15762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Start PNO scan 15862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius startSuccessfulPnoScan(scanSettings, pnoSettings, scanEventHandler, pnoEventHandler); 15962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 16062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius expectSuccessfulSwPnoScan(order, pnoEventHandler, scanResults); 16162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 16262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius verifyNoMoreInteractions(pnoEventHandler); 16362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 16462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 16562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius private void createScannerWithHwPnoScanSupport() { 16662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mResources.setBoolean(R.bool.config_wifi_background_scan_support, true); 16762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mScanner = new SupplicantWifiScannerImpl(mContext, mWifiNative, mLooper.getLooper()); 16862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 16962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 17062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius private void createScannerWithSwPnoScanSupport() { 17162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mResources.setBoolean(R.bool.config_wifi_background_scan_support, false); 17262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mScanner = new SupplicantWifiScannerImpl(mContext, mWifiNative, mLooper.getLooper()); 17362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 17462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 17562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius private WifiNative.PnoSettings createDummyPnoSettings() { 17662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.PnoSettings pnoSettings = new WifiNative.PnoSettings(); 17762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius pnoSettings.networkList = new WifiNative.PnoNetwork[2]; 17862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius pnoSettings.networkList[0] = new WifiNative.PnoNetwork(); 17962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius pnoSettings.networkList[0].ssid = "ssid_pno_1"; 18062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius pnoSettings.networkList[0].networkId = 1; 18162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius pnoSettings.networkList[0].priority = 1; 18262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius pnoSettings.networkList[1] = new WifiNative.PnoNetwork(); 18362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius pnoSettings.networkList[1].ssid = "ssid_pno_2"; 18462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius pnoSettings.networkList[1].networkId = 2; 18562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius pnoSettings.networkList[1].priority = 2; 18662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius return pnoSettings; 18762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 18862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 18962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius private WifiNative.ScanSettings createDummyScanSettings() { 19062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.ScanSettings settings = new NativeScanSettingsBuilder() 19162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius .withBasePeriod(10000) 19262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius .withMaxApPerScan(10) 19362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius .addBucketWithBand(10000, WifiScanner.REPORT_EVENT_AFTER_EACH_SCAN, 19462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiScanner.WIFI_BAND_24_GHZ) 19562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius .build(); 19662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius return settings; 19762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 19862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 19962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius private ScanResults createDummyScanResults() { 20062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius return ScanResults.create(0, 2400, 2450, 2450, 2400, 2450, 2450, 2400, 2450, 2450); 20162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 20262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 20362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius private void startSuccessfulPnoScan(WifiNative.ScanSettings scanSettings, 20462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.PnoSettings pnoSettings, WifiNative.ScanEventHandler scanEventHandler, 20562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.PnoEventHandler pnoEventHandler) { 20662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mWifiNative.setNetworkVariable(anyInt(), anyString(), anyString())).thenReturn(true); 20762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mWifiNative.enableNetworkWithoutConnect(anyInt())).thenReturn(true); 20862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Scans succeed 20962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mWifiNative.scan(any(Set.class), any(Set.class))).thenReturn(true); 21062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mWifiNative.enableBackgroundScan(anyBoolean(), anyObject())).thenReturn(true); 21162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 21262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius assertTrue(mScanner.setPnoList(pnoSettings, pnoEventHandler)); 21362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // This should happen only for SW PNO scan 21462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius if (mScanner.shouldScheduleBackgroundScanForPno()) { 21562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius assertTrue(mScanner.startBatchedScan(scanSettings, scanEventHandler)); 21662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 21762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 21862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 21962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius private Set<Integer> expectedBandScanFreqs(int band) { 22062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius ChannelCollection collection = mScanner.getChannelHelper().createChannelCollection(); 22162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius collection.addBand(band); 22262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius return collection.getSupplicantScanFreqs(); 22362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 22462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 22562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius /** 22662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * Verify that the PNO scan was successfully started. 22762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius */ 22862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius private void expectSuccessfulHwPnoScan(InOrder order, WifiNative.PnoSettings pnoSettings, 22962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.PnoEventHandler eventHandler, ScanResults scanResults) { 23062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius for (int i = 0; i < pnoSettings.networkList.length; i++) { 23162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.PnoNetwork network = pnoSettings.networkList[i]; 23262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius order.verify(mWifiNative).setNetworkVariable(network.networkId, 23362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiConfiguration.priorityVarName, Integer.toString(network.priority)); 23462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius order.verify(mWifiNative).enableNetworkWithoutConnect(network.networkId); 23562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 23662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Verify HW PNO scan started 23762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius order.verify(mWifiNative).enableBackgroundScan(true, null); 23862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 23962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Setup scan results 24062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mWifiNative.getScanResults()).thenReturn(scanResults.getScanDetailArrayList()); 24162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 24262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Notify scan has finished 24362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mWifiMonitor.sendMessage(mWifiNative.getInterfaceName(), WifiMonitor.SCAN_RESULTS_EVENT); 24462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius assertEquals("dispatch message after results event", 1, mLooper.dispatchAll()); 24562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 24662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius order.verify(eventHandler).onPnoNetworkFound(scanResults.getRawScanResults()); 24762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 24862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 24962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius /** 25062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * Verify that the single scan results were delivered and that the PNO scan was paused and 25162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * resumed either side of it. 25262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius */ 25362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius private void expectSuccessfulSingleScanWithHwPnoEnabled(InOrder order, 25462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.ScanEventHandler eventHandler, Set<Integer> expectedScanFreqs, 25562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius Set<Integer> expectedHiddenNetIds, ScanResults scanResults) { 25662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Pause PNO scan first 25762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius order.verify(mWifiNative).enableBackgroundScan(false, null); 25862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 25962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius order.verify(mWifiNative).scan(eq(expectedScanFreqs), eq(expectedHiddenNetIds)); 26062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 26162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mWifiNative.getScanResults()).thenReturn(scanResults.getScanDetailArrayList()); 26262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 26362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Notify scan has finished 26462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mWifiMonitor.sendMessage(mWifiNative.getInterfaceName(), WifiMonitor.SCAN_RESULTS_EVENT); 26562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius assertEquals("dispatch message after results event", 1, mLooper.dispatchAll()); 26662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 26762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius order.verify(eventHandler).onScanStatus(WifiNative.WIFI_SCAN_RESULTS_AVAILABLE); 26862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius assertScanDataEquals(scanResults.getScanData(), mScanner.getLatestSingleScanResults()); 26962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 27062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Resume PNO scan after the single scan results are received 27162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius order.verify(mWifiNative).enableBackgroundScan(true, null); 27262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 27362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 27462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius /** 27562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * Verify that the SW PNO scan was successfully started. 27662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * This is basically ensuring that the background scan runs successfully and invokes the 27762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius * onPnoNetworkFound callback. 27862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius */ 27962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius private void expectSuccessfulSwPnoScan(InOrder order, 28062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius WifiNative.PnoEventHandler pnoEventHandler, ScanResults scanResults) { 28162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 28262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Verify scan started 28362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius order.verify(mWifiNative).scan(any(Set.class), any(Set.class)); 28462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 28562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Make sure that HW PNO scan was not started 28662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius verify(mWifiNative, never()).enableBackgroundScan(anyBoolean(), anyObject()); 28762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 28862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Setup scan results 28962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius when(mWifiNative.getScanResults()).thenReturn(scanResults.getScanDetailArrayList()); 29062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 29162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Notify scan has finished 29262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius mWifiMonitor.sendMessage(mWifiNative.getInterfaceName(), WifiMonitor.SCAN_RESULTS_EVENT); 29362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius assertEquals("dispatch message after results event", 1, mLooper.dispatchAll()); 29462bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 29562bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius // Scan results are sorted when using background scan 29662bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius List<ScanResult> scanResultList = Arrays.asList(scanResults.getRawScanResults()); 29762bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius Collections.sort(scanResultList, WifiScannerImpl.SCAN_RESULT_SORT_COMPARATOR); 29862bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius ScanResult[] scanResultArray = new ScanResult[scanResultList.size()]; 29962bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius scanResultList.toArray(scanResultArray); 30062bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius 30162bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius order.verify(pnoEventHandler).onPnoNetworkFound(scanResultArray); 30262bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius } 30362bc101940ae1f5e60c4d8861a149b900dbf5e5cRoshan Pius} 304