Change NetworkFactory to NetworkProvider in wifi. am: f4f0b07626 am: 0eeb9f9e4a

Change-Id: Id7e55e2fee996d1d6d792731888126083638b2fd

2 files changed, 64 insertions, 94 deletions

diff --git a/tests/wifitests/src/com/android/server/wifi/WifiNetworkFactoryTest.java b/tests/wifitests/src/com/android/server/wifi/WifiNetworkFactoryTest.java
index c6341873d..7f78cd5e5 100644
--- a/tests/wifitests/src/com/android/server/wifi/WifiNetworkFactoryTest.java
+++ b/tests/wifitests/src/com/android/server/wifi/WifiNetworkFactoryTest.java
@@ -39,7 +39,7 @@ import android.content.pm.PackageManager;
 import android.net.ConnectivityManager;
 import android.net.MacAddress;
 import android.net.NetworkCapabilities;
-import android.net.NetworkFactory;
+import android.net.NetworkProvider;
 import android.net.NetworkRequest;
 import android.net.wifi.INetworkRequestMatchCallback;
 import android.net.wifi.INetworkRequestUserSelectionCallback;
@@ -63,7 +63,6 @@ import android.test.suitebuilder.annotation.SmallTest;
 import android.util.Pair;
 import android.util.Xml;
 
-import com.android.internal.util.AsyncChannel;
 import com.android.internal.util.FastXmlSerializer;
 import com.android.server.wifi.WifiNetworkFactory.AccessPoint;
 import com.android.server.wifi.nano.WifiMetricsProto;
@@ -135,7 +134,7 @@ public class WifiNetworkFactoryTest {
     @Mock ClientModeImpl mClientModeImpl;
     @Mock ConnectivityManager mConnectivityManager;
     @Mock WifiMetrics mWifiMetrics;
-    @Mock Messenger mConnectivityMessenger;
+    @Mock NetworkProvider mNetworkProvider;
     NetworkCapabilities mNetworkCapabilities;
     TestLooper mLooper;
     NetworkRequest mNetworkRequest;
@@ -201,17 +200,14 @@ public class WifiNetworkFactoryTest {
         assertNotNull(mDataSource);
         mNetworkRequestStoreData = new NetworkRequestStoreData(mDataSource);
 
-        // Register and establish full connection to connectivity manager.
+        // Register factory with connectivity manager.
         mWifiNetworkFactory.register();
-        ArgumentCaptor<Messenger> messengerArgumentCaptor =
-                ArgumentCaptor.forClass(Messenger.class);
-        verify(mConnectivityManager).registerNetworkFactory(
-                messengerArgumentCaptor.capture(), anyString());
-        assertNotNull(messengerArgumentCaptor.getValue());
-        Message fullConnectionMsg = Message.obtain();
-        fullConnectionMsg.what = AsyncChannel.CMD_CHANNEL_FULL_CONNECTION;
-        fullConnectionMsg.replyTo = mConnectivityMessenger;
-        messengerArgumentCaptor.getValue().send(fullConnectionMsg);
+        ArgumentCaptor<NetworkProvider> networkProviderArgumentCaptor =
+                ArgumentCaptor.forClass(NetworkProvider.class);
+        verify(mConnectivityManager).registerNetworkProvider(
+                networkProviderArgumentCaptor.capture());
+        mNetworkProvider = networkProviderArgumentCaptor.getValue();
+        assertNotNull(mNetworkProvider);
         mLooper.dispatchAll();
 
         mNetworkRequest = new NetworkRequest.Builder()
@@ -290,7 +286,7 @@ public class WifiNetworkFactoryTest {
 
         assertFalse(mWifiNetworkFactory.acceptRequest(mNetworkRequest, 0));
         mLooper.dispatchAll();
-        verifyUnfullfillableDispatched(mConnectivityMessenger);
+        verify(mConnectivityManager).declareNetworkRequestUnfulfillable(eq(mNetworkRequest));
     }
 
     /**
@@ -308,7 +304,7 @@ public class WifiNetworkFactoryTest {
 
         assertFalse(mWifiNetworkFactory.acceptRequest(mNetworkRequest, 0));
         mLooper.dispatchAll();
-        verifyUnfullfillableDispatched(mConnectivityMessenger);
+        verify(mConnectivityManager).declareNetworkRequestUnfulfillable(eq(mNetworkRequest));
     }
 
     /**
@@ -325,7 +321,7 @@ public class WifiNetworkFactoryTest {
 
         assertFalse(mWifiNetworkFactory.acceptRequest(mNetworkRequest, 0));
         mLooper.dispatchAll();
-        verifyUnfullfillableDispatched(mConnectivityMessenger);
+        verify(mConnectivityManager).declareNetworkRequestUnfulfillable(eq(mNetworkRequest));
     }
 
     /**
@@ -452,7 +448,7 @@ public class WifiNetworkFactoryTest {
         mNetworkRequest.networkCapabilities.setNetworkSpecifier(specifier2);
         assertFalse(mWifiNetworkFactory.acceptRequest(mNetworkRequest, 0));
         mLooper.dispatchAll();
-        verifyUnfullfillableDispatched(mConnectivityMessenger);
+        verify(mConnectivityManager).declareNetworkRequestUnfulfillable(eq(mNetworkRequest));
     }
 
     /**
@@ -508,7 +504,7 @@ public class WifiNetworkFactoryTest {
         mNetworkRequest.networkCapabilities.setNetworkSpecifier(specifier2);
         assertFalse(mWifiNetworkFactory.acceptRequest(mNetworkRequest, 0));
         mLooper.dispatchAll();
-        verifyUnfullfillableDispatched(mConnectivityMessenger);
+        verify(mConnectivityManager).declareNetworkRequestUnfulfillable(eq(mNetworkRequest));
     }
 
     /**
@@ -543,7 +539,7 @@ public class WifiNetworkFactoryTest {
 
         mWifiNetworkFactory.needNetworkFor(mNetworkRequest, 0);
         mLooper.dispatchAll();
-        verifyUnfullfillableDispatched(mConnectivityMessenger);
+        verify(mConnectivityManager).declareNetworkRequestUnfulfillable(eq(mNetworkRequest));
     }
 
     /**
@@ -1334,7 +1330,7 @@ public class WifiNetworkFactoryTest {
         verify(mAlarmManager).cancel(any(OnAlarmListener.class));
         // Verify we reset the network request handling.
         verify(mWifiConnectivityManager).setSpecificNetworkRequestInProgress(false);
-        verifyUnfullfillableDispatched(mConnectivityMessenger);
+        verify(mConnectivityManager).declareNetworkRequestUnfulfillable(eq(mNetworkRequest));
 
         verify(mWifiMetrics).incrementNetworkRequestApiNumUserReject();
 
@@ -1366,7 +1362,7 @@ public class WifiNetworkFactoryTest {
                 argThat(new WifiConfigMatcher(mSelectedNetwork)));
         // Verify we reset the network request handling.
         verify(mWifiConnectivityManager).setSpecificNetworkRequestInProgress(false);
-        verifyUnfullfillableDispatched(mConnectivityMessenger);
+        verify(mConnectivityManager).declareNetworkRequestUnfulfillable(eq(mNetworkRequest));
     }
 
     /**
@@ -1398,7 +1394,7 @@ public class WifiNetworkFactoryTest {
                 mConnectionTimeoutAlarmListenerArgumentCaptor.getValue());
         // Verify we reset the network request handling.
         verify(mWifiConnectivityManager).setSpecificNetworkRequestInProgress(false);
-        verifyUnfullfillableDispatched(mConnectivityMessenger);
+        verify(mConnectivityManager).declareNetworkRequestUnfulfillable(eq(mNetworkRequest));
     }
 
     /**
@@ -1431,7 +1427,7 @@ public class WifiNetworkFactoryTest {
                 mConnectionTimeoutAlarmListenerArgumentCaptor.getValue());
         // Verify we reset the network request handling.
         verify(mWifiConnectivityManager).setSpecificNetworkRequestInProgress(false);
-        verifyUnfullfillableDispatched(mConnectivityMessenger);
+        verify(mConnectivityManager).declareNetworkRequestUnfulfillable(eq(mNetworkRequest));
     }
 
     /**
@@ -1476,7 +1472,7 @@ public class WifiNetworkFactoryTest {
                 mConnectionTimeoutAlarmListenerArgumentCaptor.getValue());
         // Verify we reset the network request handling.
         verify(mWifiConnectivityManager).setSpecificNetworkRequestInProgress(false);
-        verifyUnfullfillableDispatched(mConnectivityMessenger);
+        verify(mConnectivityManager).declareNetworkRequestUnfulfillable(eq(mNetworkRequest));
     }
 
     /**
@@ -1681,7 +1677,7 @@ public class WifiNetworkFactoryTest {
         verify(mNetworkRequestMatchCallback).onAbort();
         verify(mWifiScanner, times(2)).getSingleScanResults();
         verify(mWifiScanner, times(2)).startScan(any(), any(), any());
-        verifyUnfullfillableDispatched(mConnectivityMessenger);
+        verify(mConnectivityManager).declareNetworkRequestUnfulfillable(eq(mNetworkRequest));
 
         // Remove the stale request1 & ensure nothing happens.
         mWifiNetworkFactory.releaseNetworkFor(mNetworkRequest);
@@ -1723,7 +1719,7 @@ public class WifiNetworkFactoryTest {
         verify(mWifiScanner, times(2)).getSingleScanResults();
         verify(mWifiScanner, times(2)).startScan(any(), any(), any());
         verify(mAlarmManager).cancel(mPeriodicScanListenerArgumentCaptor.getValue());
-        verifyUnfullfillableDispatched(mConnectivityMessenger);
+        verify(mConnectivityManager).declareNetworkRequestUnfulfillable(eq(mNetworkRequest));
 
         // Remove the stale request1 & ensure nothing happens.
         mWifiNetworkFactory.releaseNetworkFor(mNetworkRequest);
@@ -2849,15 +2845,6 @@ public class WifiNetworkFactoryTest {
         }
     }
 
-    /**
-     * Verify that an EVENT_UNFULFILLABLE_REQUEST was dispatched on the (mock) messenger.
-     */
-    private void verifyUnfullfillableDispatched(Messenger messenger) throws Exception {
-        ArgumentCaptor<Message> messageCaptor = ArgumentCaptor.forClass(Message.class);
-        verify(messenger, atLeastOnce()).send(messageCaptor.capture());
-        assertEquals(NetworkFactory.EVENT_UNFULFILLABLE_REQUEST, messageCaptor.getValue().what);
-    }
-
     private void validateUiStartParams(boolean expectedIsReqForSingeNetwork) {
         ArgumentCaptor<Intent> intentArgumentCaptor = ArgumentCaptor.forClass(Intent.class);
         verify(mContext, atLeastOnce()).startActivityAsUser(
diff --git a/tests/wifitests/src/com/android/server/wifi/aware/WifiAwareDataPathStateManagerTest.java b/tests/wifitests/src/com/android/server/wifi/aware/WifiAwareDataPathStateManagerTest.java
index 35916d2f6..2c30d5aae 100644
--- a/tests/wifitests/src/com/android/server/wifi/aware/WifiAwareDataPathStateManagerTest.java
+++ b/tests/wifitests/src/com/android/server/wifi/aware/WifiAwareDataPathStateManagerTest.java
@@ -46,6 +46,7 @@ import android.net.ConnectivityManager;
 import android.net.MacAddress;
 import android.net.NetworkCapabilities;
 import android.net.NetworkFactory;
+import android.net.NetworkProvider;
 import android.net.NetworkRequest;
 import android.net.NetworkSpecifier;
 import android.net.wifi.WifiManager;
@@ -75,7 +76,6 @@ import android.os.Messenger;
 import android.os.PowerManager;
 import android.os.Process;
 import android.os.test.TestLooper;
-import android.util.Pair;
 
 import androidx.test.filters.SmallTest;
 
@@ -307,7 +307,7 @@ public class WifiAwareDataPathStateManagerTest {
         mMockLooper.dispatchAll();
 
         // failure: no interactions with native manager
-        verifyUnfullfillableDispatched(res.mReverseMessenger);
+        verifyRequestDeclaredUnfullfillable(nr);
         verifyNoMoreInteractions(mMockNative, mMockCm, mAwareMetricsMock, mMockNwMgt);
     }
 
@@ -335,7 +335,7 @@ public class WifiAwareDataPathStateManagerTest {
                 null, passphrase, true, 0, 5, 6);
 
         Message reqNetworkMsg = Message.obtain();
-        reqNetworkMsg.what = NetworkFactory.CMD_REQUEST_NETWORK;
+        reqNetworkMsg.what = NetworkProvider.CMD_REQUEST_NETWORK;
         reqNetworkMsg.obj = nr;
         reqNetworkMsg.arg1 = 0;
         res.mMessenger.send(reqNetworkMsg);
@@ -373,7 +373,7 @@ public class WifiAwareDataPathStateManagerTest {
                 null, passphrase, false, 0, -3, 6);
 
         Message reqNetworkMsg = Message.obtain();
-        reqNetworkMsg.what = NetworkFactory.CMD_REQUEST_NETWORK;
+        reqNetworkMsg.what = NetworkProvider.CMD_REQUEST_NETWORK;
         reqNetworkMsg.obj = nr;
         reqNetworkMsg.arg1 = 0;
         res.mMessenger.send(reqNetworkMsg);
@@ -406,7 +406,7 @@ public class WifiAwareDataPathStateManagerTest {
                 null, null, false, 0, 10, 6);
 
         Message reqNetworkMsg = Message.obtain();
-        reqNetworkMsg.what = NetworkFactory.CMD_REQUEST_NETWORK;
+        reqNetworkMsg.what = NetworkProvider.CMD_REQUEST_NETWORK;
         reqNetworkMsg.obj = nr;
         reqNetworkMsg.arg1 = 0;
         res.mMessenger.send(reqNetworkMsg);
@@ -443,7 +443,7 @@ public class WifiAwareDataPathStateManagerTest {
                 null, null, true, 0);
 
         Message reqNetworkMsg = Message.obtain();
-        reqNetworkMsg.what = NetworkFactory.CMD_REQUEST_NETWORK;
+        reqNetworkMsg.what = NetworkProvider.CMD_REQUEST_NETWORK;
         reqNetworkMsg.obj = nr;
         reqNetworkMsg.arg1 = 0;
         res.mMessenger.send(reqNetworkMsg);
@@ -467,7 +467,7 @@ public class WifiAwareDataPathStateManagerTest {
         mDut.onRespondToDataPathSetupRequestResponse(transactionId.getValue(), true, 0);
         mMockLooper.dispatchAll();
 
-        verifyUnfullfillableDispatched(res.mReverseMessenger);
+        verifyRequestDeclaredUnfullfillable(nr);
 
         // failure if there's further activity
         verifyNoMoreInteractions(mMockNative, mMockCm, mAwareMetricsMock, mMockNwMgt);
@@ -528,7 +528,7 @@ public class WifiAwareDataPathStateManagerTest {
                     null, null, false, networkRequestId);
 
             Message reqNetworkMsg = Message.obtain();
-            reqNetworkMsg.what = NetworkFactory.CMD_REQUEST_NETWORK;
+            reqNetworkMsg.what = NetworkProvider.CMD_REQUEST_NETWORK;
             reqNetworkMsg.obj = nrs[i];
             reqNetworkMsg.arg1 = 0;
             messenger.send(reqNetworkMsg);
@@ -621,7 +621,7 @@ public class WifiAwareDataPathStateManagerTest {
         InOrder inOrderM = inOrder(mAwareMetricsMock);
 
         // (1) initialize all clients
-        Messenger messenger = initOobDataPathEndPoint(true, 2, clientId, inOrder, inOrderM).first;
+        Messenger messenger = initOobDataPathEndPoint(true, 2, clientId, inOrder, inOrderM);
         for (int i = 1; i < numRequestsPre + numRequestsPost; ++i) {
             initOobDataPathEndPoint(false, 1, clientId + i, inOrder, inOrderM);
         }
@@ -636,7 +636,7 @@ public class WifiAwareDataPathStateManagerTest {
                     null, i);
 
             Message reqNetworkMsg = Message.obtain();
-            reqNetworkMsg.what = NetworkFactory.CMD_REQUEST_NETWORK;
+            reqNetworkMsg.what = NetworkProvider.CMD_REQUEST_NETWORK;
             reqNetworkMsg.obj = nrs[i];
             reqNetworkMsg.arg1 = 0;
             messenger.send(reqNetworkMsg);
@@ -697,7 +697,7 @@ public class WifiAwareDataPathStateManagerTest {
                     null, i);
 
             Message reqNetworkMsg = Message.obtain();
-            reqNetworkMsg.what = NetworkFactory.CMD_REQUEST_NETWORK;
+            reqNetworkMsg.what = NetworkProvider.CMD_REQUEST_NETWORK;
             reqNetworkMsg.obj = nrs[i];
             reqNetworkMsg.arg1 = 0;
             messenger.send(reqNetworkMsg);
@@ -706,7 +706,7 @@ public class WifiAwareDataPathStateManagerTest {
                 clientId + numRequestsPre + numRequestsPost, ddepi.mSessionId, ddepi.mPeerHandle,
                 null, null, false, 11);
         Message reqNetworkMsg = Message.obtain();
-        reqNetworkMsg.what = NetworkFactory.CMD_REQUEST_NETWORK;
+        reqNetworkMsg.what = NetworkProvider.CMD_REQUEST_NETWORK;
         reqNetworkMsg.obj = nrs[numRequestsPre + numRequestsPost];
         reqNetworkMsg.arg1 = 0;
         messenger.send(reqNetworkMsg);
@@ -761,8 +761,7 @@ public class WifiAwareDataPathStateManagerTest {
         InOrder inOrderM = inOrder(mAwareMetricsMock);
 
         // (1) initialize all clients
-        Pair<Messenger, Messenger> res = initOobDataPathEndPoint(true, numNdis, clientId, inOrder,
-                inOrderM);
+        Messenger messenger = initOobDataPathEndPoint(true, numNdis, clientId, inOrder, inOrderM);
         for (int i = 1; i < numNdis + 3; ++i) {
             initOobDataPathEndPoint(false, numNdis, clientId + i, inOrder, inOrderM);
         }
@@ -778,10 +777,10 @@ public class WifiAwareDataPathStateManagerTest {
                     null, i);
 
             Message reqNetworkMsg = Message.obtain();
-            reqNetworkMsg.what = NetworkFactory.CMD_REQUEST_NETWORK;
+            reqNetworkMsg.what = NetworkProvider.CMD_REQUEST_NETWORK;
             reqNetworkMsg.obj = nr;
             reqNetworkMsg.arg1 = 0;
-            res.first.send(reqNetworkMsg);
+            messenger.send(reqNetworkMsg);
             mMockLooper.dispatchAll();
 
             if (i < numNdis) {
@@ -809,7 +808,7 @@ public class WifiAwareDataPathStateManagerTest {
                 assertEquals(0, netInfo.getPort()); // uninitialized -> 0
                 assertEquals(-1, netInfo.getTransportProtocol()); // uninitialized -> -1
             } else {
-                verifyUnfullfillableDispatched(res.second);
+                verifyRequestDeclaredUnfullfillable(nr);
             }
         }
 
@@ -1204,7 +1203,7 @@ public class WifiAwareDataPathStateManagerTest {
         nr.networkCapabilities.setNetworkSpecifier(ns);
 
         Message reqNetworkMsg = Message.obtain();
-        reqNetworkMsg.what = NetworkFactory.CMD_REQUEST_NETWORK;
+        reqNetworkMsg.what = NetworkProvider.CMD_REQUEST_NETWORK;
         reqNetworkMsg.obj = nr;
         reqNetworkMsg.arg1 = 0;
         res.mMessenger.send(reqNetworkMsg);
@@ -1213,7 +1212,7 @@ public class WifiAwareDataPathStateManagerTest {
         // consequences of failure:
         //   Responder (publisher): responds with a rejection to any data-path requests
         //   Initiator (subscribe): doesn't initiate (i.e. no HAL requests)
-        verifyUnfullfillableDispatched(res.mReverseMessenger);
+        verifyRequestDeclaredUnfullfillable(nr);
         if (doPublish) {
             // (2) get request & respond
             mDut.onDataPathRequestNotification(pubSubId, peerDiscoveryMac, ndpId, null);
@@ -1264,7 +1263,7 @@ public class WifiAwareDataPathStateManagerTest {
 
         // (3) request network
         Message reqNetworkMsg = Message.obtain();
-        reqNetworkMsg.what = NetworkFactory.CMD_REQUEST_NETWORK;
+        reqNetworkMsg.what = NetworkProvider.CMD_REQUEST_NETWORK;
         reqNetworkMsg.obj = nr;
         reqNetworkMsg.arg1 = 0;
         res.mMessenger.send(reqNetworkMsg);
@@ -1273,7 +1272,7 @@ public class WifiAwareDataPathStateManagerTest {
         // consequences of failure:
         //   Responder (publisher): responds with a rejection to any data-path requests
         //   Initiator (subscribe): doesn't initiate (i.e. no HAL requests)
-        verifyUnfullfillableDispatched(res.mReverseMessenger);
+        verifyRequestDeclaredUnfullfillable(nr);
         if (doPublish) {
             // (2) get request & respond
             mDut.onDataPathRequestNotification(pubSubId, peerDiscoveryMac, ndpId, null);
@@ -1348,7 +1347,7 @@ public class WifiAwareDataPathStateManagerTest {
         }
 
         Message reqNetworkMsg = Message.obtain();
-        reqNetworkMsg.what = NetworkFactory.CMD_REQUEST_NETWORK;
+        reqNetworkMsg.what = NetworkProvider.CMD_REQUEST_NETWORK;
         reqNetworkMsg.obj = nr;
         reqNetworkMsg.arg1 = 0;
         res.mMessenger.send(reqNetworkMsg);
@@ -1362,7 +1361,7 @@ public class WifiAwareDataPathStateManagerTest {
             // short-circuit the rest of this test
             inOrderM.verify(mAwareMetricsMock).recordNdpStatus(eq(NanStatusType.INTERNAL_FAILURE),
                     eq(useDirect), anyLong());
-            verifyUnfullfillableDispatched(res.mReverseMessenger);
+            verifyRequestDeclaredUnfullfillable(nr);
             verifyNoMoreInteractions(mMockNative, mMockCm, mAwareMetricsMock);
             return;
         }
@@ -1408,7 +1407,7 @@ public class WifiAwareDataPathStateManagerTest {
             }
 
             if (timedout) {
-                verifyUnfullfillableDispatched(res.mReverseMessenger);
+                verifyRequestDeclaredUnfullfillable(nr);
                 inOrder.verify(mMockNative).endDataPath(transactionId.capture(), eq(ndpId));
                 mDut.onEndDataPathResponse(transactionId.getValue(), true, 0);
             } else {
@@ -1426,7 +1425,7 @@ public class WifiAwareDataPathStateManagerTest {
             assertTrue(mAlarmManager.dispatch(
                     WifiAwareStateManager.HAL_DATA_PATH_CONFIRM_TIMEOUT_TAG));
             mMockLooper.dispatchAll();
-            verifyUnfullfillableDispatched(res.mReverseMessenger);
+            verifyRequestDeclaredUnfullfillable(nr);
             inOrder.verify(mMockNative).endDataPath(transactionId.capture(), eq(ndpId));
             mDut.onEndDataPathResponse(transactionId.getValue(), true, 0);
             mMockLooper.dispatchAll();
@@ -1504,7 +1503,7 @@ public class WifiAwareDataPathStateManagerTest {
         }
 
         Message reqNetworkMsg = Message.obtain();
-        reqNetworkMsg.what = NetworkFactory.CMD_REQUEST_NETWORK;
+        reqNetworkMsg.what = NetworkProvider.CMD_REQUEST_NETWORK;
         reqNetworkMsg.obj = nr;
         reqNetworkMsg.arg1 = 0;
         res.mMessenger.send(reqNetworkMsg);
@@ -1542,7 +1541,7 @@ public class WifiAwareDataPathStateManagerTest {
                 assertTrue(mAlarmManager.dispatch(
                         WifiAwareStateManager.HAL_DATA_PATH_CONFIRM_TIMEOUT_TAG));
                 mMockLooper.dispatchAll();
-                verifyUnfullfillableDispatched(res.mReverseMessenger);
+                verifyRequestDeclaredUnfullfillable(nr);
                 inOrder.verify(mMockNative).endDataPath(transactionId.capture(), eq(ndpId));
                 mDut.onEndDataPathResponse(transactionId.getValue(), true, 0);
                 mMockLooper.dispatchAll();
@@ -1570,7 +1569,7 @@ public class WifiAwareDataPathStateManagerTest {
                 inOrderM.verify(mAwareMetricsMock).recordNdpSessionDuration(anyLong());
             }
         } else {
-            verifyUnfullfillableDispatched(res.mReverseMessenger);
+            verifyRequestDeclaredUnfullfillable(nr);
             inOrder.verify(mMockNative).respondToDataPathRequest(transactionId.capture(), eq(false),
                     eq(ndpId), eq(""), eq(null), eq(null), eq(null), eq(false), any());
             mDut.onRespondToDataPathSetupRequestResponse(transactionId.getValue(), true, 0);
@@ -1733,8 +1732,8 @@ public class WifiAwareDataPathStateManagerTest {
         ArgumentCaptor<Integer> sessionId = ArgumentCaptor.forClass(Integer.class);
         ArgumentCaptor<Integer> peerIdCaptor = ArgumentCaptor.forClass(Integer.class);
 
-        Pair<Messenger, Messenger> res = initOobDataPathEndPoint(isFirstIteration, 1, clientId,
-                inOrder, inOrderM);
+        Messenger messenger = initOobDataPathEndPoint(isFirstIteration, 1, clientId, inOrder,
+                inOrderM);
 
         if (doPublish) {
             mDut.publish(clientId, publishConfig, mMockSessionCallback);
@@ -1763,10 +1762,10 @@ public class WifiAwareDataPathStateManagerTest {
                 eq(someMsg.getBytes()));
 
         return new DataPathEndPointInfo(sessionId.getValue(), peerIdCaptor.getValue(),
-                isFirstIteration ? res.first : null, isFirstIteration ? res.second : null);
+                isFirstIteration ? messenger : null);
     }
 
-    private Pair<Messenger, Messenger> initOobDataPathEndPoint(boolean startUpSequence,
+    private Messenger initOobDataPathEndPoint(boolean startUpSequence,
             int maxNdiInterfaces, int clientId, InOrder inOrder, InOrder inOrderM)
             throws Exception {
         final int pid = 2000;
@@ -1775,6 +1774,8 @@ public class WifiAwareDataPathStateManagerTest {
 
         ArgumentCaptor<Short> transactionId = ArgumentCaptor.forClass(Short.class);
         ArgumentCaptor<Messenger> messengerCaptor = ArgumentCaptor.forClass(Messenger.class);
+        ArgumentCaptor<NetworkProvider> networkProviderCaptor =
+                ArgumentCaptor.forClass(NetworkProvider.class);
         ArgumentCaptor<String> strCaptor = ArgumentCaptor.forClass(String.class);
 
         Capabilities capabilities = new Capabilities();
@@ -1782,10 +1783,9 @@ public class WifiAwareDataPathStateManagerTest {
 
         if (startUpSequence) {
             // (0) start/registrations
-            inOrder.verify(mMockCm).registerNetworkFactory(messengerCaptor.capture(),
-                    strCaptor.capture());
+            inOrder.verify(mMockCm).registerNetworkProvider(networkProviderCaptor.capture());
             collector.checkThat("factory name", "WIFI_AWARE_FACTORY",
-                    equalTo(strCaptor.getValue()));
+                    equalTo(networkProviderCaptor.getValue().getName()));
 
             // (1) get capabilities
             mDut.queryCapabilities();
@@ -1826,20 +1826,8 @@ public class WifiAwareDataPathStateManagerTest {
                 mDut.onCreateDataPathInterfaceResponse(transactionId.getValue(), true, 0);
                 mMockLooper.dispatchAll();
             }
-            Messenger messenger = messengerCaptor.getValue();
-
-            Messenger reverseMessenger = null;
-            if (messenger != null) {
-                reverseMessenger = mock(Messenger.class);
-
-                Message fullConnectionMessage = Message.obtain();
-                fullConnectionMessage.what = AsyncChannel.CMD_CHANNEL_FULL_CONNECTION;
-                fullConnectionMessage.replyTo = reverseMessenger;
-                messenger.send(fullConnectionMessage);
-                mMockLooper.dispatchAll();
-            }
-
-            return Pair.create(messenger, reverseMessenger);
+            Messenger messenger = networkProviderCaptor.getValue().getMessenger();
+            return messenger;
         }
 
         return null;
@@ -1891,25 +1879,20 @@ public class WifiAwareDataPathStateManagerTest {
         int mSessionId;
         PeerHandle mPeerHandle;
         Messenger mMessenger;
-        Messenger mReverseMessenger;
 
-        DataPathEndPointInfo(int sessionId, int peerId, Messenger messenger,
-                Messenger reverseMessenger) {
+        DataPathEndPointInfo(int sessionId, int peerId, Messenger messenger) {
             mSessionId = sessionId;
             mPeerHandle = new PeerHandle(peerId);
             mMessenger = messenger;
-            mReverseMessenger = reverseMessenger;
         }
     }
 
     /**
-     * Verify that an EVENT_UNFULFILLABLE_REQUEST was dispatched on the (mock) messenger.
+     * Verify that declareNetworkRequestUnfulfillable was called.
      */
-    private void verifyUnfullfillableDispatched(Messenger messenger) throws Exception {
-        ArgumentCaptor<Message> messageCaptor = ArgumentCaptor.forClass(Message.class);
-
-        verify(messenger, atLeastOnce()).send(messageCaptor.capture());
-        assertEquals(NetworkFactory.EVENT_UNFULFILLABLE_REQUEST, messageCaptor.getValue().what);
+    private void verifyRequestDeclaredUnfullfillable(NetworkRequest request) throws Exception {
+        mMockLooper.dispatchAll();
+        verify(mMockCm, atLeastOnce()).declareNetworkRequestUnfulfillable(any());
     }
 
     // copy of the method in WifiAwareDataPathStateManager - but without error checking (so we can