Home | History | Annotate | Download | only in doze 
      1 /*
      2  * Copyright (C) 2017 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 
     17 package com.android.systemui.doze;
     18 
     19 import static com.android.systemui.doze.DozeMachine.State.DOZE_AOD;
     20 import static com.android.systemui.doze.DozeMachine.State.DOZE_AOD_PAUSED;
     21 import static com.android.systemui.doze.DozeMachine.State.INITIALIZED;
     22 import static com.android.systemui.doze.DozeMachine.State.UNINITIALIZED;
     23 
     24 import static org.mockito.ArgumentMatchers.any;
     25 import static org.mockito.ArgumentMatchers.anyInt;
     26 import static org.mockito.ArgumentMatchers.anyLong;
     27 import static org.mockito.ArgumentMatchers.eq;
     28 import static org.mockito.Mockito.clearInvocations;
     29 import static org.mockito.Mockito.mock;
     30 import static org.mockito.Mockito.verify;
     31 
     32 import android.app.AlarmManager;
     33 import android.content.Context;
     34 import android.os.Handler;
     35 import android.os.HandlerThread;
     36 import android.support.test.InstrumentationRegistry;
     37 import android.support.test.filters.SmallTest;
     38 import android.support.test.runner.AndroidJUnit4;
     39 
     40 import com.android.systemui.SysuiTestCase;
     41 import com.android.systemui.util.wakelock.WakeLockFake;
     42 
     43 import org.junit.After;
     44 import org.junit.Before;
     45 import org.junit.Test;
     46 import org.junit.runner.RunWith;
     47 
     48 @RunWith(AndroidJUnit4.class)
     49 @SmallTest
     50 public class DozeUiTest extends SysuiTestCase {
     51 
     52     private AlarmManager mAlarmManager;
     53     private DozeMachine mMachine;
     54     private WakeLockFake mWakeLock;
     55     private DozeHostFake mHost;
     56     private Handler mHandler;
     57     private HandlerThread mHandlerThread;
     58     private DozeUi mDozeUi;
     59 
     60     @Before
     61     public void setUp() throws Exception {
     62         mHandlerThread = new HandlerThread("DozeUiTest");
     63         mHandlerThread.start();
     64         mAlarmManager = mock(AlarmManager.class);
     65         mMachine = mock(DozeMachine.class);
     66         mWakeLock = new WakeLockFake();
     67         mHost = new DozeHostFake();
     68         mHandler = mHandlerThread.getThreadHandler();
     69 
     70         mDozeUi = new DozeUi(mContext, mAlarmManager, mMachine, mWakeLock, mHost, mHandler);
     71     }
     72 
     73     @After
     74     public void tearDown() throws Exception {
     75         mHandlerThread.quit();
     76         mHandler = null;
     77         mHandlerThread = null;
     78     }
     79 
     80     @Test
     81     public void pausingAndUnpausingAod_registersTimeTickAfterUnpausing() throws Exception {
     82         mDozeUi.transitionTo(UNINITIALIZED, INITIALIZED);
     83         mDozeUi.transitionTo(INITIALIZED, DOZE_AOD);
     84         mDozeUi.transitionTo(DOZE_AOD, DOZE_AOD_PAUSED);
     85 
     86         clearInvocations(mAlarmManager);
     87 
     88         mDozeUi.transitionTo(DOZE_AOD_PAUSED, DOZE_AOD);
     89 
     90         verify(mAlarmManager).setExact(anyInt(), anyLong(), eq("doze_time_tick"), any(), any());
     91     }
     92 }