tests/tests/bluetooth/src/android/bluetooth/cts/BasicBluetoothGattTest.java

/*
 * Copyright (C) 2021 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package android.bluetooth.cts;

import android.bluetooth.BluetoothAdapter;
import android.bluetooth.BluetoothDevice;
import android.bluetooth.BluetoothGatt;
import android.bluetooth.BluetoothGattCallback;
import android.bluetooth.BluetoothGattService;
import android.bluetooth.BluetoothManager;
import android.test.AndroidTestCase;

import androidx.test.InstrumentationRegistry;

import java.util.List;

/**
 * Tests a small part of the {@link BluetoothGatt} methods without a real Bluetooth device.
 * Other tests that run with real bluetooth connections are located in CtsVerifier.
 */
public class BasicBluetoothGattTest extends AndroidTestCase {

    private BluetoothAdapter mBluetoothAdapter;
    private BluetoothDevice mBluetoothDevice;
    private BluetoothGatt mBluetoothGatt;

    @Override
    public void setUp() {
        if (!TestUtils.isBleSupported(getContext())) {
            return;
        }
        InstrumentationRegistry.getInstrumentation().getUiAutomation()
            .adoptShellPermissionIdentity(android.Manifest.permission.BLUETOOTH_CONNECT);

        mBluetoothAdapter = mContext.getSystemService(BluetoothManager.class).getAdapter();
        if (!mBluetoothAdapter.isEnabled()) {
            assertTrue(BTAdapterUtils.enableAdapter(mBluetoothAdapter, mContext));
        }
        mBluetoothDevice = mBluetoothAdapter.getRemoteDevice("00:11:22:AA:BB:CC");
        mBluetoothGatt = mBluetoothDevice.connectGatt(
                mContext, /*autoConnect=*/ true, new BluetoothGattCallback() {});
    }

    @Override
    public void tearDown() {
        if (!TestUtils.isBleSupported(getContext())) {
            // mBluetoothAdapter == null.
            return;
        }
        mBluetoothGatt.disconnect();
        assertTrue(BTAdapterUtils.disableAdapter(mBluetoothAdapter, mContext));
        InstrumentationRegistry.getInstrumentation().getUiAutomation()
            .dropShellPermissionIdentity();
    }

    public void testGetServices() throws Exception {
        if (!TestUtils.isBleSupported(getContext())) {
            return;
        }

        // getServices() returns an empty list if service discovery has not yet been performed.
        List<BluetoothGattService> services = mBluetoothGatt.getServices();
        assertNotNull(services);
        assertTrue(services.isEmpty());
    }

    public void testConnect() throws Exception {
        if (!TestUtils.isBleSupported(getContext())) {
            return;
        }

        try {
            mBluetoothGatt.connect();
        } catch (Exception e) {
            fail("Exception caught from connect(): " + e.toString());
        }
    }

    public void testSetPreferredPhy() throws Exception {
        if (!TestUtils.isBleSupported(getContext())) {
            return;
        }

        try {
            mBluetoothGatt.setPreferredPhy(BluetoothDevice.PHY_LE_1M, BluetoothDevice.PHY_LE_1M,
                    BluetoothDevice.PHY_OPTION_NO_PREFERRED);
        } catch (Exception e) {
            fail("Exception caught from setPreferredPhy(): " + e.toString());
        }
    }

    public void testGetConnectedDevices() {
        if (!TestUtils.isBleSupported(getContext())) {
            return;
        }
        try {
            mBluetoothGatt.getConnectedDevices();
            fail("Should throw UnsupportedOperationException!");
        } catch (UnsupportedOperationException ex) {
            // Expected
        }
    }

    public void testGetConnectionState() {
        if (!TestUtils.isBleSupported(getContext())) {
            return;
        }
        try {
            mBluetoothGatt.getConnectionState(null);
            fail("Should throw UnsupportedOperationException!");
        } catch (UnsupportedOperationException ex) {
            // Expected
        }
    }

    public void testGetDevicesMatchingConnectionStates() {
        if (!TestUtils.isBleSupported(getContext())) {
            return;
        }
        try {
            mBluetoothGatt.getDevicesMatchingConnectionStates(null);
            fail("Should throw UnsupportedOperationException!");
        } catch (UnsupportedOperationException ex) {
            // Expected
        }
    }
}