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.bluetooth.mapclient; 18 19 import android.bluetooth.BluetoothAdapter; 20 import android.bluetooth.BluetoothDevice; 21 import android.content.Context; 22 import android.support.test.InstrumentationRegistry; 23 import android.support.test.filters.MediumTest; 24 import android.support.test.rule.ServiceTestRule; 25 import android.support.test.runner.AndroidJUnit4; 26 27 import com.android.bluetooth.R; 28 import com.android.bluetooth.TestUtils; 29 import com.android.bluetooth.btservice.AdapterService; 30 31 import org.junit.After; 32 import org.junit.Assert; 33 import org.junit.Assume; 34 import org.junit.Before; 35 import org.junit.Rule; 36 import org.junit.Test; 37 import org.junit.runner.RunWith; 38 import org.mockito.Mock; 39 import org.mockito.MockitoAnnotations; 40 41 import java.util.ArrayList; 42 import java.util.List; 43 import java.util.Map; 44 45 @MediumTest 46 @RunWith(AndroidJUnit4.class) 47 public class MapClientTest { 48 private static final String TAG = MapClientTest.class.getSimpleName(); 49 private MapClientService mService = null; 50 private BluetoothAdapter mAdapter = null; 51 private Context mTargetContext; 52 53 @Mock private AdapterService mAdapterService; 54 @Mock private MnsService mMockMnsService; 55 56 @Rule public final ServiceTestRule mServiceRule = new ServiceTestRule(); 57 58 @Before setUp()59 public void setUp() throws Exception { 60 mTargetContext = InstrumentationRegistry.getTargetContext(); 61 Assume.assumeTrue("Ignore test when MapClientService is not enabled", 62 mTargetContext.getResources().getBoolean(R.bool.profile_supported_mapmce)); 63 MockitoAnnotations.initMocks(this); 64 TestUtils.setAdapterService(mAdapterService); 65 MapUtils.setMnsService(mMockMnsService); 66 TestUtils.startService(mServiceRule, MapClientService.class); 67 mService = MapClientService.getMapClientService(); 68 Assert.assertNotNull(mService); 69 cleanUpInstanceMap(); 70 mAdapter = BluetoothAdapter.getDefaultAdapter(); 71 } 72 73 @After tearDown()74 public void tearDown() throws Exception { 75 if (!mTargetContext.getResources().getBoolean(R.bool.profile_supported_mapmce)) { 76 return; 77 } 78 TestUtils.stopService(mServiceRule, MapClientService.class); 79 mService = MapClientService.getMapClientService(); 80 Assert.assertNull(mService); 81 TestUtils.clearAdapterService(mAdapterService); 82 } 83 cleanUpInstanceMap()84 private void cleanUpInstanceMap() { 85 if (!mService.getInstanceMap().isEmpty()) { 86 List<BluetoothDevice> deviceList = mService.getConnectedDevices(); 87 for (BluetoothDevice d : deviceList) { 88 mService.disconnect(d); 89 } 90 } 91 Assert.assertTrue(mService.getInstanceMap().isEmpty()); 92 } 93 94 @Test testInitialize()95 public void testInitialize() { 96 Assert.assertNotNull(MapClientService.getMapClientService()); 97 } 98 99 /** 100 * Test connection of one device. 101 */ 102 @Test testConnect()103 public void testConnect() { 104 // make sure there is no statemachine already defined for this device 105 BluetoothDevice device = makeBluetoothDevice("11:11:11:11:11:11"); 106 Assert.assertNull(mService.getInstanceMap().get(device)); 107 108 // connect a bluetooth device 109 Assert.assertTrue(mService.connect(device)); 110 111 // is the statemachine created 112 Map<BluetoothDevice, MceStateMachine> map = mService.getInstanceMap(); 113 Assert.assertEquals(1, map.size()); 114 Assert.assertNotNull(map.get(device)); 115 } 116 117 /** 118 * Test connecting MAXIMUM_CONNECTED_DEVICES devices. 119 */ 120 @Test testConnectMaxDevices()121 public void testConnectMaxDevices() { 122 // Create bluetoothdevice & mock statemachine objects to be used in this test 123 List<BluetoothDevice> list = new ArrayList<>(); 124 String address = "11:11:11:11:11:1"; 125 for (int i = 0; i < MapClientService.MAXIMUM_CONNECTED_DEVICES; ++i) { 126 list.add(makeBluetoothDevice(address + i)); 127 } 128 129 // make sure there is no statemachine already defined for the devices defined above 130 for (BluetoothDevice d : list) { 131 Assert.assertNull(mService.getInstanceMap().get(d)); 132 } 133 134 // run the test - connect all devices 135 for (BluetoothDevice d : list) { 136 Assert.assertTrue(mService.connect(d)); 137 } 138 139 // verify 140 Map<BluetoothDevice, MceStateMachine> map = mService.getInstanceMap(); 141 Assert.assertEquals(MapClientService.MAXIMUM_CONNECTED_DEVICES, map.size()); 142 for (BluetoothDevice d : list) { 143 Assert.assertNotNull(map.get(d)); 144 } 145 146 // Try to connect one more device. Should fail. 147 BluetoothDevice last = makeBluetoothDevice("11:22:33:44:55:66"); 148 Assert.assertFalse(mService.connect(last)); 149 } 150 makeBluetoothDevice(String address)151 private BluetoothDevice makeBluetoothDevice(String address) { 152 return mAdapter.getRemoteDevice(address); 153 } 154 } 155