Robert Snoeberger | b300a4e | 2019-02-13 20:13:53 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2019 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 | package com.android.keyguard.clock; |
| 17 | |
| 18 | import static com.google.common.truth.Truth.assertThat; |
| 19 | |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 20 | import static org.mockito.ArgumentMatchers.any; |
Robert Snoeberger | 372e13f | 2019-04-08 17:07:14 -0400 | [diff] [blame] | 21 | import static org.mockito.ArgumentMatchers.anyInt; |
Robert Snoeberger | d8445ea | 2019-03-27 10:41:23 -0400 | [diff] [blame] | 22 | import static org.mockito.Mockito.reset; |
| 23 | import static org.mockito.Mockito.verify; |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 24 | import static org.mockito.Mockito.when; |
| 25 | |
| 26 | import android.content.ContentResolver; |
| 27 | import android.database.ContentObserver; |
Robert Snoeberger | 372e13f | 2019-04-08 17:07:14 -0400 | [diff] [blame] | 28 | import android.net.Uri; |
Robert Snoeberger | b300a4e | 2019-02-13 20:13:53 +0000 | [diff] [blame] | 29 | import android.test.suitebuilder.annotation.SmallTest; |
| 30 | import android.testing.AndroidTestingRunner; |
Robert Snoeberger | b300a4e | 2019-02-13 20:13:53 +0000 | [diff] [blame] | 31 | import android.testing.TestableLooper.RunWithLooper; |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 32 | import android.view.LayoutInflater; |
Robert Snoeberger | b300a4e | 2019-02-13 20:13:53 +0000 | [diff] [blame] | 33 | |
Robert Snoeberger | be2a188 | 2019-04-10 13:48:02 -0400 | [diff] [blame] | 34 | import androidx.lifecycle.MutableLiveData; |
| 35 | |
Robert Snoeberger | b300a4e | 2019-02-13 20:13:53 +0000 | [diff] [blame] | 36 | import com.android.systemui.SysuiTestCase; |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 37 | import com.android.systemui.colorextraction.SysuiColorExtractor; |
Robert Snoeberger | b300a4e | 2019-02-13 20:13:53 +0000 | [diff] [blame] | 38 | import com.android.systemui.dock.DockManager; |
| 39 | import com.android.systemui.dock.DockManagerFake; |
Robert Snoeberger | d8445ea | 2019-03-27 10:41:23 -0400 | [diff] [blame] | 40 | import com.android.systemui.plugins.ClockPlugin; |
Robert Snoeberger | be2a188 | 2019-04-10 13:48:02 -0400 | [diff] [blame] | 41 | import com.android.systemui.settings.CurrentUserObservable; |
Robert Snoeberger | 9ad03f4 | 2019-02-28 14:47:49 -0500 | [diff] [blame] | 42 | import com.android.systemui.shared.plugins.PluginManager; |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 43 | import com.android.systemui.util.InjectionInflationController; |
Robert Snoeberger | b300a4e | 2019-02-13 20:13:53 +0000 | [diff] [blame] | 44 | |
| 45 | import org.junit.After; |
| 46 | import org.junit.Before; |
| 47 | import org.junit.Test; |
| 48 | import org.junit.runner.RunWith; |
Robert Snoeberger | d8445ea | 2019-03-27 10:41:23 -0400 | [diff] [blame] | 49 | import org.mockito.ArgumentCaptor; |
Robert Snoeberger | b300a4e | 2019-02-13 20:13:53 +0000 | [diff] [blame] | 50 | import org.mockito.Mock; |
| 51 | import org.mockito.MockitoAnnotations; |
| 52 | |
| 53 | @SmallTest |
| 54 | @RunWith(AndroidTestingRunner.class) |
Robert Snoeberger | be2a188 | 2019-04-10 13:48:02 -0400 | [diff] [blame] | 55 | // Need to run tests on main looper because LiveData operations such as setData, observe, |
| 56 | // removeObserver cannot be invoked on a background thread. |
| 57 | @RunWithLooper(setAsMainLooper = true) |
Robert Snoeberger | b300a4e | 2019-02-13 20:13:53 +0000 | [diff] [blame] | 58 | public final class ClockManagerTest extends SysuiTestCase { |
| 59 | |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 60 | private static final String BUBBLE_CLOCK = BubbleClockController.class.getName(); |
| 61 | private static final Class<?> BUBBLE_CLOCK_CLASS = BubbleClockController.class; |
Robert Snoeberger | be2a188 | 2019-04-10 13:48:02 -0400 | [diff] [blame] | 62 | private static final int MAIN_USER_ID = 0; |
| 63 | private static final int SECONDARY_USER_ID = 11; |
Robert Snoeberger | 372e13f | 2019-04-08 17:07:14 -0400 | [diff] [blame] | 64 | private static final Uri SETTINGS_URI = null; |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 65 | |
Robert Snoeberger | b300a4e | 2019-02-13 20:13:53 +0000 | [diff] [blame] | 66 | private ClockManager mClockManager; |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 67 | private ContentObserver mContentObserver; |
Robert Snoeberger | b300a4e | 2019-02-13 20:13:53 +0000 | [diff] [blame] | 68 | private DockManagerFake mFakeDockManager; |
Robert Snoeberger | be2a188 | 2019-04-10 13:48:02 -0400 | [diff] [blame] | 69 | private MutableLiveData<Integer> mCurrentUser; |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 70 | @Mock InjectionInflationController mMockInjectionInflationController; |
Robert Snoeberger | 9ad03f4 | 2019-02-28 14:47:49 -0500 | [diff] [blame] | 71 | @Mock PluginManager mMockPluginManager; |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 72 | @Mock SysuiColorExtractor mMockColorExtractor; |
| 73 | @Mock ContentResolver mMockContentResolver; |
Robert Snoeberger | be2a188 | 2019-04-10 13:48:02 -0400 | [diff] [blame] | 74 | @Mock CurrentUserObservable mMockCurrentUserObserable; |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 75 | @Mock SettingsWrapper mMockSettingsWrapper; |
Robert Snoeberger | d8445ea | 2019-03-27 10:41:23 -0400 | [diff] [blame] | 76 | @Mock ClockManager.ClockChangedListener mMockListener1; |
| 77 | @Mock ClockManager.ClockChangedListener mMockListener2; |
Robert Snoeberger | b300a4e | 2019-02-13 20:13:53 +0000 | [diff] [blame] | 78 | |
| 79 | @Before |
| 80 | public void setUp() { |
| 81 | MockitoAnnotations.initMocks(this); |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 82 | |
| 83 | LayoutInflater inflater = LayoutInflater.from(getContext()); |
| 84 | when(mMockInjectionInflationController.injectable(any())).thenReturn(inflater); |
| 85 | |
Robert Snoeberger | b300a4e | 2019-02-13 20:13:53 +0000 | [diff] [blame] | 86 | mFakeDockManager = new DockManagerFake(); |
Robert Snoeberger | 3f4650c | 2019-03-25 14:41:42 -0400 | [diff] [blame] | 87 | |
Robert Snoeberger | be2a188 | 2019-04-10 13:48:02 -0400 | [diff] [blame] | 88 | mCurrentUser = new MutableLiveData<>(); |
| 89 | mCurrentUser.setValue(MAIN_USER_ID); |
| 90 | when(mMockCurrentUserObserable.getCurrentUser()).thenReturn(mCurrentUser); |
| 91 | |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 92 | mClockManager = new ClockManager(getContext(), mMockInjectionInflationController, |
Robert Snoeberger | 3f4650c | 2019-03-25 14:41:42 -0400 | [diff] [blame] | 93 | mMockPluginManager, mMockColorExtractor, mMockContentResolver, |
Lucas Dupin | d93e95f | 2019-05-01 00:49:35 +0000 | [diff] [blame] | 94 | mMockCurrentUserObserable, mMockSettingsWrapper, mFakeDockManager); |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 95 | |
Robert Snoeberger | d8445ea | 2019-03-27 10:41:23 -0400 | [diff] [blame] | 96 | mClockManager.addOnClockChangedListener(mMockListener1); |
| 97 | mClockManager.addOnClockChangedListener(mMockListener2); |
| 98 | reset(mMockListener1, mMockListener2); |
| 99 | |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 100 | mContentObserver = mClockManager.getContentObserver(); |
Robert Snoeberger | b300a4e | 2019-02-13 20:13:53 +0000 | [diff] [blame] | 101 | } |
| 102 | |
| 103 | @After |
| 104 | public void tearDown() { |
Robert Snoeberger | d8445ea | 2019-03-27 10:41:23 -0400 | [diff] [blame] | 105 | mClockManager.removeOnClockChangedListener(mMockListener1); |
| 106 | mClockManager.removeOnClockChangedListener(mMockListener2); |
Robert Snoeberger | b300a4e | 2019-02-13 20:13:53 +0000 | [diff] [blame] | 107 | } |
| 108 | |
| 109 | @Test |
| 110 | public void dockEvent() { |
| 111 | mFakeDockManager.setDockEvent(DockManager.STATE_DOCKED); |
| 112 | assertThat(mClockManager.isDocked()).isTrue(); |
| 113 | } |
| 114 | |
| 115 | @Test |
| 116 | public void undockEvent() { |
| 117 | mFakeDockManager.setDockEvent(DockManager.STATE_NONE); |
| 118 | assertThat(mClockManager.isDocked()).isFalse(); |
| 119 | } |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 120 | |
| 121 | @Test |
| 122 | public void getCurrentClock_default() { |
| 123 | // GIVEN that settings doesn't contain any values |
Robert Snoeberger | 372e13f | 2019-04-08 17:07:14 -0400 | [diff] [blame] | 124 | when(mMockSettingsWrapper.getLockScreenCustomClockFace(anyInt())).thenReturn(null); |
| 125 | when(mMockSettingsWrapper.getDockedClockFace(anyInt())).thenReturn(null); |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 126 | // WHEN settings change event is fired |
Robert Snoeberger | be2a188 | 2019-04-10 13:48:02 -0400 | [diff] [blame] | 127 | mContentObserver.onChange(false, SETTINGS_URI, MAIN_USER_ID); |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 128 | // THEN the result is null, indicated the default clock face should be used. |
| 129 | assertThat(mClockManager.getCurrentClock()).isNull(); |
| 130 | } |
| 131 | |
| 132 | @Test |
| 133 | public void getCurrentClock_customClock() { |
| 134 | // GIVEN that settings is set to the bubble clock face |
Robert Snoeberger | 372e13f | 2019-04-08 17:07:14 -0400 | [diff] [blame] | 135 | when(mMockSettingsWrapper.getLockScreenCustomClockFace(anyInt())).thenReturn(BUBBLE_CLOCK); |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 136 | // WHEN settings change event is fired |
Robert Snoeberger | be2a188 | 2019-04-10 13:48:02 -0400 | [diff] [blame] | 137 | mContentObserver.onChange(false, SETTINGS_URI, MAIN_USER_ID); |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 138 | // THEN the plugin is the bubble clock face. |
| 139 | assertThat(mClockManager.getCurrentClock()).isInstanceOf(BUBBLE_CLOCK_CLASS); |
| 140 | } |
| 141 | |
| 142 | @Test |
Robert Snoeberger | d8445ea | 2019-03-27 10:41:23 -0400 | [diff] [blame] | 143 | public void onClockChanged_customClock() { |
| 144 | // GIVEN that settings is set to the bubble clock face |
Robert Snoeberger | 372e13f | 2019-04-08 17:07:14 -0400 | [diff] [blame] | 145 | when(mMockSettingsWrapper.getLockScreenCustomClockFace(anyInt())).thenReturn(BUBBLE_CLOCK); |
Robert Snoeberger | d8445ea | 2019-03-27 10:41:23 -0400 | [diff] [blame] | 146 | // WHEN settings change event is fired |
Robert Snoeberger | be2a188 | 2019-04-10 13:48:02 -0400 | [diff] [blame] | 147 | mContentObserver.onChange(false, SETTINGS_URI, MAIN_USER_ID); |
Robert Snoeberger | d8445ea | 2019-03-27 10:41:23 -0400 | [diff] [blame] | 148 | // THEN the plugin is the bubble clock face. |
| 149 | ArgumentCaptor<ClockPlugin> captor = ArgumentCaptor.forClass(ClockPlugin.class); |
| 150 | verify(mMockListener1).onClockChanged(captor.capture()); |
| 151 | assertThat(captor.getValue()).isInstanceOf(BUBBLE_CLOCK_CLASS); |
| 152 | } |
| 153 | |
| 154 | @Test |
| 155 | public void onClockChanged_uniqueInstances() { |
| 156 | // GIVEN that settings is set to the bubble clock face |
Robert Snoeberger | 372e13f | 2019-04-08 17:07:14 -0400 | [diff] [blame] | 157 | when(mMockSettingsWrapper.getLockScreenCustomClockFace(anyInt())).thenReturn(BUBBLE_CLOCK); |
Robert Snoeberger | d8445ea | 2019-03-27 10:41:23 -0400 | [diff] [blame] | 158 | // WHEN settings change event is fired |
Robert Snoeberger | be2a188 | 2019-04-10 13:48:02 -0400 | [diff] [blame] | 159 | mContentObserver.onChange(false, SETTINGS_URI, MAIN_USER_ID); |
Robert Snoeberger | d8445ea | 2019-03-27 10:41:23 -0400 | [diff] [blame] | 160 | // THEN the listeners receive separate instances of the Bubble clock plugin. |
| 161 | ArgumentCaptor<ClockPlugin> captor1 = ArgumentCaptor.forClass(ClockPlugin.class); |
| 162 | ArgumentCaptor<ClockPlugin> captor2 = ArgumentCaptor.forClass(ClockPlugin.class); |
| 163 | verify(mMockListener1).onClockChanged(captor1.capture()); |
| 164 | verify(mMockListener2).onClockChanged(captor2.capture()); |
| 165 | assertThat(captor1.getValue()).isInstanceOf(BUBBLE_CLOCK_CLASS); |
| 166 | assertThat(captor2.getValue()).isInstanceOf(BUBBLE_CLOCK_CLASS); |
| 167 | assertThat(captor1.getValue()).isNotSameAs(captor2.getValue()); |
| 168 | } |
| 169 | |
| 170 | @Test |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 171 | public void getCurrentClock_badSettingsValue() { |
| 172 | // GIVEN that settings contains a value that doesn't correspond to a |
| 173 | // custom clock face. |
Robert Snoeberger | 372e13f | 2019-04-08 17:07:14 -0400 | [diff] [blame] | 174 | when(mMockSettingsWrapper.getLockScreenCustomClockFace(anyInt())).thenReturn("bad value"); |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 175 | // WHEN settings change event is fired |
Robert Snoeberger | be2a188 | 2019-04-10 13:48:02 -0400 | [diff] [blame] | 176 | mContentObserver.onChange(false, SETTINGS_URI, MAIN_USER_ID); |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 177 | // THEN the result is null. |
| 178 | assertThat(mClockManager.getCurrentClock()).isNull(); |
| 179 | } |
| 180 | |
| 181 | @Test |
| 182 | public void getCurrentClock_dockedDefault() { |
| 183 | // WHEN dock event is fired |
| 184 | mFakeDockManager.setDockEvent(DockManager.STATE_DOCKED); |
| 185 | // THEN the result is null, indicating the default clock face. |
| 186 | assertThat(mClockManager.getCurrentClock()).isNull(); |
| 187 | } |
| 188 | |
| 189 | @Test |
| 190 | public void getCurrentClock_dockedCustomClock() { |
| 191 | // GIVEN settings is set to the bubble clock face |
Robert Snoeberger | 372e13f | 2019-04-08 17:07:14 -0400 | [diff] [blame] | 192 | when(mMockSettingsWrapper.getDockedClockFace(anyInt())).thenReturn(BUBBLE_CLOCK); |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 193 | // WHEN dock event fires |
| 194 | mFakeDockManager.setDockEvent(DockManager.STATE_DOCKED); |
| 195 | // THEN the plugin is the bubble clock face. |
| 196 | assertThat(mClockManager.getCurrentClock()).isInstanceOf(BUBBLE_CLOCK_CLASS); |
| 197 | } |
| 198 | |
| 199 | @Test |
| 200 | public void getCurrentClock_badDockedSettingsValue() { |
| 201 | // GIVEN settings contains a value that doesn't correspond to an available clock face. |
Robert Snoeberger | 372e13f | 2019-04-08 17:07:14 -0400 | [diff] [blame] | 202 | when(mMockSettingsWrapper.getDockedClockFace(anyInt())).thenReturn("bad value"); |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 203 | // WHEN dock event fires |
| 204 | mFakeDockManager.setDockEvent(DockManager.STATE_DOCKED); |
| 205 | // THEN the result is null. |
| 206 | assertThat(mClockManager.getCurrentClock()).isNull(); |
| 207 | } |
| 208 | |
| 209 | @Test |
| 210 | public void getCurrentClock_badDockedSettingsFallback() { |
| 211 | // GIVEN settings contains a value that doesn't correspond to an available clock face, but |
| 212 | // locked screen settings is set to bubble clock. |
Robert Snoeberger | 372e13f | 2019-04-08 17:07:14 -0400 | [diff] [blame] | 213 | when(mMockSettingsWrapper.getDockedClockFace(anyInt())).thenReturn("bad value"); |
| 214 | when(mMockSettingsWrapper.getLockScreenCustomClockFace(anyInt())).thenReturn(BUBBLE_CLOCK); |
Robert Snoeberger | 71e5079 | 2019-02-15 15:48:01 -0500 | [diff] [blame] | 215 | // WHEN dock event is fired |
| 216 | mFakeDockManager.setDockEvent(DockManager.STATE_DOCKED); |
| 217 | // THEN the plugin is the bubble clock face. |
| 218 | assertThat(mClockManager.getCurrentClock()).isInstanceOf(BUBBLE_CLOCK_CLASS); |
| 219 | } |
Robert Snoeberger | be2a188 | 2019-04-10 13:48:02 -0400 | [diff] [blame] | 220 | |
| 221 | @Test |
| 222 | public void onUserChanged_defaultClock() { |
| 223 | // WHEN the user changes |
| 224 | mCurrentUser.setValue(SECONDARY_USER_ID); |
| 225 | // THEN the plugin is null for the default clock face |
| 226 | assertThat(mClockManager.getCurrentClock()).isNull(); |
| 227 | } |
| 228 | |
| 229 | @Test |
| 230 | public void onUserChanged_customClock() { |
| 231 | // GIVEN that a second user has selected the bubble clock face |
| 232 | when(mMockSettingsWrapper.getLockScreenCustomClockFace(SECONDARY_USER_ID)).thenReturn( |
| 233 | BUBBLE_CLOCK); |
| 234 | // WHEN the user changes |
| 235 | mCurrentUser.setValue(SECONDARY_USER_ID); |
| 236 | // THEN the plugin is the bubble clock face. |
| 237 | assertThat(mClockManager.getCurrentClock()).isInstanceOf(BUBBLE_CLOCK_CLASS); |
| 238 | } |
| 239 | |
| 240 | @Test |
| 241 | public void onUserChanged_docked() { |
| 242 | // GIVEN device is docked |
| 243 | mFakeDockManager.setDockEvent(DockManager.STATE_DOCKED); |
| 244 | // AND the second user as selected the bubble clock for the dock |
| 245 | when(mMockSettingsWrapper.getDockedClockFace(SECONDARY_USER_ID)).thenReturn(BUBBLE_CLOCK); |
| 246 | // WHEN the user changes |
| 247 | mCurrentUser.setValue(SECONDARY_USER_ID); |
| 248 | // THEN the plugin is the bubble clock face. |
| 249 | assertThat(mClockManager.getCurrentClock()).isInstanceOf(BUBBLE_CLOCK_CLASS); |
| 250 | } |
Robert Snoeberger | b300a4e | 2019-02-13 20:13:53 +0000 | [diff] [blame] | 251 | } |