tests/robotests/src/com/android/settings/deviceinfo/UsbBackendTest.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
  17 package com.android.settings.deviceinfo;
  18
  19 import static org.mockito.Answers.RETURNS_DEEP_STUBS;
  20 import static org.mockito.Matchers.argThat;
  21 import static org.mockito.Mockito.eq;
  22 import static org.mockito.Mockito.verify;
  23 import static org.mockito.Mockito.when;
  24
  25 import android.content.Context;
  26 import android.content.pm.PackageManager;
  27 import android.hardware.usb.UsbManager;
  28
  29 import com.android.settings.testutils.SettingsRobolectricTestRunner;
  30 import com.android.settings.TestConfig;
  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.annotation.Config;
  38
  39 @RunWith(SettingsRobolectricTestRunner.class)
  40 @Config(manifest = TestConfig.MANIFEST_PATH, sdk = TestConfig.SDK_VERSION)
  41 public class UsbBackendTest {
  42
  43     @Mock(answer = RETURNS_DEEP_STUBS)
  44     private Context mContext;
  45     @Mock
  46     private UsbManager mUsbManager;
  47     @Mock
  48     private UsbBackend.UserRestrictionUtil mUserRestrictionUtil;
  49
  50     @Before
  51     public void setUp() {
  52         MockitoAnnotations.initMocks(this);
  53         when(mContext.getPackageManager().hasSystemFeature(PackageManager.FEATURE_MIDI))
  54             .thenReturn(true);
  55         when((Object)mContext.getSystemService(UsbManager.class)).thenReturn(mUsbManager);
  56     }
  57
  58     @Test
  59     public void constructor_noUsbPort_shouldNotCrash() {
  60         UsbBackend usbBackend = new UsbBackend(mContext, mUserRestrictionUtil);
  61         // Should not crash
  62     }
  63
  64     @Test
  65     public void getCurrentMode_shouldRegisterReceiverToGetUsbState() {
  66         UsbBackend usbBackend = new UsbBackend(mContext, mUserRestrictionUtil);
  67
  68         usbBackend.getCurrentMode();
  69
  70         verify(mContext).registerReceiver(eq(null),
  71             argThat(intentFilter -> intentFilter != null &&
  72                 UsbManager.ACTION_USB_STATE.equals(intentFilter.getAction(0))));
  73     }
  74 }