Home | History | Annotate | Download | only in phone 
      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.statusbar.phone;
     18 
     19 import static org.mockito.ArgumentMatchers.any;
     20 import static org.mockito.ArgumentMatchers.eq;
     21 import static org.mockito.Mockito.doAnswer;
     22 import static org.mockito.Mockito.mock;
     23 import static org.mockito.Mockito.never;
     24 import static org.mockito.Mockito.verify;
     25 
     26 import android.os.Debug;
     27 import android.support.test.filters.SmallTest;
     28 import android.testing.AndroidTestingRunner;
     29 import android.testing.TestableLooper;
     30 
     31 import com.android.systemui.SysuiTestCase;
     32 import com.android.systemui.doze.DozeHost;
     33 
     34 import org.junit.Before;
     35 import org.junit.Test;
     36 import org.junit.runner.RunWith;
     37 import org.mockito.Mock;
     38 import org.mockito.MockitoAnnotations;
     39 
     40 @RunWith(AndroidTestingRunner.class)
     41 @TestableLooper.RunWithLooper
     42 @SmallTest
     43 public class DozeScrimControllerTest extends SysuiTestCase {
     44 
     45     @Mock
     46     private ScrimController mScrimController;
     47     @Mock
     48     private DozeParameters mDozeParameters;
     49     private DozeScrimController mDozeScrimController;
     50 
     51     @Before
     52     public void setup() {
     53         MockitoAnnotations.initMocks(this);
     54         // Make sure callbacks will be invoked to complete the lifecycle.
     55         doAnswer(invocationOnMock -> {
     56             ScrimController.Callback callback = invocationOnMock.getArgument(1);
     57             callback.onStart();
     58             callback.onDisplayBlanked();
     59             callback.onFinished();
     60             return null;
     61         }).when(mScrimController).transitionTo(any(ScrimState.class),
     62                 any(ScrimController.Callback.class));
     63 
     64         mDozeScrimController = new DozeScrimController(mScrimController, getContext(),
     65                 mDozeParameters);
     66         mDozeScrimController.setDozing(true);
     67     }
     68 
     69     @Test
     70     public void changesScrimControllerState() {
     71         mDozeScrimController.pulse(mock(DozeHost.PulseCallback.class), 0);
     72         verify(mScrimController).transitionTo(eq(ScrimState.PULSING),
     73                 any(ScrimController.Callback.class));
     74     }
     75 
     76     @Test
     77     public void callsPulseCallback() {
     78         DozeHost.PulseCallback callback = mock(DozeHost.PulseCallback.class);
     79         mDozeScrimController.pulse(callback, 0);
     80 
     81         verify(callback).onPulseStarted();
     82         mDozeScrimController.pulseOutNow();
     83         verify(callback).onPulseFinished();
     84     }
     85 
     86     @Test
     87     public void secondPulseIsSuppressed() {
     88         DozeHost.PulseCallback callback1 = mock(DozeHost.PulseCallback.class);
     89         DozeHost.PulseCallback callback2 = mock(DozeHost.PulseCallback.class);
     90         mDozeScrimController.pulse(callback1, 0);
     91         mDozeScrimController.pulse(callback2, 0);
     92 
     93         verify(callback1, never()).onPulseFinished();
     94         verify(callback2).onPulseFinished();
     95     }
     96 
     97     @Test
     98     public void suppressesPulseIfNotDozing() {
     99         mDozeScrimController.setDozing(false);
    100         DozeHost.PulseCallback callback = mock(DozeHost.PulseCallback.class);
    101         mDozeScrimController.pulse(callback, 0);
    102 
    103         verify(callback).onPulseFinished();
    104     }
    105 }
    106