diff --git a/tests/net/java/com/android/server/ConnectivityServiceTest.java b/tests/net/java/com/android/server/ConnectivityServiceTest.java index 28f8122002..2aea1d7011 100644 --- a/tests/net/java/com/android/server/ConnectivityServiceTest.java +++ b/tests/net/java/com/android/server/ConnectivityServiceTest.java @@ -914,8 +914,10 @@ public class ConnectivityServiceTest { mock(INetworkPolicyManager.class), mock(IpConnectivityLog.class)); - mService.systemReady(); + // Create local CM before sending system ready so that we can answer + // getSystemService() correctly. mCm = new WrappedConnectivityManager(InstrumentationRegistry.getContext(), mService); + mService.systemReady(); mCm.bindProcessToNetwork(null); // Ensure that the default setting for Captive Portals is used for most tests @@ -3412,8 +3414,10 @@ public class ConnectivityServiceTest { @Test public void testNetworkCallbackMaximum() { - final int MAX_REQUESTS = 100; - final int CALLBACKS = 90; + // We can only have 99 callbacks, because MultipathPolicyTracker is + // already one of them. + final int MAX_REQUESTS = 99; + final int CALLBACKS = 89; final int INTENTS = 10; assertEquals(MAX_REQUESTS, CALLBACKS + INTENTS); diff --git a/tests/net/java/com/android/server/net/NetworkStatsServiceTest.java b/tests/net/java/com/android/server/net/NetworkStatsServiceTest.java index b1b05e8b86..49b2643fa8 100644 --- a/tests/net/java/com/android/server/net/NetworkStatsServiceTest.java +++ b/tests/net/java/com/android/server/net/NetworkStatsServiceTest.java @@ -95,6 +95,7 @@ import android.telephony.TelephonyManager; import com.android.internal.net.VpnInfo; import com.android.internal.util.test.BroadcastInterceptingContext; +import com.android.server.LocalServices; import com.android.server.net.NetworkStatsService.NetworkStatsSettings; import com.android.server.net.NetworkStatsService.NetworkStatsSettings.Config; @@ -221,6 +222,9 @@ public class NetworkStatsServiceTest { @After public void tearDown() throws Exception { + // Registered by NetworkStatsService's constructor. + LocalServices.removeServiceForTest(NetworkStatsManagerInternal.class); + IoUtils.deleteContents(mStatsDir); mServiceContext = null;