packages/SystemUI/tests/src/com/android/systemui/statusbar/policy/NetworkControllerBaseTest.java - platform/frameworks/base - Git at Google

 /*
  * 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 com.android.systemui.statusbar.policy.NetworkControllerImpl.SubscriptionDefaults;

 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;
     protected SignalStrength mSignalStrength;
     protected ServiceState mServiceState;
     protected ConnectivityManager mMockCm;
     protected WifiManager mMockWm;
     protected SubscriptionManager mMockSm;
     protected TelephonyManager mMockTm;
     protected Config mConfig;
     protected CallbackHandler mCallbackHandler;
     protected SubscriptionDefaults mMockSubDefaults;

     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);
         mMockSubDefaults = mock(SubscriptionDefaults.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),
                 mMockSubDefaults);
         setupNetworkController();

         // Trigger blank callbacks to always get the current state (some tests don't trigger
         // changes from default state).
         mNetworkController.addSignalCallback(null);
         mNetworkController.addEmergencyListener(null);
     }

     protected void setupNetworkController() {
         // For now just pretend to be the data sim, so we can test that too.
         mSubId = SubscriptionManager.DEFAULT_SUBSCRIPTION_ID;
         setDefaultSubId(mSubId);
         setSubscriptions(mSubId);
         mMobileSignalController = mNetworkController.mMobileSignalControllers.get(mSubId);
         mPhoneStateListener = mMobileSignalController.mPhoneStateListener;
     }

     protected void setDefaultSubId(int subId) {
         when(mMockSubDefaults.getDefaultDataSubId()).thenReturn(subId);
         when(mMockSubDefaults.getDefaultVoiceSubId()).thenReturn(subId);
     }

     protected void setSubscriptions(int... subIds) {
         List<SubscriptionInfo> subs = new ArrayList<SubscriptionInfo>();
         for (int subId : subIds) {
             SubscriptionInfo subscription = mock(SubscriptionInfo.class);
             when(subscription.getSubscriptionId()).thenReturn(subId);
             subs.add(subscription);
         }
         when(mMockSm.getActiveSubscriptionInfoList()).thenReturn(subs);
         mNetworkController.doUpdateMobileControllers();
     }

     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), mMockSubDefaults);

       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);
     }

     protected 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) {
         when(mServiceState.getDataNetworkType()).thenReturn(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);
    }
 }
	/*
	* 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 com.android.systemui.statusbar.policy.NetworkControllerImpl.SubscriptionDefaults;

	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;
	protected SignalStrength mSignalStrength;
	protected ServiceState mServiceState;
	protected ConnectivityManager mMockCm;
	protected WifiManager mMockWm;
	protected SubscriptionManager mMockSm;
	protected TelephonyManager mMockTm;
	protected Config mConfig;
	protected CallbackHandler mCallbackHandler;
	protected SubscriptionDefaults mMockSubDefaults;

	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);
	mMockSubDefaults = mock(SubscriptionDefaults.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),
	mMockSubDefaults);
	setupNetworkController();

	// Trigger blank callbacks to always get the current state (some tests don't trigger
	// changes from default state).
	mNetworkController.addSignalCallback(null);
	mNetworkController.addEmergencyListener(null);
	}

	protected void setupNetworkController() {
	// For now just pretend to be the data sim, so we can test that too.
	mSubId = SubscriptionManager.DEFAULT_SUBSCRIPTION_ID;
	setDefaultSubId(mSubId);
	setSubscriptions(mSubId);
	mMobileSignalController = mNetworkController.mMobileSignalControllers.get(mSubId);
	mPhoneStateListener = mMobileSignalController.mPhoneStateListener;
	}

	protected void setDefaultSubId(int subId) {
	when(mMockSubDefaults.getDefaultDataSubId()).thenReturn(subId);
	when(mMockSubDefaults.getDefaultVoiceSubId()).thenReturn(subId);
	}

	protected void setSubscriptions(int... subIds) {
	List<SubscriptionInfo> subs = new ArrayList<SubscriptionInfo>();
	for (int subId : subIds) {
	SubscriptionInfo subscription = mock(SubscriptionInfo.class);
	when(subscription.getSubscriptionId()).thenReturn(subId);
	subs.add(subscription);
	}
	when(mMockSm.getActiveSubscriptionInfoList()).thenReturn(subs);
	mNetworkController.doUpdateMobileControllers();
	}

	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), mMockSubDefaults);

	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);
	}

	protected 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) {
	when(mServiceState.getDataNetworkType()).thenReturn(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);
	}
	}