Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 0da6404c authored by Jack Yu's avatar Jack Yu Committed by Sarah Chin
Browse files

Move override logic from SysUI to Telephony 

Test: atest NetworkControllerDataTest, NetworkControllerSignalTest
Bug: none
Change-Id: I4519d6671297de10cc573b607d21ab030b497950
parent 814109d6

packages/SystemUI/src/com/android/systemui/statusbar/policy/MobileSignalController.java

+60 −218
Original line number Diff line number Diff line
@@ -15,24 +15,18 @@ 
 */

package com.android.systemui.statusbar.policy;



import static android.telephony.AccessNetworkConstants.TRANSPORT_TYPE_WWAN;

import static android.telephony.NetworkRegistrationInfo.DOMAIN_PS;



import android.content.Context;

import android.content.Intent;

import android.database.ContentObserver;

import android.net.NetworkCapabilities;

import android.os.Handler;

import android.os.Looper;

import android.os.Message;

import android.provider.Settings.Global;

import android.telephony.AccessNetworkConstants;

import android.telephony.Annotation;

import android.telephony.CdmaEriInformation;

import android.telephony.CellSignalStrength;

import android.telephony.CellSignalStrengthCdma;

import android.telephony.DataSpecificRegistrationInfo;

import android.telephony.NetworkRegistrationInfo;

import android.telephony.DisplayInfo;

import android.telephony.PhoneStateListener;

import android.telephony.ServiceState;

import android.telephony.SignalStrength;
@@ -60,16 +54,10 @@ import java.util.List; 
import java.util.Map;

import java.util.Objects;

import java.util.concurrent.Executor;

import java.util.regex.Matcher;

import java.util.regex.Pattern;





public class MobileSignalController extends SignalController<

        MobileSignalController.MobileState, MobileSignalController.MobileIconGroup> {



    // The message to display Nr5G icon gracfully by CarrierConfig timeout

    private static final int MSG_DISPLAY_GRACE = 1;



    private final TelephonyManager mPhone;

    private final SubscriptionDefaults mDefaults;

    private final String mNetworkNameDefault;
@@ -86,19 +74,15 @@ public class MobileSignalController extends SignalController< 
    // Since some pieces of the phone state are interdependent we store it locally,

    // this could potentially become part of MobileState for simplification/complication

    // of code.

    private int mDataNetType = TelephonyManager.NETWORK_TYPE_UNKNOWN;

    private boolean mCA = false;

    private boolean mCAPlus = false;

    private int mDataState = TelephonyManager.DATA_DISCONNECTED;

    private DisplayInfo mDisplayInfo = new DisplayInfo(TelephonyManager.NETWORK_TYPE_UNKNOWN,

            DisplayInfo.OVERRIDE_NETWORK_TYPE_NONE);

    private ServiceState mServiceState;

    private SignalStrength mSignalStrength;

    private MobileIconGroup mDefaultIcons;

    private Config mConfig;

    private final Handler mDisplayGraceHandler;

    @VisibleForTesting

    boolean mInflateSignalStrengths = false;

    @VisibleForTesting

    boolean mIsShowingIconGracefully = false;



    // TODO: Reduce number of vars passed in, if we have the NetworkController, probably don't

    // need listener lists anymore.
@@ -136,16 +120,6 @@ public class MobileSignalController extends SignalController< 
                updateTelephony();

            }

        };



        mDisplayGraceHandler = new Handler(receiverLooper) {

            @Override

            public void handleMessage(Message msg) {

                if (msg.what == MSG_DISPLAY_GRACE) {

                    mIsShowingIconGracefully = false;

                    updateTelephony();

                }

            }

        };

    }



    public void setConfiguration(Config config) {
@@ -190,7 +164,8 @@ public class MobileSignalController extends SignalController< 
                        | PhoneStateListener.LISTEN_DATA_CONNECTION_STATE

                        | PhoneStateListener.LISTEN_DATA_ACTIVITY

                        | PhoneStateListener.LISTEN_CARRIER_NETWORK_CHANGE

                        | PhoneStateListener.LISTEN_ACTIVE_DATA_SUBSCRIPTION_ID_CHANGE);

                        | PhoneStateListener.LISTEN_ACTIVE_DATA_SUBSCRIPTION_ID_CHANGE

                        | PhoneStateListener.LISTEN_DISPLAY_INFO_CHANGED);

        mContext.getContentResolver().registerContentObserver(Global.getUriFor(Global.MOBILE_DATA),

                true, mObserver);

        mContext.getContentResolver().registerContentObserver(Global.getUriFor(
@@ -268,50 +243,58 @@ public class MobileSignalController extends SignalController< 
            mNetworkToIconLookup.put(toIconKey(TelephonyManager.NETWORK_TYPE_LTE),

                    TelephonyIcons.FOUR_G);

            if (mConfig.hideLtePlus) {

                mNetworkToIconLookup.put(toIconKeyCA(TelephonyManager.NETWORK_TYPE_LTE),

                        TelephonyIcons.FOUR_G);

                mNetworkToIconLookup.put(toDisplayIconKey(

                        DisplayInfo.OVERRIDE_NETWORK_TYPE_LTE_CA), TelephonyIcons.FOUR_G);

            } else {

                mNetworkToIconLookup.put(toIconKeyCA(TelephonyManager.NETWORK_TYPE_LTE),

                        TelephonyIcons.FOUR_G_PLUS);

                mNetworkToIconLookup.put(toDisplayIconKey(

                        DisplayInfo.OVERRIDE_NETWORK_TYPE_LTE_CA), TelephonyIcons.FOUR_G_PLUS);

            }

        } else {

            mNetworkToIconLookup.put(toIconKey(TelephonyManager.NETWORK_TYPE_LTE),

                    TelephonyIcons.LTE);

            if (mConfig.hideLtePlus) {

                mNetworkToIconLookup.put(toIconKeyCA(TelephonyManager.NETWORK_TYPE_LTE),

                        TelephonyIcons.LTE);

                mNetworkToIconLookup.put(toDisplayIconKey(

                        DisplayInfo.OVERRIDE_NETWORK_TYPE_LTE_CA), TelephonyIcons.LTE);

            } else {

                mNetworkToIconLookup.put(toIconKeyCA(TelephonyManager.NETWORK_TYPE_LTE),

                        TelephonyIcons.LTE_PLUS);

                mNetworkToIconLookup.put(toDisplayIconKey(

                        DisplayInfo.OVERRIDE_NETWORK_TYPE_LTE_CA), TelephonyIcons.LTE_PLUS);

            }

        }

        mNetworkToIconLookup.put(toIconKeyCAPlus(TelephonyManager.NETWORK_TYPE_LTE),

                TelephonyIcons.LTE_CA_5G_E);

        mNetworkToIconLookup.put(toIconKey(TelephonyManager.NETWORK_TYPE_IWLAN),

                TelephonyIcons.WFC);

        mNetworkToIconLookup.put(toDisplayIconKey(

                DisplayInfo.OVERRIDE_NETWORK_TYPE_LTE_ADVANCED_PRO), TelephonyIcons.LTE_CA_5G_E);

        mNetworkToIconLookup.put(toDisplayIconKey(

                DisplayInfo.OVERRIDE_NETWORK_TYPE_NR_NSA), TelephonyIcons.NR_5G);

        mNetworkToIconLookup.put(toDisplayIconKey(

                DisplayInfo.OVERRIDE_NETWORK_TYPE_NR_NSA_MMWAVE), TelephonyIcons.NR_5G_PLUS);

    }



    private String getIconKey() {

        if (mCA) {

            return toIconKeyCA(mDataNetType);

        } else if (mCAPlus) {

            return toIconKeyCAPlus(mDataNetType);

        if (mDisplayInfo.getOverrideNetworkType() == DisplayInfo.OVERRIDE_NETWORK_TYPE_NONE) {

            return toIconKey(mDisplayInfo.getNetworkType());

        } else {

            return toIconKey(mDataNetType);

            return toDisplayIconKey(mDisplayInfo.getOverrideNetworkType());

        }

    }



    // Some specific carriers have 5GE network which is special CA network.

    private String toIconKeyCAPlus(@Annotation.NetworkType int networkType) {

        return toIconKeyCA(networkType) + "_Plus";

    private String toIconKey(@Annotation.NetworkType int networkType) {

        return Integer.toString(networkType);

    }



    private String toIconKeyCA(@Annotation.NetworkType int networkType) {

        return toIconKey(networkType) + "_CA";

    private String toDisplayIconKey(@Annotation.OverrideNetworkType int displayNetworkType) {

        switch (displayNetworkType) {

            case DisplayInfo.OVERRIDE_NETWORK_TYPE_LTE_CA:

                return toIconKey(TelephonyManager.NETWORK_TYPE_LTE) + "_CA";

            case DisplayInfo.OVERRIDE_NETWORK_TYPE_LTE_ADVANCED_PRO:

                return toIconKey(TelephonyManager.NETWORK_TYPE_LTE) + "_CA_Plus";

            case DisplayInfo.OVERRIDE_NETWORK_TYPE_NR_NSA:

                return "5G";

            case DisplayInfo.OVERRIDE_NETWORK_TYPE_NR_NSA_MMWAVE:

                return "5G_Plus";

            default:

                return "unsupported";

        }



    private String toIconKey(@Annotation.NetworkType int networkType) {

        return Integer.toString(networkType);

    }



    private void updateInflateSignalStrength() {
@@ -465,26 +448,6 @@ public class MobileSignalController extends SignalController< 
        }

    }



    private boolean isCarrierSpecificDataIcon() {

        if (mConfig.patternOfCarrierSpecificDataIcon == null

                || mConfig.patternOfCarrierSpecificDataIcon.length() == 0) {

            return false;

        }



        Pattern stringPattern = Pattern.compile(mConfig.patternOfCarrierSpecificDataIcon);

        String[] operatorNames = new String[]{mServiceState.getOperatorAlphaLongRaw(),

                mServiceState.getOperatorAlphaShortRaw()};

        for (String opName : operatorNames) {

            if (!TextUtils.isEmpty(opName)) {

                Matcher matcher = stringPattern.matcher(opName);

                if (matcher.find()) {

                    return true;

                }

            }

        }

        return false;

    }



    /**

     * Updates the network's name based on incoming spn and plmn.

     */
@@ -538,18 +501,18 @@ public class MobileSignalController extends SignalController< 
    }



    /**

     * Updates the current state based on mServiceState, mSignalStrength, mDataNetType,

     * mDataState, and mSimState.  It should be called any time one of these is updated.

     * Updates the current state based on mServiceState, mSignalStrength, mDataState,

     * mDisplayInfo, and mSimState.  It should be called any time one of these is updated.

     * This will call listeners if necessary.

     */

    private final void updateTelephony() {

        if (DEBUG) {

            Log.d(mTag, "updateTelephonySignalStrength: hasService=" +

                    Utils.isInService(mServiceState) + " ss=" + mSignalStrength);

                    Utils.isInService(mServiceState) + " ss=" + mSignalStrength

                    + " displayInfo=" + mDisplayInfo);

        }

        checkDefaultData();

        mCurrentState.connected = Utils.isInService(mServiceState)

                && mSignalStrength != null;

        mCurrentState.connected = Utils.isInService(mServiceState) && mSignalStrength != null;

        if (mCurrentState.connected) {

            if (!mSignalStrength.isGsm() && mConfig.alwaysShowCdmaRssi) {

                mCurrentState.level = getCdmaLevel();
@@ -558,17 +521,8 @@ public class MobileSignalController extends SignalController< 
            }

        }



        // When the device is camped on a 5G Non-Standalone network, the data network type is still

        // LTE. In this case, we first check which 5G icon should be shown.

        MobileIconGroup nr5GIconGroup = getNr5GIconGroup();

        if (mConfig.nrIconDisplayGracePeriodMs > 0) {

            nr5GIconGroup = adjustNr5GIconGroupByDisplayGraceTime(nr5GIconGroup);

        }



        String iconKey = getIconKey();

        if (nr5GIconGroup != null) {

            mCurrentState.iconGroup = nr5GIconGroup;

        } else if (mNetworkToIconLookup.get(iconKey) != null) {

        if (mNetworkToIconLookup.get(iconKey) != null) {

            mCurrentState.iconGroup = mNetworkToIconLookup.get(iconKey);

        } else {

            mCurrentState.iconGroup = mDefaultIcons;
@@ -580,8 +534,7 @@ public class MobileSignalController extends SignalController< 
        if (isCarrierNetworkChangeActive()) {

            mCurrentState.iconGroup = TelephonyIcons.CARRIER_NETWORK_CHANGE;

        } else if (isDataDisabled() && !mConfig.alwaysShowDataRatIcon) {

            if (mSubscriptionInfo.getSubscriptionId()

                    != mDefaults.getDefaultDataSubId()) {

            if (mSubscriptionInfo.getSubscriptionId() != mDefaults.getDefaultDataSubId()) {

                mCurrentState.iconGroup = TelephonyIcons.NOT_DEFAULT_DATA;

            } else {

                mCurrentState.iconGroup = TelephonyIcons.DATA_DISABLED;
@@ -623,91 +576,6 @@ public class MobileSignalController extends SignalController< 
        notifyListenersIfNecessary();

    }



    private int getNrState(ServiceState serviceState) {

        NetworkRegistrationInfo nri = serviceState.getNetworkRegistrationInfo(

                NetworkRegistrationInfo.DOMAIN_PS, AccessNetworkConstants.TRANSPORT_TYPE_WWAN);

        if (nri != null) {

            return nri.getNrState();

        }

        return NetworkRegistrationInfo.NR_STATE_NONE;

    }



    private MobileIconGroup getNr5GIconGroup() {

        if (mServiceState == null) return null;



        int nrState = getNrState(mServiceState);

        if (nrState == NetworkRegistrationInfo.NR_STATE_CONNECTED) {

            // Check if the NR 5G is using millimeter wave and the icon is config.

            if (mServiceState.getNrFrequencyRange() == ServiceState.FREQUENCY_RANGE_MMWAVE) {

                if (mConfig.nr5GIconMap.containsKey(Config.NR_CONNECTED_MMWAVE)) {

                    return mConfig.nr5GIconMap.get(Config.NR_CONNECTED_MMWAVE);

                }

            }



            // If NR 5G is not using millimeter wave or there is no icon for millimeter wave, we

            // check the normal 5G icon.

            if (mConfig.nr5GIconMap.containsKey(Config.NR_CONNECTED)) {

                return mConfig.nr5GIconMap.get(Config.NR_CONNECTED);

            }

        } else if (nrState == NetworkRegistrationInfo.NR_STATE_NOT_RESTRICTED) {

            if (mCurrentState.activityDormant) {

                if (mConfig.nr5GIconMap.containsKey(Config.NR_NOT_RESTRICTED_RRC_IDLE)) {

                    return mConfig.nr5GIconMap.get(Config.NR_NOT_RESTRICTED_RRC_IDLE);

                }

            } else {

                if (mConfig.nr5GIconMap.containsKey(Config.NR_NOT_RESTRICTED_RRC_CON)) {

                    return mConfig.nr5GIconMap.get(Config.NR_NOT_RESTRICTED_RRC_CON);

                }

            }

        } else if (nrState == NetworkRegistrationInfo.NR_STATE_RESTRICTED) {

            if (mConfig.nr5GIconMap.containsKey(Config.NR_RESTRICTED)) {

                return mConfig.nr5GIconMap.get(Config.NR_RESTRICTED);

            }

        }



        return null;

    }



    /**

     * The function to adjust MobileIconGroup depend on CarrierConfig's time

     * nextIconGroup == null imply next state could be 2G/3G/4G/4G+

     * nextIconGroup != null imply next state will be 5G/5G+

     * Flag : mIsShowingIconGracefully

     * ---------------------------------------------------------------------------------

     * |   Last state   |  Current state  | Flag |       Action                        |

     * ---------------------------------------------------------------------------------

     * |     5G/5G+     | 2G/3G/4G/4G+    | true | return previous IconGroup           |

     * |     5G/5G+     |     5G/5G+      | true | Bypass                              |

     * |  2G/3G/4G/4G+  |     5G/5G+      | true | Bypass                              |

     * |  2G/3G/4G/4G+  | 2G/3G/4G/4G+    | true | Bypass                              |

     * |  SS.connected  | SS.disconnect   |  T|F | Reset timer                         |

     * |NETWORK_TYPE_LTE|!NETWORK_TYPE_LTE|  T|F | Reset timer                         |

     * |     5G/5G+     | 2G/3G/4G/4G+    | false| Bypass                              |

     * |     5G/5G+     |     5G/5G+      | false| Bypass                              |

     * |  2G/3G/4G/4G+  |     5G/5G+      | false| SendMessageDelay(time), flag->true  |

     * |  2G/3G/4G/4G+  | 2G/3G/4G/4G+    | false| Bypass                              |

     * ---------------------------------------------------------------------------------

     */

    private MobileIconGroup adjustNr5GIconGroupByDisplayGraceTime(

            MobileIconGroup candidateIconGroup) {

        if (mIsShowingIconGracefully && candidateIconGroup == null) {

            candidateIconGroup = (MobileIconGroup) mCurrentState.iconGroup;

        } else if (!mIsShowingIconGracefully && candidateIconGroup != null

                && mLastState.iconGroup != candidateIconGroup) {

            mDisplayGraceHandler.sendMessageDelayed(

                    mDisplayGraceHandler.obtainMessage(MSG_DISPLAY_GRACE),

                    mConfig.nrIconDisplayGracePeriodMs);

            mIsShowingIconGracefully = true;

        } else if (!mCurrentState.connected || mDataState == TelephonyManager.DATA_DISCONNECTED

                || candidateIconGroup == null) {

            mDisplayGraceHandler.removeMessages(MSG_DISPLAY_GRACE);

            mIsShowingIconGracefully = false;

            candidateIconGroup = null;

        }



        return candidateIconGroup;

    }



    boolean isDataDisabled() {

        return !mPhone.isDataConnectionEnabled();

    }
@@ -718,8 +586,6 @@ public class MobileSignalController extends SignalController< 
                || activity == TelephonyManager.DATA_ACTIVITY_IN;

        mCurrentState.activityOut = activity == TelephonyManager.DATA_ACTIVITY_INOUT

                || activity == TelephonyManager.DATA_ACTIVITY_OUT;

        mCurrentState.activityDormant = activity == TelephonyManager.DATA_ACTIVITY_DORMANT;



        notifyListenersIfNecessary();

    }
@@ -729,13 +595,10 @@ public class MobileSignalController extends SignalController< 
        pw.println("  mSubscription=" + mSubscriptionInfo + ",");

        pw.println("  mServiceState=" + mServiceState + ",");

        pw.println("  mSignalStrength=" + mSignalStrength + ",");

        pw.println("  mDisplayInfo=" + mDisplayInfo + ",");

        pw.println("  mDataState=" + mDataState + ",");

        pw.println("  mDataNetType=" + mDataNetType + ",");

        pw.println("  mCA=" + mCA + ",");

        pw.println("  mCAPlus=" + mCAPlus + ",");

        pw.println("  mInflateSignalStrengths=" + mInflateSignalStrengths + ",");

        pw.println("  isDataDisabled=" + isDataDisabled() + ",");

        pw.println("  mIsShowingIconGracefully=" + mIsShowingIconGracefully + ",");

    }



    class MobilePhoneStateListener extends PhoneStateListener {
@@ -760,14 +623,8 @@ public class MobileSignalController extends SignalController< 
                        + " dataState=" + state.getDataRegistrationState());

            }

            mServiceState = state;

            if (mServiceState != null) {

                NetworkRegistrationInfo regInfo = mServiceState.getNetworkRegistrationInfo(

                        DOMAIN_PS, TRANSPORT_TYPE_WWAN);

                if (regInfo != null) {

                    updateDataNetType(regInfo.getAccessNetworkTechnology());

                }

            }

            updateTelephony();

            // onDisplayInfoChanged is invoked directly after onServiceStateChanged, so not calling

            // updateTelephony() to prevent icon flickering in case of overrides.

        }



        @Override
@@ -777,33 +634,10 @@ public class MobileSignalController extends SignalController< 
                        + " type=" + networkType);

            }

            mDataState = state;

            updateDataNetType(networkType);

            updateTelephony();

            if (networkType != mDisplayInfo.getNetworkType()) {

                mDisplayInfo = new DisplayInfo(networkType, DisplayInfo.OVERRIDE_NETWORK_TYPE_NONE);

            }



        private void updateDataNetType(int networkType) {

            mDataNetType = networkType;

            mCA = false;

            mCAPlus = false;

            if (mDataNetType == TelephonyManager.NETWORK_TYPE_LTE) {

                if (isCarrierSpecificDataIcon()) {

                    mCAPlus = true;

                } else if (mServiceState != null && isUsingCarrierAggregation(mServiceState)) {

                    mCA = true;

                }

            }

        }



        private boolean isUsingCarrierAggregation(ServiceState serviceState) {

            NetworkRegistrationInfo nri = serviceState.getNetworkRegistrationInfo(

                    NetworkRegistrationInfo.DOMAIN_PS, AccessNetworkConstants.TRANSPORT_TYPE_WWAN);

            if (nri != null) {

                DataSpecificRegistrationInfo dsri = nri.getDataSpecificInfo();

                if (dsri != null) {

                    return dsri.isUsingCarrierAggregation();

                }

            }

            return false;

            updateTelephony();

        }



        @Override
@@ -820,7 +654,6 @@ public class MobileSignalController extends SignalController< 
                Log.d(mTag, "onCarrierNetworkChange: active=" + active);

            }

            mCurrentState.carrierNetworkChangeMode = active;



            updateTelephony();

        }
@@ -830,7 +663,16 @@ public class MobileSignalController extends SignalController< 
            updateDataSim();

            updateTelephony();

        }

    };



        @Override

        public void onDisplayInfoChanged(DisplayInfo displayInfo) {

            if (DEBUG) {

                Log.d(mTag, "onDisplayInfoChanged: displayInfo=" + displayInfo);

            }

            mDisplayInfo = displayInfo;

            updateTelephony();

        }

    }



    static class MobileIconGroup extends SignalController.IconGroup {

        final int mDataContentDescription; // mContentDescriptionDataType

packages/SystemUI/src/com/android/systemui/statusbar/policy/NetworkControllerImpl.java

+0 −89

File changed.

Preview size limit exceeded, changes collapsed.

packages/SystemUI/src/com/android/systemui/statusbar/policy/SignalController.java

+0 −4
Original line number Diff line number Diff line
@@ -261,7 +261,6 @@ public abstract class SignalController<T extends SignalController.State, 
        boolean enabled;

        boolean activityIn;

        boolean activityOut;

        public boolean activityDormant;

        int level;

        IconGroup iconGroup;

        int inetCondition;
@@ -278,7 +277,6 @@ public abstract class SignalController<T extends SignalController.State, 
            inetCondition = state.inetCondition;

            activityIn = state.activityIn;

            activityOut = state.activityOut;

            activityDormant = state.activityDormant;

            rssi = state.rssi;

            time = state.time;

        }
@@ -302,7 +300,6 @@ public abstract class SignalController<T extends SignalController.State, 
                    .append("iconGroup=").append(iconGroup).append(',')

                    .append("activityIn=").append(activityIn).append(',')

                    .append("activityOut=").append(activityOut).append(',')

                    .append("activityDormant=").append(activityDormant).append(',')

                    .append("rssi=").append(rssi).append(',')

                    .append("lastModified=").append(sSDF.format(time));

        }
@@ -320,7 +317,6 @@ public abstract class SignalController<T extends SignalController.State, 
                    && other.iconGroup == iconGroup

                    && other.activityIn == activityIn

                    && other.activityOut == activityOut

                    && other.activityDormant == activityDormant

                    && other.rssi == rssi;

        }

    }

packages/SystemUI/tests/src/com/android/systemui/statusbar/policy/NetworkControllerBaseTest.java

+8 −27
Original line number Diff line number Diff line
@@ -45,6 +45,7 @@ import android.provider.Settings; 
import android.provider.Settings.Global;

import android.telephony.CdmaEriInformation;

import android.telephony.CellSignalStrength;

import android.telephony.DisplayInfo;

import android.telephony.NetworkRegistrationInfo;

import android.telephony.PhoneStateListener;

import android.telephony.ServiceState;
@@ -96,6 +97,7 @@ public class NetworkControllerBaseTest extends SysuiTestCase { 
    protected PhoneStateListener mPhoneStateListener;

    protected SignalStrength mSignalStrength;

    protected ServiceState mServiceState;

    protected DisplayInfo mDisplayInfo;

    protected NetworkRegistrationInfo mFakeRegInfo;

    protected ConnectivityManager mMockCm;

    protected WifiManager mMockWm;
@@ -159,6 +161,7 @@ public class NetworkControllerBaseTest extends SysuiTestCase { 


        mSignalStrength = mock(SignalStrength.class);

        mServiceState = mock(ServiceState.class);

        mDisplayInfo = mock(DisplayInfo.class);



        mFakeRegInfo = new NetworkRegistrationInfo.Builder()

                .setTransportType(TRANSPORT_TYPE_WWAN)
@@ -167,6 +170,9 @@ public class NetworkControllerBaseTest extends SysuiTestCase { 
                .build();

        doReturn(mFakeRegInfo).when(mServiceState)

                .getNetworkRegistrationInfo(DOMAIN_PS, TRANSPORT_TYPE_WWAN);

        doReturn(TelephonyManager.NETWORK_TYPE_LTE).when(mDisplayInfo).getNetworkType();

        doReturn(DisplayInfo.OVERRIDE_NETWORK_TYPE_NONE).when(mDisplayInfo)

                .getOverrideNetworkType();



        mEriInformation = new CdmaEriInformation(CdmaEriInformation.ERI_OFF,

                CdmaEriInformation.ERI_ICON_MODE_NORMAL);
@@ -260,33 +266,6 @@ public class NetworkControllerBaseTest extends SysuiTestCase { 
            NetworkCapabilities.TRANSPORT_CELLULAR, true, true);

    }



    public void setupDefaultNr5GIconConfiguration() {

        NetworkControllerImpl.Config.add5GIconMapping("connected_mmwave:5g_plus", mConfig);

        NetworkControllerImpl.Config.add5GIconMapping("connected:5g", mConfig);

    }



    public void setupNr5GIconConfigurationForNotRestrictedRrcCon() {

        NetworkControllerImpl.Config.add5GIconMapping("connected_mmwave:5g_plus", mConfig);

        NetworkControllerImpl.Config.add5GIconMapping("connected:5g_plus", mConfig);

        NetworkControllerImpl.Config.add5GIconMapping("not_restricted_rrc_con:5g", mConfig);

    }



    public void setupNr5GIconConfigurationForNotRestrictedRrcIdle() {

        NetworkControllerImpl.Config.add5GIconMapping("connected_mmwave:5g_plus", mConfig);

        NetworkControllerImpl.Config.add5GIconMapping("connected:5g_plus", mConfig);

        NetworkControllerImpl.Config.add5GIconMapping("not_restricted_rrc_idle:5g", mConfig);

    }



    public void setupDefaultNr5GIconDisplayGracePeriodTime_enableThirtySeconds() {

        final int enableDisplayGraceTimeSec = 30;

        NetworkControllerImpl.Config.setDisplayGraceTime(enableDisplayGraceTimeSec, mConfig);

    }



    public void setupDefaultNr5GIconDisplayGracePeriodTime_disabled() {

        final int disableDisplayGraceTimeSec = 0;

        NetworkControllerImpl.Config.setDisplayGraceTime(disableDisplayGraceTimeSec, mConfig);

    }



    public void setConnectivityViaBroadcast(

        int networkType, boolean validated, boolean isConnected) {

        setConnectivityCommon(networkType, validated, isConnected);
@@ -369,6 +348,7 @@ public class NetworkControllerBaseTest extends SysuiTestCase { 
    protected void updateServiceState() {

        Log.d(TAG, "Sending Service State: " + mServiceState);

        mPhoneStateListener.onServiceStateChanged(mServiceState);

        mPhoneStateListener.onDisplayInfoChanged(mDisplayInfo);

    }



    public void updateCallState(int state) {
@@ -384,6 +364,7 @@ public class NetworkControllerBaseTest extends SysuiTestCase { 
                .build();

        when(mServiceState.getNetworkRegistrationInfo(DOMAIN_PS, TRANSPORT_TYPE_WWAN))

                .thenReturn(fakeRegInfo);

        when(mDisplayInfo.getNetworkType()).thenReturn(dataNetType);

        mPhoneStateListener.onDataConnectionStateChanged(dataState, dataNetType);

    }

packages/SystemUI/tests/src/com/android/systemui/statusbar/policy/NetworkControllerDataTest.java

+1 −259

File changed.

Preview size limit exceeded, changes collapsed.