Merge "Check feature availability before running tests"

2020-10-29 02:57:23 +00:00
parent 9d851f9329 259ce71793
commit eacac95375
1 changed files with 186 additions and 16 deletions
--- a/services/tests/servicestests/src/com/android/server/display/color/ColorDisplayServiceTest.java
+++ b/services/tests/servicestests/src/com/android/server/display/color/ColorDisplayServiceTest.java
@@ -18,6 +18,7 @@ package com.android.server.display.color;

 import static com.google.common.truth.Truth.assertWithMessage;

+import static org.junit.Assume.assumeTrue;
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.doReturn;
@@ -74,6 +75,10 @@ public class ColorDisplayServiceTest {

    private Context mContext;
    private int mUserId;
+    private boolean mNightDisplayAvailable;
+    private boolean mDisplayWhiteBalanceAvailable;
+    private boolean mScreenWideColorGamutAvailable;
+    private boolean mColorModeAvailable;

    private MockTwilightManager mTwilightManager;

@@ -90,7 +95,18 @@ public class ColorDisplayServiceTest {

    @Before
    public void setUp() {
-        mContext = Mockito.spy(new ContextWrapper(InstrumentationRegistry.getTargetContext()));
+        final Context targetContext = InstrumentationRegistry.getTargetContext();
+        mNightDisplayAvailable =
+                ColorDisplayManager.isNightDisplayAvailable(targetContext);
+        mDisplayWhiteBalanceAvailable =
+                ColorDisplayManager.isDisplayWhiteBalanceAvailable(targetContext);
+        mScreenWideColorGamutAvailable =
+                targetContext.getResources().getConfiguration().isScreenWideColorGamut();
+        final int[] availableColorModes = targetContext.getResources().getIntArray(
+                R.array.config_availableColorModes);
+        mColorModeAvailable = availableColorModes != null && availableColorModes.length > 0;
+
+        mContext = Mockito.spy(new ContextWrapper(targetContext));
        doReturn(mContext).when(mContext).getApplicationContext();

        mResourcesSpy = Mockito.spy(mContext.getResources());
@@ -137,6 +153,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedAfterNight_ifOffAfterNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(-120 /* startTimeOffset */, -60 /* endTimeOffset */);
        setNightDisplayActivated(false /* activated */, -30 /* lastActivatedTimeOffset */);

@@ -146,6 +164,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedAfterNight_ifOffBeforeNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(-120 /* startTimeOffset */, -60 /* endTimeOffset */);
        setNightDisplayActivated(false /* activated */, -180 /* lastActivatedTimeOffset */);

@@ -155,6 +175,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedAfterNight_ifOffDuringNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(-120 /* startTimeOffset */, -60 /* endTimeOffset */);
        setNightDisplayActivated(false /* activated */, -90 /* lastActivatedTimeOffset */);

@@ -164,6 +186,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedAfterNight_ifOffInFuture_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(-120 /* startTimeOffset */, -60 /* endTimeOffset */);
        setNightDisplayActivated(false /* activated */, 30 /* lastActivatedTimeOffset */);

@@ -173,6 +197,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedAfterNight_ifOnAfterNight_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(-120 /* startTimeOffset */, -60 /* endTimeOffset */);
        setNightDisplayActivated(true /* activated */, -30 /* lastActivatedTimeOffset */);

@@ -182,6 +208,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedAfterNight_ifOnBeforeNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(-120 /* startTimeOffset */, -60 /* endTimeOffset */);
        setNightDisplayActivated(true /* activated */, -180 /* lastActivatedTimeOffset */);

@@ -191,6 +219,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedAfterNight_ifOnDuringNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(-120 /* startTimeOffset */, -60 /* endTimeOffset */);
        setNightDisplayActivated(true /* activated */, -90 /* lastActivatedTimeOffset */);

@@ -200,6 +230,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedAfterNight_ifOnInFuture_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(-120 /* startTimeOffset */, -60 /* endTimeOffset */);
        setNightDisplayActivated(true /* activated */, 30 /* lastActivatedTimeOffset */);

@@ -209,6 +241,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedBeforeNight_ifOffAfterNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(60 /* startTimeOffset */, 120 /* endTimeOffset */);
        setNightDisplayActivated(false /* activated */, 180 /* lastActivatedTimeOffset */);

@@ -218,6 +252,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedBeforeNight_ifOffBeforeNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(60 /* startTimeOffset */, 120 /* endTimeOffset */);
        setNightDisplayActivated(false /* activated */, 30 /* lastActivatedTimeOffset */);

@@ -227,6 +263,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedBeforeNight_ifOffDuringNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(60 /* startTimeOffset */, 120 /* endTimeOffset */);
        setNightDisplayActivated(false /* activated */, 90 /* lastActivatedTimeOffset */);

@@ -236,6 +274,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedBeforeNight_ifOffInPast_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(60 /* startTimeOffset */, 120 /* endTimeOffset */);
        setNightDisplayActivated(false /* activated */, -30 /* lastActivatedTimeOffset */);

@@ -245,6 +285,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedBeforeNight_ifOnAfterNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(60 /* startTimeOffset */, 120 /* endTimeOffset */);
        setNightDisplayActivated(true /* activated */, 180 /* lastActivatedTimeOffset */);

@@ -254,6 +296,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedBeforeNight_ifOnBeforeNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(60 /* startTimeOffset */, 120 /* endTimeOffset */);
        setNightDisplayActivated(true /* activated */, 30 /* lastActivatedTimeOffset */);

@@ -263,6 +307,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedBeforeNight_ifOnDuringNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(60 /* startTimeOffset */, 120 /* endTimeOffset */);
        setNightDisplayActivated(true /* activated */, 90 /* lastActivatedTimeOffset */);

@@ -272,6 +318,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedBeforeNight_ifOnInPast_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(60 /* startTimeOffset */, 120 /* endTimeOffset */);
        setNightDisplayActivated(true /* activated */, -30 /* lastActivatedTimeOffset */);

@@ -281,6 +329,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedDuringNight_ifOffAfterNight_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(-60 /* startTimeOffset */, 60 /* endTimeOffset */);
        setNightDisplayActivated(false /* activated */, 90 /* lastActivatedTimeOffset */);

@@ -290,6 +340,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedDuringNight_ifOffBeforeNight_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(-60 /* startTimeOffset */, 60 /* endTimeOffset */);
        setNightDisplayActivated(false /* activated */, -90 /* lastActivatedTimeOffset */);

@@ -299,6 +351,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedDuringNight_ifOffDuringNightInFuture_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(-60 /* startTimeOffset */, 60 /* endTimeOffset */);
        setNightDisplayActivated(false /* activated */, 30 /* lastActivatedTimeOffset */);

@@ -308,6 +362,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedDuringNight_ifOffDuringNightInPast_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(-60 /* startTimeOffset */, 60 /* endTimeOffset */);
        setNightDisplayActivated(false /* activated */, -30 /* lastActivatedTimeOffset */);

@@ -317,6 +373,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedDuringNight_ifOnAfterNight_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(-60 /* startTimeOffset */, 60 /* endTimeOffset */);
        setNightDisplayActivated(true /* activated */, 90 /* lastActivatedTimeOffset */);

@@ -326,6 +384,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedDuringNight_ifOnBeforeNight_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(-60 /* startTimeOffset */, 60 /* endTimeOffset */);
        setNightDisplayActivated(true /* activated */, -90 /* lastActivatedTimeOffset */);

@@ -335,6 +395,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedDuringNight_ifOnDuringNightInFuture_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(-60 /* startTimeOffset */, 60 /* endTimeOffset */);
        setNightDisplayActivated(true /* activated */, 30 /* lastActivatedTimeOffset */);

@@ -344,6 +406,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void customSchedule_whenStartedDuringNight_ifOnDuringNightInPast_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeCustom(-60 /* startTimeOffset */, 60 /* endTimeOffset */);
        setNightDisplayActivated(true /* activated */, -30 /* lastActivatedTimeOffset */);

@@ -353,6 +417,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedAfterNight_ifOffAfterNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-120 /* sunsetOffset */, -60 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, -30 /* lastActivatedTimeOffset */);

@@ -362,6 +428,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedAfterNight_ifOffBeforeNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-120 /* sunsetOffset */, -60 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, -180 /* lastActivatedTimeOffset */);

@@ -371,6 +439,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedAfterNight_ifOffDuringNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-120 /* sunsetOffset */, -60 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, -90 /* lastActivatedTimeOffset */);

@@ -380,6 +450,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedAfterNight_ifOffInFuture_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-120 /* sunsetOffset */, -60 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, 30 /* lastActivatedTimeOffset */);

@@ -389,6 +461,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedAfterNight_ifOnAfterNight_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-120 /* sunsetOffset */, -60 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, -30 /* lastActivatedTimeOffset */);

@@ -398,6 +472,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedAfterNight_ifOnBeforeNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-120 /* sunsetOffset */, -60 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, -180 /* lastActivatedTimeOffset */);

@@ -407,6 +483,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedAfterNight_ifOnDuringNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-120 /* sunsetOffset */, -60 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, -90 /* lastActivatedTimeOffset */);

@@ -416,6 +494,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedAfterNight_ifOnInFuture_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-120 /* sunsetOffset */, -60 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, 30 /* lastActivatedTimeOffset */);

@@ -425,6 +505,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedBeforeNight_ifOffAfterNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(60 /* sunsetOffset */, 120 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, 180 /* lastActivatedTimeOffset */);

@@ -434,6 +516,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedBeforeNight_ifOffBeforeNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(60 /* sunsetOffset */, 120 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, 30 /* lastActivatedTimeOffset */);

@@ -443,6 +527,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedBeforeNight_ifOffDuringNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(60 /* sunsetOffset */, 120 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, 90 /* lastActivatedTimeOffset */);

@@ -452,6 +538,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedBeforeNight_ifOffInPast_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(60 /* sunsetOffset */, 120 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, -30 /* lastActivatedTimeOffset */);

@@ -461,6 +549,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedBeforeNight_ifOnAfterNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(60 /* sunsetOffset */, 120 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, 180 /* lastActivatedTimeOffset */);

@@ -470,6 +560,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedBeforeNight_ifOnBeforeNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(60 /* sunsetOffset */, 120 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, 30 /* lastActivatedTimeOffset */);

@@ -479,6 +571,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedBeforeNight_ifOnDuringNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(60 /* sunsetOffset */, 120 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, 90 /* lastActivatedTimeOffset */);

@@ -488,6 +582,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedBeforeNight_ifOnInPast_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(60 /* sunsetOffset */, 120 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, -30 /* lastActivatedTimeOffset */);

@@ -497,6 +593,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedDuringNight_ifOffAfterNight_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-60 /* sunsetOffset */, 60 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, 90 /* lastActivatedTimeOffset */);

@@ -506,6 +604,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedDuringNight_ifOffBeforeNight_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-60 /* sunsetOffset */, 60 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, -90 /* lastActivatedTimeOffset */);

@@ -515,6 +615,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedDuringNight_ifOffDuringNightInFuture_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-60 /* sunsetOffset */, 60 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, 30 /* lastActivatedTimeOffset */);

@@ -524,6 +626,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedDuringNight_ifOffDuringNightInPast_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-60 /* sunsetOffset */, 60 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, -30 /* lastActivatedTimeOffset */);

@@ -533,6 +637,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedDuringNight_ifOnAfterNight_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-60 /* sunsetOffset */, 60 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, 90 /* lastActivatedTimeOffset */);

@@ -542,6 +648,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedDuringNight_ifOnBeforeNight_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-60 /* sunsetOffset */, 60 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, -90 /* lastActivatedTimeOffset */);

@@ -551,6 +659,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedDuringNight_ifOnDuringNightInFuture_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-60 /* sunsetOffset */, 60 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, 30 /* lastActivatedTimeOffset */);

@@ -560,6 +670,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenStartedDuringNight_ifOnDuringNightInPast_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-60 /* sunsetOffset */, 60 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, -30 /* lastActivatedTimeOffset */);

@@ -569,6 +681,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedAfterNight_ifOffAfterNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-120 /* sunsetOffset */, -60 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, -30 /* lastActivatedTimeOffset */);

@@ -584,6 +698,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedAfterNight_ifOffBeforeNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-120 /* sunsetOffset */, -60 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, -180 /* lastActivatedTimeOffset */);

@@ -599,6 +715,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedAfterNight_ifOffDuringNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-120 /* sunsetOffset */, -60 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, -90 /* lastActivatedTimeOffset */);

@@ -614,6 +732,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedAfterNight_ifOffInFuture_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-120 /* sunsetOffset */, -60 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, 30 /* lastActivatedTimeOffset */);

@@ -629,6 +749,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedAfterNight_ifOnAfterNight_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-120 /* sunsetOffset */, -60 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, -30 /* lastActivatedTimeOffset */);

@@ -644,6 +766,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedAfterNight_ifOnBeforeNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-120 /* sunsetOffset */, -60 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, -180 /* lastActivatedTimeOffset */);

@@ -659,6 +783,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedAfterNight_ifOnDuringNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-120 /* sunsetOffset */, -60 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, -90 /* lastActivatedTimeOffset */);

@@ -674,6 +800,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedAfterNight_ifOnInFuture_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-120 /* sunsetOffset */, -60 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, 30 /* lastActivatedTimeOffset */);

@@ -689,6 +817,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedBeforeNight_ifOffAfterNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(60 /* sunsetOffset */, 120 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, 180 /* lastActivatedTimeOffset */);

@@ -704,6 +834,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedBeforeNight_ifOffBeforeNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(60 /* sunsetOffset */, 120 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, 30 /* lastActivatedTimeOffset */);

@@ -719,6 +851,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedBeforeNight_ifOffDuringNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(60 /* sunsetOffset */, 120 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, 90 /* lastActivatedTimeOffset */);

@@ -734,6 +868,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedBeforeNight_ifOffInPast_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(60 /* sunsetOffset */, 120 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, -30 /* lastActivatedTimeOffset */);

@@ -749,6 +885,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedBeforeNight_ifOnAfterNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(60 /* sunsetOffset */, 120 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, 180 /* lastActivatedTimeOffset */);

@@ -764,6 +902,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedBeforeNight_ifOnBeforeNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(60 /* sunsetOffset */, 120 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, 30 /* lastActivatedTimeOffset */);

@@ -779,6 +919,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedBeforeNight_ifOnDuringNight_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(60 /* sunsetOffset */, 120 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, 90 /* lastActivatedTimeOffset */);

@@ -794,6 +936,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedBeforeNight_ifOnInPast_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(60 /* sunsetOffset */, 120 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, -30 /* lastActivatedTimeOffset */);

@@ -809,6 +953,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedDuringNight_ifOffAfterNight_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-60 /* sunsetOffset */, 60 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, 90 /* lastActivatedTimeOffset */);

@@ -824,6 +970,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedDuringNight_ifOffBeforeNight_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-60 /* sunsetOffset */, 60 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, -90 /* lastActivatedTimeOffset */);

@@ -839,6 +987,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedDuringNight_ifOffDuringNightInFuture_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-60 /* sunsetOffset */, 60 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, 30 /* lastActivatedTimeOffset */);

@@ -854,6 +1004,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedDuringNight_ifOffDuringNightInPast_turnsOff() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-60 /* sunsetOffset */, 60 /* sunriseOffset */);
        setNightDisplayActivated(false /* activated */, -30 /* lastActivatedTimeOffset */);

@@ -869,6 +1021,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedDuringNight_ifOnAfterNight_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-60 /* sunsetOffset */, 60 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, 90 /* lastActivatedTimeOffset */);

@@ -884,6 +1038,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedDuringNight_ifOnBeforeNight_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-60 /* sunsetOffset */, 60 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, -90 /* lastActivatedTimeOffset */);

@@ -899,6 +1055,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedDuringNight_ifOnDuringNightInFuture_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-60 /* sunsetOffset */, 60 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, 30 /* lastActivatedTimeOffset */);

@@ -914,6 +1072,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void twilightSchedule_whenRebootedDuringNight_ifOnDuringNightInPast_turnsOn() {
+        assumeTrue(mNightDisplayAvailable);
+
        setAutoModeTwilight(-60 /* sunsetOffset */, 60 /* sunriseOffset */);
        setNightDisplayActivated(true /* activated */, -30 /* lastActivatedTimeOffset */);

@@ -929,9 +1089,7 @@ public class ColorDisplayServiceTest {

    @Test
    public void accessibility_colorInversion_transformActivated() {
-        if (!mContext.getResources().getConfiguration().isScreenWideColorGamut()) {
-            return;
-        }
+        assumeTrue(mScreenWideColorGamutAvailable);

        setAccessibilityColorInversion(true);
        setColorMode(ColorDisplayManager.COLOR_MODE_NATURAL);
@@ -944,9 +1102,7 @@ public class ColorDisplayServiceTest {

    @Test
    public void accessibility_colorCorrection_transformActivated() {
-        if (!mContext.getResources().getConfiguration().isScreenWideColorGamut()) {
-            return;
-        }
+        assumeTrue(mScreenWideColorGamutAvailable);

        setAccessibilityColorCorrection(true);
        setColorMode(ColorDisplayManager.COLOR_MODE_NATURAL);
@@ -959,9 +1115,7 @@ public class ColorDisplayServiceTest {

    @Test
    public void accessibility_all_transformActivated() {
-        if (!mContext.getResources().getConfiguration().isScreenWideColorGamut()) {
-            return;
-        }
+        assumeTrue(mScreenWideColorGamutAvailable);

        setAccessibilityColorCorrection(true);
        setAccessibilityColorInversion(true);
@@ -975,9 +1129,7 @@ public class ColorDisplayServiceTest {

    @Test
    public void accessibility_none_transformActivated() {
-        if (!mContext.getResources().getConfiguration().isScreenWideColorGamut()) {
-            return;
-        }
+        assumeTrue(mScreenWideColorGamutAvailable);

        setAccessibilityColorCorrection(false);
        setAccessibilityColorInversion(false);
@@ -990,6 +1142,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void displayWhiteBalance_enabled() {
+        assumeTrue(mDisplayWhiteBalanceAvailable);
+
        setDisplayWhiteBalanceEnabled(true);
        setNightDisplayActivated(false /* activated */, -30 /* lastActivatedTimeOffset */);
        mBinderService.setColorMode(ColorDisplayManager.COLOR_MODE_NATURAL);
@@ -999,6 +1153,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void displayWhiteBalance_disabledAfterNightDisplayEnabled() {
+        assumeTrue(mDisplayWhiteBalanceAvailable);
+
        setDisplayWhiteBalanceEnabled(true);
        startService();
        setNightDisplayActivated(true /* activated */, -30 /* lastActivatedTimeOffset */);
@@ -1011,6 +1167,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void displayWhiteBalance_enabledAfterNightDisplayDisabled() {
+        assumeTrue(mDisplayWhiteBalanceAvailable);
+
        setDisplayWhiteBalanceEnabled(true);
        startService();
        setNightDisplayActivated(true /* activated */, -30 /* lastActivatedTimeOffset */);
@@ -1025,9 +1183,9 @@ public class ColorDisplayServiceTest {

    @Test
    public void displayWhiteBalance_enabledAfterLinearColorModeSelected() {
-        if (!isColorModeValid(ColorDisplayManager.COLOR_MODE_SATURATED)) {
-            return;
-        }
+        assumeTrue(mDisplayWhiteBalanceAvailable);
+        assumeTrue(isColorModeValid(ColorDisplayManager.COLOR_MODE_SATURATED));
+
        setDisplayWhiteBalanceEnabled(true);
        mBinderService.setColorMode(ColorDisplayManager.COLOR_MODE_SATURATED);
        startService();
@@ -1040,6 +1198,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void displayWhiteBalance_disabledWhileAccessibilityColorCorrectionEnabled() {
+        assumeTrue(mDisplayWhiteBalanceAvailable);
+
        setDisplayWhiteBalanceEnabled(true);
        setAccessibilityColorCorrection(true);
        startService();
@@ -1052,6 +1212,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void displayWhiteBalance_disabledWhileAccessibilityColorInversionEnabled() {
+        assumeTrue(mDisplayWhiteBalanceAvailable);
+
        setDisplayWhiteBalanceEnabled(true);
        setAccessibilityColorInversion(true);
        startService();
@@ -1064,6 +1226,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void compositionColorSpaces_noResources() {
+        assumeTrue(mColorModeAvailable);
+
        final DisplayTransformManager dtm = LocalServices.getService(DisplayTransformManager.class);
        reset(dtm);

@@ -1079,6 +1243,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void compositionColorSpaces_invalidResources() {
+        assumeTrue(mColorModeAvailable);
+
        final DisplayTransformManager dtm = LocalServices.getService(DisplayTransformManager.class);
        reset(dtm);

@@ -1100,6 +1266,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void compositionColorSpaces_validResources_validColorMode() {
+        assumeTrue(mColorModeAvailable);
+
        final DisplayTransformManager dtm = LocalServices.getService(DisplayTransformManager.class);
        reset(dtm);

@@ -1119,6 +1287,8 @@ public class ColorDisplayServiceTest {

    @Test
    public void compositionColorSpaces_validResources_invalidColorMode() {
+        assumeTrue(mColorModeAvailable);
+
        final DisplayTransformManager dtm = LocalServices.getService(DisplayTransformManager.class);
        reset(dtm);