| /* |
| * Copyright (C) 2015 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 com.android.systemui.statusbar.policy; |
| |
| import static org.mockito.Mockito.mock; |
| import static org.mockito.Mockito.when; |
| |
| import android.content.Intent; |
| import android.net.ConnectivityManager; |
| import android.net.NetworkCapabilities; |
| import android.net.wifi.WifiManager; |
| import android.os.Looper; |
| import android.telephony.PhoneStateListener; |
| import android.telephony.ServiceState; |
| import android.telephony.SignalStrength; |
| import android.telephony.SubscriptionInfo; |
| import android.telephony.SubscriptionManager; |
| import android.telephony.TelephonyManager; |
| import android.util.Log; |
| |
| import com.android.internal.telephony.cdma.EriInfo; |
| import com.android.systemui.SysuiTestCase; |
| import com.android.systemui.statusbar.policy.NetworkController.IconState; |
| import com.android.systemui.statusbar.policy.NetworkControllerImpl.Config; |
| |
| import org.mockito.ArgumentCaptor; |
| import org.mockito.Mockito; |
| |
| import java.io.PrintWriter; |
| import java.io.StringWriter; |
| import java.util.ArrayList; |
| import java.util.List; |
| |
| public class NetworkControllerBaseTest extends SysuiTestCase { |
| private static final String TAG = "NetworkControllerBaseTest"; |
| protected static final int DEFAULT_LEVEL = 2; |
| protected static final int DEFAULT_SIGNAL_STRENGTH = |
| TelephonyIcons.TELEPHONY_SIGNAL_STRENGTH[1][DEFAULT_LEVEL]; |
| protected static final int DEFAULT_QS_SIGNAL_STRENGTH = |
| TelephonyIcons.QS_TELEPHONY_SIGNAL_STRENGTH[1][DEFAULT_LEVEL]; |
| protected static final int DEFAULT_ICON = TelephonyIcons.ICON_3G; |
| protected static final int DEFAULT_QS_ICON = TelephonyIcons.QS_ICON_3G; |
| |
| protected NetworkControllerImpl mNetworkController; |
| protected MobileSignalController mMobileSignalController; |
| protected PhoneStateListener mPhoneStateListener; |
| private SignalStrength mSignalStrength; |
| private ServiceState mServiceState; |
| protected ConnectivityManager mMockCm; |
| protected WifiManager mMockWm; |
| protected SubscriptionManager mMockSm; |
| protected TelephonyManager mMockTm; |
| protected Config mConfig; |
| protected CallbackHandler mCallbackHandler; |
| |
| protected int mSubId; |
| |
| private NetworkCapabilities mNetCapabilities; |
| |
| @Override |
| protected void setUp() throws Exception { |
| super.setUp(); |
| |
| mMockWm = mock(WifiManager.class); |
| mMockTm = mock(TelephonyManager.class); |
| mMockSm = mock(SubscriptionManager.class); |
| mMockCm = mock(ConnectivityManager.class); |
| mNetCapabilities = new NetworkCapabilities(); |
| when(mMockCm.isNetworkSupported(ConnectivityManager.TYPE_MOBILE)).thenReturn(true); |
| when(mMockCm.getDefaultNetworkCapabilitiesForUser(0)).thenReturn( |
| new NetworkCapabilities[] { mNetCapabilities }); |
| |
| mSignalStrength = mock(SignalStrength.class); |
| mServiceState = mock(ServiceState.class); |
| |
| mConfig = new Config(); |
| mConfig.hspaDataDistinguishable = true; |
| mCallbackHandler = mock(CallbackHandler.class); |
| mNetworkController = new NetworkControllerImpl(mContext, mMockCm, mMockTm, mMockWm, mMockSm, |
| mConfig, Looper.getMainLooper(), mCallbackHandler, |
| mock(AccessPointControllerImpl.class), mock(MobileDataControllerImpl.class)); |
| setupNetworkController(); |
| } |
| |
| protected void setupNetworkController() { |
| // For now just pretend to be the data sim, so we can test that too. |
| mSubId = SubscriptionManager.DEFAULT_SUBSCRIPTION_ID; |
| SubscriptionInfo subscription = mock(SubscriptionInfo.class); |
| List<SubscriptionInfo> subs = new ArrayList<SubscriptionInfo>(); |
| when(subscription.getSubscriptionId()).thenReturn(mSubId); |
| subs.add(subscription); |
| mNetworkController.setCurrentSubscriptions(subs); |
| mMobileSignalController = mNetworkController.mMobileSignalControllers.get(mSubId); |
| mMobileSignalController.getState().dataSim = true; |
| mPhoneStateListener = mMobileSignalController.mPhoneStateListener; |
| |
| // Trigger blank callbacks to always get the current state (some tests don't trigger |
| // changes from default state). |
| mNetworkController.addSignalCallback(null); |
| } |
| |
| protected NetworkControllerImpl setUpNoMobileData() { |
| when(mMockCm.isNetworkSupported(ConnectivityManager.TYPE_MOBILE)).thenReturn(false); |
| NetworkControllerImpl networkControllerNoMobile |
| = new NetworkControllerImpl(mContext, mMockCm, mMockTm, mMockWm, mMockSm, |
| mConfig, Looper.getMainLooper(), mCallbackHandler, |
| mock(AccessPointControllerImpl.class), |
| mock(MobileDataControllerImpl.class)); |
| |
| setupNetworkController(); |
| |
| return networkControllerNoMobile; |
| |
| } |
| |
| @Override |
| protected void tearDown() throws Exception { |
| StringWriter sw = new StringWriter(); |
| PrintWriter pw = new PrintWriter(sw); |
| mNetworkController.dump(null, pw, null); |
| pw.flush(); |
| Log.d(TAG, sw.toString()); |
| super.tearDown(); |
| } |
| |
| // 2 Bars 3G GSM. |
| public void setupDefaultSignal() { |
| setIsGsm(true); |
| setVoiceRegState(ServiceState.STATE_IN_SERVICE); |
| setGsmRoaming(false); |
| setLevel(DEFAULT_LEVEL); |
| updateDataConnectionState(TelephonyManager.DATA_CONNECTED, |
| TelephonyManager.NETWORK_TYPE_UMTS); |
| setConnectivity(NetworkCapabilities.TRANSPORT_CELLULAR, true, true); |
| } |
| |
| public void setConnectivity(int networkType, boolean inetCondition, boolean isConnected) { |
| Intent i = new Intent(ConnectivityManager.INET_CONDITION_ACTION); |
| // TODO: Separate out into several NetworkCapabilities. |
| if (isConnected) { |
| mNetCapabilities.addTransportType(networkType); |
| } else { |
| mNetCapabilities.removeTransportType(networkType); |
| } |
| if (inetCondition) { |
| mNetCapabilities.addCapability(NetworkCapabilities.NET_CAPABILITY_VALIDATED); |
| } else { |
| mNetCapabilities.removeCapability(NetworkCapabilities.NET_CAPABILITY_VALIDATED); |
| } |
| |
| mNetworkController.onReceive(mContext, i); |
| } |
| |
| public void setGsmRoaming(boolean isRoaming) { |
| when(mServiceState.getRoaming()).thenReturn(isRoaming); |
| updateServiceState(); |
| } |
| |
| public void setCdmaRoaming(boolean isRoaming) { |
| when(mServiceState.getCdmaEriIconIndex()).thenReturn(isRoaming ? |
| EriInfo.ROAMING_INDICATOR_ON : EriInfo.ROAMING_INDICATOR_OFF); |
| when(mServiceState.getCdmaEriIconMode()).thenReturn(isRoaming ? |
| EriInfo.ROAMING_ICON_MODE_NORMAL : -1); |
| updateServiceState(); |
| } |
| |
| public void setVoiceRegState(int voiceRegState) { |
| when(mServiceState.getVoiceRegState()).thenReturn(voiceRegState); |
| updateServiceState(); |
| } |
| |
| public void setDataRegState(int dataRegState) { |
| when(mServiceState.getDataRegState()).thenReturn(dataRegState); |
| updateServiceState(); |
| } |
| |
| public void setIsEmergencyOnly(boolean isEmergency) { |
| when(mServiceState.isEmergencyOnly()).thenReturn(isEmergency); |
| updateServiceState(); |
| } |
| |
| public void setCdmaLevel(int level) { |
| when(mSignalStrength.getCdmaLevel()).thenReturn(level); |
| updateSignalStrength(); |
| } |
| |
| public void setLevel(int level) { |
| when(mSignalStrength.getLevel()).thenReturn(level); |
| updateSignalStrength(); |
| } |
| |
| public void setIsGsm(boolean gsm) { |
| when(mSignalStrength.isGsm()).thenReturn(gsm); |
| updateSignalStrength(); |
| } |
| |
| public void setCdmaEri(int index, int mode) { |
| // TODO: Figure this out. |
| } |
| |
| private void updateSignalStrength() { |
| Log.d(TAG, "Sending Signal Strength: " + mSignalStrength); |
| mPhoneStateListener.onSignalStrengthsChanged(mSignalStrength); |
| } |
| |
| private void updateServiceState() { |
| Log.d(TAG, "Sending Service State: " + mServiceState); |
| mPhoneStateListener.onServiceStateChanged(mServiceState); |
| } |
| |
| public void updateCallState(int state) { |
| // Inputs not currently used in NetworkControllerImpl. |
| mPhoneStateListener.onCallStateChanged(state, "0123456789"); |
| } |
| |
| public void updateDataConnectionState(int dataState, int dataNetType) { |
| mPhoneStateListener.onDataConnectionStateChanged(dataState, dataNetType); |
| } |
| |
| public void updateDataActivity(int dataActivity) { |
| mPhoneStateListener.onDataActivity(dataActivity); |
| } |
| |
| public void setCarrierNetworkChange(boolean enable) { |
| Log.d(TAG, "setCarrierNetworkChange(" + enable + ")"); |
| mPhoneStateListener.onCarrierNetworkChange(enable); |
| } |
| |
| protected void verifyHasNoSims(boolean hasNoSimsVisible) { |
| ArgumentCaptor<Boolean> hasNoSimsArg = ArgumentCaptor.forClass(Boolean.class); |
| |
| Mockito.verify(mCallbackHandler, Mockito.atLeastOnce()).setNoSims(hasNoSimsArg.capture()); |
| assertEquals("No sims", hasNoSimsVisible, (boolean) hasNoSimsArg.getValue()); |
| } |
| |
| protected void verifyLastQsMobileDataIndicators(boolean visible, int icon, int typeIcon, |
| boolean dataIn, boolean dataOut) { |
| ArgumentCaptor<IconState> iconArg = ArgumentCaptor.forClass(IconState.class); |
| ArgumentCaptor<Integer> typeIconArg = ArgumentCaptor.forClass(Integer.class); |
| ArgumentCaptor<Boolean> dataInArg = ArgumentCaptor.forClass(Boolean.class); |
| ArgumentCaptor<Boolean> dataOutArg = ArgumentCaptor.forClass(Boolean.class); |
| |
| Mockito.verify(mCallbackHandler, Mockito.atLeastOnce()).setMobileDataIndicators( |
| ArgumentCaptor.forClass(IconState.class).capture(), |
| iconArg.capture(), |
| ArgumentCaptor.forClass(Integer.class).capture(), |
| typeIconArg.capture(), dataInArg.capture(), dataOutArg.capture(), |
| ArgumentCaptor.forClass(String.class).capture(), |
| ArgumentCaptor.forClass(String.class).capture(), |
| ArgumentCaptor.forClass(Boolean.class).capture(), |
| ArgumentCaptor.forClass(Integer.class).capture()); |
| IconState iconState = iconArg.getValue(); |
| assertEquals("Visibility in, quick settings", visible, iconState.visible); |
| assertEquals("Signal icon in, quick settings", icon, iconState.icon); |
| assertEquals("Data icon in, quick settings", typeIcon, (int) typeIconArg.getValue()); |
| assertEquals("Data direction in, in quick settings", dataIn, |
| (boolean) dataInArg.getValue()); |
| assertEquals("Data direction out, in quick settings", dataOut, |
| (boolean) dataOutArg.getValue()); |
| } |
| |
| protected void verifyLastMobileDataIndicators(boolean visible, int icon, int typeIcon) { |
| ArgumentCaptor<IconState> iconArg = ArgumentCaptor.forClass(IconState.class); |
| ArgumentCaptor<Integer> typeIconArg = ArgumentCaptor.forClass(Integer.class); |
| |
| // TODO: Verify all fields. |
| Mockito.verify(mCallbackHandler, Mockito.atLeastOnce()).setMobileDataIndicators( |
| iconArg.capture(), |
| ArgumentCaptor.forClass(IconState.class).capture(), |
| typeIconArg.capture(), |
| ArgumentCaptor.forClass(Integer.class).capture(), |
| ArgumentCaptor.forClass(Boolean.class).capture(), |
| ArgumentCaptor.forClass(Boolean.class).capture(), |
| ArgumentCaptor.forClass(String.class).capture(), |
| ArgumentCaptor.forClass(String.class).capture(), |
| ArgumentCaptor.forClass(Boolean.class).capture(), |
| ArgumentCaptor.forClass(Integer.class).capture()); |
| IconState iconState = iconArg.getValue(); |
| |
| assertEquals("Data icon in status bar", typeIcon, (int) typeIconArg.getValue()); |
| assertEquals("Visibility in status bar", visible, iconState.visible); |
| } |
| |
| protected void assertNetworkNameEquals(String expected) { |
| assertEquals("Network name", expected, mMobileSignalController.getState().networkName); |
| } |
| } |