1 /*
2  * Copyright (C) 2022 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.dreams;
18 
19 import static org.mockito.ArgumentMatchers.any;
20 import static org.mockito.Mockito.never;
21 import static org.mockito.Mockito.times;
22 import static org.mockito.Mockito.verify;
23 
24 import android.testing.AndroidTestingRunner;
25 
26 import androidx.test.filters.SmallTest;
27 
28 import com.android.systemui.SysuiTestCase;
29 
30 import org.junit.Before;
31 import org.junit.Test;
32 import org.junit.runner.RunWith;
33 import org.mockito.Mock;
34 import org.mockito.MockitoAnnotations;
35 
36 import java.util.List;
37 import java.util.concurrent.Executor;
38 
39 @SmallTest
40 @RunWith(AndroidTestingRunner.class)
41 public class DreamOverlayStatusBarItemsProviderTest extends SysuiTestCase {
42     @Mock
43     DreamOverlayStatusBarItemsProvider.Callback mCallback;
44     @Mock
45     DreamOverlayStatusBarItemsProvider.StatusBarItem mStatusBarItem;
46 
47     private final Executor mMainExecutor = Runnable::run;
48 
49     DreamOverlayStatusBarItemsProvider mProvider;
50 
51     @Before
setup()52     public void setup() {
53         MockitoAnnotations.initMocks(this);
54         mProvider = new DreamOverlayStatusBarItemsProvider(mMainExecutor);
55     }
56 
57     @Test
addingCallbackCallsOnStatusBarItemsChanged()58     public void addingCallbackCallsOnStatusBarItemsChanged() {
59         mProvider.addStatusBarItem(mStatusBarItem);
60         mProvider.addCallback(mCallback);
61         verify(mCallback).onStatusBarItemsChanged(List.of(mStatusBarItem));
62     }
63 
64     @Test
addingStatusBarItemCallsOnStatusBarItemsChanged()65     public void addingStatusBarItemCallsOnStatusBarItemsChanged() {
66         mProvider.addCallback(mCallback);
67         mProvider.addStatusBarItem(mStatusBarItem);
68         verify(mCallback).onStatusBarItemsChanged(List.of(mStatusBarItem));
69     }
70 
71     @Test
addingDuplicateStatusBarItemDoesNotCallOnStatusBarItemsChanged()72     public void addingDuplicateStatusBarItemDoesNotCallOnStatusBarItemsChanged() {
73         mProvider.addCallback(mCallback);
74         mProvider.addStatusBarItem(mStatusBarItem);
75         mProvider.addStatusBarItem(mStatusBarItem);
76         // Called only once for addStatusBarItem.
77         verify(mCallback, times(1))
78                 .onStatusBarItemsChanged(List.of(mStatusBarItem));
79     }
80 
81     @Test
removingStatusBarItemCallsOnStatusBarItemsChanged()82     public void removingStatusBarItemCallsOnStatusBarItemsChanged() {
83         mProvider.addCallback(mCallback);
84         mProvider.addStatusBarItem(mStatusBarItem);
85         mProvider.removeStatusBarItem(mStatusBarItem);
86         // Called once for addStatusBarItem and once for removeStatusBarItem.
87         verify(mCallback, times(2)).onStatusBarItemsChanged(any());
88     }
89 
90     @Test
removingNonexistentStatusBarItemDoesNotCallOnStatusBarItemsChanged()91     public void removingNonexistentStatusBarItemDoesNotCallOnStatusBarItemsChanged() {
92         mProvider.addCallback(mCallback);
93         mProvider.removeStatusBarItem(mStatusBarItem);
94         verify(mCallback, never()).onStatusBarItemsChanged(any());
95     }
96 }
97