tests/robotests/src/com/android/settings/connecteddevice/ConnectedDeviceGroupControllerTest.java

   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 package com.android.settings.connecteddevice;
  17
  18 import static com.google.common.truth.Truth.assertThat;
  19
  20 import static org.mockito.Matchers.anyString;
  21 import static org.mockito.Mockito.doReturn;
  22 import static org.mockito.Mockito.spy;
  23 import static org.mockito.Mockito.verify;
  24
  25 import android.arch.lifecycle.LifecycleOwner;
  26 import android.content.Context;
  27 import android.support.v7.preference.Preference;
  28 import android.support.v7.preference.PreferenceGroup;
  29 import android.support.v7.preference.PreferenceManager;
  30 import android.support.v7.preference.PreferenceScreen;
  31
  32 import com.android.settings.TestConfig;
  33 import com.android.settings.bluetooth.ConnectedBluetoothDeviceUpdater;
  34 import com.android.settings.connecteddevice.usb.ConnectedUsbDeviceUpdater;
  35 import com.android.settings.dashboard.DashboardFragment;
  36 import com.android.settings.testutils.SettingsRobolectricTestRunner;
  37 import com.android.settingslib.core.lifecycle.Lifecycle;
  38
  39 import org.junit.Before;
  40 import org.junit.Test;
  41 import org.junit.runner.RunWith;
  42 import org.mockito.Answers;
  43 import org.mockito.Mock;
  44 import org.mockito.MockitoAnnotations;
  45 import org.robolectric.RuntimeEnvironment;
  46 import org.robolectric.annotation.Config;
  47
  48 @RunWith(SettingsRobolectricTestRunner.class)
  49 @Config(manifest = TestConfig.MANIFEST_PATH, sdk = TestConfig.SDK_VERSION)
  50 public class ConnectedDeviceGroupControllerTest {
  51     private static final String PREFERENCE_KEY_1 = "pref_key_1";
  52
  53     @Mock
  54     private DashboardFragment mDashboardFragment;
  55     @Mock
  56     private ConnectedBluetoothDeviceUpdater mConnectedBluetoothDeviceUpdater;
  57     @Mock
  58     private ConnectedUsbDeviceUpdater mConnectedUsbDeviceUpdater;
  59     @Mock
  60     private PreferenceScreen mPreferenceScreen;
  61     @Mock(answer = Answers.RETURNS_DEEP_STUBS)
  62     private PreferenceManager mPreferenceManager;
  63
  64     private PreferenceGroup mPreferenceGroup;
  65     private Context mContext;
  66     private Preference mPreference;
  67     private ConnectedDeviceGroupController mConnectedDeviceGroupController;
  68     private LifecycleOwner mLifecycleOwner;
  69     private Lifecycle mLifecycle;
  70
  71     @Before
  72     public void setUp() {
  73         MockitoAnnotations.initMocks(this);
  74
  75         mContext = RuntimeEnvironment.application;
  76         mPreference = new Preference(mContext);
  77         mPreference.setKey(PREFERENCE_KEY_1);
  78         mLifecycleOwner = () -> mLifecycle;
  79         mLifecycle = new Lifecycle(mLifecycleOwner);
  80         mPreferenceGroup = spy(new PreferenceScreen(mContext, null));
  81         doReturn(mPreferenceManager).when(mPreferenceGroup).getPreferenceManager();
  82         doReturn(mContext).when(mDashboardFragment).getContext();
  83
  84         mConnectedDeviceGroupController = new ConnectedDeviceGroupController(mDashboardFragment,
  85                 mLifecycle, mConnectedBluetoothDeviceUpdater, mConnectedUsbDeviceUpdater);
  86         mConnectedDeviceGroupController.mPreferenceGroup = mPreferenceGroup;
  87     }
  88
  89     @Test
  90     public void testOnDeviceAdded_firstAdd_becomeVisibleAndPreferenceAdded() {
  91         mConnectedDeviceGroupController.onDeviceAdded(mPreference);
  92
  93         assertThat(mPreferenceGroup.isVisible()).isTrue();
  94         assertThat(mPreferenceGroup.findPreference(PREFERENCE_KEY_1)).isEqualTo(mPreference);
  95     }
  96
  97     @Test
  98     public void testOnDeviceRemoved_lastRemove_becomeInvisibleAndPreferenceRemoved() {
  99         mPreferenceGroup.addPreference(mPreference);
 100
 101         mConnectedDeviceGroupController.onDeviceRemoved(mPreference);
 102
 103         assertThat(mPreferenceGroup.isVisible()).isFalse();
 104         assertThat(mPreferenceGroup.getPreferenceCount()).isEqualTo(0);
 105     }
 106
 107     @Test
 108     public void testOnDeviceRemoved_notLastRemove_stillVisible() {
 109         mPreferenceGroup.setVisible(true);
 110         mPreferenceGroup.addPreference(mPreference);
 111         mPreferenceGroup.addPreference(new Preference(mContext));
 112
 113         mConnectedDeviceGroupController.onDeviceRemoved(mPreference);
 114
 115         assertThat(mPreferenceGroup.isVisible()).isTrue();
 116     }
 117
 118     @Test
 119     public void testDisplayPreference_becomeInvisible() {
 120         doReturn(mPreferenceGroup).when(mPreferenceScreen).findPreference(anyString());
 121
 122         mConnectedDeviceGroupController.displayPreference(mPreferenceScreen);
 123
 124         assertThat(mPreferenceGroup.isVisible()).isFalse();
 125     }
 126
 127     @Test
 128     public void testLifecycle() {
 129         // register the callback in onStart()
 130         mLifecycle.handleLifecycleEvent(android.arch.lifecycle.Lifecycle.Event.ON_START);
 131         verify(mConnectedBluetoothDeviceUpdater).registerCallback();
 132         verify(mConnectedUsbDeviceUpdater).registerCallback();
 133
 134         // unregister the callback in onStop()
 135         mLifecycle.handleLifecycleEvent(android.arch.lifecycle.Lifecycle.Event.ON_STOP);
 136         verify(mConnectedBluetoothDeviceUpdater).unregisterCallback();
 137         verify(mConnectedUsbDeviceUpdater).unregisterCallback();
 138     }
 139 }