From cc693e950f1160fd903686c562cd8e2d61d87ab7 Mon Sep 17 00:00:00 2001
From: Jacky Wang <jiannan@google.com>
Date: Fri, 18 Oct 2024 19:55:14 +0800
Subject: [PATCH] Enable catalyst test for PowerUsageSummaryScreenTest

Bug: 372774754
Flag: EXEMPT test
Test: atest
Change-Id: I5b35b0ff3416dde4781b7874f5bbc12f6b3f2475
---
 .../PowerUsageSummaryScreenTest.kt            | 23 ++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/tests/robotests/src/com/android/settings/fuelgauge/batteryusage/PowerUsageSummaryScreenTest.kt b/tests/robotests/src/com/android/settings/fuelgauge/batteryusage/PowerUsageSummaryScreenTest.kt
index 993d39e81b0..80a6b2ef8f3 100644
--- a/tests/robotests/src/com/android/settings/fuelgauge/batteryusage/PowerUsageSummaryScreenTest.kt
+++ b/tests/robotests/src/com/android/settings/fuelgauge/batteryusage/PowerUsageSummaryScreenTest.kt
@@ -19,16 +19,22 @@ import android.content.ContextWrapper
 import android.content.res.Resources
 import android.platform.test.annotations.DisableFlags
 import android.platform.test.annotations.EnableFlags
+import androidx.fragment.app.testing.FragmentScenario
+import androidx.preference.PreferenceFragmentCompat
 import com.android.settings.R
 import com.android.settings.flags.Flags
+import com.android.settings.testutils.shadow.ShadowUtils
 import com.android.settingslib.preference.CatalystScreenTestCase
 import com.google.common.truth.Truth.assertThat
+import org.junit.After
 import org.junit.Test
 import org.mockito.ArgumentMatchers.anyInt
 import org.mockito.kotlin.doReturn
 import org.mockito.kotlin.mock
 import org.mockito.kotlin.stub
+import org.robolectric.annotation.Config
 
+@Config(shadows = [ShadowUtils::class])
 class PowerUsageSummaryScreenTest : CatalystScreenTestCase() {
 
     override val preferenceScreenCreator = PowerUsageSummaryScreen()
@@ -43,6 +49,11 @@ class PowerUsageSummaryScreenTest : CatalystScreenTestCase() {
             override fun getResources(): Resources = mockResources
         }
 
+    @After
+    fun tearDown() {
+        ShadowUtils.reset()
+    }
+
     @Test
     fun key() {
         assertThat(preferenceScreenCreator.key).isEqualTo(PowerUsageSummaryScreen.KEY)
@@ -76,5 +87,15 @@ class PowerUsageSummaryScreenTest : CatalystScreenTestCase() {
             .isEqualTo(R.drawable.ic_settings_battery_white)
     }
 
-    override fun migration() {}
+    override fun migration() {
+        ShadowUtils.setIsBatteryPresent(false)
+
+        super.migration()
+    }
+
+    override fun launchFragmentScenario(fragmentClass: Class<PreferenceFragmentCompat>) =
+        FragmentScenario.launch(
+            fragmentClass,
+            themeResId = R.style.Theme_CollapsingToolbar_Settings,
+        )
 }