Merge "Fix the StockTilesRepository tests with different feature configs" into main

1 files changed, 43 insertions, 3 deletions

diff --git a/packages/SystemUI/multivalentTests/src/com/android/systemui/qs/panels/data/repository/StockTilesRepositoryTest.kt b/packages/SystemUI/multivalentTests/src/com/android/systemui/qs/panels/data/repository/StockTilesRepositoryTest.kt
index 56cead19d1df..0d2f4eec370d 100644
--- a/packages/SystemUI/multivalentTests/src/com/android/systemui/qs/panels/data/repository/StockTilesRepositoryTest.kt
+++ b/packages/SystemUI/multivalentTests/src/com/android/systemui/qs/panels/data/repository/StockTilesRepositoryTest.kt
@@ -17,8 +17,11 @@
 package com.android.systemui.qs.panels.data.repository
 
 import android.content.res.mainResources
+import android.platform.test.annotations.DisableFlags
+import android.platform.test.annotations.EnableFlags
 import androidx.test.ext.junit.runners.AndroidJUnit4
 import androidx.test.filters.SmallTest
+import com.android.server.display.feature.flags.Flags
 import com.android.systemui.SysuiTestCase
 import com.android.systemui.qs.pipeline.shared.TileSpec
 import com.android.systemui.res.R
@@ -30,12 +33,49 @@ import org.junit.runner.RunWith
 @RunWith(AndroidJUnit4::class)
 @SmallTest
 class StockTilesRepositoryTest : SysuiTestCase() {
-    private val kosmos = testKosmos()
+    private val kosmos =
+        testKosmos().apply { mainResources = mContext.orCreateTestableResources.resources }
 
-    private val underTest = StockTilesRepository(kosmos.mainResources)
+    @Test
+    @EnableFlags(Flags.FLAG_EVEN_DIMMER)
+    fun stockTilesMatchesResources_evenDimmerFlagOn_configOn() {
+        // Enable the EvenDimmer config
+        mContext
+            .getOrCreateTestableResources()
+            .addOverride(com.android.internal.R.bool.config_evenDimmerEnabled, true)
+        val underTest = StockTilesRepository(kosmos.mainResources)
+
+        val expected =
+            kosmos.mainResources
+                .getString(R.string.quick_settings_tiles_stock)
+                .split(",")
+                .filterNot { it.equals("reduce_brightness") }
+                .map(TileSpec::create)
+        assertThat(underTest.stockTiles).isEqualTo(expected)
+    }
+
+    @Test
+    @EnableFlags(Flags.FLAG_EVEN_DIMMER)
+    fun stockTilesMatchesResources_evenDimmerFlagOn_configOff() {
+        // Disable the EvenDimmer config
+        mContext
+            .getOrCreateTestableResources()
+            .addOverride(com.android.internal.R.bool.config_evenDimmerEnabled, false)
+        val underTest = StockTilesRepository(kosmos.mainResources)
+
+        val expected =
+            kosmos.mainResources
+                .getString(R.string.quick_settings_tiles_stock)
+                .split(",")
+                .map(TileSpec::create)
+        assertThat(underTest.stockTiles).isEqualTo(expected)
+    }
 
     @Test
-    fun stockTilesMatchesResources() {
+    @DisableFlags(Flags.FLAG_EVEN_DIMMER)
+    fun stockTilesMatchesResources_evenDimmerFlagOff() {
+        val underTest = StockTilesRepository(kosmos.mainResources)
+
         val expected =
             kosmos.mainResources
                 .getString(R.string.quick_settings_tiles_stock)