Enable catalyst test for PowerUsageSummaryScreenTest

Bug: 372774754
Flag: EXEMPT test
Test: atest
Change-Id: I5b35b0ff3416dde4781b7874f5bbc12f6b3f2475
This commit is contained in:
Jacky Wang
2024-10-18 19:55:14 +08:00
parent 46c45023d2
commit cc693e950f

View File

@@ -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,
)
}