1 /* 2 * Copyright 2018 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.settings.connecteddevice; 17 18 import static com.android.settings.core.BasePreferenceController.AVAILABLE; 19 import static com.android.settings.core.BasePreferenceController.UNSUPPORTED_ON_DEVICE; 20 import static com.google.common.truth.Truth.assertThat; 21 22 import static org.mockito.Mockito.doReturn; 23 import static org.mockito.Mockito.spy; 24 import static org.mockito.Mockito.verify; 25 import android.arch.lifecycle.LifecycleOwner; 26 import android.content.Context; 27 import android.content.pm.PackageManager; 28 import android.support.v7.preference.PreferenceManager; 29 import com.android.settings.bluetooth.BluetoothDeviceUpdater; 30 import com.android.settings.connecteddevice.dock.DockUpdater; 31 import com.android.settings.dashboard.DashboardFragment; 32 import com.android.settings.testutils.SettingsRobolectricTestRunner; 33 import com.android.settingslib.core.lifecycle.Lifecycle; 34 35 import org.junit.Before; 36 import org.junit.Test; 37 import org.junit.runner.RunWith; 38 import org.mockito.Answers; 39 import org.mockito.Mock; 40 import org.mockito.MockitoAnnotations; 41 import org.robolectric.RuntimeEnvironment; 42 43 @RunWith(SettingsRobolectricTestRunner.class) 44 public class SavedDeviceGroupControllerTest { 45 private static final String PREFERENCE_KEY_1 = "pref_key_1"; 46 47 @Mock 48 private DashboardFragment mDashboardFragment; 49 @Mock 50 private BluetoothDeviceUpdater mBluetoothDeviceUpdater; 51 @Mock 52 private DockUpdater mSavedDockUpdater; 53 @Mock(answer = Answers.RETURNS_DEEP_STUBS) 54 private PreferenceManager mPreferenceManager; 55 @Mock 56 private PackageManager mPackageManager; 57 58 private Context mContext; 59 private SavedDeviceGroupController mSavedDeviceGroupController; 60 private LifecycleOwner mLifecycleOwner; 61 private Lifecycle mLifecycle; 62 63 @Before setUp()64 public void setUp() { 65 MockitoAnnotations.initMocks(this); 66 mContext = spy(RuntimeEnvironment.application); 67 mLifecycleOwner = () -> mLifecycle; 68 mLifecycle = new Lifecycle(mLifecycleOwner); 69 doReturn(mContext).when(mDashboardFragment).getContext(); 70 doReturn(mPackageManager).when(mContext).getPackageManager(); 71 mSavedDeviceGroupController = new SavedDeviceGroupController(mContext); 72 mSavedDeviceGroupController.setBluetoothDeviceUpdater(mBluetoothDeviceUpdater); 73 mSavedDeviceGroupController.setSavedDockUpdater(mSavedDockUpdater); 74 } 75 76 @Test testRegister()77 public void testRegister() { 78 // register the callback in onStart() 79 mSavedDeviceGroupController.onStart(); 80 verify(mBluetoothDeviceUpdater).registerCallback(); 81 verify(mSavedDockUpdater).registerCallback(); 82 } 83 @Test testUnregister()84 public void testUnregister() { 85 // unregister the callback in onStop() 86 mSavedDeviceGroupController.onStop(); 87 verify(mBluetoothDeviceUpdater).unregisterCallback(); 88 verify(mSavedDockUpdater).unregisterCallback(); 89 } 90 @Test testGetAvailabilityStatus_noBluetoothFeature_returnUnSupported()91 public void testGetAvailabilityStatus_noBluetoothFeature_returnUnSupported() { 92 doReturn(false).when(mPackageManager).hasSystemFeature(PackageManager.FEATURE_BLUETOOTH); 93 assertThat(mSavedDeviceGroupController.getAvailabilityStatus()).isEqualTo( 94 UNSUPPORTED_ON_DEVICE); 95 } 96 @Test testGetAvailabilityStatus_BluetoothFeature_returnSupported()97 public void testGetAvailabilityStatus_BluetoothFeature_returnSupported() { 98 doReturn(true).when(mPackageManager).hasSystemFeature(PackageManager.FEATURE_BLUETOOTH); 99 assertThat(mSavedDeviceGroupController.getAvailabilityStatus()).isEqualTo( 100 AVAILABLE); 101 } 102 } 103