WifiConfigManagerNew: Hidden network list retrieval

am: 5c6da02cf7

Change-Id: Ib5e49bd86191ccb9f5e8ce0ca26cf631aa45d880

1 files changed, 48 insertions, 0 deletions

diff --git a/tests/wifitests/src/com/android/server/wifi/WifiConfigManagerTest.java b/tests/wifitests/src/com/android/server/wifi/WifiConfigManagerTest.java
index 4edf324df..e6132c166 100644
--- a/tests/wifitests/src/com/android/server/wifi/WifiConfigManagerTest.java
+++ b/tests/wifitests/src/com/android/server/wifi/WifiConfigManagerTest.java
@@ -2045,6 +2045,54 @@ public class WifiConfigManagerTest {
         assertFalse(dumpString.contains(eapNetwork.enterpriseConfig.getPassword()));
     }
 
+    /**
+     * Verifies the ordering of network list generated using
+     * {@link WifiConfigManager#retrieveHiddenNetworkList()}.
+     */
+    @Test
+    public void testRetrieveHiddenList() {
+        // Create and add 3 networks.
+        WifiConfiguration network1 = WifiConfigurationTestUtil.createWepHiddenNetwork();
+        WifiConfiguration network2 = WifiConfigurationTestUtil.createPskHiddenNetwork();
+        WifiConfiguration network3 = WifiConfigurationTestUtil.createOpenHiddenNetwork();
+        verifyAddNetworkToWifiConfigManager(network1);
+        verifyAddNetworkToWifiConfigManager(network2);
+        verifyAddNetworkToWifiConfigManager(network3);
+
+        // Enable all of them.
+        assertTrue(mWifiConfigManager.enableNetwork(network1.networkId, false, TEST_CREATOR_UID));
+        assertTrue(mWifiConfigManager.enableNetwork(network2.networkId, false, TEST_CREATOR_UID));
+        assertTrue(mWifiConfigManager.enableNetwork(network3.networkId, false, TEST_CREATOR_UID));
+
+        // Now set scan results in 2 of them to set the corresponding
+        // {@link NetworkSelectionStatus#mSeenInLastQualifiedNetworkSelection} field.
+        assertTrue(mWifiConfigManager.setNetworkCandidateScanResult(
+                network1.networkId, createScanDetailForNetwork(network1).getScanResult(), 54));
+        assertTrue(mWifiConfigManager.setNetworkCandidateScanResult(
+                network3.networkId, createScanDetailForNetwork(network3).getScanResult(), 54));
+
+        // Now increment |network3|'s association count. This should ensure that this network
+        // is preferred over |network1|.
+        assertTrue(mWifiConfigManager.updateNetworkAfterConnect(network3.networkId));
+
+        // Retrieve the hidden network list & verify the order of the networks returned.
+        List<WifiScanner.ScanSettings.HiddenNetwork> hiddenNetworks =
+                mWifiConfigManager.retrieveHiddenNetworkList();
+        assertEquals(3, hiddenNetworks.size());
+        assertEquals(network3.SSID, hiddenNetworks.get(0).ssid);
+        assertEquals(network1.SSID, hiddenNetworks.get(1).ssid);
+        assertEquals(network2.SSID, hiddenNetworks.get(2).ssid);
+
+        // Now permanently disable |network3|. This should remove network 3 from the list.
+        assertTrue(mWifiConfigManager.disableNetwork(network3.networkId, TEST_CREATOR_UID));
+
+        // Retrieve the hidden network list again & verify the order of the networks returned.
+        hiddenNetworks = mWifiConfigManager.retrieveHiddenNetworkList();
+        assertEquals(2, hiddenNetworks.size());
+        assertEquals(network1.SSID, hiddenNetworks.get(0).ssid);
+        assertEquals(network2.SSID, hiddenNetworks.get(1).ssid);
+    }
+
     private void createWifiConfigManager() {
         mWifiConfigManager =
                 new WifiConfigManager(