diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/wifitests/src/com/android/server/wifi/WifiConfigManagerNewTest.java | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/tests/wifitests/src/com/android/server/wifi/WifiConfigManagerNewTest.java b/tests/wifitests/src/com/android/server/wifi/WifiConfigManagerNewTest.java index a1aa158b1..119f6c744 100644 --- a/tests/wifitests/src/com/android/server/wifi/WifiConfigManagerNewTest.java +++ b/tests/wifitests/src/com/android/server/wifi/WifiConfigManagerNewTest.java @@ -19,9 +19,11 @@ package com.android.server.wifi; import static org.junit.Assert.*; import static org.mockito.Mockito.*; +import android.app.Application; import android.app.test.MockAnswerUtil.AnswerWithArguments; import android.content.Context; import android.content.Intent; +import android.content.pm.ApplicationInfo; import android.content.pm.PackageManager; import android.content.pm.UserInfo; import android.net.IpConfiguration; @@ -1879,12 +1881,52 @@ public class WifiConfigManagerNewTest { assertTrue(mWifiConfigManager.enableNetwork( result.getNetworkId(), true, TEST_CREATOR_UID)); assertEquals(result.getNetworkId(), mWifiConfigManager.getLastSelectedNetwork()); + assertEquals(openNetwork.configKey(), mWifiConfigManager.getLastSelectedNetworkConfigKey()); assertTrue(mWifiConfigManager.removeNetwork(result.getNetworkId(), TEST_CREATOR_UID)); assertEquals( WifiConfiguration.INVALID_NETWORK_ID, mWifiConfigManager.getLastSelectedNetwork()); } + /** + * Verifies that all the networks for the provided app is removed when + * {@link WifiConfigManagerNew#removeNetworksForApp(ApplicationInfo)} is invoked. + */ + @Test + public void testRemoveNetworksForApp() throws Exception { + verifyAddNetworkToWifiConfigManager(WifiConfigurationTestUtil.createOpenNetwork()); + verifyAddNetworkToWifiConfigManager(WifiConfigurationTestUtil.createPskNetwork()); + verifyAddNetworkToWifiConfigManager(WifiConfigurationTestUtil.createWepNetwork()); + + assertFalse(mWifiConfigManager.getConfiguredNetworks().isEmpty()); + + ApplicationInfo app = new ApplicationInfo(); + app.uid = TEST_CREATOR_UID; + app.packageName = TEST_CREATOR_NAME; + assertTrue(mWifiConfigManager.removeNetworksForApp(app)); + + // Ensure all the networks are removed now. + assertTrue(mWifiConfigManager.getConfiguredNetworks().isEmpty()); + } + + /** + * Verifies that all the networks for the provided user is removed when + * {@link WifiConfigManagerNew#removeNetworksForUser(int)} is invoked. + */ + @Test + public void testRemoveNetworksForUser() throws Exception { + verifyAddNetworkToWifiConfigManager(WifiConfigurationTestUtil.createOpenNetwork()); + verifyAddNetworkToWifiConfigManager(WifiConfigurationTestUtil.createPskNetwork()); + verifyAddNetworkToWifiConfigManager(WifiConfigurationTestUtil.createWepNetwork()); + + assertFalse(mWifiConfigManager.getConfiguredNetworks().isEmpty()); + + assertTrue(mWifiConfigManager.removeNetworksForUser(TEST_DEFAULT_USER)); + + // Ensure all the networks are removed now. + assertTrue(mWifiConfigManager.getConfiguredNetworks().isEmpty()); + } + private void createWifiConfigManager() { mWifiConfigManager = new WifiConfigManagerNew( |