blob: b130c1737c6298422d3b09e4034c0a6fa975efa9 [file] [log] [blame]
/*
* Copyright (C) 2017 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.wifi;
import static android.net.wifi.WifiManager.HOTSPOT_FAILED;
import static android.net.wifi.WifiManager.HOTSPOT_STARTED;
import static android.net.wifi.WifiManager.HOTSPOT_STOPPED;
import static android.net.wifi.WifiManager.LocalOnlyHotspotCallback.ERROR_GENERIC;
import static android.net.wifi.WifiManager.LocalOnlyHotspotCallback.ERROR_INCOMPATIBLE_MODE;
import static android.net.wifi.WifiManager.LocalOnlyHotspotCallback.ERROR_NO_CHANNEL;
import static android.net.wifi.WifiManager.LocalOnlyHotspotCallback.ERROR_TETHERING_DISALLOWED;
import static android.net.wifi.WifiManager.LocalOnlyHotspotCallback.REQUEST_REGISTERED;
import static android.net.wifi.WifiManager.SAP_START_FAILURE_GENERAL;
import static android.net.wifi.WifiManager.WIFI_AP_STATE_ENABLED;
import static android.net.wifi.WifiManager.WIFI_AP_STATE_ENABLING;
import static android.net.wifi.WifiManager.WIFI_AP_STATE_FAILED;
import static org.junit.Assert.assertArrayEquals;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import static org.mockito.ArgumentMatchers.anyBoolean;
import static org.mockito.Mockito.any;
import static org.mockito.Mockito.anyInt;
import static org.mockito.Mockito.anyList;
import static org.mockito.Mockito.anyString;
import static org.mockito.Mockito.doThrow;
import static org.mockito.Mockito.eq;
import static org.mockito.Mockito.inOrder;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.verifyNoMoreInteractions;
import static org.mockito.Mockito.verifyZeroInteractions;
import static org.mockito.Mockito.when;
import android.content.Context;
import android.content.pm.ApplicationInfo;
import android.net.DhcpInfo;
import android.net.wifi.WifiManager.LocalOnlyHotspotCallback;
import android.net.wifi.WifiManager.LocalOnlyHotspotObserver;
import android.net.wifi.WifiManager.LocalOnlyHotspotReservation;
import android.net.wifi.WifiManager.LocalOnlyHotspotSubscription;
import android.net.wifi.WifiManager.NetworkRequestMatchCallback;
import android.net.wifi.WifiManager.NetworkRequestUserSelectionCallback;
import android.net.wifi.WifiManager.OnWifiUsabilityStatsListener;
import android.net.wifi.WifiManager.SoftApCallback;
import android.net.wifi.WifiManager.TrafficStateCallback;
import android.os.Build;
import android.os.Handler;
import android.os.HandlerExecutor;
import android.os.IBinder;
import android.os.Message;
import android.os.Messenger;
import android.os.test.TestLooper;
import androidx.test.filters.SmallTest;
import org.junit.Before;
import org.junit.Test;
import org.mockito.ArgumentCaptor;
import org.mockito.InOrder;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.Executor;
/**
* Unit tests for {@link android.net.wifi.WifiManager}.
*/
@SmallTest
public class WifiManagerTest {
private static final int ERROR_NOT_SET = -1;
private static final int ERROR_TEST_REASON = 5;
private static final int TEST_UID = 14553;
private static final int TEST_NETWORK_ID = 143;
private static final String TEST_PACKAGE_NAME = "TestPackage";
private static final String TEST_COUNTRY_CODE = "US";
private static final String[] TEST_MAC_ADDRESSES = {"da:a1:19:0:0:0"};
@Mock Context mContext;
@Mock android.net.wifi.IWifiManager mWifiService;
@Mock ApplicationInfo mApplicationInfo;
@Mock WifiConfiguration mApConfig;
@Mock IBinder mAppBinder;
@Mock SoftApCallback mSoftApCallback;
@Mock TrafficStateCallback mTrafficStateCallback;
@Mock NetworkRequestMatchCallback mNetworkRequestMatchCallback;
@Mock OnWifiUsabilityStatsListener mOnWifiUsabilityStatsListener;
private Executor mExecutor;
private Handler mHandler;
private TestLooper mLooper;
private WifiManager mWifiManager;
private Messenger mWifiServiceMessenger;
final ArgumentCaptor<Messenger> mMessengerCaptor = ArgumentCaptor.forClass(Messenger.class);
@Before
public void setUp() throws Exception {
MockitoAnnotations.initMocks(this);
mLooper = new TestLooper();
mHandler = spy(new Handler(mLooper.getLooper()));
mApplicationInfo.targetSdkVersion = Build.VERSION_CODES.Q;
when(mContext.getApplicationInfo()).thenReturn(mApplicationInfo);
when(mContext.getOpPackageName()).thenReturn(TEST_PACKAGE_NAME);
mWifiServiceMessenger = new Messenger(mHandler);
mWifiManager = new WifiManager(mContext, mWifiService, mLooper.getLooper());
verify(mWifiService).getVerboseLoggingLevel();
}
/**
* Check the call to startSoftAp calls WifiService to startSoftAp with the provided
* WifiConfiguration. Verify that the return value is propagated to the caller.
*/
@Test
public void testStartSoftApCallsServiceWithWifiConfig() throws Exception {
when(mWifiService.startSoftAp(eq(mApConfig))).thenReturn(true);
assertTrue(mWifiManager.startSoftAp(mApConfig));
when(mWifiService.startSoftAp(eq(mApConfig))).thenReturn(false);
assertFalse(mWifiManager.startSoftAp(mApConfig));
}
/**
* Check the call to startSoftAp calls WifiService to startSoftAp with a null config. Verify
* that the return value is propagated to the caller.
*/
@Test
public void testStartSoftApCallsServiceWithNullConfig() throws Exception {
when(mWifiService.startSoftAp(eq(null))).thenReturn(true);
assertTrue(mWifiManager.startSoftAp(null));
when(mWifiService.startSoftAp(eq(null))).thenReturn(false);
assertFalse(mWifiManager.startSoftAp(null));
}
/**
* Check the call to stopSoftAp calls WifiService to stopSoftAp.
*/
@Test
public void testStopSoftApCallsService() throws Exception {
when(mWifiService.stopSoftAp()).thenReturn(true);
assertTrue(mWifiManager.stopSoftAp());
when(mWifiService.stopSoftAp()).thenReturn(false);
assertFalse(mWifiManager.stopSoftAp());
}
/**
* Test creation of a LocalOnlyHotspotReservation and verify that close properly calls
* WifiService.stopLocalOnlyHotspot.
*/
@Test
public void testCreationAndCloseOfLocalOnlyHotspotReservation() throws Exception {
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
when(mWifiService.startLocalOnlyHotspot(any(Messenger.class), any(IBinder.class),
anyString())).thenReturn(REQUEST_REGISTERED);
mWifiManager.startLocalOnlyHotspot(callback, mHandler);
callback.onStarted(mWifiManager.new LocalOnlyHotspotReservation(mApConfig));
assertEquals(mApConfig, callback.mRes.getWifiConfiguration());
callback.mRes.close();
verify(mWifiService).stopLocalOnlyHotspot();
}
/**
* Verify stopLOHS is called when try-with-resources is used properly.
*/
@Test
public void testLocalOnlyHotspotReservationCallsStopProperlyInTryWithResources()
throws Exception {
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
when(mWifiService.startLocalOnlyHotspot(any(Messenger.class), any(IBinder.class),
anyString())).thenReturn(REQUEST_REGISTERED);
mWifiManager.startLocalOnlyHotspot(callback, mHandler);
callback.onStarted(mWifiManager.new LocalOnlyHotspotReservation(mApConfig));
try (WifiManager.LocalOnlyHotspotReservation res = callback.mRes) {
assertEquals(mApConfig, res.getWifiConfiguration());
}
verify(mWifiService).stopLocalOnlyHotspot();
}
/**
* Test creation of a LocalOnlyHotspotSubscription.
* TODO: when registrations are tracked, verify removal on close.
*/
@Test
public void testCreationOfLocalOnlyHotspotSubscription() throws Exception {
try (WifiManager.LocalOnlyHotspotSubscription sub =
mWifiManager.new LocalOnlyHotspotSubscription()) {
sub.close();
}
}
public class TestLocalOnlyHotspotCallback extends LocalOnlyHotspotCallback {
public boolean mOnStartedCalled = false;
public boolean mOnStoppedCalled = false;
public int mFailureReason = -1;
public LocalOnlyHotspotReservation mRes = null;
public long mCallingThreadId = -1;
@Override
public void onStarted(LocalOnlyHotspotReservation r) {
mRes = r;
mOnStartedCalled = true;
mCallingThreadId = Thread.currentThread().getId();
}
@Override
public void onStopped() {
mOnStoppedCalled = true;
mCallingThreadId = Thread.currentThread().getId();
}
@Override
public void onFailed(int reason) {
mFailureReason = reason;
mCallingThreadId = Thread.currentThread().getId();
}
}
/**
* Verify callback is properly plumbed when called.
*/
@Test
public void testLocalOnlyHotspotCallback() {
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
assertFalse(callback.mOnStartedCalled);
assertFalse(callback.mOnStoppedCalled);
assertEquals(ERROR_NOT_SET, callback.mFailureReason);
assertEquals(null, callback.mRes);
// test onStarted
WifiManager.LocalOnlyHotspotReservation res =
mWifiManager.new LocalOnlyHotspotReservation(mApConfig);
callback.onStarted(res);
assertEquals(res, callback.mRes);
assertTrue(callback.mOnStartedCalled);
assertFalse(callback.mOnStoppedCalled);
assertEquals(ERROR_NOT_SET, callback.mFailureReason);
// test onStopped
callback.onStopped();
assertEquals(res, callback.mRes);
assertTrue(callback.mOnStartedCalled);
assertTrue(callback.mOnStoppedCalled);
assertEquals(ERROR_NOT_SET, callback.mFailureReason);
// test onFailed
callback.onFailed(ERROR_TEST_REASON);
assertEquals(res, callback.mRes);
assertTrue(callback.mOnStartedCalled);
assertTrue(callback.mOnStoppedCalled);
assertEquals(ERROR_TEST_REASON, callback.mFailureReason);
}
public class TestLocalOnlyHotspotObserver extends LocalOnlyHotspotObserver {
public boolean mOnRegistered = false;
public boolean mOnStartedCalled = false;
public boolean mOnStoppedCalled = false;
public WifiConfiguration mConfig = null;
public LocalOnlyHotspotSubscription mSub = null;
public long mCallingThreadId = -1;
@Override
public void onRegistered(LocalOnlyHotspotSubscription sub) {
mOnRegistered = true;
mSub = sub;
mCallingThreadId = Thread.currentThread().getId();
}
@Override
public void onStarted(WifiConfiguration config) {
mOnStartedCalled = true;
mConfig = config;
mCallingThreadId = Thread.currentThread().getId();
}
@Override
public void onStopped() {
mOnStoppedCalled = true;
mCallingThreadId = Thread.currentThread().getId();
}
}
/**
* Verify observer is properly plumbed when called.
*/
@Test
public void testLocalOnlyHotspotObserver() {
TestLocalOnlyHotspotObserver observer = new TestLocalOnlyHotspotObserver();
assertFalse(observer.mOnRegistered);
assertFalse(observer.mOnStartedCalled);
assertFalse(observer.mOnStoppedCalled);
assertEquals(null, observer.mConfig);
assertEquals(null, observer.mSub);
WifiManager.LocalOnlyHotspotSubscription sub =
mWifiManager.new LocalOnlyHotspotSubscription();
observer.onRegistered(sub);
assertTrue(observer.mOnRegistered);
assertFalse(observer.mOnStartedCalled);
assertFalse(observer.mOnStoppedCalled);
assertEquals(null, observer.mConfig);
assertEquals(sub, observer.mSub);
observer.onStarted(mApConfig);
assertTrue(observer.mOnRegistered);
assertTrue(observer.mOnStartedCalled);
assertFalse(observer.mOnStoppedCalled);
assertEquals(mApConfig, observer.mConfig);
assertEquals(sub, observer.mSub);
observer.onStopped();
assertTrue(observer.mOnRegistered);
assertTrue(observer.mOnStartedCalled);
assertTrue(observer.mOnStoppedCalled);
assertEquals(mApConfig, observer.mConfig);
assertEquals(sub, observer.mSub);
}
/**
* Verify call to startLocalOnlyHotspot goes to WifiServiceImpl.
*/
@Test
public void testStartLocalOnlyHotspot() throws Exception {
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
mWifiManager.startLocalOnlyHotspot(callback, mHandler);
verify(mWifiService)
.startLocalOnlyHotspot(any(Messenger.class), any(IBinder.class), anyString());
}
/**
* Verify a SecurityException is thrown for callers without proper permissions for
* startLocalOnlyHotspot.
*/
@Test(expected = SecurityException.class)
public void testStartLocalOnlyHotspotThrowsSecurityException() throws Exception {
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
doThrow(new SecurityException()).when(mWifiService)
.startLocalOnlyHotspot(any(Messenger.class), any(IBinder.class), anyString());
mWifiManager.startLocalOnlyHotspot(callback, mHandler);
}
/**
* Verify an IllegalStateException is thrown for callers that already have a pending request for
* startLocalOnlyHotspot.
*/
@Test(expected = IllegalStateException.class)
public void testStartLocalOnlyHotspotThrowsIllegalStateException() throws Exception {
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
doThrow(new IllegalStateException()).when(mWifiService)
.startLocalOnlyHotspot(any(Messenger.class), any(IBinder.class), anyString());
mWifiManager.startLocalOnlyHotspot(callback, mHandler);
}
/**
* Verify that the handler provided by the caller is used for the callbacks.
*/
@Test
public void testCorrectLooperIsUsedForHandler() throws Exception {
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
when(mWifiService.startLocalOnlyHotspot(any(Messenger.class), any(IBinder.class),
anyString())).thenReturn(ERROR_INCOMPATIBLE_MODE);
mWifiManager.startLocalOnlyHotspot(callback, mHandler);
mLooper.dispatchAll();
assertEquals(ERROR_INCOMPATIBLE_MODE, callback.mFailureReason);
verify(mContext, never()).getMainLooper();
}
/**
* Verify that the main looper's thread is used if a handler is not provided by the reqiestomg
* application.
*/
@Test
public void testMainLooperIsUsedWhenHandlerNotProvided() throws Exception {
// record thread from looper.getThread and check ids.
TestLooper altLooper = new TestLooper();
when(mContext.getMainLooper()).thenReturn(altLooper.getLooper());
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
when(mWifiService.startLocalOnlyHotspot(any(Messenger.class), any(IBinder.class),
anyString())).thenReturn(ERROR_INCOMPATIBLE_MODE);
mWifiManager.startLocalOnlyHotspot(callback, null);
altLooper.dispatchAll();
assertEquals(ERROR_INCOMPATIBLE_MODE, callback.mFailureReason);
assertEquals(altLooper.getLooper().getThread().getId(), callback.mCallingThreadId);
verify(mContext).getMainLooper();
}
/**
* Verify the LOHS onStarted callback is triggered when WifiManager receives a HOTSPOT_STARTED
* message from WifiServiceImpl.
*/
@Test
public void testOnStartedIsCalledWithReservation() throws Exception {
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
TestLooper callbackLooper = new TestLooper();
Handler callbackHandler = new Handler(callbackLooper.getLooper());
when(mWifiService.startLocalOnlyHotspot(mMessengerCaptor.capture(),
any(IBinder.class), anyString())).thenReturn(REQUEST_REGISTERED);
mWifiManager.startLocalOnlyHotspot(callback, callbackHandler);
callbackLooper.dispatchAll();
mLooper.dispatchAll();
assertFalse(callback.mOnStartedCalled);
assertEquals(null, callback.mRes);
// now trigger the callback
Message msg = new Message();
msg.what = HOTSPOT_STARTED;
msg.obj = mApConfig;
mMessengerCaptor.getValue().send(msg);
mLooper.dispatchAll();
callbackLooper.dispatchAll();
assertTrue(callback.mOnStartedCalled);
assertEquals(mApConfig, callback.mRes.getWifiConfiguration());
}
/**
* Verify onFailed is called if WifiServiceImpl sends a HOTSPOT_STARTED message with a null
* config.
*/
@Test
public void testOnStartedIsCalledWithNullConfig() throws Exception {
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
TestLooper callbackLooper = new TestLooper();
Handler callbackHandler = new Handler(callbackLooper.getLooper());
when(mWifiService.startLocalOnlyHotspot(mMessengerCaptor.capture(),
any(IBinder.class), anyString())).thenReturn(REQUEST_REGISTERED);
mWifiManager.startLocalOnlyHotspot(callback, callbackHandler);
callbackLooper.dispatchAll();
mLooper.dispatchAll();
assertFalse(callback.mOnStartedCalled);
assertEquals(null, callback.mRes);
// now trigger the callback
Message msg = new Message();
msg.what = HOTSPOT_STARTED;
mMessengerCaptor.getValue().send(msg);
mLooper.dispatchAll();
callbackLooper.dispatchAll();
assertFalse(callback.mOnStartedCalled);
assertEquals(ERROR_GENERIC, callback.mFailureReason);
}
/**
* Verify onStopped is called if WifiServiceImpl sends a HOTSPOT_STOPPED message.
*/
@Test
public void testOnStoppedIsCalled() throws Exception {
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
TestLooper callbackLooper = new TestLooper();
Handler callbackHandler = new Handler(callbackLooper.getLooper());
when(mWifiService.startLocalOnlyHotspot(mMessengerCaptor.capture(),
any(IBinder.class), anyString())).thenReturn(REQUEST_REGISTERED);
mWifiManager.startLocalOnlyHotspot(callback, callbackHandler);
callbackLooper.dispatchAll();
mLooper.dispatchAll();
assertFalse(callback.mOnStoppedCalled);
// now trigger the callback
Message msg = new Message();
msg.what = HOTSPOT_STOPPED;
mMessengerCaptor.getValue().send(msg);
mLooper.dispatchAll();
callbackLooper.dispatchAll();
assertTrue(callback.mOnStoppedCalled);
}
/**
* Verify onFailed is called if WifiServiceImpl sends a HOTSPOT_FAILED message.
*/
@Test
public void testOnFailedIsCalled() throws Exception {
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
TestLooper callbackLooper = new TestLooper();
Handler callbackHandler = new Handler(callbackLooper.getLooper());
when(mWifiService.startLocalOnlyHotspot(mMessengerCaptor.capture(),
any(IBinder.class), anyString())).thenReturn(REQUEST_REGISTERED);
mWifiManager.startLocalOnlyHotspot(callback, callbackHandler);
callbackLooper.dispatchAll();
mLooper.dispatchAll();
assertEquals(ERROR_NOT_SET, callback.mFailureReason);
// now trigger the callback
Message msg = new Message();
msg.what = HOTSPOT_FAILED;
msg.arg1 = ERROR_NO_CHANNEL;
mMessengerCaptor.getValue().send(msg);
mLooper.dispatchAll();
callbackLooper.dispatchAll();
assertEquals(ERROR_NO_CHANNEL, callback.mFailureReason);
}
/**
* Verify callback triggered from startLocalOnlyHotspot with an incompatible mode failure.
*/
@Test
public void testLocalOnlyHotspotCallbackFullOnIncompatibleMode() throws Exception {
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
when(mWifiService.startLocalOnlyHotspot(any(Messenger.class), any(IBinder.class),
anyString())).thenReturn(ERROR_INCOMPATIBLE_MODE);
mWifiManager.startLocalOnlyHotspot(callback, mHandler);
mLooper.dispatchAll();
assertEquals(ERROR_INCOMPATIBLE_MODE, callback.mFailureReason);
assertFalse(callback.mOnStartedCalled);
assertFalse(callback.mOnStoppedCalled);
assertEquals(null, callback.mRes);
}
/**
* Verify callback triggered from startLocalOnlyHotspot with a tethering disallowed failure.
*/
@Test
public void testLocalOnlyHotspotCallbackFullOnTetheringDisallowed() throws Exception {
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
when(mWifiService.startLocalOnlyHotspot(any(Messenger.class), any(IBinder.class),
anyString())).thenReturn(ERROR_TETHERING_DISALLOWED);
mWifiManager.startLocalOnlyHotspot(callback, mHandler);
mLooper.dispatchAll();
assertEquals(ERROR_TETHERING_DISALLOWED, callback.mFailureReason);
assertFalse(callback.mOnStartedCalled);
assertFalse(callback.mOnStoppedCalled);
assertEquals(null, callback.mRes);
}
/**
* Verify a SecurityException resulting from an application without necessary permissions will
* bubble up through the call to start LocalOnlyHotspot and will not trigger other callbacks.
*/
@Test(expected = SecurityException.class)
public void testLocalOnlyHotspotCallbackFullOnSecurityException() throws Exception {
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
doThrow(new SecurityException()).when(mWifiService)
.startLocalOnlyHotspot(any(Messenger.class), any(IBinder.class), anyString());
try {
mWifiManager.startLocalOnlyHotspot(callback, mHandler);
} catch (SecurityException e) {
assertEquals(ERROR_NOT_SET, callback.mFailureReason);
assertFalse(callback.mOnStartedCalled);
assertFalse(callback.mOnStoppedCalled);
assertEquals(null, callback.mRes);
throw e;
}
}
/**
* Verify the handler passed to startLocalOnlyHotspot is correctly used for callbacks when
* SoftApMode fails due to a underlying error.
*/
@Test
public void testLocalOnlyHotspotCallbackFullOnNoChannelError() throws Exception {
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
when(mWifiService.startLocalOnlyHotspot(any(Messenger.class), any(IBinder.class),
anyString())).thenReturn(REQUEST_REGISTERED);
mWifiManager.startLocalOnlyHotspot(callback, mHandler);
mLooper.dispatchAll();
//assertEquals(ERROR_NO_CHANNEL, callback.mFailureReason);
assertFalse(callback.mOnStartedCalled);
assertFalse(callback.mOnStoppedCalled);
assertEquals(null, callback.mRes);
}
/**
* Verify that the call to cancel a LOHS request does call stopLOHS.
*/
@Test
public void testCancelLocalOnlyHotspotRequestCallsStopOnWifiService() throws Exception {
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
when(mWifiService.startLocalOnlyHotspot(any(Messenger.class), any(IBinder.class),
anyString())).thenReturn(REQUEST_REGISTERED);
mWifiManager.startLocalOnlyHotspot(callback, mHandler);
mWifiManager.cancelLocalOnlyHotspotRequest();
verify(mWifiService).stopLocalOnlyHotspot();
}
/**
* Verify that we do not crash if cancelLocalOnlyHotspotRequest is called without an existing
* callback stored.
*/
@Test
public void testCancelLocalOnlyHotspotReturnsWithoutExistingRequest() {
mWifiManager.cancelLocalOnlyHotspotRequest();
}
/**
* Verify that the callback is not triggered if the LOHS request was already cancelled.
*/
@Test
public void testCallbackAfterLocalOnlyHotspotWasCancelled() throws Exception {
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
when(mWifiService.startLocalOnlyHotspot(any(Messenger.class), any(IBinder.class),
anyString())).thenReturn(REQUEST_REGISTERED);
mWifiManager.startLocalOnlyHotspot(callback, mHandler);
mWifiManager.cancelLocalOnlyHotspotRequest();
verify(mWifiService).stopLocalOnlyHotspot();
mLooper.dispatchAll();
assertEquals(ERROR_NOT_SET, callback.mFailureReason);
assertFalse(callback.mOnStartedCalled);
assertFalse(callback.mOnStoppedCalled);
assertEquals(null, callback.mRes);
}
/**
* Verify that calling cancel LOHS request does not crash if an error callback was already
* handled.
*/
@Test
public void testCancelAfterLocalOnlyHotspotCallbackTriggered() throws Exception {
TestLocalOnlyHotspotCallback callback = new TestLocalOnlyHotspotCallback();
when(mWifiService.startLocalOnlyHotspot(any(Messenger.class), any(IBinder.class),
anyString())).thenReturn(ERROR_INCOMPATIBLE_MODE);
mWifiManager.startLocalOnlyHotspot(callback, mHandler);
mLooper.dispatchAll();
assertEquals(ERROR_INCOMPATIBLE_MODE, callback.mFailureReason);
assertFalse(callback.mOnStartedCalled);
assertFalse(callback.mOnStoppedCalled);
assertEquals(null, callback.mRes);
mWifiManager.cancelLocalOnlyHotspotRequest();
verify(mWifiService, never()).stopLocalOnlyHotspot();
}
/**
* Verify the watchLocalOnlyHotspot call goes to WifiServiceImpl.
*/
@Test
public void testWatchLocalOnlyHotspot() throws Exception {
TestLocalOnlyHotspotObserver observer = new TestLocalOnlyHotspotObserver();
mWifiManager.watchLocalOnlyHotspot(observer, mHandler);
verify(mWifiService).startWatchLocalOnlyHotspot(any(Messenger.class), any(IBinder.class));
}
/**
* Verify a SecurityException is thrown for callers without proper permissions for
* startWatchLocalOnlyHotspot.
*/
@Test(expected = SecurityException.class)
public void testStartWatchLocalOnlyHotspotThrowsSecurityException() throws Exception {
TestLocalOnlyHotspotObserver observer = new TestLocalOnlyHotspotObserver();
doThrow(new SecurityException()).when(mWifiService)
.startWatchLocalOnlyHotspot(any(Messenger.class), any(IBinder.class));
mWifiManager.watchLocalOnlyHotspot(observer, mHandler);
}
/**
* Verify an IllegalStateException is thrown for callers that already have a pending request for
* watchLocalOnlyHotspot.
*/
@Test(expected = IllegalStateException.class)
public void testStartWatchLocalOnlyHotspotThrowsIllegalStateException() throws Exception {
TestLocalOnlyHotspotObserver observer = new TestLocalOnlyHotspotObserver();
doThrow(new IllegalStateException()).when(mWifiService)
.startWatchLocalOnlyHotspot(any(Messenger.class), any(IBinder.class));
mWifiManager.watchLocalOnlyHotspot(observer, mHandler);
}
/**
* Verify an IllegalArgumentException is thrown if callback is not provided.
*/
@Test
public void registerSoftApCallbackThrowsIllegalArgumentExceptionOnNullArgumentForCallback() {
try {
mWifiManager.registerSoftApCallback(new HandlerExecutor(mHandler), null);
fail("expected IllegalArgumentException");
} catch (IllegalArgumentException expected) {
}
}
/**
* Verify an IllegalArgumentException is thrown if executor is null.
*/
@Test
public void registerSoftApCallbackThrowsIllegalArgumentExceptionOnNullArgumentForExecutor() {
try {
mWifiManager.registerSoftApCallback(null, mSoftApCallback);
fail("expected IllegalArgumentException");
} catch (IllegalArgumentException expected) {
}
}
/**
* Verify an IllegalArgumentException is thrown if callback is not provided.
*/
@Test
public void unregisterSoftApCallbackThrowsIllegalArgumentExceptionOnNullArgumentForCallback() {
try {
mWifiManager.unregisterSoftApCallback(null);
fail("expected IllegalArgumentException");
} catch (IllegalArgumentException expected) {
}
}
/**
* Verify the call to registerSoftApCallback goes to WifiServiceImpl.
*/
@Test
public void registerSoftApCallbackCallGoesToWifiServiceImpl() throws Exception {
mWifiManager.registerSoftApCallback(new HandlerExecutor(mHandler), mSoftApCallback);
verify(mWifiService).registerSoftApCallback(any(IBinder.class),
any(ISoftApCallback.Stub.class), anyInt());
}
/**
* Verify the call to unregisterSoftApCallback goes to WifiServiceImpl.
*/
@Test
public void unregisterSoftApCallbackCallGoesToWifiServiceImpl() throws Exception {
ArgumentCaptor<Integer> callbackIdentifier = ArgumentCaptor.forClass(Integer.class);
mWifiManager.registerSoftApCallback(new HandlerExecutor(mHandler), mSoftApCallback);
verify(mWifiService).registerSoftApCallback(any(IBinder.class),
any(ISoftApCallback.Stub.class), callbackIdentifier.capture());
mWifiManager.unregisterSoftApCallback(mSoftApCallback);
verify(mWifiService).unregisterSoftApCallback(eq((int) callbackIdentifier.getValue()));
}
/*
* Verify client-provided callback is being called through callback proxy
*/
@Test
public void softApCallbackProxyCallsOnStateChanged() throws Exception {
ArgumentCaptor<ISoftApCallback.Stub> callbackCaptor =
ArgumentCaptor.forClass(ISoftApCallback.Stub.class);
mWifiManager.registerSoftApCallback(new HandlerExecutor(mHandler), mSoftApCallback);
verify(mWifiService).registerSoftApCallback(any(IBinder.class), callbackCaptor.capture(),
anyInt());
callbackCaptor.getValue().onStateChanged(WIFI_AP_STATE_ENABLED, 0);
mLooper.dispatchAll();
verify(mSoftApCallback).onStateChanged(WIFI_AP_STATE_ENABLED, 0);
}
/*
* Verify client-provided callback is being called through callback proxy
*/
@Test
public void softApCallbackProxyCallsOnConnectedClientsChanged() throws Exception {
ArgumentCaptor<ISoftApCallback.Stub> callbackCaptor =
ArgumentCaptor.forClass(ISoftApCallback.Stub.class);
mWifiManager.registerSoftApCallback(new HandlerExecutor(mHandler), mSoftApCallback);
verify(mWifiService).registerSoftApCallback(any(IBinder.class), callbackCaptor.capture(),
anyInt());
final List<WifiClient> testClients = new ArrayList();
callbackCaptor.getValue().onConnectedClientsChanged(testClients);
mLooper.dispatchAll();
verify(mSoftApCallback).onConnectedClientsChanged(testClients);
}
/*
* Verify client-provided callback is being called through callback proxy on multiple events
*/
@Test
public void softApCallbackProxyCallsOnMultipleUpdates() throws Exception {
ArgumentCaptor<ISoftApCallback.Stub> callbackCaptor =
ArgumentCaptor.forClass(ISoftApCallback.Stub.class);
mWifiManager.registerSoftApCallback(new HandlerExecutor(mHandler), mSoftApCallback);
verify(mWifiService).registerSoftApCallback(any(IBinder.class), callbackCaptor.capture(),
anyInt());
final List<WifiClient> testClients = new ArrayList();
callbackCaptor.getValue().onStateChanged(WIFI_AP_STATE_ENABLING, 0);
callbackCaptor.getValue().onConnectedClientsChanged(testClients);
callbackCaptor.getValue().onStateChanged(WIFI_AP_STATE_FAILED, SAP_START_FAILURE_GENERAL);
mLooper.dispatchAll();
verify(mSoftApCallback).onStateChanged(WIFI_AP_STATE_ENABLING, 0);
verify(mSoftApCallback).onConnectedClientsChanged(testClients);
verify(mSoftApCallback).onStateChanged(WIFI_AP_STATE_FAILED, SAP_START_FAILURE_GENERAL);
}
/*
* Verify client-provided callback is being called on the correct thread
*/
@Test
public void softApCallbackIsCalledOnCorrectThread() throws Exception {
ArgumentCaptor<ISoftApCallback.Stub> callbackCaptor =
ArgumentCaptor.forClass(ISoftApCallback.Stub.class);
TestLooper altLooper = new TestLooper();
Handler altHandler = new Handler(altLooper.getLooper());
mWifiManager.registerSoftApCallback(new HandlerExecutor(altHandler), mSoftApCallback);
verify(mWifiService).registerSoftApCallback(any(IBinder.class), callbackCaptor.capture(),
anyInt());
callbackCaptor.getValue().onStateChanged(WIFI_AP_STATE_ENABLED, 0);
altLooper.dispatchAll();
verify(mSoftApCallback).onStateChanged(WIFI_AP_STATE_ENABLED, 0);
}
/**
* Verify that the handler provided by the caller is used for registering soft AP callback.
*/
@Test
public void testCorrectLooperIsUsedForSoftApCallbackHandler() throws Exception {
mWifiManager.registerSoftApCallback(new HandlerExecutor(mHandler), mSoftApCallback);
mLooper.dispatchAll();
verify(mWifiService).registerSoftApCallback(any(IBinder.class),
any(ISoftApCallback.Stub.class), anyInt());
verify(mContext, never()).getMainLooper();
}
/**
* Verify that the handler provided by the caller is used for the observer.
*/
@Test
public void testCorrectLooperIsUsedForObserverHandler() throws Exception {
TestLocalOnlyHotspotObserver observer = new TestLocalOnlyHotspotObserver();
mWifiManager.watchLocalOnlyHotspot(observer, mHandler);
mLooper.dispatchAll();
assertTrue(observer.mOnRegistered);
verify(mContext, never()).getMainLooper();
}
/**
* Verify that the main looper's thread is used if a handler is not provided by the requesting
* application.
*/
@Test
public void testMainLooperIsUsedWhenHandlerNotProvidedForObserver() throws Exception {
// record thread from looper.getThread and check ids.
TestLooper altLooper = new TestLooper();
when(mContext.getMainLooper()).thenReturn(altLooper.getLooper());
TestLocalOnlyHotspotObserver observer = new TestLocalOnlyHotspotObserver();
mWifiManager.watchLocalOnlyHotspot(observer, null);
altLooper.dispatchAll();
assertTrue(observer.mOnRegistered);
assertEquals(altLooper.getLooper().getThread().getId(), observer.mCallingThreadId);
verify(mContext).getMainLooper();
}
/**
* Verify the LOHS onRegistered observer callback is triggered when WifiManager receives a
* HOTSPOT_OBSERVER_REGISTERED message from WifiServiceImpl.
*/
@Test
public void testOnRegisteredIsCalledWithSubscription() throws Exception {
TestLocalOnlyHotspotObserver observer = new TestLocalOnlyHotspotObserver();
TestLooper observerLooper = new TestLooper();
Handler observerHandler = new Handler(observerLooper.getLooper());
assertFalse(observer.mOnRegistered);
assertEquals(null, observer.mSub);
mWifiManager.watchLocalOnlyHotspot(observer, observerHandler);
verify(mWifiService).startWatchLocalOnlyHotspot(mMessengerCaptor.capture(),
any(IBinder.class));
// now trigger the callback
observerLooper.dispatchAll();
mLooper.dispatchAll();
assertTrue(observer.mOnRegistered);
assertNotNull(observer.mSub);
}
/**
* Verify the LOHS onStarted observer callback is triggered when WifiManager receives a
* HOTSPOT_STARTED message from WifiServiceImpl.
*/
@Test
public void testObserverOnStartedIsCalledWithWifiConfig() throws Exception {
TestLocalOnlyHotspotObserver observer = new TestLocalOnlyHotspotObserver();
TestLooper observerLooper = new TestLooper();
Handler observerHandler = new Handler(observerLooper.getLooper());
mWifiManager.watchLocalOnlyHotspot(observer, observerHandler);
verify(mWifiService).startWatchLocalOnlyHotspot(mMessengerCaptor.capture(),
any(IBinder.class));
observerLooper.dispatchAll();
mLooper.dispatchAll();
assertFalse(observer.mOnStartedCalled);
// now trigger the callback
Message msg = new Message();
msg.what = HOTSPOT_STARTED;
msg.obj = mApConfig;
mMessengerCaptor.getValue().send(msg);
mLooper.dispatchAll();
observerLooper.dispatchAll();
assertTrue(observer.mOnStartedCalled);
assertEquals(mApConfig, observer.mConfig);
}
/**
* Verify the LOHS onStarted observer callback is triggered not when WifiManager receives a
* HOTSPOT_STARTED message from WifiServiceImpl with a null config.
*/
@Test
public void testObserverOnStartedNotCalledWithNullConfig() throws Exception {
TestLocalOnlyHotspotObserver observer = new TestLocalOnlyHotspotObserver();
TestLooper observerLooper = new TestLooper();
Handler observerHandler = new Handler(observerLooper.getLooper());
mWifiManager.watchLocalOnlyHotspot(observer, observerHandler);
verify(mWifiService).startWatchLocalOnlyHotspot(mMessengerCaptor.capture(),
any(IBinder.class));
observerLooper.dispatchAll();
mLooper.dispatchAll();
assertFalse(observer.mOnStartedCalled);
// now trigger the callback
Message msg = new Message();
msg.what = HOTSPOT_STARTED;
mMessengerCaptor.getValue().send(msg);
mLooper.dispatchAll();
observerLooper.dispatchAll();
assertFalse(observer.mOnStartedCalled);
assertEquals(null, observer.mConfig);
}
/**
* Verify the LOHS onStopped observer callback is triggered when WifiManager receives a
* HOTSPOT_STOPPED message from WifiServiceImpl.
*/
@Test
public void testObserverOnStoppedIsCalled() throws Exception {
TestLocalOnlyHotspotObserver observer = new TestLocalOnlyHotspotObserver();
TestLooper observerLooper = new TestLooper();
Handler observerHandler = new Handler(observerLooper.getLooper());
mWifiManager.watchLocalOnlyHotspot(observer, observerHandler);
verify(mWifiService).startWatchLocalOnlyHotspot(mMessengerCaptor.capture(),
any(IBinder.class));
observerLooper.dispatchAll();
mLooper.dispatchAll();
assertFalse(observer.mOnStoppedCalled);
// now trigger the callback
Message msg = new Message();
msg.what = HOTSPOT_STOPPED;
mMessengerCaptor.getValue().send(msg);
mLooper.dispatchAll();
observerLooper.dispatchAll();
assertTrue(observer.mOnStoppedCalled);
}
/**
* Verify WifiServiceImpl is not called if there is not a registered LOHS observer callback.
*/
@Test
public void testUnregisterWifiServiceImplNotCalledWithoutRegisteredObserver() throws Exception {
mWifiManager.unregisterLocalOnlyHotspotObserver();
verifyZeroInteractions(mWifiService);
}
/**
* Verify WifiServiceImpl is called when there is a registered LOHS observer callback.
*/
@Test
public void testUnregisterWifiServiceImplCalledWithRegisteredObserver() throws Exception {
TestLocalOnlyHotspotObserver observer = new TestLocalOnlyHotspotObserver();
TestLooper observerLooper = new TestLooper();
Handler observerHandler = new Handler(observerLooper.getLooper());
mWifiManager.watchLocalOnlyHotspot(observer, observerHandler);
mWifiManager.unregisterLocalOnlyHotspotObserver();
verify(mWifiService).stopWatchLocalOnlyHotspot();
}
/**
* Verify that calls WifiServiceImpl to set country code when no exception happens.
*/
@Test
public void testSetWifiCountryCode() throws Exception {
mWifiManager.setCountryCode(TEST_COUNTRY_CODE);
verify(mWifiService).setCountryCode(TEST_COUNTRY_CODE);
}
/**
* Verify that WifiManager.setCountryCode() rethrows exceptions if caller does not
* have necessary permissions.
*/
@Test(expected = SecurityException.class)
public void testSetWifiCountryCodeFailedOnSecurityException() throws Exception {
doThrow(new SecurityException()).when(mWifiService).setCountryCode(anyString());
mWifiManager.setCountryCode(TEST_COUNTRY_CODE);
}
/**
* Test that calls to get the current WPS config token return null and do not have any
* interactions with WifiServiceImpl.
*/
@Test
public void testGetCurrentNetworkWpsNfcConfigurationTokenReturnsNull() {
assertNull(mWifiManager.getCurrentNetworkWpsNfcConfigurationToken());
verifyNoMoreInteractions(mWifiService);
}
class WpsCallbackTester extends WifiManager.WpsCallback {
public boolean mStarted = false;
public boolean mSucceeded = false;
public boolean mFailed = false;
public int mFailureCode = -1;
@Override
public void onStarted(String pin) {
mStarted = true;
}
@Override
public void onSucceeded() {
mSucceeded = true;
}
@Override
public void onFailed(int reason) {
mFailed = true;
mFailureCode = reason;
}
}
/**
* Verify that a call to start WPS immediately returns a failure.
*/
@Test
public void testStartWpsImmediatelyFailsWithCallback() {
WpsCallbackTester wpsCallback = new WpsCallbackTester();
mWifiManager.startWps(null, wpsCallback);
assertTrue(wpsCallback.mFailed);
assertEquals(WifiManager.ERROR, wpsCallback.mFailureCode);
assertFalse(wpsCallback.mStarted);
assertFalse(wpsCallback.mSucceeded);
verifyNoMoreInteractions(mWifiService);
}
/**
* Verify that a call to start WPS does not go to WifiServiceImpl if we do not have a callback.
*/
@Test
public void testStartWpsDoesNotCallWifiServiceImpl() {
mWifiManager.startWps(null, null);
verifyNoMoreInteractions(mWifiService);
}
/**
* Verify that a call to cancel WPS immediately returns a failure.
*/
@Test
public void testCancelWpsImmediatelyFailsWithCallback() {
WpsCallbackTester wpsCallback = new WpsCallbackTester();
mWifiManager.cancelWps(wpsCallback);
assertTrue(wpsCallback.mFailed);
assertEquals(WifiManager.ERROR, wpsCallback.mFailureCode);
assertFalse(wpsCallback.mStarted);
assertFalse(wpsCallback.mSucceeded);
verifyNoMoreInteractions(mWifiService);
}
/**
* Verify that a call to cancel WPS does not go to WifiServiceImpl if we do not have a callback.
*/
@Test
public void testCancelWpsDoesNotCallWifiServiceImpl() {
mWifiManager.cancelWps(null);
verifyNoMoreInteractions(mWifiService);
}
/**
* Verify that a successful call properly returns true.
*/
@Test
public void testSetWifiApConfigurationSuccessReturnsTrue() throws Exception {
WifiConfiguration apConfig = new WifiConfiguration();
when(mWifiService.setWifiApConfiguration(eq(apConfig), eq(TEST_PACKAGE_NAME)))
.thenReturn(true);
assertTrue(mWifiManager.setWifiApConfiguration(apConfig));
}
/**
* Verify that a failed call properly returns false.
*/
@Test
public void testSetWifiApConfigurationFailureReturnsFalse() throws Exception {
WifiConfiguration apConfig = new WifiConfiguration();
when(mWifiService.setWifiApConfiguration(eq(apConfig), eq(TEST_PACKAGE_NAME)))
.thenReturn(false);
assertFalse(mWifiManager.setWifiApConfiguration(apConfig));
}
/**
* Verify Exceptions are rethrown when underlying calls to WifiService throw exceptions.
*/
@Test
public void testSetWifiApConfigurationRethrowsException() throws Exception {
doThrow(new SecurityException()).when(mWifiService).setWifiApConfiguration(any(), any());
try {
mWifiManager.setWifiApConfiguration(new WifiConfiguration());
fail("setWifiApConfiguration should rethrow Exceptions from WifiService");
} catch (SecurityException e) { }
}
/**
* Check the call to startScan calls WifiService.
*/
@Test
public void testStartScan() throws Exception {
when(mWifiService.startScan(TEST_PACKAGE_NAME)).thenReturn(true);
assertTrue(mWifiManager.startScan());
when(mWifiService.startScan(TEST_PACKAGE_NAME)).thenReturn(false);
assertFalse(mWifiManager.startScan());
}
/**
* Verify main looper is used when handler is not provided.
*/
@Test
public void registerTrafficStateCallbackUsesMainLooperOnNullArgumentForHandler()
throws Exception {
when(mContext.getMainLooper()).thenReturn(mLooper.getLooper());
ArgumentCaptor<ITrafficStateCallback.Stub> callbackCaptor =
ArgumentCaptor.forClass(ITrafficStateCallback.Stub.class);
mWifiManager.registerTrafficStateCallback(mTrafficStateCallback, null);
verify(mWifiService).registerTrafficStateCallback(
any(IBinder.class), callbackCaptor.capture(), anyInt());
assertEquals(0, mLooper.dispatchAll());
callbackCaptor.getValue().onStateChanged(TrafficStateCallback.DATA_ACTIVITY_INOUT);
assertEquals(1, mLooper.dispatchAll());
verify(mTrafficStateCallback).onStateChanged(TrafficStateCallback.DATA_ACTIVITY_INOUT);
}
/**
* Verify the call to unregisterTrafficStateCallback goes to WifiServiceImpl.
*/
@Test
public void unregisterTrafficStateCallbackCallGoesToWifiServiceImpl() throws Exception {
ArgumentCaptor<Integer> callbackIdentifier = ArgumentCaptor.forClass(Integer.class);
mWifiManager.registerTrafficStateCallback(mTrafficStateCallback, mHandler);
verify(mWifiService).registerTrafficStateCallback(any(IBinder.class),
any(ITrafficStateCallback.Stub.class), callbackIdentifier.capture());
mWifiManager.unregisterTrafficStateCallback(mTrafficStateCallback);
verify(mWifiService).unregisterTrafficStateCallback(
eq((int) callbackIdentifier.getValue()));
}
/*
* Verify client-provided callback is being called through callback proxy on multiple events
*/
@Test
public void trafficStateCallbackProxyCallsOnMultipleUpdates() throws Exception {
ArgumentCaptor<ITrafficStateCallback.Stub> callbackCaptor =
ArgumentCaptor.forClass(ITrafficStateCallback.Stub.class);
mWifiManager.registerTrafficStateCallback(mTrafficStateCallback, mHandler);
verify(mWifiService).registerTrafficStateCallback(
any(IBinder.class), callbackCaptor.capture(), anyInt());
InOrder inOrder = inOrder(mTrafficStateCallback);
callbackCaptor.getValue().onStateChanged(TrafficStateCallback.DATA_ACTIVITY_IN);
callbackCaptor.getValue().onStateChanged(TrafficStateCallback.DATA_ACTIVITY_INOUT);
callbackCaptor.getValue().onStateChanged(TrafficStateCallback.DATA_ACTIVITY_OUT);
mLooper.dispatchAll();
inOrder.verify(mTrafficStateCallback).onStateChanged(
TrafficStateCallback.DATA_ACTIVITY_IN);
inOrder.verify(mTrafficStateCallback).onStateChanged(
TrafficStateCallback.DATA_ACTIVITY_INOUT);
inOrder.verify(mTrafficStateCallback).onStateChanged(
TrafficStateCallback.DATA_ACTIVITY_OUT);
}
/*
* Verify client-provided callback is being called on the correct thread
*/
@Test
public void trafficStateCallbackIsCalledOnCorrectThread() throws Exception {
ArgumentCaptor<ITrafficStateCallback.Stub> callbackCaptor =
ArgumentCaptor.forClass(ITrafficStateCallback.Stub.class);
TestLooper altLooper = new TestLooper();
Handler altHandler = new Handler(altLooper.getLooper());
mWifiManager.registerTrafficStateCallback(mTrafficStateCallback, altHandler);
verify(mContext, never()).getMainLooper();
verify(mWifiService).registerTrafficStateCallback(
any(IBinder.class), callbackCaptor.capture(), anyInt());
assertEquals(0, altLooper.dispatchAll());
callbackCaptor.getValue().onStateChanged(TrafficStateCallback.DATA_ACTIVITY_INOUT);
assertEquals(1, altLooper.dispatchAll());
verify(mTrafficStateCallback).onStateChanged(TrafficStateCallback.DATA_ACTIVITY_INOUT);
}
/**
* Verify the call to registerNetworkRequestMatchCallback goes to WifiServiceImpl.
*/
@Test
public void registerNetworkRequestMatchCallbackCallGoesToWifiServiceImpl()
throws Exception {
when(mContext.getMainLooper()).thenReturn(mLooper.getLooper());
ArgumentCaptor<INetworkRequestMatchCallback.Stub> callbackCaptor =
ArgumentCaptor.forClass(INetworkRequestMatchCallback.Stub.class);
mWifiManager.registerNetworkRequestMatchCallback(mNetworkRequestMatchCallback, null);
verify(mWifiService).registerNetworkRequestMatchCallback(
any(IBinder.class), callbackCaptor.capture(), anyInt());
INetworkRequestUserSelectionCallback iUserSelectionCallback =
mock(INetworkRequestUserSelectionCallback.class);
assertEquals(0, mLooper.dispatchAll());
callbackCaptor.getValue().onAbort();
assertEquals(1, mLooper.dispatchAll());
verify(mNetworkRequestMatchCallback).onAbort();
callbackCaptor.getValue().onMatch(new ArrayList<ScanResult>());
assertEquals(1, mLooper.dispatchAll());
verify(mNetworkRequestMatchCallback).onMatch(anyList());
callbackCaptor.getValue().onUserSelectionConnectSuccess(new WifiConfiguration());
assertEquals(1, mLooper.dispatchAll());
verify(mNetworkRequestMatchCallback).onUserSelectionConnectSuccess(
any(WifiConfiguration.class));
callbackCaptor.getValue().onUserSelectionConnectFailure(new WifiConfiguration());
assertEquals(1, mLooper.dispatchAll());
verify(mNetworkRequestMatchCallback).onUserSelectionConnectFailure(
any(WifiConfiguration.class));
}
/**
* Verify the call to unregisterNetworkRequestMatchCallback goes to WifiServiceImpl.
*/
@Test
public void unregisterNetworkRequestMatchCallbackCallGoesToWifiServiceImpl() throws Exception {
ArgumentCaptor<Integer> callbackIdentifier = ArgumentCaptor.forClass(Integer.class);
mWifiManager.registerNetworkRequestMatchCallback(mNetworkRequestMatchCallback, mHandler);
verify(mWifiService).registerNetworkRequestMatchCallback(
any(IBinder.class), any(INetworkRequestMatchCallback.class),
callbackIdentifier.capture());
mWifiManager.unregisterNetworkRequestMatchCallback(mNetworkRequestMatchCallback);
verify(mWifiService).unregisterNetworkRequestMatchCallback(
eq((int) callbackIdentifier.getValue()));
}
/**
* Verify the call to NetworkRequestUserSelectionCallback goes to
* WifiServiceImpl.
*/
@Test
public void networkRequestUserSelectionCallbackCallGoesToWifiServiceImpl()
throws Exception {
when(mContext.getMainLooper()).thenReturn(mLooper.getLooper());
ArgumentCaptor<INetworkRequestMatchCallback.Stub> callbackCaptor =
ArgumentCaptor.forClass(INetworkRequestMatchCallback.Stub.class);
mWifiManager.registerNetworkRequestMatchCallback(mNetworkRequestMatchCallback, null);
verify(mWifiService).registerNetworkRequestMatchCallback(
any(IBinder.class), callbackCaptor.capture(), anyInt());
INetworkRequestUserSelectionCallback iUserSelectionCallback =
mock(INetworkRequestUserSelectionCallback.class);
ArgumentCaptor<NetworkRequestUserSelectionCallback> userSelectionCallbackCaptor =
ArgumentCaptor.forClass(NetworkRequestUserSelectionCallback.class);
callbackCaptor.getValue().onUserSelectionCallbackRegistration(
iUserSelectionCallback);
assertEquals(1, mLooper.dispatchAll());
verify(mNetworkRequestMatchCallback).onUserSelectionCallbackRegistration(
userSelectionCallbackCaptor.capture());
WifiConfiguration selected = new WifiConfiguration();
userSelectionCallbackCaptor.getValue().select(selected);
verify(iUserSelectionCallback).select(selected);
userSelectionCallbackCaptor.getValue().reject();
verify(iUserSelectionCallback).reject();
}
/**
* Check the call to getAllMatchingWifiConfigs calls getAllMatchingFqdnsForScanResults and
* getWifiConfigsForPasspointProfiles of WifiService in order.
*/
@Test
public void testGetAllMatchingWifiConfigs() throws Exception {
Map<String, List<ScanResult>> fqdns = new HashMap<>();
fqdns.put("www.test.com", new ArrayList<>());
when(mWifiService.getAllMatchingFqdnsForScanResults(any(List.class))).thenReturn(fqdns);
InOrder inOrder = inOrder(mWifiService);
mWifiManager.getAllMatchingWifiConfigs(new ArrayList<>());
inOrder.verify(mWifiService).getAllMatchingFqdnsForScanResults(any(List.class));
inOrder.verify(mWifiService).getWifiConfigsForPasspointProfiles(any(List.class));
}
/**
* Check the call to getMatchingOsuProviders calls getMatchingOsuProviders of WifiService
* with the provided a list of ScanResult.
*/
@Test
public void testGetMatchingOsuProviders() throws Exception {
mWifiManager.getMatchingOsuProviders(new ArrayList<>());
verify(mWifiService).getMatchingOsuProviders(any(List.class));
}
/**
* Verify calls to {@link WifiManager#addNetworkSuggestions(List)} and
* {@link WifiManager#removeNetworkSuggestions(List)}.
*/
@Test
public void addRemoveNetworkSuggestions() throws Exception {
when(mWifiService.addNetworkSuggestions(any(List.class), anyString()))
.thenReturn(WifiManager.STATUS_NETWORK_SUGGESTIONS_SUCCESS);
when(mWifiService.removeNetworkSuggestions(any(List.class), anyString()))
.thenReturn(WifiManager.STATUS_NETWORK_SUGGESTIONS_SUCCESS);
assertEquals(WifiManager.STATUS_NETWORK_SUGGESTIONS_SUCCESS,
mWifiManager.addNetworkSuggestions(new ArrayList<>()));
verify(mWifiService).addNetworkSuggestions(anyList(), eq(TEST_PACKAGE_NAME));
assertEquals(WifiManager.STATUS_NETWORK_SUGGESTIONS_SUCCESS,
mWifiManager.removeNetworkSuggestions(new ArrayList<>()));
verify(mWifiService).removeNetworkSuggestions(anyList(), eq(TEST_PACKAGE_NAME));
}
/**
* Verify call to {@link WifiManager#getMaxNumberOfNetworkSuggestionsPerApp()}.
*/
@Test
public void getMaxNumberOfNetworkSuggestionsPerApp() {
assertEquals(WifiManager.NETWORK_SUGGESTIONS_MAX_PER_APP,
mWifiManager.getMaxNumberOfNetworkSuggestionsPerApp());
}
/**
* Verify getting the factory MAC address.
*/
@Test
public void testGetFactoryMacAddress() throws Exception {
when(mWifiService.getFactoryMacAddresses()).thenReturn(TEST_MAC_ADDRESSES);
assertArrayEquals(TEST_MAC_ADDRESSES, mWifiManager.getFactoryMacAddresses());
verify(mWifiService).getFactoryMacAddresses();
}
/**
* Verify the call to addOnWifiUsabilityStatsListener goes to WifiServiceImpl.
*/
@Test
public void addOnWifiUsabilityStatsListeneroesToWifiServiceImpl() throws Exception {
mExecutor = new SynchronousExecutor();
mWifiManager.addOnWifiUsabilityStatsListener(mExecutor, mOnWifiUsabilityStatsListener);
verify(mWifiService).addOnWifiUsabilityStatsListener(any(IBinder.class),
any(IOnWifiUsabilityStatsListener.Stub.class), anyInt());
}
/**
* Verify the call to removeOnWifiUsabilityStatsListener goes to WifiServiceImpl.
*/
@Test
public void removeOnWifiUsabilityListenerGoesToWifiServiceImpl() throws Exception {
ArgumentCaptor<Integer> listenerIdentifier = ArgumentCaptor.forClass(Integer.class);
mExecutor = new SynchronousExecutor();
mWifiManager.addOnWifiUsabilityStatsListener(mExecutor, mOnWifiUsabilityStatsListener);
verify(mWifiService).addOnWifiUsabilityStatsListener(any(IBinder.class),
any(IOnWifiUsabilityStatsListener.Stub.class), listenerIdentifier.capture());
mWifiManager.removeOnWifiUsabilityStatsListener(mOnWifiUsabilityStatsListener);
verify(mWifiService).removeOnWifiUsabilityStatsListener(
eq((int) listenerIdentifier.getValue()));
}
/**
* Defined for testing purpose.
*/
class SynchronousExecutor implements Executor {
public void execute(Runnable r) {
r.run();
}
}
/**
* Test behavior of isEnhancedOpenSupported
*/
@Test
public void testIsEnhancedOpenSupported() throws Exception {
when(mWifiService.getSupportedFeatures())
.thenReturn(new Long(WifiManager.WIFI_FEATURE_OWE));
assertTrue(mWifiManager.isEnhancedOpenSupported());
when(mWifiService.getSupportedFeatures())
.thenReturn(new Long(~WifiManager.WIFI_FEATURE_OWE));
assertFalse(mWifiManager.isEnhancedOpenSupported());
}
/**
* Test behavior of isWpa3SaeSupported
*/
@Test
public void testIsWpa3SaeSupported() throws Exception {
when(mWifiService.getSupportedFeatures())
.thenReturn(new Long(WifiManager.WIFI_FEATURE_WPA3_SAE));
assertTrue(mWifiManager.isWpa3SaeSupported());
when(mWifiService.getSupportedFeatures())
.thenReturn(new Long(~WifiManager.WIFI_FEATURE_WPA3_SAE));
assertFalse(mWifiManager.isWpa3SaeSupported());
}
/**
* Test behavior of isWpa3SuiteBSupported
*/
@Test
public void testIsWpa3SuiteBSupported() throws Exception {
when(mWifiService.getSupportedFeatures())
.thenReturn(new Long(WifiManager.WIFI_FEATURE_WPA3_SUITE_B));
assertTrue(mWifiManager.isWpa3SuiteBSupported());
when(mWifiService.getSupportedFeatures())
.thenReturn(new Long(~WifiManager.WIFI_FEATURE_WPA3_SUITE_B));
assertFalse(mWifiManager.isWpa3SuiteBSupported());
}
/**
* Test behavior of isEasyConnectSupported
*/
@Test
public void testIsEasyConnectSupported() throws Exception {
when(mWifiService.getSupportedFeatures())
.thenReturn(new Long(WifiManager.WIFI_FEATURE_DPP));
assertTrue(mWifiManager.isEasyConnectSupported());
when(mWifiService.getSupportedFeatures())
.thenReturn(new Long(~WifiManager.WIFI_FEATURE_DPP));
assertFalse(mWifiManager.isEasyConnectSupported());
}
/**
* Test behavior of {@link WifiManager#addNetwork(WifiConfiguration)}
*/
@Test
public void testAddNetwork() throws Exception {
WifiConfiguration configuration = new WifiConfiguration();
when(mWifiService.addOrUpdateNetwork(any(), anyString()))
.thenReturn(TEST_NETWORK_ID);
assertEquals(mWifiManager.addNetwork(configuration), TEST_NETWORK_ID);
verify(mWifiService).addOrUpdateNetwork(configuration, mContext.getOpPackageName());
// send a null config
assertEquals(mWifiManager.addNetwork(null), -1);
}
/**
* Test behavior of {@link WifiManager#addNetwork(WifiConfiguration)}
*/
@Test
public void testUpdateNetwork() throws Exception {
WifiConfiguration configuration = new WifiConfiguration();
when(mWifiService.addOrUpdateNetwork(any(), anyString()))
.thenReturn(TEST_NETWORK_ID);
configuration.networkId = TEST_NETWORK_ID;
assertEquals(mWifiManager.updateNetwork(configuration), TEST_NETWORK_ID);
verify(mWifiService).addOrUpdateNetwork(configuration, mContext.getOpPackageName());
// config with invalid network ID
configuration.networkId = -1;
assertEquals(mWifiManager.updateNetwork(configuration), -1);
// send a null config
assertEquals(mWifiManager.updateNetwork(null), -1);
}
/**
* Test behavior of {@link WifiManager#enableNetwork(int, boolean)}
*/
@Test
public void testEnableNetwork() throws Exception {
when(mWifiService.enableNetwork(anyInt(), anyBoolean(), anyString()))
.thenReturn(true);
assertTrue(mWifiManager.enableNetwork(TEST_NETWORK_ID, true));
verify(mWifiService).enableNetwork(TEST_NETWORK_ID, true, mContext.getOpPackageName());
}
/**
* Test behavior of {@link WifiManager#disableNetwork(int)}
*/
@Test
public void testDisableNetwork() throws Exception {
when(mWifiService.disableNetwork(anyInt(), anyString()))
.thenReturn(true);
assertTrue(mWifiManager.disableNetwork(TEST_NETWORK_ID));
verify(mWifiService).disableNetwork(TEST_NETWORK_ID, mContext.getOpPackageName());
}
/**
* Test behavior of {@link WifiManager#disconnect()}
*/
@Test
public void testDisconnect() throws Exception {
when(mWifiService.disconnect(anyString())).thenReturn(true);
assertTrue(mWifiManager.disconnect());
verify(mWifiService).disconnect(mContext.getOpPackageName());
}
/**
* Test behavior of {@link WifiManager#reconnect()}
*/
@Test
public void testReconnect() throws Exception {
when(mWifiService.reconnect(anyString())).thenReturn(true);
assertTrue(mWifiManager.reconnect());
verify(mWifiService).reconnect(mContext.getOpPackageName());
}
/**
* Test behavior of {@link WifiManager#reassociate()}
*/
@Test
public void testReassociate() throws Exception {
when(mWifiService.reassociate(anyString())).thenReturn(true);
assertTrue(mWifiManager.reassociate());
verify(mWifiService).reassociate(mContext.getOpPackageName());
}
/**
* Test behavior of {@link WifiManager#getSupportedFeatures()}
*/
@Test
public void testGetSupportedFeatures() throws Exception {
long supportedFeatures =
WifiManager.WIFI_FEATURE_SCANNER
| WifiManager.WIFI_FEATURE_PASSPOINT
| WifiManager.WIFI_FEATURE_P2P;
when(mWifiService.getSupportedFeatures())
.thenReturn(Long.valueOf(supportedFeatures));
assertTrue(mWifiManager.isWifiScannerSupported());
assertTrue(mWifiManager.isPasspointSupported());
assertTrue(mWifiManager.isP2pSupported());
assertFalse(mWifiManager.isPortableHotspotSupported());
assertFalse(mWifiManager.is5GHzBandSupported());
assertFalse(mWifiManager.isDeviceToDeviceRttSupported());
assertFalse(mWifiManager.isDeviceToApRttSupported());
assertFalse(mWifiManager.isPreferredNetworkOffloadSupported());
assertFalse(mWifiManager.isAdditionalStaSupported());
assertFalse(mWifiManager.isTdlsSupported());
assertFalse(mWifiManager.isOffChannelTdlsSupported());
assertFalse(mWifiManager.isEnhancedPowerReportingSupported());
}
/**
* Test behavior of {@link WifiManager#getControllerActivityEnergyInfo()}
*/
@Test
public void testGetControllerActivityEnergyInfo() throws Exception {
WifiActivityEnergyInfo activityEnergyInfo =
new WifiActivityEnergyInfo(5, 3, 3, new long[]{5L, 5L, 5L}, 5, 5, 5, 5);
when(mWifiService.reportActivityInfo()).thenReturn(activityEnergyInfo);
assertEquals(activityEnergyInfo, mWifiManager.getControllerActivityEnergyInfo());
}
/**
* Test behavior of {@link WifiManager#getConnectionInfo()}
*/
@Test
public void testGetConnectionInfo() throws Exception {
WifiInfo wifiInfo = new WifiInfo();
when(mWifiService.getConnectionInfo(anyString())).thenReturn(wifiInfo);
assertEquals(wifiInfo, mWifiManager.getConnectionInfo());
}
/**
* Test behavior of {@link WifiManager#isDualModeSupported()} ()}
*/
@Test
public void testIsDualModeSupported() throws Exception {
when(mWifiService.needs5GHzToAnyApBandConversion()).thenReturn(true);
assertTrue(mWifiManager.isDualModeSupported());
verify(mWifiService).needs5GHzToAnyApBandConversion();
}
/**
* Test behavior of {@link WifiManager#isDualBandSupported()}
*/
@Test
public void testIsDualBandSupported() throws Exception {
when(mWifiService.isDualBandSupported()).thenReturn(true);
assertTrue(mWifiManager.isDualBandSupported());
verify(mWifiService).isDualBandSupported();
}
/**
* Test behavior of {@link WifiManager#getDhcpInfo()}
*/
@Test
public void testGetDhcpInfo() throws Exception {
DhcpInfo dhcpInfo = new DhcpInfo();
when(mWifiService.getDhcpInfo()).thenReturn(dhcpInfo);
assertEquals(dhcpInfo, mWifiManager.getDhcpInfo());
verify(mWifiService).getDhcpInfo();
}
/**
* Test behavior of {@link WifiManager#setWifiEnabled(boolean)}
*/
@Test
public void testSetWifiEnabled() throws Exception {
when(mWifiService.setWifiEnabled(anyString(), anyBoolean())).thenReturn(true);
assertTrue(mWifiManager.setWifiEnabled(true));
verify(mWifiService).setWifiEnabled(mContext.getOpPackageName(), true);
assertTrue(mWifiManager.setWifiEnabled(false));
verify(mWifiService).setWifiEnabled(mContext.getOpPackageName(), false);
}
}