Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 8d9ca3b3 authored by Fiona Campbell's avatar Fiona Campbell
Browse files

Sleeplocks test 

- add new test
- amend owners

Bug: 432241094
Flag: com.android.server.power.feature.flags.partial_sleep_wakelocks
Test: atest PowerManagerServiceTest
Change-Id: If06242b4b5fa902d129d887bb54db47a39676486
parent 10aa6d61

core/proto/android/server/OWNERS

+1 −0
Original line number Diff line number Diff line 
per-file activitymanagerservice.proto = file:/services/core/java/com/android/server/am/OWNERS

per-file window*.proto = file:/services/core/java/com/android/server/wm/OWNERS

per-file power*.proto = file:/services/core/java/com/android/server/power/OWNERS

services/tests/powerservicetests/src/com/android/server/power/PowerManagerServiceTest.java

+87 −5
Original line number Diff line number Diff line
@@ -832,10 +832,9 @@ public class PowerManagerServiceTest { 


    @RequiresFlagsEnabled(Flags.FLAG_PARTIAL_SLEEP_WAKELOCKS)

    @Test

    public void testPartialSleepWakelock_multiplePowerGroups() {

    public void testPartialSleepWakelock_multiplePowerGroups_sleepDefault() {

        // setup

        final int nonDefaultPowerGroupId = Display.DEFAULT_DISPLAY_GROUP + 1;

        int displayInNonDefaultGroup = 1;

        final AtomicReference<DisplayManagerInternal.DisplayGroupListener> listener =

                new AtomicReference<>();

        long eventTime1 = 10;
@@ -858,12 +857,12 @@ public class PowerManagerServiceTest { 
        assertThat(mService.getWakefulnessLocked(nonDefaultPowerGroupId))

                .isEqualTo(WAKEFULNESS_AWAKE);



        // Transition default display to doze, and verify the global wakefulness

        mService.setWakefulnessLocked(Display.DEFAULT_DISPLAY_GROUP, WAKEFULNESS_DOZING, eventTime1,

        // Transition default display to sleep, and verify the global wakefulness

        mService.setWakefulnessLocked(Display.DEFAULT_DISPLAY_GROUP, WAKEFULNESS_ASLEEP, eventTime1,

                0, PowerManager.GO_TO_SLEEP_REASON_INATTENTIVE, 0, null, null);

        advanceTime(1000);



        // assert that default group is alseep

        // assert that default group is asleep

        assertWithMessage("Global wakefulness is not awake")

                .that(mService.getGlobalWakefulnessLocked())

                .isEqualTo(WAKEFULNESS_AWAKE);
@@ -906,6 +905,89 @@ public class PowerManagerServiceTest { 
        mService.getBinderServiceInstance().releaseWakeLock(token, /* flags= */ 0);

    }



    @RequiresFlagsEnabled(Flags.FLAG_PARTIAL_SLEEP_WAKELOCKS)

    @Test

    public void testPartialSleepWakelock_multiplePowerGroups_sleepNonDefault() {

        // setup

        final int nonDefaultPowerGroupId = Display.DEFAULT_DISPLAY_GROUP + 1;

        int displayInNonDefaultGroup = 1;

        final AtomicReference<DisplayManagerInternal.DisplayGroupListener> listener =

                new AtomicReference<>();

        long eventTime1 = 10;

        final DisplayInfo infoDefault = new DisplayInfo();

        infoDefault.displayGroupId = Display.DEFAULT_DISPLAY_GROUP;

        final DisplayInfo infoSecondary = new DisplayInfo();

        infoSecondary.displayGroupId = nonDefaultPowerGroupId;

        when(mDisplayManagerInternalMock.getDisplayInfo(Display.DEFAULT_DISPLAY))

                .thenReturn(infoDefault);

        when(mDisplayManagerInternalMock.getDisplayInfo(nonDefaultPowerGroupId))

                .thenReturn(infoSecondary);

        doAnswer((Answer<Void>) invocation -> {

            listener.set(invocation.getArgument(0));

            return null;

        }).when(mDisplayManagerInternalMock).registerDisplayGroupListener(any());



        createService();

        startSystem();

        listener.get().onDisplayGroupAdded(nonDefaultPowerGroupId);



        // Verify all displays are awake.

        assertThat(mService.getGlobalWakefulnessLocked())

                .isEqualTo(WAKEFULNESS_AWAKE);

        assertThat(mService.getWakefulnessLocked(Display.DEFAULT_DISPLAY_GROUP))

                .isEqualTo(WAKEFULNESS_AWAKE);

        assertThat(mService.getWakefulnessLocked(nonDefaultPowerGroupId))

                .isEqualTo(WAKEFULNESS_AWAKE);



        // Transition secondary display to sleep, and verify the global wakefulness

        mService.setWakefulnessLocked(nonDefaultPowerGroupId, WAKEFULNESS_ASLEEP, eventTime1,

                0, PowerManager.GO_TO_SLEEP_REASON_INATTENTIVE, 0, null, null);

        advanceTime(1000);



        // assert that default group is asleep

        assertWithMessage("Global wakefulness is not awake")

                .that(mService.getGlobalWakefulnessLocked())

                .isEqualTo(WAKEFULNESS_AWAKE);

        assertWithMessage("Default group is not awake")

                .that(mService.getWakefulnessLocked(Display.DEFAULT_DISPLAY_GROUP))

                .isEqualTo(WAKEFULNESS_AWAKE);

        assertWithMessage("Secondary group is not asleep")

                .that(mService.getWakefulnessLocked(nonDefaultPowerGroupId))

                .isEqualTo(WAKEFULNESS_ASLEEP);



        // grab sleeplock on non default group

        // Grab a wakelock

        final String tag = "sleeplock1";

        final String packageName = "pkg.name";

        final IBinder token = new Binder();

        final int flags = PowerManager.PARTIAL_SLEEP_WAKE_LOCK;

        mService.getBinderServiceInstance().acquireWakeLock(token, flags, tag, packageName,

                null /* workSource */, null /* historyTag */, displayInNonDefaultGroup,

                null /* callback */);

        advanceTime(1000);



        // try to wake non default group

        mService.getBinderServiceInstance().wakeUpWithDisplayId(mClock.now(),

                PowerManager.WAKE_REASON_WAKE_KEY, "", "",

                displayInNonDefaultGroup);



        advanceTime(1000);





        // verify non default group is still asleep + default display is awake

        assertWithMessage("Global wakefulness is not awake")

                .that(mService.getGlobalWakefulnessLocked())

                .isEqualTo(WAKEFULNESS_AWAKE);

        assertWithMessage("Default group is not awake")

                .that(mService.getWakefulnessLocked(Display.DEFAULT_DISPLAY_GROUP))

                .isEqualTo(WAKEFULNESS_AWAKE);

        assertWithMessage("Secondary group is not asleep")

                .that(mService.getWakefulnessLocked(nonDefaultPowerGroupId))

                .isEqualTo(WAKEFULNESS_ASLEEP);



        mService.getBinderServiceInstance().releaseWakeLock(token, /* flags= */ 0);

    }





    @Test

    public void testWakefulnessSleep_SoftSleepFlag_WithScreenDimWakelock() {