Merge "add unit tests for screen-on blocker when change screen state" into main

        verify(mHolder.displayPowerState, never()).setScreenState(anyInt());

    }

    @Test

    public void testOffloadBlocker_turnON_screenOnBlocked() {

        // set up.

        int initState = Display.STATE_OFF;

        mHolder = createDisplayPowerController(DISPLAY_ID, UNIQUE_ID);

        mHolder.dpc.setDisplayOffloadSession(mDisplayOffloadSession);

        // start with OFF.

        when(mHolder.displayPowerState.getScreenState()).thenReturn(initState);

        DisplayPowerRequest dpr = new DisplayPowerRequest();

        dpr.policy = DisplayPowerRequest.POLICY_OFF;

        mHolder.dpc.requestPowerState(dpr, /* waitForNegativeProximity= */ false);

        advanceTime(1); // Run updatePowerState

        // go to ON.

        dpr.policy = DisplayPowerRequest.POLICY_BRIGHT;

        mHolder.dpc.requestPowerState(dpr, /* waitForNegativeProximity= */ false);

        advanceTime(1); // Run updatePowerState

        verify(mDisplayOffloadSession).blockScreenOn(any(Runnable.class));

    }

    @Test

    public void testOffloadBlocker_turnOFF_screenOnNotBlocked() {

        // set up.

        int initState = Display.STATE_ON;

        mHolder.dpc.setDisplayOffloadSession(mDisplayOffloadSession);

        // start with ON.

        when(mHolder.displayPowerState.getScreenState()).thenReturn(initState);

        DisplayPowerRequest dpr = new DisplayPowerRequest();

        dpr.policy = DisplayPowerRequest.POLICY_BRIGHT;

        mHolder.dpc.requestPowerState(dpr, /* waitForNegativeProximity= */ false);

        advanceTime(1); // Run updatePowerState

        // go to OFF.

        dpr.policy = DisplayPowerRequest.POLICY_OFF;

        mHolder.dpc.requestPowerState(dpr, /* waitForNegativeProximity= */ false);

        advanceTime(1); // Run updatePowerState

        verify(mDisplayOffloadSession, never()).blockScreenOn(any(Runnable.class));

    }

    @Test

    public void testBrightnessFromOffload() {

        when(mDisplayManagerFlagsMock.isDisplayOffloadEnabled()).thenReturn(true);