tests/robotests/src/com/android/settings/connecteddevice/usb/ConnectedUsbDeviceUpdaterTest.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.usb;
  17
  18 import static com.google.common.truth.Truth.assertThat;
  19
  20 import static org.mockito.Mockito.doReturn;
  21 import static org.mockito.Mockito.verify;
  22 import static org.mockito.Mockito.when;
  23
  24 import android.content.Context;
  25
  26 import com.android.settings.R;
  27 import com.android.settings.TestConfig;
  28 import com.android.settings.connecteddevice.DevicePreferenceCallback;
  29 import com.android.settings.dashboard.DashboardFragment;
  30 import com.android.settings.testutils.SettingsRobolectricTestRunner;
  31
  32 import org.junit.Before;
  33 import org.junit.Test;
  34 import org.junit.runner.RunWith;
  35 import org.mockito.Mock;
  36 import org.mockito.MockitoAnnotations;
  37 import org.robolectric.RuntimeEnvironment;
  38 import org.robolectric.annotation.Config;
  39
  40 @RunWith(SettingsRobolectricTestRunner.class)
  41 @Config(manifest = TestConfig.MANIFEST_PATH, sdk = TestConfig.SDK_VERSION)
  42 public class ConnectedUsbDeviceUpdaterTest {
  43     private Context mContext;
  44     private ConnectedUsbDeviceUpdater mDeviceUpdater;
  45
  46     @Mock
  47     private DashboardFragment mFragment;
  48     @Mock
  49     private UsbConnectionBroadcastReceiver mUsbReceiver;
  50     @Mock
  51     private DevicePreferenceCallback mDevicePreferenceCallback;
  52     @Mock
  53     private UsbBackend mUsbBackend;
  54
  55     @Before
  56     public void setUp() {
  57         MockitoAnnotations.initMocks(this);
  58
  59         mContext = RuntimeEnvironment.application;
  60         when(mFragment.getContext()).thenReturn(mContext);
  61         mDeviceUpdater = new ConnectedUsbDeviceUpdater(mFragment, mDevicePreferenceCallback,
  62                 mUsbBackend);
  63         mDeviceUpdater.mUsbReceiver = mUsbReceiver;
  64     }
  65
  66     @Test
  67     public void testInitUsbPreference_preferenceInit() {
  68         mDeviceUpdater.initUsbPreference(mContext);
  69
  70         assertThat(mDeviceUpdater.mUsbPreference.getTitle()).isEqualTo("USB");
  71         assertThat(mDeviceUpdater.mUsbPreference.getIcon()).isEqualTo(mContext.getDrawable(
  72                 R.drawable.ic_usb));
  73         assertThat(mDeviceUpdater.mUsbPreference.isSelectable()).isFalse();
  74     }
  75
  76     @Test
  77     public void testInitUsbPreference_usbConnected_preferenceAdded() {
  78         mDeviceUpdater.initUsbPreference(mContext);
  79         mDeviceUpdater.mUsbConnectionListener.onUsbConnectionChanged(true /* connected */,
  80                 UsbBackend.MODE_DATA_NONE);
  81
  82         verify(mDevicePreferenceCallback).onDeviceAdded(mDeviceUpdater.mUsbPreference);
  83     }
  84
  85     @Test
  86     public void testInitUsbPreference_usbDisconnected_preferenceRemoved() {
  87         mDeviceUpdater.initUsbPreference(mContext);
  88         mDeviceUpdater.mUsbConnectionListener.onUsbConnectionChanged(false /* connected */,
  89                 UsbBackend.MODE_DATA_NONE);
  90
  91         verify(mDevicePreferenceCallback).onDeviceRemoved(mDeviceUpdater.mUsbPreference);
  92     }
  93
  94 }