packages/SystemUI/tests/src/com/android/systemui/qs/QSTileHostTest.java - platform/frameworks/base - Git at Google

 /*
  * Copyright (C) 2018 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.qs;


 import static junit.framework.Assert.assertEquals;
 import static junit.framework.Assert.assertTrue;
 import static junit.framework.TestCase.assertFalse;

 import static org.mockito.ArgumentMatchers.anyInt;
 import static org.mockito.ArgumentMatchers.anyString;
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.ArgumentMatchers.isNull;
 import static org.mockito.Mockito.atLeastOnce;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.never;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;

 import android.content.ComponentName;
 import android.content.Context;
 import android.content.Intent;
 import android.os.Handler;
 import android.os.Looper;
 import android.testing.AndroidTestingRunner;
 import android.testing.TestableLooper;
 import android.testing.TestableLooper.RunWithLooper;
 import android.view.View;

 import androidx.annotation.Nullable;
 import androidx.test.filters.SmallTest;

 import com.android.internal.logging.MetricsLogger;
 import com.android.internal.logging.UiEventLogger;
 import com.android.internal.util.CollectionUtils;
 import com.android.systemui.R;
 import com.android.systemui.SysuiTestCase;
 import com.android.systemui.broadcast.BroadcastDispatcher;
 import com.android.systemui.classifier.FalsingManagerFake;
 import com.android.systemui.dump.DumpManager;
 import com.android.systemui.plugins.ActivityStarter;
 import com.android.systemui.plugins.qs.QSFactory;
 import com.android.systemui.plugins.qs.QSTile;
 import com.android.systemui.plugins.statusbar.StatusBarStateController;
 import com.android.systemui.qs.external.CustomTile;
 import com.android.systemui.qs.external.CustomTileStatePersister;
 import com.android.systemui.qs.external.TileServiceKey;
 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;
 import com.android.systemui.tuner.TunerService;
 import com.android.systemui.util.settings.SecureSettings;

 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mockito.ArgumentCaptor;
 import org.mockito.Mock;
 import org.mockito.MockitoAnnotations;

 import java.io.FileDescriptor;
 import java.io.PrintWriter;
 import java.io.StringWriter;
 import java.util.List;
 import java.util.Optional;

 import javax.inject.Provider;

 @RunWith(AndroidTestingRunner.class)
 @SmallTest
 @RunWithLooper(setAsMainLooper = true)
 public class QSTileHostTest extends SysuiTestCase {

     private static String MOCK_STATE_STRING = "MockState";
     private static ComponentName CUSTOM_TILE =
             ComponentName.unflattenFromString("TEST_PKG/.TEST_CLS");
     private static final String CUSTOM_TILE_SPEC = CustomTile.toSpec(CUSTOM_TILE);

     @Mock
     private StatusBarIconController mIconController;
     @Mock
     private QSFactory mDefaultFactory;
     @Mock
     private PluginManager mPluginManager;
     @Mock
     private TunerService mTunerService;
     @Mock
     private Provider<AutoTileManager> mAutoTiles;
     @Mock
     private DumpManager mDumpManager;
     @Mock
     private BroadcastDispatcher mBroadcastDispatcher;
     @Mock
     private QSTile.State mMockState;
     @Mock
     private StatusBar mStatusBar;
     @Mock
     private QSLogger mQSLogger;
     @Mock
     private CustomTile mCustomTile;
     @Mock
     private UiEventLogger mUiEventLogger;
     @Mock
     private UserTracker mUserTracker;
     @Mock
     private SecureSettings mSecureSettings;
     @Mock
     private CustomTileStatePersister mCustomTileStatePersister;
     @Mock
     private FeatureFlags mFeatureFlags;

     private Handler mHandler;
     private TestableLooper mLooper;
     private QSTileHost mQSTileHost;

     @Before
     public void setUp() {
         MockitoAnnotations.initMocks(this);
         mLooper = TestableLooper.get(this);
         mHandler = new Handler(mLooper.getLooper());
         mQSTileHost = new TestQSTileHost(mContext, mIconController, mDefaultFactory, mHandler,
                 mLooper.getLooper(), mPluginManager, mTunerService, mAutoTiles, mDumpManager,
                 mBroadcastDispatcher, mStatusBar, mQSLogger, mUiEventLogger, mUserTracker,
                 mSecureSettings, mCustomTileStatePersister, mFeatureFlags);
         setUpTileFactory();
         when(mFeatureFlags.isProviderModelSettingEnabled()).thenReturn(false);
         when(mSecureSettings.getStringForUser(eq(QSTileHost.TILES_SETTING), anyInt()))
                 .thenReturn("");
     }

     private void setUpTileFactory() {
         when(mMockState.toString()).thenReturn(MOCK_STATE_STRING);
         // Only create this kind of tiles
         when(mDefaultFactory.createTile(anyString())).thenAnswer(
                 invocation -> {
                     String spec = invocation.getArgument(0);
                     if ("spec1".equals(spec)) {
                         return new TestTile1(mQSTileHost);
                     } else if ("spec2".equals(spec)) {
                         return new TestTile2(mQSTileHost);
                     } else if ("spec3".equals(spec)) {
                         return new TestTile3(mQSTileHost);
                     } else if ("na".equals(spec)) {
                         return new NotAvailableTile(mQSTileHost);
                     } else if (CUSTOM_TILE_SPEC.equals(spec)) {
                         return mCustomTile;
                     } else {
                         return null;
                     }
                 });
         when(mCustomTile.isAvailable()).thenReturn(true);
     }

     @Test
     public void testLoadTileSpecs_emptySetting() {
         List<String> tiles = QSTileHost.loadTileSpecs(mContext, "", mFeatureFlags);
         assertFalse(tiles.isEmpty());
     }

     @Test
     public void testLoadTileSpecs_nullSetting() {
         List<String> tiles = QSTileHost.loadTileSpecs(mContext, null, mFeatureFlags);
         assertFalse(tiles.isEmpty());
     }

     @Test
     public void testInvalidSpecUsesDefault() {
         mContext.getOrCreateTestableResources()
                 .addOverride(R.string.quick_settings_tiles, "spec1,spec2");
         mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "not-valid");

         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()
                 .addOverride(R.string.quick_settings_tiles, "spec1,spec2");
         mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec2,not-valid");

         assertEquals(1, mQSTileHost.getTiles().size());
         QSTile element = CollectionUtils.firstOrNull(mQSTileHost.getTiles());
         assertTrue(element instanceof TestTile2);
     }

     @Test
     public void testDump() {
         mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1,spec2");
         StringWriter w = new StringWriter();
         PrintWriter pw = new PrintWriter(w);
         mQSTileHost.dump(mock(FileDescriptor.class), pw, new String[]{});
         String output = "QSTileHost:\n"
                 + TestTile1.class.getSimpleName() + ":\n"
                 + "    " + MOCK_STATE_STRING + "\n"
                 + TestTile2.class.getSimpleName() + ":\n"
                 + "    " + MOCK_STATE_STRING + "\n";
         assertEquals(output, w.getBuffer().toString());
     }

     @Test
     public void testDefault() {
         mContext.getOrCreateTestableResources()
                 .addOverride(R.string.quick_settings_tiles_default, "spec1");
         mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "default");
         assertEquals(1, mQSTileHost.getTiles().size());
         QSTile element = CollectionUtils.firstOrNull(mQSTileHost.getTiles());
         assertTrue(element instanceof TestTile1);
         verify(mQSLogger).logTileAdded("spec1");
     }

     @Test
     public void testNoRepeatedSpecs_addTile() {
         mContext.getOrCreateTestableResources()
                 .addOverride(R.string.quick_settings_tiles, "spec1,spec2");
         mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1,spec2");

         mQSTileHost.addTile("spec1");

         assertEquals(2, mQSTileHost.mTileSpecs.size());
         assertEquals("spec1", mQSTileHost.mTileSpecs.get(0));
         assertEquals("spec2", mQSTileHost.mTileSpecs.get(1));
     }

     @Test
     public void testAddTileAtValidPosition() {
         mContext.getOrCreateTestableResources()
                 .addOverride(R.string.quick_settings_tiles, "spec1,spec3");
         mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1,spec3");

         mQSTileHost.addTile("spec2", 1);

         assertEquals(3, mQSTileHost.mTileSpecs.size());
         assertEquals("spec1", mQSTileHost.mTileSpecs.get(0));
         assertEquals("spec2", mQSTileHost.mTileSpecs.get(1));
         assertEquals("spec3", mQSTileHost.mTileSpecs.get(2));
     }

     @Test
     public void testAddTileAtInvalidPositionAddsToEnd() {
         mContext.getOrCreateTestableResources()
                 .addOverride(R.string.quick_settings_tiles, "spec1,spec3");
         mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1,spec3");

         mQSTileHost.addTile("spec2", 100);

         assertEquals(3, mQSTileHost.mTileSpecs.size());
         assertEquals("spec1", mQSTileHost.mTileSpecs.get(0));
         assertEquals("spec3", mQSTileHost.mTileSpecs.get(1));
         assertEquals("spec2", mQSTileHost.mTileSpecs.get(2));
     }

     @Test
     public void testAddTileAtEnd() {
         mContext.getOrCreateTestableResources()
                 .addOverride(R.string.quick_settings_tiles, "spec1,spec3");
         mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1,spec3");

         mQSTileHost.addTile("spec2", QSTileHost.POSITION_AT_END);

         assertEquals(3, mQSTileHost.mTileSpecs.size());
         assertEquals("spec1", mQSTileHost.mTileSpecs.get(0));
         assertEquals("spec3", mQSTileHost.mTileSpecs.get(1));
         assertEquals("spec2", mQSTileHost.mTileSpecs.get(2));
     }

     @Test
     public void testNoRepeatedSpecs_customTile() {
         mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, CUSTOM_TILE_SPEC);

         mQSTileHost.addTile(CUSTOM_TILE, /* end */ false);

         assertEquals(1, mQSTileHost.mTileSpecs.size());
         assertEquals(CUSTOM_TILE_SPEC, mQSTileHost.mTileSpecs.get(0));
     }

     @Test
     public void testAddedAtBeginningOnDefault_customTile() {
         mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1"); // seed

         mQSTileHost.addTile(CUSTOM_TILE);

         assertEquals(2, mQSTileHost.mTileSpecs.size());
         assertEquals(CUSTOM_TILE_SPEC, mQSTileHost.mTileSpecs.get(0));
     }

     @Test
     public void testAddedAtBeginning_customTile() {
         mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1"); // seed

         mQSTileHost.addTile(CUSTOM_TILE, /* end */ false);

         assertEquals(2, mQSTileHost.mTileSpecs.size());
         assertEquals(CUSTOM_TILE_SPEC, mQSTileHost.mTileSpecs.get(0));
     }

     @Test
     public void testAddedAtEnd_customTile() {
         mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1"); // seed

         mQSTileHost.addTile(CUSTOM_TILE, /* end */ true);

         assertEquals(2, mQSTileHost.mTileSpecs.size());
         assertEquals(CUSTOM_TILE_SPEC, mQSTileHost.mTileSpecs.get(1));
     }

     @Test
     public void testLoadTileSpec_repeated() {
         List<String> specs = QSTileHost.loadTileSpecs(mContext, "spec1,spec1,spec2", mFeatureFlags);

         assertEquals(2, specs.size());
         assertEquals("spec1", specs.get(0));
         assertEquals("spec2", specs.get(1));
     }

     @Test
     public void testLoadTileSpec_repeatedInDefault() {
         mContext.getOrCreateTestableResources()
                 .addOverride(R.string.quick_settings_tiles_default, "spec1,spec1");
         List<String> specs = QSTileHost.loadTileSpecs(mContext, "default", mFeatureFlags);

         // Remove spurious tiles, like dbg:mem
         specs.removeIf(spec -> !"spec1".equals(spec));
         assertEquals(1, specs.size());
     }

     @Test
     public void testLoadTileSpec_repeatedDefaultAndSetting() {
         mContext.getOrCreateTestableResources()
                 .addOverride(R.string.quick_settings_tiles_default, "spec1");
         List<String> specs = QSTileHost.loadTileSpecs(mContext, "default,spec1", mFeatureFlags);

         // Remove spurious tiles, like dbg:mem
         specs.removeIf(spec -> !"spec1".equals(spec));
         assertEquals(1, specs.size());
     }

     @Test
     public void testNotAvailableTile_specNotNull() {
         mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "na");
         verify(mQSLogger, never()).logTileDestroyed(isNull(), anyString());
     }

     @Test
     public void testCustomTileRemoved_stateDeleted() {
         mQSTileHost.changeTiles(List.of(CUSTOM_TILE_SPEC), List.of());

         verify(mCustomTileStatePersister)
                 .removeState(new TileServiceKey(CUSTOM_TILE, mQSTileHost.getUserId()));
     }

     private class TestQSTileHost extends QSTileHost {
         TestQSTileHost(Context context, StatusBarIconController iconController,
                 QSFactory defaultFactory, Handler mainHandler, Looper bgLooper,
                 PluginManager pluginManager, TunerService tunerService,
                 Provider<AutoTileManager> autoTiles, DumpManager dumpManager,
                 BroadcastDispatcher broadcastDispatcher, StatusBar statusBar, QSLogger qsLogger,
                 UiEventLogger uiEventLogger, UserTracker userTracker,
                 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, featureFlags);
         }

         @Override
         public void onPluginConnected(QSFactory plugin, Context pluginContext) {
         }

         @Override
         public void onPluginDisconnected(QSFactory plugin) {
         }

         @Override
         void saveTilesToSettings(List<String> tileSpecs) {
             super.saveTilesToSettings(tileSpecs);

             ArgumentCaptor<String> specs = ArgumentCaptor.forClass(String.class);
             verify(mSecureSettings, atLeastOnce()).putStringForUser(eq(QSTileHost.TILES_SETTING),
                     specs.capture(), isNull(), eq(false), anyInt(), eq(true));

             // After tiles are changed, make sure to call onTuningChanged with the new setting if it
             // changed
             onTuningChanged(TILES_SETTING, specs.getValue());
         }
     }

     private class TestTile extends QSTileImpl<QSTile.State> {

         protected TestTile(QSHost host) {
             super(
                     host,
                     mLooper.getLooper(),
                     new Handler(mLooper.getLooper()),
                     new FalsingManagerFake(),
                     mock(MetricsLogger.class),
                     mock(StatusBarStateController.class),
                     mock(ActivityStarter.class),
                     mQSLogger
             );
         }

         @Override
         public State newTileState() {
             return mMockState;
         }

         @Override
         public State getState() {
             return mMockState;
         }

         @Override
         protected void handleClick(@Nullable View view) {}

         @Override
         protected void handleUpdateState(State state, Object arg) {}

         @Override
         public int getMetricsCategory() {
             return 0;
         }

         @Override
         public Intent getLongClickIntent() {
             return null;
         }

         @Override
         public CharSequence getTileLabel() {
             return null;
         }
     }

     private class TestTile1 extends TestTile {

         protected TestTile1(QSHost host) {
             super(host);
         }
     }

     private class TestTile2 extends TestTile {

         protected TestTile2(QSHost host) {
             super(host);
         }
     }

     private class TestTile3 extends TestTile {

         protected TestTile3(QSHost host) {
             super(host);
         }
     }

     private class NotAvailableTile extends TestTile {

         protected NotAvailableTile(QSHost host) {
             super(host);
         }

         @Override
         public boolean isAvailable() {
             return false;
         }
     }
 }
	/*
	* Copyright (C) 2018 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.qs;


	import static junit.framework.Assert.assertEquals;
	import static junit.framework.Assert.assertTrue;
	import static junit.framework.TestCase.assertFalse;

	import static org.mockito.ArgumentMatchers.anyInt;
	import static org.mockito.ArgumentMatchers.anyString;
	import static org.mockito.ArgumentMatchers.eq;
	import static org.mockito.ArgumentMatchers.isNull;
	import static org.mockito.Mockito.atLeastOnce;
	import static org.mockito.Mockito.mock;
	import static org.mockito.Mockito.never;
	import static org.mockito.Mockito.verify;
	import static org.mockito.Mockito.when;

	import android.content.ComponentName;
	import android.content.Context;
	import android.content.Intent;
	import android.os.Handler;
	import android.os.Looper;
	import android.testing.AndroidTestingRunner;
	import android.testing.TestableLooper;
	import android.testing.TestableLooper.RunWithLooper;
	import android.view.View;

	import androidx.annotation.Nullable;
	import androidx.test.filters.SmallTest;

	import com.android.internal.logging.MetricsLogger;
	import com.android.internal.logging.UiEventLogger;
	import com.android.internal.util.CollectionUtils;
	import com.android.systemui.R;
	import com.android.systemui.SysuiTestCase;
	import com.android.systemui.broadcast.BroadcastDispatcher;
	import com.android.systemui.classifier.FalsingManagerFake;
	import com.android.systemui.dump.DumpManager;
	import com.android.systemui.plugins.ActivityStarter;
	import com.android.systemui.plugins.qs.QSFactory;
	import com.android.systemui.plugins.qs.QSTile;
	import com.android.systemui.plugins.statusbar.StatusBarStateController;
	import com.android.systemui.qs.external.CustomTile;
	import com.android.systemui.qs.external.CustomTileStatePersister;
	import com.android.systemui.qs.external.TileServiceKey;
	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;
	import com.android.systemui.tuner.TunerService;
	import com.android.systemui.util.settings.SecureSettings;

	import org.junit.Before;
	import org.junit.Test;
	import org.junit.runner.RunWith;
	import org.mockito.ArgumentCaptor;
	import org.mockito.Mock;
	import org.mockito.MockitoAnnotations;

	import java.io.FileDescriptor;
	import java.io.PrintWriter;
	import java.io.StringWriter;
	import java.util.List;
	import java.util.Optional;

	import javax.inject.Provider;

	@RunWith(AndroidTestingRunner.class)
	@SmallTest
	@RunWithLooper(setAsMainLooper = true)
	public class QSTileHostTest extends SysuiTestCase {

	private static String MOCK_STATE_STRING = "MockState";
	private static ComponentName CUSTOM_TILE =
	ComponentName.unflattenFromString("TEST_PKG/.TEST_CLS");
	private static final String CUSTOM_TILE_SPEC = CustomTile.toSpec(CUSTOM_TILE);

	@Mock
	private StatusBarIconController mIconController;
	@Mock
	private QSFactory mDefaultFactory;
	@Mock
	private PluginManager mPluginManager;
	@Mock
	private TunerService mTunerService;
	@Mock
	private Provider<AutoTileManager> mAutoTiles;
	@Mock
	private DumpManager mDumpManager;
	@Mock
	private BroadcastDispatcher mBroadcastDispatcher;
	@Mock
	private QSTile.State mMockState;
	@Mock
	private StatusBar mStatusBar;
	@Mock
	private QSLogger mQSLogger;
	@Mock
	private CustomTile mCustomTile;
	@Mock
	private UiEventLogger mUiEventLogger;
	@Mock
	private UserTracker mUserTracker;
	@Mock
	private SecureSettings mSecureSettings;
	@Mock
	private CustomTileStatePersister mCustomTileStatePersister;
	@Mock
	private FeatureFlags mFeatureFlags;

	private Handler mHandler;
	private TestableLooper mLooper;
	private QSTileHost mQSTileHost;

	@Before
	public void setUp() {
	MockitoAnnotations.initMocks(this);
	mLooper = TestableLooper.get(this);
	mHandler = new Handler(mLooper.getLooper());
	mQSTileHost = new TestQSTileHost(mContext, mIconController, mDefaultFactory, mHandler,
	mLooper.getLooper(), mPluginManager, mTunerService, mAutoTiles, mDumpManager,
	mBroadcastDispatcher, mStatusBar, mQSLogger, mUiEventLogger, mUserTracker,
	mSecureSettings, mCustomTileStatePersister, mFeatureFlags);
	setUpTileFactory();
	when(mFeatureFlags.isProviderModelSettingEnabled()).thenReturn(false);
	when(mSecureSettings.getStringForUser(eq(QSTileHost.TILES_SETTING), anyInt()))
	.thenReturn("");
	}

	private void setUpTileFactory() {
	when(mMockState.toString()).thenReturn(MOCK_STATE_STRING);
	// Only create this kind of tiles
	when(mDefaultFactory.createTile(anyString())).thenAnswer(
	invocation -> {
	String spec = invocation.getArgument(0);
	if ("spec1".equals(spec)) {
	return new TestTile1(mQSTileHost);
	} else if ("spec2".equals(spec)) {
	return new TestTile2(mQSTileHost);
	} else if ("spec3".equals(spec)) {
	return new TestTile3(mQSTileHost);
	} else if ("na".equals(spec)) {
	return new NotAvailableTile(mQSTileHost);
	} else if (CUSTOM_TILE_SPEC.equals(spec)) {
	return mCustomTile;
	} else {
	return null;
	}
	});
	when(mCustomTile.isAvailable()).thenReturn(true);
	}

	@Test
	public void testLoadTileSpecs_emptySetting() {
	List<String> tiles = QSTileHost.loadTileSpecs(mContext, "", mFeatureFlags);
	assertFalse(tiles.isEmpty());
	}

	@Test
	public void testLoadTileSpecs_nullSetting() {
	List<String> tiles = QSTileHost.loadTileSpecs(mContext, null, mFeatureFlags);
	assertFalse(tiles.isEmpty());
	}

	@Test
	public void testInvalidSpecUsesDefault() {
	mContext.getOrCreateTestableResources()
	.addOverride(R.string.quick_settings_tiles, "spec1,spec2");
	mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "not-valid");

	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()
	.addOverride(R.string.quick_settings_tiles, "spec1,spec2");
	mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec2,not-valid");

	assertEquals(1, mQSTileHost.getTiles().size());
	QSTile element = CollectionUtils.firstOrNull(mQSTileHost.getTiles());
	assertTrue(element instanceof TestTile2);
	}

	@Test
	public void testDump() {
	mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1,spec2");
	StringWriter w = new StringWriter();
	PrintWriter pw = new PrintWriter(w);
	mQSTileHost.dump(mock(FileDescriptor.class), pw, new String[]{});
	String output = "QSTileHost:\n"
	+ TestTile1.class.getSimpleName() + ":\n"
	+ " " + MOCK_STATE_STRING + "\n"
	+ TestTile2.class.getSimpleName() + ":\n"
	+ " " + MOCK_STATE_STRING + "\n";
	assertEquals(output, w.getBuffer().toString());
	}

	@Test
	public void testDefault() {
	mContext.getOrCreateTestableResources()
	.addOverride(R.string.quick_settings_tiles_default, "spec1");
	mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "default");
	assertEquals(1, mQSTileHost.getTiles().size());
	QSTile element = CollectionUtils.firstOrNull(mQSTileHost.getTiles());
	assertTrue(element instanceof TestTile1);
	verify(mQSLogger).logTileAdded("spec1");
	}

	@Test
	public void testNoRepeatedSpecs_addTile() {
	mContext.getOrCreateTestableResources()
	.addOverride(R.string.quick_settings_tiles, "spec1,spec2");
	mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1,spec2");

	mQSTileHost.addTile("spec1");

	assertEquals(2, mQSTileHost.mTileSpecs.size());
	assertEquals("spec1", mQSTileHost.mTileSpecs.get(0));
	assertEquals("spec2", mQSTileHost.mTileSpecs.get(1));
	}

	@Test
	public void testAddTileAtValidPosition() {
	mContext.getOrCreateTestableResources()
	.addOverride(R.string.quick_settings_tiles, "spec1,spec3");
	mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1,spec3");

	mQSTileHost.addTile("spec2", 1);

	assertEquals(3, mQSTileHost.mTileSpecs.size());
	assertEquals("spec1", mQSTileHost.mTileSpecs.get(0));
	assertEquals("spec2", mQSTileHost.mTileSpecs.get(1));
	assertEquals("spec3", mQSTileHost.mTileSpecs.get(2));
	}

	@Test
	public void testAddTileAtInvalidPositionAddsToEnd() {
	mContext.getOrCreateTestableResources()
	.addOverride(R.string.quick_settings_tiles, "spec1,spec3");
	mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1,spec3");

	mQSTileHost.addTile("spec2", 100);

	assertEquals(3, mQSTileHost.mTileSpecs.size());
	assertEquals("spec1", mQSTileHost.mTileSpecs.get(0));
	assertEquals("spec3", mQSTileHost.mTileSpecs.get(1));
	assertEquals("spec2", mQSTileHost.mTileSpecs.get(2));
	}

	@Test
	public void testAddTileAtEnd() {
	mContext.getOrCreateTestableResources()
	.addOverride(R.string.quick_settings_tiles, "spec1,spec3");
	mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1,spec3");

	mQSTileHost.addTile("spec2", QSTileHost.POSITION_AT_END);

	assertEquals(3, mQSTileHost.mTileSpecs.size());
	assertEquals("spec1", mQSTileHost.mTileSpecs.get(0));
	assertEquals("spec3", mQSTileHost.mTileSpecs.get(1));
	assertEquals("spec2", mQSTileHost.mTileSpecs.get(2));
	}

	@Test
	public void testNoRepeatedSpecs_customTile() {
	mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, CUSTOM_TILE_SPEC);

	mQSTileHost.addTile(CUSTOM_TILE, /* end */ false);

	assertEquals(1, mQSTileHost.mTileSpecs.size());
	assertEquals(CUSTOM_TILE_SPEC, mQSTileHost.mTileSpecs.get(0));
	}

	@Test
	public void testAddedAtBeginningOnDefault_customTile() {
	mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1"); // seed

	mQSTileHost.addTile(CUSTOM_TILE);

	assertEquals(2, mQSTileHost.mTileSpecs.size());
	assertEquals(CUSTOM_TILE_SPEC, mQSTileHost.mTileSpecs.get(0));
	}

	@Test
	public void testAddedAtBeginning_customTile() {
	mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1"); // seed

	mQSTileHost.addTile(CUSTOM_TILE, /* end */ false);

	assertEquals(2, mQSTileHost.mTileSpecs.size());
	assertEquals(CUSTOM_TILE_SPEC, mQSTileHost.mTileSpecs.get(0));
	}

	@Test
	public void testAddedAtEnd_customTile() {
	mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "spec1"); // seed

	mQSTileHost.addTile(CUSTOM_TILE, /* end */ true);

	assertEquals(2, mQSTileHost.mTileSpecs.size());
	assertEquals(CUSTOM_TILE_SPEC, mQSTileHost.mTileSpecs.get(1));
	}

	@Test
	public void testLoadTileSpec_repeated() {
	List<String> specs = QSTileHost.loadTileSpecs(mContext, "spec1,spec1,spec2", mFeatureFlags);

	assertEquals(2, specs.size());
	assertEquals("spec1", specs.get(0));
	assertEquals("spec2", specs.get(1));
	}

	@Test
	public void testLoadTileSpec_repeatedInDefault() {
	mContext.getOrCreateTestableResources()
	.addOverride(R.string.quick_settings_tiles_default, "spec1,spec1");
	List<String> specs = QSTileHost.loadTileSpecs(mContext, "default", mFeatureFlags);

	// Remove spurious tiles, like dbg:mem
	specs.removeIf(spec -> !"spec1".equals(spec));
	assertEquals(1, specs.size());
	}

	@Test
	public void testLoadTileSpec_repeatedDefaultAndSetting() {
	mContext.getOrCreateTestableResources()
	.addOverride(R.string.quick_settings_tiles_default, "spec1");
	List<String> specs = QSTileHost.loadTileSpecs(mContext, "default,spec1", mFeatureFlags);

	// Remove spurious tiles, like dbg:mem
	specs.removeIf(spec -> !"spec1".equals(spec));
	assertEquals(1, specs.size());
	}

	@Test
	public void testNotAvailableTile_specNotNull() {
	mQSTileHost.onTuningChanged(QSTileHost.TILES_SETTING, "na");
	verify(mQSLogger, never()).logTileDestroyed(isNull(), anyString());
	}

	@Test
	public void testCustomTileRemoved_stateDeleted() {
	mQSTileHost.changeTiles(List.of(CUSTOM_TILE_SPEC), List.of());

	verify(mCustomTileStatePersister)
	.removeState(new TileServiceKey(CUSTOM_TILE, mQSTileHost.getUserId()));
	}

	private class TestQSTileHost extends QSTileHost {
	TestQSTileHost(Context context, StatusBarIconController iconController,
	QSFactory defaultFactory, Handler mainHandler, Looper bgLooper,
	PluginManager pluginManager, TunerService tunerService,
	Provider<AutoTileManager> autoTiles, DumpManager dumpManager,
	BroadcastDispatcher broadcastDispatcher, StatusBar statusBar, QSLogger qsLogger,
	UiEventLogger uiEventLogger, UserTracker userTracker,
	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, featureFlags);
	}

	@Override
	public void onPluginConnected(QSFactory plugin, Context pluginContext) {
	}

	@Override
	public void onPluginDisconnected(QSFactory plugin) {
	}

	@Override
	void saveTilesToSettings(List<String> tileSpecs) {
	super.saveTilesToSettings(tileSpecs);

	ArgumentCaptor<String> specs = ArgumentCaptor.forClass(String.class);
	verify(mSecureSettings, atLeastOnce()).putStringForUser(eq(QSTileHost.TILES_SETTING),
	specs.capture(), isNull(), eq(false), anyInt(), eq(true));

	// After tiles are changed, make sure to call onTuningChanged with the new setting if it
	// changed
	onTuningChanged(TILES_SETTING, specs.getValue());
	}
	}

	private class TestTile extends QSTileImpl<QSTile.State> {

	protected TestTile(QSHost host) {
	super(
	host,
	mLooper.getLooper(),
	new Handler(mLooper.getLooper()),
	new FalsingManagerFake(),
	mock(MetricsLogger.class),
	mock(StatusBarStateController.class),
	mock(ActivityStarter.class),
	mQSLogger
	);
	}

	@Override
	public State newTileState() {
	return mMockState;
	}

	@Override
	public State getState() {
	return mMockState;
	}

	@Override
	protected void handleClick(@Nullable View view) {}

	@Override
	protected void handleUpdateState(State state, Object arg) {}

	@Override
	public int getMetricsCategory() {
	return 0;
	}

	@Override
	public Intent getLongClickIntent() {
	return null;
	}

	@Override
	public CharSequence getTileLabel() {
	return null;
	}
	}

	private class TestTile1 extends TestTile {

	protected TestTile1(QSHost host) {
	super(host);
	}
	}

	private class TestTile2 extends TestTile {

	protected TestTile2(QSHost host) {
	super(host);
	}
	}

	private class TestTile3 extends TestTile {

	protected TestTile3(QSHost host) {
	super(host);
	}
	}

	private class NotAvailableTile extends TestTile {

	protected NotAvailableTile(QSHost host) {
	super(host);
	}

	@Override
	public boolean isAvailable() {
	return false;
	}
	}
	}