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.CONDITIONALLY_UNAVAILABLE; 20 21 import static com.google.common.truth.Truth.assertThat; 22 23 import static org.mockito.Mockito.doReturn; 24 import static org.mockito.Mockito.spy; 25 import static org.mockito.Mockito.verify; 26 import static org.mockito.Mockito.when; 27 28 import android.bluetooth.BluetoothAdapter; 29 import android.bluetooth.BluetoothDevice; 30 import android.content.Context; 31 import android.content.pm.PackageManager; 32 import android.graphics.drawable.Drawable; 33 import android.util.Pair; 34 35 import androidx.preference.Preference; 36 import androidx.preference.PreferenceCategory; 37 import androidx.preference.PreferenceGroup; 38 import androidx.preference.PreferenceManager; 39 40 import com.android.settings.R; 41 import com.android.settings.bluetooth.BluetoothDevicePreference; 42 import com.android.settings.bluetooth.BluetoothDeviceUpdater; 43 import com.android.settings.connecteddevice.dock.DockUpdater; 44 import com.android.settings.dashboard.DashboardFragment; 45 import com.android.settings.testutils.shadow.ShadowBluetoothAdapter; 46 import com.android.settings.widget.SingleTargetGearPreference; 47 import com.android.settingslib.bluetooth.CachedBluetoothDevice; 48 49 import org.junit.Before; 50 import org.junit.Test; 51 import org.junit.runner.RunWith; 52 import org.mockito.Mock; 53 import org.mockito.MockitoAnnotations; 54 import org.robolectric.RobolectricTestRunner; 55 import org.robolectric.RuntimeEnvironment; 56 import org.robolectric.annotation.Config; 57 import org.robolectric.shadow.api.Shadow; 58 59 import java.util.ArrayList; 60 import java.util.List; 61 62 @RunWith(RobolectricTestRunner.class) 63 @Config(shadows = ShadowBluetoothAdapter.class) 64 public class PreviouslyConnectedDevicePreferenceControllerTest { 65 66 private static final String KEY = "test_key"; 67 private static final String FAKE_ADDRESS_1 = "AA:AA:AA:AA:AA:01"; 68 private static final String FAKE_ADDRESS_2 = "AA:AA:AA:AA:AA:02"; 69 private static final String FAKE_ADDRESS_3 = "AA:AA:AA:AA:AA:03"; 70 private static final String FAKE_ADDRESS_4 = "AA:AA:AA:AA:AA:04"; 71 private static final String FAKE_ADDRESS_5 = "AA:AA:AA:AA:AA:05"; 72 73 @Mock 74 private DashboardFragment mDashboardFragment; 75 @Mock 76 private BluetoothDeviceUpdater mBluetoothDeviceUpdater; 77 @Mock 78 private DockUpdater mDockUpdater; 79 @Mock 80 private PackageManager mPackageManager; 81 @Mock 82 private PreferenceManager mPreferenceManager; 83 @Mock 84 private Preference mSeeAllPreference; 85 @Mock 86 private CachedBluetoothDevice mCachedDevice1; 87 @Mock 88 private CachedBluetoothDevice mCachedDevice2; 89 @Mock 90 private CachedBluetoothDevice mCachedDevice3; 91 @Mock 92 private CachedBluetoothDevice mCachedDevice4; 93 @Mock 94 private CachedBluetoothDevice mCachedDevice5; 95 @Mock 96 private BluetoothDevice mBluetoothDevice1; 97 @Mock 98 private BluetoothDevice mBluetoothDevice2; 99 @Mock 100 private BluetoothDevice mBluetoothDevice3; 101 @Mock 102 private BluetoothDevice mBluetoothDevice4; 103 @Mock 104 private BluetoothDevice mBluetoothDevice5; 105 @Mock 106 private Drawable mDrawable; 107 108 private Context mContext; 109 private PreviouslyConnectedDevicePreferenceController mPreConnectedDeviceController; 110 private PreferenceGroup mPreferenceGroup; 111 private ShadowBluetoothAdapter mShadowBluetoothAdapter; 112 113 @Before setUp()114 public void setUp() { 115 MockitoAnnotations.initMocks(this); 116 Pair<Drawable, String> pairs = new Pair<>(mDrawable, "fake_device"); 117 mContext = spy(RuntimeEnvironment.application); 118 doReturn(mContext).when(mDashboardFragment).getContext(); 119 doReturn(mPackageManager).when(mContext).getPackageManager(); 120 mPreConnectedDeviceController = 121 new PreviouslyConnectedDevicePreferenceController(mContext, KEY); 122 mPreConnectedDeviceController.setBluetoothDeviceUpdater(mBluetoothDeviceUpdater); 123 mPreConnectedDeviceController.setSavedDockUpdater(mDockUpdater); 124 mShadowBluetoothAdapter = Shadow.extract(BluetoothAdapter.getDefaultAdapter()); 125 126 when(mCachedDevice1.getDevice()).thenReturn(mBluetoothDevice1); 127 when(mCachedDevice1.getAddress()).thenReturn(FAKE_ADDRESS_1); 128 when(mCachedDevice1.getDrawableWithDescription()).thenReturn(pairs); 129 when(mCachedDevice2.getDevice()).thenReturn(mBluetoothDevice2); 130 when(mCachedDevice2.getAddress()).thenReturn(FAKE_ADDRESS_2); 131 when(mCachedDevice2.getDrawableWithDescription()).thenReturn(pairs); 132 when(mCachedDevice3.getDevice()).thenReturn(mBluetoothDevice3); 133 when(mCachedDevice3.getAddress()).thenReturn(FAKE_ADDRESS_3); 134 when(mCachedDevice3.getDrawableWithDescription()).thenReturn(pairs); 135 when(mCachedDevice4.getDevice()).thenReturn(mBluetoothDevice4); 136 when(mCachedDevice4.getAddress()).thenReturn(FAKE_ADDRESS_4); 137 when(mCachedDevice4.getDrawableWithDescription()).thenReturn(pairs); 138 when(mCachedDevice5.getDevice()).thenReturn(mBluetoothDevice5); 139 when(mCachedDevice5.getAddress()).thenReturn(FAKE_ADDRESS_5); 140 when(mCachedDevice5.getDrawableWithDescription()).thenReturn(pairs); 141 142 final List<BluetoothDevice> mMostRecentlyConnectedDevices = new ArrayList<>(); 143 mMostRecentlyConnectedDevices.add(mBluetoothDevice1); 144 mMostRecentlyConnectedDevices.add(mBluetoothDevice2); 145 mMostRecentlyConnectedDevices.add(mBluetoothDevice4); 146 mMostRecentlyConnectedDevices.add(mBluetoothDevice3); 147 mShadowBluetoothAdapter.setMostRecentlyConnectedDevices(mMostRecentlyConnectedDevices); 148 149 mPreferenceGroup = spy(new PreferenceCategory(mContext)); 150 doReturn(mPreferenceManager).when(mPreferenceGroup).getPreferenceManager(); 151 mPreferenceGroup.setVisible(false); 152 mPreConnectedDeviceController.setPreferenceGroup(mPreferenceGroup); 153 mPreConnectedDeviceController.mSeeAllPreference = mSeeAllPreference; 154 } 155 156 @Test onStart_registerCallback()157 public void onStart_registerCallback() { 158 // register the callback in onStart() 159 mPreConnectedDeviceController.onStart(); 160 161 verify(mBluetoothDeviceUpdater).registerCallback(); 162 verify(mDockUpdater).registerCallback(); 163 verify(mContext).registerReceiver(mPreConnectedDeviceController.mReceiver, 164 mPreConnectedDeviceController.mIntentFilter, Context.RECEIVER_EXPORTED_UNAUDITED); 165 verify(mBluetoothDeviceUpdater).refreshPreference(); 166 } 167 168 @Test onStop_unregisterCallback()169 public void onStop_unregisterCallback() { 170 // register it first 171 mContext.registerReceiver(mPreConnectedDeviceController.mReceiver, null, 172 Context.RECEIVER_EXPORTED/*UNAUDITED*/); 173 174 // unregister the callback in onStop() 175 mPreConnectedDeviceController.onStop(); 176 177 verify(mBluetoothDeviceUpdater).unregisterCallback(); 178 verify(mDockUpdater).unregisterCallback(); 179 verify(mContext).unregisterReceiver(mPreConnectedDeviceController.mReceiver); 180 } 181 182 @Test getAvailabilityStatus_noBluetoothDockFeature_returnUnSupported()183 public void getAvailabilityStatus_noBluetoothDockFeature_returnUnSupported() { 184 doReturn(false).when(mPackageManager).hasSystemFeature(PackageManager.FEATURE_BLUETOOTH); 185 mPreConnectedDeviceController.setSavedDockUpdater(null); 186 187 assertThat(mPreConnectedDeviceController.getAvailabilityStatus()).isEqualTo( 188 CONDITIONALLY_UNAVAILABLE); 189 } 190 191 @Test getAvailabilityStatus_hasBluetoothFeature_returnSupported()192 public void getAvailabilityStatus_hasBluetoothFeature_returnSupported() { 193 doReturn(true).when(mPackageManager).hasSystemFeature(PackageManager.FEATURE_BLUETOOTH); 194 mPreConnectedDeviceController.setSavedDockUpdater(null); 195 196 assertThat(mPreConnectedDeviceController.getAvailabilityStatus()).isEqualTo( 197 AVAILABLE); 198 } 199 200 @Test getAvailabilityStatus_haveDockFeature_returnSupported()201 public void getAvailabilityStatus_haveDockFeature_returnSupported() { 202 doReturn(false).when(mPackageManager).hasSystemFeature(PackageManager.FEATURE_BLUETOOTH); 203 204 assertThat(mPreConnectedDeviceController.getAvailabilityStatus()).isEqualTo( 205 AVAILABLE); 206 } 207 208 @Test onDeviceAdded_addDevicePreference_displayIt()209 public void onDeviceAdded_addDevicePreference_displayIt() { 210 final BluetoothDevicePreference preference1 = new BluetoothDevicePreference( 211 mContext, mCachedDevice1, true, BluetoothDevicePreference.SortType.TYPE_NO_SORT); 212 213 mPreConnectedDeviceController.onDeviceAdded(preference1); 214 215 assertThat(mPreferenceGroup.getPreferenceCount()).isEqualTo(2); 216 } 217 218 @Test onDeviceAdded_addDockDevicePreference_displayIt()219 public void onDeviceAdded_addDockDevicePreference_displayIt() { 220 final SingleTargetGearPreference dockPreference = new SingleTargetGearPreference( 221 mContext, null /* AttributeSet */); 222 223 mPreConnectedDeviceController.onDeviceAdded(dockPreference); 224 225 assertThat(mPreferenceGroup.getPreferenceCount()).isEqualTo(2); 226 } 227 228 @Test onDeviceAdded_addFourDevicePreference_onlyDisplayThree()229 public void onDeviceAdded_addFourDevicePreference_onlyDisplayThree() { 230 final BluetoothDevicePreference preference1 = new BluetoothDevicePreference( 231 mContext, mCachedDevice1, true, BluetoothDevicePreference.SortType.TYPE_NO_SORT); 232 final BluetoothDevicePreference preference2 = new BluetoothDevicePreference( 233 mContext, mCachedDevice2, true, BluetoothDevicePreference.SortType.TYPE_NO_SORT); 234 final BluetoothDevicePreference preference3 = new BluetoothDevicePreference( 235 mContext, mCachedDevice3, true, BluetoothDevicePreference.SortType.TYPE_NO_SORT); 236 final BluetoothDevicePreference preference4 = new BluetoothDevicePreference( 237 mContext, mCachedDevice4, true, BluetoothDevicePreference.SortType.TYPE_NO_SORT); 238 final SingleTargetGearPreference dockPreference = new SingleTargetGearPreference( 239 mContext, null /* AttributeSet */); 240 241 mPreConnectedDeviceController.onDeviceAdded(preference1); 242 mPreConnectedDeviceController.onDeviceAdded(preference2); 243 mPreConnectedDeviceController.onDeviceAdded(preference3); 244 mPreConnectedDeviceController.onDeviceAdded(preference4); 245 mPreConnectedDeviceController.onDeviceAdded(dockPreference); 246 247 // 3 BluetoothDevicePreference and 1 see all preference 248 assertThat(mPreferenceGroup.getPreferenceCount()).isEqualTo(4); 249 } 250 251 @Test onDeviceAdded_addPreferenceNotExistInRecentlyDevices_noCrash()252 public void onDeviceAdded_addPreferenceNotExistInRecentlyDevices_noCrash() { 253 final BluetoothDevicePreference preference = new BluetoothDevicePreference( 254 mContext, mCachedDevice5, true, BluetoothDevicePreference.SortType.TYPE_NO_SORT); 255 256 mPreConnectedDeviceController.onDeviceAdded(preference); 257 258 // 1 BluetoothDevicePreference and 1 see all preference 259 assertThat(mPreferenceGroup.getPreferenceCount()).isEqualTo(2); 260 } 261 262 @Test onDeviceRemoved_removeLastDevice_showSeeAllPreference()263 public void onDeviceRemoved_removeLastDevice_showSeeAllPreference() { 264 final BluetoothDevicePreference preference1 = new BluetoothDevicePreference( 265 mContext, mCachedDevice1, true, BluetoothDevicePreference.SortType.TYPE_NO_SORT); 266 final SingleTargetGearPreference dockPreference = new SingleTargetGearPreference( 267 mContext, null /* AttributeSet */); 268 mPreferenceGroup.addPreference(preference1); 269 mPreferenceGroup.addPreference(dockPreference); 270 271 mPreConnectedDeviceController.onDeviceRemoved(preference1); 272 mPreConnectedDeviceController.onDeviceRemoved(dockPreference); 273 274 assertThat(mPreferenceGroup.getPreferenceCount()).isEqualTo(1); 275 } 276 277 @Test updatePreferenceVisibility_bluetoothIsEnable_shouldShowCorrectText()278 public void updatePreferenceVisibility_bluetoothIsEnable_shouldShowCorrectText() { 279 mShadowBluetoothAdapter.setEnabled(true); 280 mPreConnectedDeviceController.updatePreferenceVisibility(); 281 282 verify(mSeeAllPreference).setSummary(""); 283 } 284 285 @Test updatePreferenceVisibility_bluetoothIsDisable_shouldShowCorrectText()286 public void updatePreferenceVisibility_bluetoothIsDisable_shouldShowCorrectText() { 287 mShadowBluetoothAdapter.setEnabled(false); 288 mPreConnectedDeviceController.updatePreferenceVisibility(); 289 290 verify(mSeeAllPreference).setSummary( 291 mContext.getString(R.string.connected_device_see_all_summary)); 292 } 293 } 294