app_Settings/tests/robotests/src/com/android/settings/bluetooth/BluetoothPairingControllerTest.java

/*
 * Copyright (C) 2018 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 com.android.settings.bluetooth;

import static android.bluetooth.BluetoothDevice.PAIRING_VARIANT_CONSENT;

import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import android.bluetooth.BluetoothClass;
import android.bluetooth.BluetoothDevice;
import android.content.Context;
import android.content.Intent;
import android.os.Parcel;

import com.android.settings.testutils.shadow.ShadowBluetoothAdapter;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.robolectric.RobolectricTestRunner;
import org.robolectric.RuntimeEnvironment;
import org.robolectric.annotation.Config;

@RunWith(RobolectricTestRunner.class)
@Config(shadows = {ShadowBluetoothAdapter.class})
public class BluetoothPairingControllerTest {
    private final BluetoothClass mBluetoothClass =
            createBtClass(BluetoothClass.Device.AUDIO_VIDEO_HANDSFREE);
    @Mock
    private BluetoothDevice mBluetoothDevice;
    private Context mContext;
    private BluetoothPairingController mBluetoothPairingController;

    private BluetoothClass createBtClass(int deviceClass) {
        Parcel p = Parcel.obtain();
        p.writeInt(deviceClass);
        p.setDataPosition(0); // reset position of parcel before passing to constructor

        BluetoothClass bluetoothClass = BluetoothClass.CREATOR.createFromParcel(p);
        p.recycle();
        return bluetoothClass;
    }

    @Before
    public void setUp() {
        MockitoAnnotations.initMocks(this);

        mContext = RuntimeEnvironment.application;
        final Intent intent = new Intent();
        intent.putExtra(BluetoothDevice.EXTRA_DEVICE, mBluetoothDevice);
        mBluetoothPairingController = new BluetoothPairingController(intent, mContext);
    }

    @Test
    public void onDialogPositiveClick_confirmationDialog_setPBAP() {
        mBluetoothPairingController.mType = PAIRING_VARIANT_CONSENT;
        mBluetoothPairingController.onCheckedChanged(null, true);

        mBluetoothPairingController.onDialogPositiveClick(null);

        verify(mBluetoothDevice).setPhonebookAccessPermission(BluetoothDevice.ACCESS_ALLOWED);
    }

    @Test
    public void onSetContactSharingState_permissionAllowed_setPBAPAllowed() {
        when(mBluetoothDevice.getPhonebookAccessPermission()).thenReturn(
                BluetoothDevice.ACCESS_ALLOWED);
        mBluetoothPairingController.setContactSharingState();
        mBluetoothPairingController.onDialogPositiveClick(null);

        verify(mBluetoothDevice).setPhonebookAccessPermission(BluetoothDevice.ACCESS_ALLOWED);
    }

    @Test
    public void onSetContactSharingState_permissionUnknown_audioVideoHandsfree_setPBAPAllowed() {
        when(mBluetoothDevice.getPhonebookAccessPermission()).thenReturn(
                BluetoothDevice.ACCESS_UNKNOWN);
        when(mBluetoothDevice.getBluetoothClass()).thenReturn(mBluetoothClass);
        mBluetoothPairingController.setContactSharingState();
        mBluetoothPairingController.onDialogPositiveClick(null);

        verify(mBluetoothDevice).setPhonebookAccessPermission(BluetoothDevice.ACCESS_ALLOWED);
    }

    @Test
    public void onSetContactSharingState_permissionRejected_setPBAPRejected() {
        when(mBluetoothDevice.getPhonebookAccessPermission()).thenReturn(
                BluetoothDevice.ACCESS_REJECTED);
        when(mBluetoothDevice.getBluetoothClass()).thenReturn(mBluetoothClass);
        mBluetoothPairingController.setContactSharingState();
        mBluetoothPairingController.onDialogPositiveClick(null);

        verify(mBluetoothDevice).setPhonebookAccessPermission(BluetoothDevice.ACCESS_REJECTED);
    }
}