Switch to the two-argument version of explicitlySelected.

This allows wifi to pass in the value of noInternetAccessExpected
even if the network was not manually selected by the user.

This is needed on partial connectivity networks where the user
has responded "yes, don't ask again" to the dialog that asks
whether the device should stay connected. Such networks will
validate and thus re-enable autojoin.

When wifi autojoins such a network, ConnectivityService must be
informed that the user has accepted partial connectivity so it
can switch to the network as soon as partial connectivity is
detected.

The code change is very simple and most of this CL consists of
tests. The tests are a bit tricky because WifiNetworkAgent is
private to ClientModeImpl, and thus cannot just be mocked out.

Bug: 130766237
Test: atest FrameworksWifiTests
Change-Id: I04be21743faa88063a2aa89720b70bc61f9d1c64

1 files changed, 119 insertions, 34 deletions

diff --git a/tests/wifitests/src/com/android/server/wifi/ClientModeImplTest.java b/tests/wifitests/src/com/android/server/wifi/ClientModeImplTest.java
index e839796c9..8dfd3e99f 100644
--- a/tests/wifitests/src/com/android/server/wifi/ClientModeImplTest.java
+++ b/tests/wifitests/src/com/android/server/wifi/ClientModeImplTest.java
@@ -123,6 +123,7 @@ import java.lang.reflect.Method;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
+import java.util.Objects;
 import java.util.concurrent.CountDownLatch;
 import java.util.function.Consumer;
 
@@ -328,6 +329,7 @@ public class ClientModeImplTest {
     HandlerThread mP2pThread;
     HandlerThread mSyncThread;
     AsyncChannel  mCmiAsyncChannel;
+    AsyncChannel  mNetworkAgentAsyncChannel;
     TestAlarmManager mAlarmManager;
     MockWifiMonitor mWifiMonitor;
     TestLooper mLooper;
@@ -381,6 +383,8 @@ public class ClientModeImplTest {
     @Mock WifiLockManager mWifiLockManager;
     @Mock AsyncChannel mNullAsyncChannel;
     @Mock CarrierNetworkConfig mCarrierNetworkConfig;
+    @Mock Handler mNetworkAgentHandler;
+
 
     final ArgumentCaptor<WifiNative.InterfaceCallback> mInterfaceCallbackCaptor =
             ArgumentCaptor.forClass(WifiNative.InterfaceCallback.class);
@@ -500,7 +504,8 @@ public class ClientModeImplTest {
         when(mWifiScoreCard.getL2KeyAndGroupHint(any())).thenReturn(new Pair<>(null, null));
     }
 
-    private void registerAsyncChannel(Consumer<AsyncChannel> consumer, Messenger messenger) {
+    private void registerAsyncChannel(Consumer<AsyncChannel> consumer, Messenger messenger,
+            Handler wrappedHandler) {
         final AsyncChannel channel = new AsyncChannel();
         Handler handler = new Handler(mLooper.getLooper()) {
             @Override
@@ -514,7 +519,15 @@ public class ClientModeImplTest {
                         }
                         break;
                     case AsyncChannel.CMD_CHANNEL_DISCONNECTED:
-                        Log.d(TAG, "Command channel disconnected" + this);
+                        Log.d(TAG, "Command channel disconnected " + this);
+                        break;
+                    case AsyncChannel.CMD_CHANNEL_FULLY_CONNECTED:
+                        Log.d(TAG, "Command channel fully connected " + this);
+                        break;
+                    default:
+                        if (wrappedHandler != null) {
+                            wrappedHandler.handleMessage(msg);
+                        }
                         break;
                 }
             }
@@ -524,6 +537,10 @@ public class ClientModeImplTest {
         mLooper.dispatchAll();
     }
 
+    private void registerAsyncChannel(Consumer<AsyncChannel> consumer, Messenger messenger) {
+        registerAsyncChannel(consumer, messenger, null /* wrappedHandler */);
+    }
+
     private void initializeCmi() throws Exception {
         mCmi = new ClientModeImpl(mContext, mFrameworkFacade, mLooper.getLooper(),
                 mUserManager, mWifiInjector, mBackupManagerProxy, mCountryCode, mWifiNative,
@@ -561,6 +578,8 @@ public class ClientModeImplTest {
         mP2pThread = null;
         mSyncThread = null;
         mCmiAsyncChannel = null;
+        mNetworkAgentAsyncChannel = null;
+        mNetworkAgentHandler = null;
         mCmi = null;
     }
 
@@ -2237,25 +2256,30 @@ public class ClientModeImplTest {
         verify(mWifiConfigManager, never()).setRecentFailureAssociationStatus(anyInt(), anyInt());
     }
 
-    /**
-     * Test that the helper method
-     * {@link ClientModeImpl#shouldEvaluateWhetherToSendExplicitlySelected(WifiConfiguration)}
-     * returns true when we connect to the last selected network before expiration of
-     * {@link ClientModeImpl#LAST_SELECTED_NETWORK_EXPIRATION_AGE_MILLIS}.
-     */
-    @Test
-    public void testShouldEvaluateWhetherToSendExplicitlySelected_SameNetworkNotExpired() {
+    private WifiConfiguration makeLastSelectedWifiConfiguration(int lastSelectedNetworkId,
+            long timeSinceLastSelected) {
         long lastSelectedTimestamp = 45666743454L;
-        int lastSelectedNetworkId = 5;
 
         when(mClock.getElapsedSinceBootMillis()).thenReturn(
-                lastSelectedTimestamp
-                        + ClientModeImpl.LAST_SELECTED_NETWORK_EXPIRATION_AGE_MILLIS - 1);
+                lastSelectedTimestamp + timeSinceLastSelected);
         when(mWifiConfigManager.getLastSelectedTimeStamp()).thenReturn(lastSelectedTimestamp);
         when(mWifiConfigManager.getLastSelectedNetwork()).thenReturn(lastSelectedNetworkId);
 
         WifiConfiguration currentConfig = new WifiConfiguration();
         currentConfig.networkId = lastSelectedNetworkId;
+        return currentConfig;
+    }
+
+    /**
+     * Test that the helper method
+     * {@link ClientModeImpl#shouldEvaluateWhetherToSendExplicitlySelected(WifiConfiguration)}
+     * returns true when we connect to the last selected network before expiration of
+     * {@link ClientModeImpl#LAST_SELECTED_NETWORK_EXPIRATION_AGE_MILLIS}.
+     */
+    @Test
+    public void testShouldEvaluateWhetherToSendExplicitlySelected_SameNetworkNotExpired() {
+        WifiConfiguration currentConfig = makeLastSelectedWifiConfiguration(5,
+                ClientModeImpl.LAST_SELECTED_NETWORK_EXPIRATION_AGE_MILLIS - 1);
         assertTrue(mCmi.shouldEvaluateWhetherToSendExplicitlySelected(currentConfig));
     }
 
@@ -2267,17 +2291,8 @@ public class ClientModeImplTest {
      */
     @Test
     public void testShouldEvaluateWhetherToSendExplicitlySelected_SameNetworkExpired() {
-        long lastSelectedTimestamp = 45666743454L;
-        int lastSelectedNetworkId = 5;
-
-        when(mClock.getElapsedSinceBootMillis()).thenReturn(
-                lastSelectedTimestamp
-                        + ClientModeImpl.LAST_SELECTED_NETWORK_EXPIRATION_AGE_MILLIS + 1);
-        when(mWifiConfigManager.getLastSelectedTimeStamp()).thenReturn(lastSelectedTimestamp);
-        when(mWifiConfigManager.getLastSelectedNetwork()).thenReturn(lastSelectedNetworkId);
-
-        WifiConfiguration currentConfig = new WifiConfiguration();
-        currentConfig.networkId = lastSelectedNetworkId;
+        WifiConfiguration currentConfig = makeLastSelectedWifiConfiguration(5,
+                ClientModeImpl.LAST_SELECTED_NETWORK_EXPIRATION_AGE_MILLIS + 1);
         assertFalse(mCmi.shouldEvaluateWhetherToSendExplicitlySelected(currentConfig));
     }
 
@@ -2288,18 +2303,88 @@ public class ClientModeImplTest {
      */
     @Test
     public void testShouldEvaluateWhetherToSendExplicitlySelected_DifferentNetwork() {
-        long lastSelectedTimestamp = 45666743454L;
-        int lastSelectedNetworkId = 5;
+        WifiConfiguration currentConfig = makeLastSelectedWifiConfiguration(5,
+                ClientModeImpl.LAST_SELECTED_NETWORK_EXPIRATION_AGE_MILLIS - 1);
+        currentConfig.networkId = 4;
+        assertFalse(mCmi.shouldEvaluateWhetherToSendExplicitlySelected(currentConfig));
+    }
 
-        when(mClock.getElapsedSinceBootMillis()).thenReturn(
-                lastSelectedTimestamp
-                        + ClientModeImpl.LAST_SELECTED_NETWORK_EXPIRATION_AGE_MILLIS - 1);
-        when(mWifiConfigManager.getLastSelectedTimeStamp()).thenReturn(lastSelectedTimestamp);
-        when(mWifiConfigManager.getLastSelectedNetwork()).thenReturn(lastSelectedNetworkId);
+    private void expectRegisterNetworkAgent() {
+        // Expects that the code calls registerNetworkAgent and provides a way for the test to
+        // verify the messages sent through the NetworkAgent to ConnectivityService.
+        // We cannot just use a mock object here because mWifiNetworkAgent is private to CMI.
+        // TODO: consider exposing WifiNetworkAgent and using mocks.
+        ArgumentCaptor<Messenger> messengerCaptor = ArgumentCaptor.forClass(Messenger.class);
+        verify(mConnectivityManager).registerNetworkAgent(messengerCaptor.capture(),
+                any(NetworkInfo.class), any(LinkProperties.class), any(NetworkCapabilities.class),
+                anyInt(), any(NetworkMisc.class), anyInt());
 
-        WifiConfiguration currentConfig = new WifiConfiguration();
-        currentConfig.networkId = lastSelectedNetworkId - 1;
-        assertFalse(mCmi.shouldEvaluateWhetherToSendExplicitlySelected(currentConfig));
+        registerAsyncChannel((x) -> {
+            mNetworkAgentAsyncChannel = x;
+        }, messengerCaptor.getValue(), mNetworkAgentHandler);
+
+        mNetworkAgentAsyncChannel.sendMessage(AsyncChannel.CMD_CHANNEL_FULL_CONNECTION);
+        mLooper.dispatchAll();
+    }
+
+    private void expectNetworkAgentMessage(int what, int arg1, int arg2, Object obj) {
+        verify(mNetworkAgentHandler).handleMessage(argThat(msg ->
+                what == msg.what && arg1 == msg.arg1 && arg2 == msg.arg2
+                && Objects.equals(obj, msg.obj)));
+    }
+
+    /**
+     * Verify that when a network is explicitly selected, but noInternetAccessExpected is false,
+     * {@link NetworkAgent#explicitlySelected} is called with explicitlySelected=false and
+     * acceptUnvalidated=false.
+     */
+    @Test
+    public void testExplicitlySelected_ExplicitInternetExpected() throws Exception {
+        // Network is explicitly selected.
+        WifiConfiguration config = makeLastSelectedWifiConfiguration(FRAMEWORK_NETWORK_ID,
+                ClientModeImpl.LAST_SELECTED_NETWORK_EXPIRATION_AGE_MILLIS - 1);
+        mConnectedNetwork.noInternetAccessExpected = false;
+
+        connect();
+        expectRegisterNetworkAgent();
+        expectNetworkAgentMessage(NetworkAgent.EVENT_SET_EXPLICITLY_SELECTED,
+                1 /* explicitlySelected */, 0 /* acceptUnvalidated */, null);
+    }
+
+    /**
+     * Verify that when a network is not explicitly selected, but noInternetAccessExpected is true,
+     * {@link NetworkAgent#explicitlySelected} is called with explicitlySelected=false and
+     * acceptUnvalidated=true.
+     */
+    @Test
+    public void testExplicitlySelected_NotExplicitNoInternetExpected() throws Exception {
+        // Network is no longer explicitly selected.
+        WifiConfiguration config = makeLastSelectedWifiConfiguration(FRAMEWORK_NETWORK_ID,
+                ClientModeImpl.LAST_SELECTED_NETWORK_EXPIRATION_AGE_MILLIS + 1);
+        mConnectedNetwork.noInternetAccessExpected = true;
+
+        connect();
+        expectRegisterNetworkAgent();
+        expectNetworkAgentMessage(NetworkAgent.EVENT_SET_EXPLICITLY_SELECTED,
+                0 /* explicitlySelected */, 1 /* acceptUnvalidated */, null);
+    }
+
+    /**
+     * Verify that when a network is explicitly selected, and noInternetAccessExpected is true,
+     * {@link NetworkAgent#explicitlySelected} is called with explicitlySelected=true and
+     * acceptUnvalidated=true.
+     */
+    @Test
+    public void testExplicitlySelected_ExplicitNoInternetExpected() throws Exception {
+        // Network is explicitly selected.
+        WifiConfiguration config = makeLastSelectedWifiConfiguration(FRAMEWORK_NETWORK_ID,
+                ClientModeImpl.LAST_SELECTED_NETWORK_EXPIRATION_AGE_MILLIS - 1);
+        mConnectedNetwork.noInternetAccessExpected = true;
+
+        connect();
+        expectRegisterNetworkAgent();
+        expectNetworkAgentMessage(NetworkAgent.EVENT_SET_EXPLICITLY_SELECTED,
+                1 /* explicitlySelected */, 1 /* acceptUnvalidated */, null);
     }
 
     /**