Merge "Cherry-pick SapServiceTest from AOSP" into tm-qpr-dev (55efd1b7) · Commits · e / os / android_packages_modules_Bluetooth

android/app/tests/unit/src/com/android/bluetooth/sap/SapServiceTest.java

+83 −7

Original line number	Diff line number	Diff line
		@@ -13,12 +13,18 @@
		* See the License for the specific language governing permissions and
		* limitations under the License.
		*/

		package com.android.bluetooth.sap;

		import static com.google.common.truth.Truth.assertThat;

		import static org.mockito.Mockito.anyString;
		import static org.mockito.Mockito.doReturn;
		import static org.mockito.Mockito.when;

		import android.bluetooth.BluetoothAdapter;
		import android.bluetooth.BluetoothDevice;
		import android.bluetooth.BluetoothProfile;
		import android.content.Context;

		import androidx.test.InstrumentationRegistry;
		@@ -26,12 +32,11 @@ import androidx.test.filters.MediumTest;
		import androidx.test.rule.ServiceTestRule;
		import androidx.test.runner.AndroidJUnit4;

		import com.android.bluetooth.R;
		import com.android.bluetooth.TestUtils;
		import com.android.bluetooth.btservice.AdapterService;
		import com.android.bluetooth.btservice.storage.DatabaseManager;

		import org.junit.After;
		import org.junit.Assert;
		import org.junit.Assume;
		import org.junit.Before;
		import org.junit.Rule;
		@@ -40,9 +45,15 @@ import org.junit.runner.RunWith;
		import org.mockito.Mock;
		import org.mockito.MockitoAnnotations;

		import java.util.concurrent.CountDownLatch;
		import java.util.concurrent.TimeUnit;
		import java.util.concurrent.atomic.AtomicBoolean;

		@MediumTest
		@RunWith(AndroidJUnit4.class)
		public class SapServiceTest {
		private static final int TIMEOUT_MS = 5_000;

		private SapService mService = null;
		private BluetoothAdapter mAdapter = null;
		private Context mTargetContext;
		@@ -50,6 +61,8 @@ public class SapServiceTest {
		@Rule public final ServiceTestRule mServiceRule = new ServiceTestRule();

		@Mock private AdapterService mAdapterService;
		@Mock private DatabaseManager mDatabaseManager;
		private BluetoothDevice mDevice;

		@Before
		public void setUp() throws Exception {
		@@ -61,10 +74,11 @@ public class SapServiceTest {
		doReturn(true, false).when(mAdapterService).isStartedProfile(anyString());
		TestUtils.startService(mServiceRule, SapService.class);
		mService = SapService.getSapService();
		Assert.assertNotNull(mService);
		assertThat(mService).isNotNull();
		// Try getting the Bluetooth adapter
		mAdapter = BluetoothAdapter.getDefaultAdapter();
		Assert.assertNotNull(mAdapter);
		assertThat(mAdapter).isNotNull();
		mDevice = TestUtils.getTestDevice(mAdapter, 0);
		}

		@After
		@@ -74,12 +88,74 @@ public class SapServiceTest {
		}
		TestUtils.stopService(mServiceRule, SapService.class);
		mService = SapService.getSapService();
		Assert.assertNull(mService);
		assertThat(mService).isNull();
		TestUtils.clearAdapterService(mAdapterService);
		}

		@Test
		public void testInitialize() {
		Assert.assertNotNull(SapService.getSapService());
		public void testGetSapService() {
		assertThat(mService).isEqualTo(SapService.getSapService());
		assertThat(mService.getConnectedDevices()).isEmpty();
		}

		/**
		* Test stop SAP Service
		*/
		@Test
		public void testStopSapService() throws Exception {
		AtomicBoolean stopResult = new AtomicBoolean();
		AtomicBoolean startResult = new AtomicBoolean();
		CountDownLatch latch = new CountDownLatch(1);

		// SAP Service is already running: test stop(). Note: must be done on the main thread
		InstrumentationRegistry.getInstrumentation().runOnMainSync(new Runnable() {
		public void run() {
		stopResult.set(mService.stop());
		startResult.set(mService.start());
		latch.countDown();
		}
		});

		assertThat(latch.await(TIMEOUT_MS, TimeUnit.MILLISECONDS)).isTrue();
		assertThat(stopResult.get()).isTrue();
		assertThat(startResult.get()).isTrue();
		}

		/**
		* Test get connection policy for BluetoothDevice
		*/
		@Test
		public void testGetConnectionPolicy() {
		when(mAdapterService.getDatabase()).thenReturn(mDatabaseManager);
		when(mDatabaseManager
		.getProfileConnectionPolicy(mDevice, BluetoothProfile.SAP))
		.thenReturn(BluetoothProfile.CONNECTION_POLICY_UNKNOWN);
		assertThat(mService.getConnectionPolicy(mDevice))
		.isEqualTo(BluetoothProfile.CONNECTION_POLICY_UNKNOWN);

		when(mDatabaseManager
		.getProfileConnectionPolicy(mDevice, BluetoothProfile.SAP))
		.thenReturn(BluetoothProfile.CONNECTION_POLICY_FORBIDDEN);
		assertThat(mService.getConnectionPolicy(mDevice))
		.isEqualTo(BluetoothProfile.CONNECTION_POLICY_FORBIDDEN);

		when(mDatabaseManager
		.getProfileConnectionPolicy(mDevice, BluetoothProfile.SAP))
		.thenReturn(BluetoothProfile.CONNECTION_POLICY_ALLOWED);

		assertThat(mService.getConnectionPolicy(mDevice))
		.isEqualTo(BluetoothProfile.CONNECTION_POLICY_ALLOWED);
		}

		@Test
		public void testGetRemoteDevice() {
		assertThat(mService.getRemoteDevice()).isNull();
		}

		@Test
		public void testGetRemoteDeviceName() {
		assertThat(SapService.getRemoteDeviceName()).isNull();
		}
		}