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

Commit fd48e9f7 authored by Ying Xu's avatar Ying Xu Committed by Android (Google) Code Review
Browse files

Merge "[DO NOT MERGE] Make sure WiFi and Cell tiles are not created in Provider Model" into sc-dev

parents 4a8d463e bc6f74db

packages/SystemUI/src/com/android/systemui/qs/QSTileHost.java

+25 −5
Original line number Diff line number Diff line
@@ -51,6 +51,7 @@ import com.android.systemui.qs.external.TileServices; 
import com.android.systemui.qs.logging.QSLogger;

import com.android.systemui.settings.UserTracker;

import com.android.systemui.shared.plugins.PluginManager;

import com.android.systemui.statusbar.FeatureFlags;

import com.android.systemui.statusbar.phone.AutoTileManager;

import com.android.systemui.statusbar.phone.StatusBar;

import com.android.systemui.statusbar.phone.StatusBarIconController;
@@ -95,6 +96,7 @@ public class QSTileHost implements QSHost, Tunable, PluginListener<QSFactory>, D 
    private final UiEventLogger mUiEventLogger;

    private final InstanceIdSequence mInstanceIdSequence;

    private final CustomTileStatePersister mCustomTileStatePersister;

    private final FeatureFlags mFeatureFlags;



    private final List<Callback> mCallbacks = new ArrayList<>();

    private AutoTileManager mAutoTiles;
@@ -122,7 +124,8 @@ public class QSTileHost implements QSHost, Tunable, PluginListener<QSFactory>, D 
            UiEventLogger uiEventLogger,

            UserTracker userTracker,

            SecureSettings secureSettings,

            CustomTileStatePersister customTileStatePersister

            CustomTileStatePersister customTileStatePersister,

            FeatureFlags featureFlags

    ) {

        mIconController = iconController;

        mContext = context;
@@ -144,6 +147,7 @@ public class QSTileHost implements QSHost, Tunable, PluginListener<QSFactory>, D 
        mUserTracker = userTracker;

        mSecureSettings = secureSettings;

        mCustomTileStatePersister = customTileStatePersister;

        mFeatureFlags = featureFlags;



        mainHandler.post(() -> {

            // This is technically a hack to avoid circular dependency of
@@ -265,7 +269,7 @@ public class QSTileHost implements QSHost, Tunable, PluginListener<QSFactory>, D 
        if (newValue == null && UserManager.isDeviceInDemoMode(mContext)) {

            newValue = mContext.getResources().getString(R.string.quick_settings_tiles_retail_mode);

        }

        final List<String> tileSpecs = loadTileSpecs(mContext, newValue);

        final List<String> tileSpecs = loadTileSpecs(mContext, newValue, mFeatureFlags);

        int currentUser = mUserTracker.getUserId();

        if (currentUser != mCurrentUser) {

            mUserContext = mUserTracker.getUserContext();
@@ -334,7 +338,7 @@ public class QSTileHost implements QSHost, Tunable, PluginListener<QSFactory>, D 
        if (newTiles.isEmpty() && !tileSpecs.isEmpty()) {

            // If we didn't manage to create any tiles, set it to empty (default)

            Log.d(TAG, "No valid tiles on tuning changed. Setting to default.");

            changeTiles(currentSpecs, loadTileSpecs(mContext, ""));

            changeTiles(currentSpecs, loadTileSpecs(mContext, "", mFeatureFlags));

        } else {

            for (int i = 0; i < mCallbacks.size(); i++) {

                mCallbacks.get(i).onTilesChanged();
@@ -389,7 +393,7 @@ public class QSTileHost implements QSHost, Tunable, PluginListener<QSFactory>, D 


    private void changeTileSpecs(Predicate<List<String>> changeFunction) {

        final String setting = mSecureSettings.getStringForUser(TILES_SETTING, mCurrentUser);

        final List<String> tileSpecs = loadTileSpecs(mContext, setting);

        final List<String> tileSpecs = loadTileSpecs(mContext, setting, mFeatureFlags);

        if (changeFunction.test(tileSpecs)) {

            saveTilesToSettings(tileSpecs);

        }
@@ -478,7 +482,8 @@ public class QSTileHost implements QSHost, Tunable, PluginListener<QSFactory>, D 
        throw new RuntimeException("Default factory didn't create view for " + tile.getTileSpec());

    }



    protected static List<String> loadTileSpecs(Context context, String tileList) {

    protected static List<String> loadTileSpecs(

            Context context, String tileList, FeatureFlags featureFlags) {

        final Resources res = context.getResources();



        if (TextUtils.isEmpty(tileList)) {
@@ -511,6 +516,21 @@ public class QSTileHost implements QSHost, Tunable, PluginListener<QSFactory>, D 
                }

            }

        }

        if (featureFlags.isProviderModelSettingEnabled()) {

            if (!tiles.contains("internet")) {

                if (tiles.contains("wifi")) {

                    // Replace the WiFi with Internet, and remove the Cell

                    tiles.set(tiles.indexOf("wifi"), "internet");

                    tiles.remove("cell");

                } else if (tiles.contains("cell")) {

                    // Replace the Cell with Internet

                    tiles.set(tiles.indexOf("cell"), "internet");

                }

            } else {

                tiles.remove("wifi");

                tiles.remove("cell");

            }

        }

        return tiles;

    }

packages/SystemUI/src/com/android/systemui/qs/customize/TileQueryHelper.java

+9 −1
Original line number Diff line number Diff line
@@ -41,6 +41,7 @@ import com.android.systemui.qs.dagger.QSScope; 
import com.android.systemui.qs.external.CustomTile;

import com.android.systemui.qs.tileimpl.QSTileImpl.DrawableIcon;

import com.android.systemui.settings.UserTracker;

import com.android.systemui.statusbar.FeatureFlags;

import com.android.systemui.util.leak.GarbageMonitor;



import java.util.ArrayList;
@@ -62,6 +63,7 @@ public class TileQueryHelper { 
    private final Executor mBgExecutor;

    private final Context mContext;

    private final UserTracker mUserTracker;

    private final FeatureFlags mFeatureFlags;

    private TileStateListener mListener;



    private boolean mFinished;
@@ -71,12 +73,14 @@ public class TileQueryHelper { 
            Context context,

            UserTracker userTracker,

            @Main Executor mainExecutor,

            @Background Executor bgExecutor

            @Background Executor bgExecutor,

            FeatureFlags featureFlags

    ) {

        mContext = context;

        mMainExecutor = mainExecutor;

        mBgExecutor = bgExecutor;

        mUserTracker = userTracker;

        mFeatureFlags = featureFlags;

    }



    public void setListener(TileStateListener listener) {
@@ -117,6 +121,10 @@ public class TileQueryHelper { 
        }



        final ArrayList<QSTile> tilesToAdd = new ArrayList<>();

        if (mFeatureFlags.isProviderModelSettingEnabled()) {

            possibleTiles.remove("cell");

            possibleTiles.remove("wifi");

        }



        for (String spec : possibleTiles) {

            // Only add current and stock tiles that can be created from QSFactoryImpl.

packages/SystemUI/tests/src/com/android/systemui/qs/QSFragmentTest.java

+1 −1
Original line number Diff line number Diff line
@@ -139,7 +139,7 @@ public class QSFragmentTest extends SysuiBaseFragmentTest { 
                () -> mock(AutoTileManager.class), mock(DumpManager.class),

                mock(BroadcastDispatcher.class), Optional.of(mock(StatusBar.class)),

                mock(QSLogger.class), mock(UiEventLogger.class), mock(UserTracker.class),

                mock(SecureSettings.class), mock(CustomTileStatePersister.class));

                mock(SecureSettings.class), mock(CustomTileStatePersister.class), mFeatureFlags);

        qs.setHost(host);



        qs.setListening(true);

packages/SystemUI/tests/src/com/android/systemui/qs/QSTileHostTest.java

+62 −9
Original line number Diff line number Diff line
@@ -63,6 +63,7 @@ import com.android.systemui.qs.logging.QSLogger; 
import com.android.systemui.qs.tileimpl.QSTileImpl;

import com.android.systemui.settings.UserTracker;

import com.android.systemui.shared.plugins.PluginManager;

import com.android.systemui.statusbar.FeatureFlags;

import com.android.systemui.statusbar.phone.AutoTileManager;

import com.android.systemui.statusbar.phone.StatusBar;

import com.android.systemui.statusbar.phone.StatusBarIconController;
@@ -124,6 +125,8 @@ public class QSTileHostTest extends SysuiTestCase { 
    private SecureSettings mSecureSettings;

    @Mock

    private CustomTileStatePersister mCustomTileStatePersister;

    @Mock

    private FeatureFlags mFeatureFlags;



    private Handler mHandler;

    private TestableLooper mLooper;
@@ -137,9 +140,9 @@ public class QSTileHostTest extends SysuiTestCase { 
        mQSTileHost = new TestQSTileHost(mContext, mIconController, mDefaultFactory, mHandler,

                mLooper.getLooper(), mPluginManager, mTunerService, mAutoTiles, mDumpManager,

                mBroadcastDispatcher, mStatusBar, mQSLogger, mUiEventLogger, mUserTracker,

                mSecureSettings, mCustomTileStatePersister);

                mSecureSettings, mCustomTileStatePersister, mFeatureFlags);

        setUpTileFactory();



        when(mFeatureFlags.isProviderModelSettingEnabled()).thenReturn(false);

        when(mSecureSettings.getStringForUser(eq(QSTileHost.TILES_SETTING), anyInt()))

                .thenReturn("");

    }
@@ -169,13 +172,13 @@ public class QSTileHostTest extends SysuiTestCase { 


    @Test

    public void testLoadTileSpecs_emptySetting() {

        List<String> tiles = QSTileHost.loadTileSpecs(mContext, "");

        List<String> tiles = QSTileHost.loadTileSpecs(mContext, "", mFeatureFlags);

        assertFalse(tiles.isEmpty());

    }



    @Test

    public void testLoadTileSpecs_nullSetting() {

        List<String> tiles = QSTileHost.loadTileSpecs(mContext, null);

        List<String> tiles = QSTileHost.loadTileSpecs(mContext, null, mFeatureFlags);

        assertFalse(tiles.isEmpty());

    }
@@ -188,6 +191,55 @@ public class QSTileHostTest extends SysuiTestCase { 
        assertEquals(2, mQSTileHost.getTiles().size());

    }



    @Test

    public void testRemoveWifiAndCellularWithoutInternet() {

        when(mFeatureFlags.isProviderModelSettingEnabled()).thenReturn(true);

        mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "wifi, spec1, cell, spec2");



        assertEquals("internet", mQSTileHost.mTileSpecs.get(0));

        assertEquals("spec1", mQSTileHost.mTileSpecs.get(1));

        assertEquals("spec2", mQSTileHost.mTileSpecs.get(2));

    }



    @Test

    public void testRemoveWifiAndCellularWithInternet() {

        when(mFeatureFlags.isProviderModelSettingEnabled()).thenReturn(true);

        mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "wifi, spec1, cell, spec2, internet");



        assertEquals("spec1", mQSTileHost.mTileSpecs.get(0));

        assertEquals("spec2", mQSTileHost.mTileSpecs.get(1));

        assertEquals("internet", mQSTileHost.mTileSpecs.get(2));

    }



    @Test

    public void testRemoveWifiWithoutInternet() {

        when(mFeatureFlags.isProviderModelSettingEnabled()).thenReturn(true);

        mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1, wifi, spec2");



        assertEquals("spec1", mQSTileHost.mTileSpecs.get(0));

        assertEquals("internet", mQSTileHost.mTileSpecs.get(1));

        assertEquals("spec2", mQSTileHost.mTileSpecs.get(2));

    }



    @Test

    public void testRemoveCellWithInternet() {

        when(mFeatureFlags.isProviderModelSettingEnabled()).thenReturn(true);

        mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1, spec2, cell, internet");



        assertEquals("spec1", mQSTileHost.mTileSpecs.get(0));

        assertEquals("spec2", mQSTileHost.mTileSpecs.get(1));

        assertEquals("internet", mQSTileHost.mTileSpecs.get(2));

    }



    @Test

    public void testNoWifiNoCellularNoInternet() {

        when(mFeatureFlags.isProviderModelSettingEnabled()).thenReturn(true);

        mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1,spec2");



        assertEquals("spec1", mQSTileHost.mTileSpecs.get(0));

        assertEquals("spec2", mQSTileHost.mTileSpecs.get(1));

    }



    @Test

    public void testSpecWithInvalidDoesNotUseDefault() {

        mContext.getOrCreateTestableResources()
@@ -321,7 +373,7 @@ public class QSTileHostTest extends SysuiTestCase { 


    @Test

    public void testLoadTileSpec_repeated() {

        List<String> specs = QSTileHost.loadTileSpecs(mContext, "spec1,spec1,spec2");

        List<String> specs = QSTileHost.loadTileSpecs(mContext, "spec1,spec1,spec2", mFeatureFlags);



        assertEquals(2, specs.size());

        assertEquals("spec1", specs.get(0));
@@ -332,7 +384,7 @@ public class QSTileHostTest extends SysuiTestCase { 
    public void testLoadTileSpec_repeatedInDefault() {

        mContext.getOrCreateTestableResources()

                .addOverride(R.string.quick_settings_tiles_default, "spec1,spec1");

        List<String> specs = QSTileHost.loadTileSpecs(mContext, "default");

        List<String> specs = QSTileHost.loadTileSpecs(mContext, "default", mFeatureFlags);



        // Remove spurious tiles, like dbg:mem

        specs.removeIf(spec -> !"spec1".equals(spec));
@@ -343,7 +395,7 @@ public class QSTileHostTest extends SysuiTestCase { 
    public void testLoadTileSpec_repeatedDefaultAndSetting() {

        mContext.getOrCreateTestableResources()

                .addOverride(R.string.quick_settings_tiles_default, "spec1");

        List<String> specs = QSTileHost.loadTileSpecs(mContext, "default,spec1");

        List<String> specs = QSTileHost.loadTileSpecs(mContext, "default,spec1", mFeatureFlags);



        // Remove spurious tiles, like dbg:mem

        specs.removeIf(spec -> !"spec1".equals(spec));
@@ -371,11 +423,12 @@ public class QSTileHostTest extends SysuiTestCase { 
                Provider<AutoTileManager> autoTiles, DumpManager dumpManager,

                BroadcastDispatcher broadcastDispatcher, StatusBar statusBar, QSLogger qsLogger,

                UiEventLogger uiEventLogger, UserTracker userTracker,

                SecureSettings secureSettings, CustomTileStatePersister customTileStatePersister) {

                SecureSettings secureSettings, CustomTileStatePersister customTileStatePersister,

                FeatureFlags featureFlags) {

            super(context, iconController, defaultFactory, mainHandler, bgLooper, pluginManager,

                    tunerService, autoTiles, dumpManager, broadcastDispatcher,

                    Optional.of(statusBar), qsLogger, uiEventLogger, userTracker, secureSettings,

                    customTileStatePersister);

                    customTileStatePersister, featureFlags);

        }



        @Override

packages/SystemUI/tests/src/com/android/systemui/qs/customize/TileQueryHelperTest.java

+5 −2
Original line number Diff line number Diff line
@@ -58,6 +58,7 @@ import com.android.systemui.plugins.qs.QSIconView; 
import com.android.systemui.plugins.qs.QSTile;

import com.android.systemui.qs.QSTileHost;

import com.android.systemui.settings.UserTracker;

import com.android.systemui.statusbar.FeatureFlags;

import com.android.systemui.util.concurrency.FakeExecutor;

import com.android.systemui.util.time.FakeSystemClock;
@@ -108,6 +109,8 @@ public class TileQueryHelperTest extends SysuiTestCase { 
    private PackageManager mPackageManager;

    @Mock

    private UserTracker mUserTracker;

    @Mock

    private FeatureFlags mFeatureFlags;

    @Captor

    private ArgumentCaptor<List<TileQueryHelper.TileInfo>> mCaptor;
@@ -133,12 +136,12 @@ public class TileQueryHelperTest extends SysuiTestCase { 
                    }

                }

        ).when(mQSTileHost).createTile(anyString());



        when(mFeatureFlags.isProviderModelSettingEnabled()).thenReturn(false);

        FakeSystemClock clock = new FakeSystemClock();

        mMainExecutor = new FakeExecutor(clock);

        mBgExecutor = new FakeExecutor(clock);

        mTileQueryHelper = new TileQueryHelper(

                mContext, mUserTracker, mMainExecutor, mBgExecutor);

                mContext, mUserTracker, mMainExecutor, mBgExecutor, mFeatureFlags);

        mTileQueryHelper.setListener(mListener);

    }