73708bff55
ConnectivityDiagnosticsManager will send callbacks to ConnectivityService for registering and unregistering them with the system. ConnectivityService needs to do the processing for persisting (and deleting) these callbacks on the ConnectivityService Thread, so messages are sent to the Connectivity Diagnostics Handler, which runs on the ConnectivityService Thread. Bug: 146444622 Bug: 143187964 Bug: 147848028 Test: compiles Test: atest FrameworksNetTests Change-Id: Ia5c8f90a60c050504e8676de9564a7607a9b03bc
316 lines
12 KiB
Java
316 lines
12 KiB
Java
/*
|
|
* Copyright (C) 2020 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package android.net;
|
|
|
|
import static android.net.ConnectivityDiagnosticsManager.ConnectivityDiagnosticsBinder;
|
|
import static android.net.ConnectivityDiagnosticsManager.ConnectivityDiagnosticsCallback;
|
|
import static android.net.ConnectivityDiagnosticsManager.ConnectivityReport;
|
|
import static android.net.ConnectivityDiagnosticsManager.DataStallReport;
|
|
|
|
import static com.android.testutils.ParcelUtilsKt.assertParcelSane;
|
|
|
|
import static org.junit.Assert.assertEquals;
|
|
import static org.junit.Assert.assertFalse;
|
|
import static org.junit.Assert.assertNotEquals;
|
|
import static org.junit.Assert.assertTrue;
|
|
import static org.junit.Assert.fail;
|
|
import static org.mockito.ArgumentMatchers.any;
|
|
import static org.mockito.ArgumentMatchers.eq;
|
|
import static org.mockito.Mockito.mock;
|
|
import static org.mockito.Mockito.times;
|
|
import static org.mockito.Mockito.verify;
|
|
import static org.mockito.Mockito.verifyNoMoreInteractions;
|
|
|
|
import android.content.Context;
|
|
import android.os.PersistableBundle;
|
|
|
|
import org.junit.After;
|
|
import org.junit.Before;
|
|
import org.junit.Test;
|
|
import org.junit.runner.RunWith;
|
|
import org.junit.runners.JUnit4;
|
|
import org.mockito.Mock;
|
|
|
|
import java.util.concurrent.Executor;
|
|
|
|
@RunWith(JUnit4.class)
|
|
public class ConnectivityDiagnosticsManagerTest {
|
|
private static final int NET_ID = 1;
|
|
private static final int DETECTION_METHOD = 2;
|
|
private static final long TIMESTAMP = 10L;
|
|
private static final String INTERFACE_NAME = "interface";
|
|
private static final String BUNDLE_KEY = "key";
|
|
private static final String BUNDLE_VALUE = "value";
|
|
|
|
private static final Executor INLINE_EXECUTOR = x -> x.run();
|
|
|
|
@Mock private Context mContext;
|
|
@Mock private IConnectivityManager mService;
|
|
@Mock private ConnectivityDiagnosticsCallback mCb;
|
|
|
|
private ConnectivityDiagnosticsBinder mBinder;
|
|
private ConnectivityDiagnosticsManager mManager;
|
|
|
|
@Before
|
|
public void setUp() {
|
|
mContext = mock(Context.class);
|
|
mService = mock(IConnectivityManager.class);
|
|
mCb = mock(ConnectivityDiagnosticsCallback.class);
|
|
|
|
mBinder = new ConnectivityDiagnosticsBinder(mCb, INLINE_EXECUTOR);
|
|
mManager = new ConnectivityDiagnosticsManager(mContext, mService);
|
|
}
|
|
|
|
@After
|
|
public void tearDown() {
|
|
// clear ConnectivityDiagnosticsManager callbacks map
|
|
ConnectivityDiagnosticsManager.sCallbacks.clear();
|
|
}
|
|
|
|
private ConnectivityReport createSampleConnectivityReport() {
|
|
final LinkProperties linkProperties = new LinkProperties();
|
|
linkProperties.setInterfaceName(INTERFACE_NAME);
|
|
|
|
final NetworkCapabilities networkCapabilities = new NetworkCapabilities();
|
|
networkCapabilities.addCapability(NetworkCapabilities.NET_CAPABILITY_IMS);
|
|
|
|
final PersistableBundle bundle = new PersistableBundle();
|
|
bundle.putString(BUNDLE_KEY, BUNDLE_VALUE);
|
|
|
|
return new ConnectivityReport(
|
|
new Network(NET_ID), TIMESTAMP, linkProperties, networkCapabilities, bundle);
|
|
}
|
|
|
|
private ConnectivityReport createDefaultConnectivityReport() {
|
|
return new ConnectivityReport(
|
|
new Network(0),
|
|
0L,
|
|
new LinkProperties(),
|
|
new NetworkCapabilities(),
|
|
PersistableBundle.EMPTY);
|
|
}
|
|
|
|
@Test
|
|
public void testPersistableBundleEquals() {
|
|
assertFalse(
|
|
ConnectivityDiagnosticsManager.persistableBundleEquals(
|
|
null, PersistableBundle.EMPTY));
|
|
assertFalse(
|
|
ConnectivityDiagnosticsManager.persistableBundleEquals(
|
|
PersistableBundle.EMPTY, null));
|
|
assertTrue(
|
|
ConnectivityDiagnosticsManager.persistableBundleEquals(
|
|
PersistableBundle.EMPTY, PersistableBundle.EMPTY));
|
|
|
|
final PersistableBundle a = new PersistableBundle();
|
|
a.putString(BUNDLE_KEY, BUNDLE_VALUE);
|
|
|
|
final PersistableBundle b = new PersistableBundle();
|
|
b.putString(BUNDLE_KEY, BUNDLE_VALUE);
|
|
|
|
final PersistableBundle c = new PersistableBundle();
|
|
c.putString(BUNDLE_KEY, null);
|
|
|
|
assertFalse(
|
|
ConnectivityDiagnosticsManager.persistableBundleEquals(PersistableBundle.EMPTY, a));
|
|
assertFalse(
|
|
ConnectivityDiagnosticsManager.persistableBundleEquals(a, PersistableBundle.EMPTY));
|
|
|
|
assertTrue(ConnectivityDiagnosticsManager.persistableBundleEquals(a, b));
|
|
assertTrue(ConnectivityDiagnosticsManager.persistableBundleEquals(b, a));
|
|
|
|
assertFalse(ConnectivityDiagnosticsManager.persistableBundleEquals(a, c));
|
|
assertFalse(ConnectivityDiagnosticsManager.persistableBundleEquals(c, a));
|
|
}
|
|
|
|
@Test
|
|
public void testConnectivityReportEquals() {
|
|
assertEquals(createSampleConnectivityReport(), createSampleConnectivityReport());
|
|
assertEquals(createDefaultConnectivityReport(), createDefaultConnectivityReport());
|
|
|
|
final LinkProperties linkProperties = new LinkProperties();
|
|
linkProperties.setInterfaceName(INTERFACE_NAME);
|
|
|
|
final NetworkCapabilities networkCapabilities = new NetworkCapabilities();
|
|
networkCapabilities.addCapability(NetworkCapabilities.NET_CAPABILITY_IMS);
|
|
|
|
final PersistableBundle bundle = new PersistableBundle();
|
|
bundle.putString(BUNDLE_KEY, BUNDLE_VALUE);
|
|
|
|
assertNotEquals(
|
|
createDefaultConnectivityReport(),
|
|
new ConnectivityReport(
|
|
new Network(NET_ID),
|
|
0L,
|
|
new LinkProperties(),
|
|
new NetworkCapabilities(),
|
|
PersistableBundle.EMPTY));
|
|
assertNotEquals(
|
|
createDefaultConnectivityReport(),
|
|
new ConnectivityReport(
|
|
new Network(0),
|
|
TIMESTAMP,
|
|
new LinkProperties(),
|
|
new NetworkCapabilities(),
|
|
PersistableBundle.EMPTY));
|
|
assertNotEquals(
|
|
createDefaultConnectivityReport(),
|
|
new ConnectivityReport(
|
|
new Network(0),
|
|
0L,
|
|
linkProperties,
|
|
new NetworkCapabilities(),
|
|
PersistableBundle.EMPTY));
|
|
assertNotEquals(
|
|
createDefaultConnectivityReport(),
|
|
new ConnectivityReport(
|
|
new Network(0),
|
|
TIMESTAMP,
|
|
new LinkProperties(),
|
|
networkCapabilities,
|
|
PersistableBundle.EMPTY));
|
|
assertNotEquals(
|
|
createDefaultConnectivityReport(),
|
|
new ConnectivityReport(
|
|
new Network(0),
|
|
TIMESTAMP,
|
|
new LinkProperties(),
|
|
new NetworkCapabilities(),
|
|
bundle));
|
|
}
|
|
|
|
@Test
|
|
public void testConnectivityReportParcelUnparcel() {
|
|
assertParcelSane(createSampleConnectivityReport(), 5);
|
|
}
|
|
|
|
private DataStallReport createSampleDataStallReport() {
|
|
final PersistableBundle bundle = new PersistableBundle();
|
|
bundle.putString(BUNDLE_KEY, BUNDLE_VALUE);
|
|
return new DataStallReport(new Network(NET_ID), TIMESTAMP, DETECTION_METHOD, bundle);
|
|
}
|
|
|
|
private DataStallReport createDefaultDataStallReport() {
|
|
return new DataStallReport(new Network(0), 0L, 0, PersistableBundle.EMPTY);
|
|
}
|
|
|
|
@Test
|
|
public void testDataStallReportEquals() {
|
|
assertEquals(createSampleDataStallReport(), createSampleDataStallReport());
|
|
assertEquals(createDefaultDataStallReport(), createDefaultDataStallReport());
|
|
|
|
final PersistableBundle bundle = new PersistableBundle();
|
|
bundle.putString(BUNDLE_KEY, BUNDLE_VALUE);
|
|
|
|
assertNotEquals(
|
|
createDefaultDataStallReport(),
|
|
new DataStallReport(new Network(NET_ID), 0L, 0, PersistableBundle.EMPTY));
|
|
assertNotEquals(
|
|
createDefaultDataStallReport(),
|
|
new DataStallReport(new Network(0), TIMESTAMP, 0, PersistableBundle.EMPTY));
|
|
assertNotEquals(
|
|
createDefaultDataStallReport(),
|
|
new DataStallReport(new Network(0), 0L, DETECTION_METHOD, PersistableBundle.EMPTY));
|
|
assertNotEquals(
|
|
createDefaultDataStallReport(), new DataStallReport(new Network(0), 0L, 0, bundle));
|
|
}
|
|
|
|
@Test
|
|
public void testDataStallReportParcelUnparcel() {
|
|
assertParcelSane(createSampleDataStallReport(), 4);
|
|
}
|
|
|
|
@Test
|
|
public void testConnectivityDiagnosticsCallbackOnConnectivityReport() {
|
|
mBinder.onConnectivityReport(createSampleConnectivityReport());
|
|
|
|
// The callback will be invoked synchronously by inline executor. Immediately check the
|
|
// latch without waiting.
|
|
verify(mCb).onConnectivityReport(eq(createSampleConnectivityReport()));
|
|
}
|
|
|
|
@Test
|
|
public void testConnectivityDiagnosticsCallbackOnDataStallSuspected() {
|
|
mBinder.onDataStallSuspected(createSampleDataStallReport());
|
|
|
|
// The callback will be invoked synchronously by inline executor. Immediately check the
|
|
// latch without waiting.
|
|
verify(mCb).onDataStallSuspected(eq(createSampleDataStallReport()));
|
|
}
|
|
|
|
@Test
|
|
public void testConnectivityDiagnosticsCallbackOnNetworkConnectivityReported() {
|
|
final Network n = new Network(NET_ID);
|
|
final boolean connectivity = true;
|
|
|
|
mBinder.onNetworkConnectivityReported(n, connectivity);
|
|
|
|
// The callback will be invoked synchronously by inline executor. Immediately check the
|
|
// latch without waiting.
|
|
verify(mCb).onNetworkConnectivityReported(eq(n), eq(connectivity));
|
|
}
|
|
|
|
@Test
|
|
public void testRegisterConnectivityDiagnosticsCallback() throws Exception {
|
|
final NetworkRequest request = new NetworkRequest.Builder().build();
|
|
|
|
mManager.registerConnectivityDiagnosticsCallback(request, INLINE_EXECUTOR, mCb);
|
|
|
|
verify(mService).registerConnectivityDiagnosticsCallback(
|
|
any(ConnectivityDiagnosticsBinder.class), eq(request));
|
|
assertTrue(ConnectivityDiagnosticsManager.sCallbacks.containsKey(mCb));
|
|
}
|
|
|
|
@Test
|
|
public void testRegisterDuplicateConnectivityDiagnosticsCallback() throws Exception {
|
|
final NetworkRequest request = new NetworkRequest.Builder().build();
|
|
|
|
mManager.registerConnectivityDiagnosticsCallback(request, INLINE_EXECUTOR, mCb);
|
|
|
|
try {
|
|
mManager.registerConnectivityDiagnosticsCallback(request, INLINE_EXECUTOR, mCb);
|
|
fail("Duplicate callback registration should fail");
|
|
} catch (IllegalArgumentException expected) {
|
|
}
|
|
}
|
|
|
|
@Test
|
|
public void testUnregisterConnectivityDiagnosticsCallback() throws Exception {
|
|
final NetworkRequest request = new NetworkRequest.Builder().build();
|
|
mManager.registerConnectivityDiagnosticsCallback(request, INLINE_EXECUTOR, mCb);
|
|
|
|
mManager.unregisterConnectivityDiagnosticsCallback(mCb);
|
|
|
|
verify(mService).unregisterConnectivityDiagnosticsCallback(
|
|
any(ConnectivityDiagnosticsBinder.class));
|
|
assertFalse(ConnectivityDiagnosticsManager.sCallbacks.containsKey(mCb));
|
|
|
|
// verify that re-registering is successful
|
|
mManager.registerConnectivityDiagnosticsCallback(request, INLINE_EXECUTOR, mCb);
|
|
verify(mService, times(2)).registerConnectivityDiagnosticsCallback(
|
|
any(ConnectivityDiagnosticsBinder.class), eq(request));
|
|
assertTrue(ConnectivityDiagnosticsManager.sCallbacks.containsKey(mCb));
|
|
}
|
|
|
|
@Test
|
|
public void testUnregisterUnknownConnectivityDiagnosticsCallback() throws Exception {
|
|
mManager.unregisterConnectivityDiagnosticsCallback(mCb);
|
|
|
|
verifyNoMoreInteractions(mService);
|
|
}
|
|
}
|