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.settings.dream; 18 19 import static com.google.common.truth.Truth.assertThat; 20 21 import static org.mockito.Mockito.spy; 22 import static org.mockito.Mockito.verify; 23 import static org.mockito.Mockito.when; 24 25 import android.content.Context; 26 import android.os.UserManager; 27 28 import androidx.test.core.app.ApplicationProvider; 29 30 import com.android.settings.testutils.FakeFeatureFactory; 31 import com.android.settings.testutils.shadow.SettingsShadowResources; 32 import com.android.settingslib.dream.DreamBackend; 33 34 import org.junit.Before; 35 import org.junit.Test; 36 import org.junit.runner.RunWith; 37 import org.mockito.Answers; 38 import org.mockito.Mock; 39 import org.mockito.MockitoAnnotations; 40 import org.robolectric.RobolectricTestRunner; 41 import org.robolectric.annotation.Config; 42 import org.robolectric.util.ReflectionHelpers; 43 44 @RunWith(RobolectricTestRunner.class) 45 @Config(shadows = SettingsShadowResources.class) 46 public class WhenToDreamPickerTest { 47 48 private WhenToDreamPicker mPicker; 49 @Mock(answer = Answers.RETURNS_DEEP_STUBS) 50 private DreamBackend mBackend; 51 @Mock 52 private UserManager mUserManager; 53 54 @Before setup()55 public void setup() { 56 MockitoAnnotations.initMocks(this); 57 final Context context = spy(ApplicationProvider.getApplicationContext()); 58 59 SettingsShadowResources.overrideResource( 60 com.android.internal.R.bool.config_dreamsEnabledOnBattery, 61 true); 62 63 when(context.getSystemService(Context.USER_SERVICE)).thenReturn(mUserManager); 64 FakeFeatureFactory.setupForTest(); 65 66 mPicker = spy(new WhenToDreamPicker()); 67 when(mPicker.getContext()).thenReturn(context); 68 mPicker.onAttach(context); 69 70 ReflectionHelpers.setField(mPicker, "mBackend", mBackend); 71 } 72 73 @Test getDefaultKeyReturnsCurrentWhenToDreamSetting()74 public void getDefaultKeyReturnsCurrentWhenToDreamSetting() { 75 when(mBackend.getWhenToDreamSetting()).thenReturn(DreamBackend.WHILE_CHARGING); 76 assertThat(mPicker.getDefaultKey()) 77 .isEqualTo(DreamSettings.getKeyFromSetting(DreamBackend.WHILE_CHARGING)); 78 79 when(mBackend.getWhenToDreamSetting()).thenReturn(DreamBackend.WHILE_DOCKED); 80 assertThat(mPicker.getDefaultKey()) 81 .isEqualTo(DreamSettings.getKeyFromSetting(DreamBackend.WHILE_DOCKED)); 82 83 when(mBackend.getWhenToDreamSetting()).thenReturn(DreamBackend.WHILE_CHARGING_OR_DOCKED); 84 assertThat(mPicker.getDefaultKey()) 85 .isEqualTo(DreamSettings.getKeyFromSetting(DreamBackend.WHILE_CHARGING_OR_DOCKED)); 86 87 when(mBackend.getWhenToDreamSetting()).thenReturn(DreamBackend.WHILE_POSTURED); 88 assertThat(mPicker.getDefaultKey()) 89 .isEqualTo(DreamSettings.getKeyFromSetting(DreamBackend.WHILE_POSTURED)); 90 91 when(mBackend.getWhenToDreamSetting()).thenReturn(DreamBackend.NEVER); 92 assertThat(mPicker.getDefaultKey()) 93 .isEqualTo(DreamSettings.getKeyFromSetting(DreamBackend.NEVER)); 94 } 95 96 @Test setDreamWhileCharging()97 public void setDreamWhileCharging() { 98 final String key = DreamSettings.getKeyFromSetting(DreamBackend.WHILE_CHARGING); 99 mPicker.setDefaultKey(key); 100 verify(mBackend).setWhenToDream(DreamBackend.WHILE_CHARGING); 101 } 102 103 @Test setDreamWhileDocked()104 public void setDreamWhileDocked() { 105 final String key = DreamSettings.getKeyFromSetting(DreamBackend.WHILE_DOCKED); 106 mPicker.setDefaultKey(key); 107 verify(mBackend).setWhenToDream(DreamBackend.WHILE_DOCKED); 108 } 109 110 @Test setDreamWhileChargingOrDocked()111 public void setDreamWhileChargingOrDocked() { 112 final String key = DreamSettings.getKeyFromSetting(DreamBackend.WHILE_CHARGING_OR_DOCKED); 113 mPicker.setDefaultKey(key); 114 verify(mBackend).setWhenToDream(DreamBackend.WHILE_CHARGING_OR_DOCKED); 115 } 116 117 @Test setDreamWhilePostured()118 public void setDreamWhilePostured() { 119 final String key = DreamSettings.getKeyFromSetting(DreamBackend.WHILE_POSTURED); 120 mPicker.setDefaultKey(key); 121 verify(mBackend).setWhenToDream(DreamBackend.WHILE_POSTURED); 122 } 123 124 @Test setDreamNever()125 public void setDreamNever() { 126 final String key = DreamSettings.getKeyFromSetting(DreamBackend.NEVER); 127 mPicker.setDefaultKey(key); 128 verify(mBackend).setWhenToDream(DreamBackend.NEVER); 129 } 130 } 131