Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit fde4e995 authored by Ze Li's avatar Ze Li
Browse files

[Hide DCK Device] Hide exclusively managed devices from the Bluetooth chip of QS SysUI 

When only 1 Bluetooth device is connected, its name will be shown on the Bluetooth chip. If the device is exclusively managed the name should not be shown.

Test: manual
Bug: 323298503
Bug: 324475542
Flag: com.android.settingslib.flags.enable_hide_exclusively_managed_bluetooth_device
Change-Id: I1398da940cc65be25675f5dffbbe8c00338315ba
parent 12a0c012

packages/SystemUI/src/com/android/systemui/statusbar/policy/BluetoothControllerImpl.java

+18 −3
Original line number Diff line number Diff line
@@ -31,6 +31,7 @@ import androidx.annotation.WorkerThread; 


import com.android.internal.annotations.GuardedBy;

import com.android.settingslib.bluetooth.BluetoothCallback;

import com.android.settingslib.bluetooth.BluetoothUtils;

import com.android.settingslib.bluetooth.CachedBluetoothDevice;

import com.android.settingslib.bluetooth.LocalBluetoothManager;

import com.android.settingslib.bluetooth.LocalBluetoothProfile;
@@ -71,6 +72,7 @@ public class BluetoothControllerImpl implements BluetoothController, BluetoothCa 
    private final LocalBluetoothManager mLocalBluetoothManager;

    private final UserManager mUserManager;

    private final int mCurrentUser;

    private final Context mContext;

    @GuardedBy("mConnectedDevices")

    private final List<CachedBluetoothDevice> mConnectedDevices = new ArrayList<>();
@@ -99,6 +101,7 @@ public class BluetoothControllerImpl implements BluetoothController, BluetoothCa 
            @Main Looper mainLooper,

            @Nullable LocalBluetoothManager localBluetoothManager,

            @Nullable BluetoothAdapter bluetoothAdapter) {

        mContext = context;

        mDumpManager = dumpManager;

        mLogger = logger;

        mBluetoothRepository = bluetoothRepository;
@@ -262,9 +265,21 @@ public class BluetoothControllerImpl implements BluetoothController, BluetoothCa 
    }



    private Collection<CachedBluetoothDevice> getDevices() {

        return mLocalBluetoothManager != null

        Collection<CachedBluetoothDevice> devices =

                mLocalBluetoothManager != null

                        ? mLocalBluetoothManager.getCachedDeviceManager().getCachedDevicesCopy()

                        : Collections.emptyList();

        if (com.android.settingslib.flags.Flags.enableHideExclusivelyManagedBluetoothDevice()) {

            // When the device is exclusively managed by its owner app it needs to be hidden.

            devices =

                    devices.stream()

                            .filter(

                                    device ->

                                            !BluetoothUtils.isExclusivelyManagedBluetoothDevice(

                                                    mContext, device.getDevice()))

                            .toList();

        }

        return devices;

    }



    private void updateConnected() {

packages/SystemUI/tests/src/com/android/systemui/statusbar/policy/BluetoothControllerImplTest.java

+78 −2
Original line number Diff line number Diff line
@@ -14,6 +14,10 @@ 


package com.android.systemui.statusbar.policy;



import static android.platform.test.flag.junit.FlagsParameterization.allCombinationsOf;



import static com.android.settingslib.flags.Flags.FLAG_ENABLE_HIDE_EXCLUSIVELY_MANAGED_BLUETOOTH_DEVICE;



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



import static org.junit.Assert.assertEquals;
@@ -21,6 +25,7 @@ import static org.junit.Assert.assertFalse; 
import static org.junit.Assert.assertTrue;

import static org.mockito.ArgumentMatchers.anyBoolean;

import static org.mockito.Mockito.atLeastOnce;

import static org.mockito.Mockito.doReturn;

import static org.mockito.Mockito.mock;

import static org.mockito.Mockito.reset;

import static org.mockito.Mockito.times;
@@ -30,7 +35,11 @@ import static org.mockito.Mockito.when; 
import android.bluetooth.BluetoothAdapter;

import android.bluetooth.BluetoothDevice;

import android.bluetooth.BluetoothProfile;

import android.testing.AndroidTestingRunner;

import android.content.pm.ApplicationInfo;

import android.content.pm.PackageManager;

import android.platform.test.annotations.DisableFlags;

import android.platform.test.annotations.EnableFlags;

import android.platform.test.flag.junit.FlagsParameterization;

import android.testing.TestableLooper;

import android.testing.TestableLooper.RunWithLooper;
@@ -55,16 +64,31 @@ import com.android.systemui.util.time.FakeSystemClock; 
import org.junit.Before;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.mockito.Mock;

import org.mockito.MockitoAnnotations;



import java.util.ArrayList;

import java.util.List;

import java.util.concurrent.Executor;



@RunWith(AndroidTestingRunner.class)

import platform.test.runner.parameterized.ParameterizedAndroidJunit4;

import platform.test.runner.parameterized.Parameters;



@RunWith(ParameterizedAndroidJunit4.class)

@RunWithLooper

@SmallTest

public class BluetoothControllerImplTest extends SysuiTestCase {



    @Parameters(name = "{0}")

    public static List<FlagsParameterization> getParams() {

        return allCombinationsOf(FLAG_ENABLE_HIDE_EXCLUSIVELY_MANAGED_BLUETOOTH_DEVICE);

    }



    private static final String TEST_EXCLUSIVE_MANAGER = "com.test.manager";



    @Mock

    private PackageManager mPackageManager;



    private UserTracker mUserTracker;

    private LocalBluetoothManager mMockBluetoothManager;

    private CachedBluetoothDeviceManager mMockDeviceManager;
@@ -77,14 +101,21 @@ public class BluetoothControllerImplTest extends SysuiTestCase { 


    private FakeExecutor mBackgroundExecutor;



    public BluetoothControllerImplTest(FlagsParameterization flags) {

        super();

        mSetFlagsRule.setFlagsParameterization(flags);

    }



    @Before

    public void setup() throws Exception {

        MockitoAnnotations.initMocks(this);

        mTestableLooper = TestableLooper.get(this);

        mMockBluetoothManager = mDependency.injectMockDependency(LocalBluetoothManager.class);

        mDevices = new ArrayList<>();

        mUserTracker = mock(UserTracker.class);

        mMockDeviceManager = mock(CachedBluetoothDeviceManager.class);

        mMockAdapter = mock(BluetoothAdapter.class);

        mContext.setMockPackageManager(mPackageManager);

        when(mMockDeviceManager.getCachedDevicesCopy()).thenReturn(mDevices);

        when(mMockBluetoothManager.getCachedDeviceManager()).thenReturn(mMockDeviceManager);

        mMockLocalAdapter = mock(LocalBluetoothAdapter.class);
@@ -114,6 +145,7 @@ public class BluetoothControllerImplTest extends SysuiTestCase { 
        CachedBluetoothDevice device = mock(CachedBluetoothDevice.class);

        when(device.isConnected()).thenReturn(true);

        when(device.getMaxConnectionState()).thenReturn(BluetoothProfile.STATE_CONNECTED);

        when(device.getDevice()).thenReturn(mock(BluetoothDevice.class));



        mDevices.add(device);

        when(mMockLocalAdapter.getConnectionState())
@@ -139,10 +171,12 @@ public class BluetoothControllerImplTest extends SysuiTestCase { 
    public void getConnectedDevices_onlyReturnsConnected() {

        CachedBluetoothDevice device1Disconnected = mock(CachedBluetoothDevice.class);

        when(device1Disconnected.isConnected()).thenReturn(false);

        when(device1Disconnected.getDevice()).thenReturn(mock(BluetoothDevice.class));

        mDevices.add(device1Disconnected);



        CachedBluetoothDevice device2Connected = mock(CachedBluetoothDevice.class);

        when(device2Connected.isConnected()).thenReturn(true);

        when(device2Connected.getDevice()).thenReturn(mock(BluetoothDevice.class));

        mDevices.add(device2Connected);



        mBluetoothControllerImpl.onDeviceAdded(device1Disconnected);
@@ -153,6 +187,46 @@ public class BluetoothControllerImplTest extends SysuiTestCase { 
                .isEqualTo(device2Connected);

    }



    @Test

    @EnableFlags(FLAG_ENABLE_HIDE_EXCLUSIVELY_MANAGED_BLUETOOTH_DEVICE)

    public void getConnectedDevice_exclusivelyManagedDevice_doNotReturn()

            throws PackageManager.NameNotFoundException {

        CachedBluetoothDevice cachedDevice = mock(CachedBluetoothDevice.class);

        when(cachedDevice.isConnected()).thenReturn(true);

        BluetoothDevice device = mock(BluetoothDevice.class);

        when(device.getMetadata(BluetoothDevice.METADATA_EXCLUSIVE_MANAGER)).thenReturn(

                TEST_EXCLUSIVE_MANAGER.getBytes());

        when(cachedDevice.getDevice()).thenReturn(device);

        doReturn(new ApplicationInfo()).when(mPackageManager).getApplicationInfo(

                TEST_EXCLUSIVE_MANAGER, 0);



        mDevices.add(cachedDevice);

        mBluetoothControllerImpl.onDeviceAdded(cachedDevice);



        assertThat(mBluetoothControllerImpl.getConnectedDevices()).isEmpty();

    }



    @Test

    @DisableFlags(FLAG_ENABLE_HIDE_EXCLUSIVELY_MANAGED_BLUETOOTH_DEVICE)

    public void getConnectedDevice_exclusivelyManagedDevice_returnsConnected()

            throws PackageManager.NameNotFoundException {

        CachedBluetoothDevice cachedDevice = mock(CachedBluetoothDevice.class);

        when(cachedDevice.isConnected()).thenReturn(true);

        BluetoothDevice device = mock(BluetoothDevice.class);

        when(device.getMetadata(BluetoothDevice.METADATA_EXCLUSIVE_MANAGER)).thenReturn(

                TEST_EXCLUSIVE_MANAGER.getBytes());

        when(cachedDevice.getDevice()).thenReturn(device);

        doReturn(new ApplicationInfo()).when(mPackageManager).getApplicationInfo(

                TEST_EXCLUSIVE_MANAGER, 0);



        mDevices.add(cachedDevice);

        mBluetoothControllerImpl.onDeviceAdded(cachedDevice);



        assertThat(mBluetoothControllerImpl.getConnectedDevices()).hasSize(1);

        assertThat(mBluetoothControllerImpl.getConnectedDevices().get(0))

                .isEqualTo(cachedDevice);

    }



    @Test

    public void testOnBluetoothStateChange_updatesBluetoothState() {

        mBluetoothControllerImpl.onBluetoothStateChanged(BluetoothAdapter.STATE_OFF);
@@ -184,6 +258,7 @@ public class BluetoothControllerImplTest extends SysuiTestCase { 


        assertFalse(mBluetoothControllerImpl.isBluetoothConnected());

        CachedBluetoothDevice device = mock(CachedBluetoothDevice.class);

        when(device.getDevice()).thenReturn(mock(BluetoothDevice.class));

        mDevices.add(device);

        when(device.isConnected()).thenReturn(true);

        when(device.getMaxConnectionState()).thenReturn(BluetoothProfile.STATE_CONNECTED);
@@ -402,6 +477,7 @@ public class BluetoothControllerImplTest extends SysuiTestCase { 
    private CachedBluetoothDevice createBluetoothDevice(

            int profile, boolean isConnected, boolean isActive) {

        CachedBluetoothDevice device = mock(CachedBluetoothDevice.class);

        when(device.getDevice()).thenReturn(mock(BluetoothDevice.class));

        mDevices.add(device);

        when(device.isActiveDevice(profile)).thenReturn(isActive);