Fix PowerManager tests.

import android.os.UserHandle;

import android.platform.test.annotations.FlakyTest;

import android.provider.Settings;

import android.service.dreams.DreamManagerInternal;

import android.test.mock.MockContentResolver;

import android.view.Display;

    @Mock private BatteryManagerInternal mBatteryManagerInternalMock;

    @Mock private ActivityManagerInternal mActivityManagerInternalMock;

    @Mock private AttentionManagerInternal mAttentionManagerInternalMock;

    @Mock private DreamManagerInternal mDreamManagerInternalMock;

    @Mock private PowerManagerService.NativeWrapper mNativeWrapperMock;

    @Mock private Notifier mNotifierMock;

    @Mock private WirelessChargerDetector mWirelessChargerDetectorMock;

        addLocalServiceMock(BatteryManagerInternal.class, mBatteryManagerInternalMock);

        addLocalServiceMock(ActivityManagerInternal.class, mActivityManagerInternalMock);

        addLocalServiceMock(AttentionManagerInternal.class, mAttentionManagerInternalMock);

        addLocalServiceMock(DreamManagerInternal.class, mDreamManagerInternalMock);

        mContextSpy = spy(new ContextWrapper(InstrumentationRegistry.getContext()));

        mResourcesSpy = spy(mContextSpy.getResources());

        int flags = PowerManager.DOZE_WAKE_LOCK;

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

                null /* workSource */, null /* historyTag */);

        when(mDreamManagerInternalMock.isDreaming()).thenReturn(true);

        mService.getBinderServiceInstance().goToSleep(SystemClock.uptimeMillis(),

                PowerManager.GO_TO_SLEEP_REASON_APPLICATION, 0);

        assertThat(mService.getWakefulness()).isEqualTo(WAKEFULNESS_DOZING);

        setAttentiveTimeout(5);

        createService();

        startSystem();

        SystemClock.sleep(8);

        SystemClock.sleep(20);

        assertThat(mService.getWakefulness()).isEqualTo(WAKEFULNESS_ASLEEP);

    }