Loading services/core/java/com/android/server/VcnManagementService.java +2 −2 Original line number Diff line number Diff line Loading @@ -335,8 +335,8 @@ public class VcnManagementService extends IVcnManagementService.Stub { @NonNull Context context, @NonNull Looper looper, @NonNull VcnNetworkProvider vcnNetworkProvider, boolean getIsInTestMode) { return new VcnContext(context, looper, vcnNetworkProvider, getIsInTestMode); boolean isInTestMode) { return new VcnContext(context, looper, vcnNetworkProvider, isInTestMode); } /** Creates a new Vcn instance using the provided configuration */ Loading services/core/java/com/android/server/vcn/UnderlyingNetworkTracker.java +2 −3 Original line number Diff line number Diff line Loading @@ -219,10 +219,9 @@ public class UnderlyingNetworkTracker { /** Builds and returns a NetworkRequest for the given subIds to match Test Networks. */ private NetworkRequest getTestNetworkRequest(@NonNull Set<Integer> subIds) { return getBaseNetworkRequestBuilder() return new NetworkRequest.Builder() .clearCapabilities() .addTransportType(NetworkCapabilities.TRANSPORT_TEST) .removeCapability(NetworkCapabilities.NET_CAPABILITY_INTERNET) .removeCapability(NetworkCapabilities.NET_CAPABILITY_NOT_VPN) .setSubscriptionIds(subIds) .build(); } Loading tests/vcn/java/com/android/server/vcn/UnderlyingNetworkTrackerTest.java +16 −19 Original line number Diff line number Diff line Loading @@ -163,27 +163,30 @@ public class UnderlyingNetworkTrackerTest { @Test public void testNetworkCallbacksRegisteredOnStartupForTestMode() { resetVcnContext(); when(mVcnContext.isInTestMode()).thenReturn(true); reset(mConnectivityManager); final VcnContext vcnContext = spy( new VcnContext( mContext, mTestLooper.getLooper(), mVcnNetworkProvider, true /* isInTestMode */)); mUnderlyingNetworkTracker = new UnderlyingNetworkTracker( mVcnContext, vcnContext, SUB_GROUP, mSubscriptionSnapshot, Collections.singleton(NetworkCapabilities.NET_CAPABILITY_INTERNET), mNetworkTrackerCb); verifyNetworkRequestsRegistered(INITIAL_SUB_IDS, true /* expectTestMode */); verify(mConnectivityManager) .requestBackgroundNetwork( eq(getTestNetworkRequest(INITIAL_SUB_IDS)), any(RouteSelectionCallback.class), any()); } private void verifyNetworkRequestsRegistered(Set<Integer> expectedSubIds) { verifyNetworkRequestsRegistered(expectedSubIds, false /* expectTestMode */); } private void verifyNetworkRequestsRegistered( Set<Integer> expectedSubIds, boolean expectTestMode) { verify(mConnectivityManager) .requestBackgroundNetwork( eq(getWifiRequest(expectedSubIds)), Loading @@ -196,14 +199,9 @@ public class UnderlyingNetworkTrackerTest { any(NetworkBringupCallback.class), any()); } final NetworkRequest expectedRouteSelectionRequest = expectTestMode ? getTestNetworkRequest(expectedSubIds) : getRouteSelectionRequest(expectedSubIds); verify(mConnectivityManager) .requestBackgroundNetwork( eq(expectedRouteSelectionRequest), eq(getRouteSelectionRequest(expectedSubIds)), any(RouteSelectionCallback.class), any()); } Loading Loading @@ -245,10 +243,9 @@ public class UnderlyingNetworkTrackerTest { } private NetworkRequest getTestNetworkRequest(Set<Integer> netCapsSubIds) { return getExpectedRequestBase() return new NetworkRequest.Builder() .clearCapabilities() .addTransportType(NetworkCapabilities.TRANSPORT_TEST) .removeCapability(NetworkCapabilities.NET_CAPABILITY_INTERNET) .removeCapability(NetworkCapabilities.NET_CAPABILITY_NOT_VPN) .setSubscriptionIds(netCapsSubIds) .build(); } Loading Loading
services/core/java/com/android/server/VcnManagementService.java +2 −2 Original line number Diff line number Diff line Loading @@ -335,8 +335,8 @@ public class VcnManagementService extends IVcnManagementService.Stub { @NonNull Context context, @NonNull Looper looper, @NonNull VcnNetworkProvider vcnNetworkProvider, boolean getIsInTestMode) { return new VcnContext(context, looper, vcnNetworkProvider, getIsInTestMode); boolean isInTestMode) { return new VcnContext(context, looper, vcnNetworkProvider, isInTestMode); } /** Creates a new Vcn instance using the provided configuration */ Loading
services/core/java/com/android/server/vcn/UnderlyingNetworkTracker.java +2 −3 Original line number Diff line number Diff line Loading @@ -219,10 +219,9 @@ public class UnderlyingNetworkTracker { /** Builds and returns a NetworkRequest for the given subIds to match Test Networks. */ private NetworkRequest getTestNetworkRequest(@NonNull Set<Integer> subIds) { return getBaseNetworkRequestBuilder() return new NetworkRequest.Builder() .clearCapabilities() .addTransportType(NetworkCapabilities.TRANSPORT_TEST) .removeCapability(NetworkCapabilities.NET_CAPABILITY_INTERNET) .removeCapability(NetworkCapabilities.NET_CAPABILITY_NOT_VPN) .setSubscriptionIds(subIds) .build(); } Loading
tests/vcn/java/com/android/server/vcn/UnderlyingNetworkTrackerTest.java +16 −19 Original line number Diff line number Diff line Loading @@ -163,27 +163,30 @@ public class UnderlyingNetworkTrackerTest { @Test public void testNetworkCallbacksRegisteredOnStartupForTestMode() { resetVcnContext(); when(mVcnContext.isInTestMode()).thenReturn(true); reset(mConnectivityManager); final VcnContext vcnContext = spy( new VcnContext( mContext, mTestLooper.getLooper(), mVcnNetworkProvider, true /* isInTestMode */)); mUnderlyingNetworkTracker = new UnderlyingNetworkTracker( mVcnContext, vcnContext, SUB_GROUP, mSubscriptionSnapshot, Collections.singleton(NetworkCapabilities.NET_CAPABILITY_INTERNET), mNetworkTrackerCb); verifyNetworkRequestsRegistered(INITIAL_SUB_IDS, true /* expectTestMode */); verify(mConnectivityManager) .requestBackgroundNetwork( eq(getTestNetworkRequest(INITIAL_SUB_IDS)), any(RouteSelectionCallback.class), any()); } private void verifyNetworkRequestsRegistered(Set<Integer> expectedSubIds) { verifyNetworkRequestsRegistered(expectedSubIds, false /* expectTestMode */); } private void verifyNetworkRequestsRegistered( Set<Integer> expectedSubIds, boolean expectTestMode) { verify(mConnectivityManager) .requestBackgroundNetwork( eq(getWifiRequest(expectedSubIds)), Loading @@ -196,14 +199,9 @@ public class UnderlyingNetworkTrackerTest { any(NetworkBringupCallback.class), any()); } final NetworkRequest expectedRouteSelectionRequest = expectTestMode ? getTestNetworkRequest(expectedSubIds) : getRouteSelectionRequest(expectedSubIds); verify(mConnectivityManager) .requestBackgroundNetwork( eq(expectedRouteSelectionRequest), eq(getRouteSelectionRequest(expectedSubIds)), any(RouteSelectionCallback.class), any()); } Loading Loading @@ -245,10 +243,9 @@ public class UnderlyingNetworkTrackerTest { } private NetworkRequest getTestNetworkRequest(Set<Integer> netCapsSubIds) { return getExpectedRequestBase() return new NetworkRequest.Builder() .clearCapabilities() .addTransportType(NetworkCapabilities.TRANSPORT_TEST) .removeCapability(NetworkCapabilities.NET_CAPABILITY_INTERNET) .removeCapability(NetworkCapabilities.NET_CAPABILITY_NOT_VPN) .setSubscriptionIds(netCapsSubIds) .build(); } Loading