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

Commit b85941de authored by Yiyi Shen's avatar Yiyi Shen
Browse files

[Audiosharing] Sync temp bond state when CSIP grouping 

Test: atest
Flag: com.android.settingslib.flags.enable_temporary_bond_devices_ui
Bug: 392004799
Change-Id: I86266765d6636e6db11516e413b5d2172a71c387
parent aff1400d

packages/SettingsLib/src/com/android/settingslib/bluetooth/CsipDeviceManager.java

+26 −8
Original line number Diff line number Diff line
@@ -30,11 +30,13 @@ import android.util.Log; 
import androidx.annotation.ChecksSdkIntAtLeast;



import com.android.internal.annotations.VisibleForTesting;

import com.android.settingslib.flags.Flags;



import java.util.ArrayList;

import java.util.HashSet;

import java.util.List;

import java.util.Map;

import java.util.Objects;

import java.util.Set;

import java.util.stream.Collectors;
@@ -385,7 +387,7 @@ public class CsipDeviceManager { 
                preferredMainDevice.refresh();

                hasChanged = true;

            }

            syncAudioSharingSourceIfNeeded(preferredMainDevice);

            syncAudioSharingStatusIfNeeded(preferredMainDevice);

        }

        if (hasChanged) {

            log("addMemberDevicesIntoMainDevice: After changed, CachedBluetoothDevice list: "
@@ -399,13 +401,16 @@ public class CsipDeviceManager { 
        return userManager != null && userManager.isManagedProfile();

    }



    private void syncAudioSharingSourceIfNeeded(CachedBluetoothDevice mainDevice) {

    private void syncAudioSharingStatusIfNeeded(CachedBluetoothDevice mainDevice) {

        boolean isAudioSharingEnabled = BluetoothUtils.isAudioSharingUIAvailable(mContext);

        if (isAudioSharingEnabled) {

        if (isAudioSharingEnabled && mainDevice != null) {

            if (isWorkProfile()) {

                log("addMemberDevicesIntoMainDevice: skip sync source for work profile");

                log("addMemberDevicesIntoMainDevice: skip sync audio sharing status, work profile");

                return;

            }

            Set<CachedBluetoothDevice> deviceSet = new HashSet<>();

            deviceSet.add(mainDevice);

            deviceSet.addAll(mainDevice.getMemberDevice());

            boolean hasBroadcastSource = BluetoothUtils.isBroadcasting(mBtManager)

                    && BluetoothUtils.hasConnectedBroadcastSource(

                    mainDevice, mBtManager);
@@ -419,9 +424,6 @@ public class CsipDeviceManager { 
                if (metadata != null && assistant != null) {

                    log("addMemberDevicesIntoMainDevice: sync audio sharing source after "

                            + "combining the top level devices.");

                    Set<CachedBluetoothDevice> deviceSet = new HashSet<>();

                    deviceSet.add(mainDevice);

                    deviceSet.addAll(mainDevice.getMemberDevice());

                    Set<BluetoothDevice> sinksToSync = deviceSet.stream()

                            .map(CachedBluetoothDevice::getDevice)

                            .filter(device ->
@@ -435,8 +437,24 @@ public class CsipDeviceManager { 
                    }

                }

            }

            if (Flags.enableTemporaryBondDevicesUi()) {

                log("addMemberDevicesIntoMainDevice: sync temp bond metadata for audio sharing "

                        + "sinks after combining the top level devices.");

                Set<BluetoothDevice> sinksToSync = deviceSet.stream()

                        .map(CachedBluetoothDevice::getDevice).filter(Objects::nonNull).collect(

                                Collectors.toSet());

                if (sinksToSync.stream().anyMatch(BluetoothUtils::isTemporaryBondDevice)) {

                    for (BluetoothDevice device : sinksToSync) {

                        if (!BluetoothUtils.isTemporaryBondDevice(device)) {

                            log("addMemberDevicesIntoMainDevice: sync temp bond metadata for "

                                    + device.getAnonymizedAddress());

                            BluetoothUtils.setTemporaryBondMetadata(device);

                        }

                    }

                }

            }

        } else {

            log("addMemberDevicesIntoMainDevice: skip sync source, flag disabled");

            log("addMemberDevicesIntoMainDevice: skip sync audio sharing status, flag disabled");

        }

    }

packages/SettingsLib/tests/robotests/src/com/android/settingslib/bluetooth/CsipDeviceManagerTest.java

+30 −7
Original line number Diff line number Diff line
@@ -40,6 +40,8 @@ import android.content.Context; 
import android.os.Looper;

import android.os.Parcel;

import android.os.UserManager;

import android.platform.test.annotations.DisableFlags;

import android.platform.test.annotations.EnableFlags;

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



import com.android.settingslib.flags.Flags;
@@ -74,6 +76,9 @@ public class CsipDeviceManagerTest { 
    private final static String DEVICE_ADDRESS_1 = "AA:BB:CC:DD:EE:11";

    private final static String DEVICE_ADDRESS_2 = "AA:BB:CC:DD:EE:22";

    private final static String DEVICE_ADDRESS_3 = "AA:BB:CC:DD:EE:33";

    private static final int METADATA_FAST_PAIR_CUSTOMIZED_FIELDS = 25;

    private static final String TEMP_BOND_METADATA =

            "<TEMP_BOND_TYPE>le_audio_sharing</TEMP_BOND_TYPE>";

    private final static int GROUP1 = 1;

    private final BluetoothClass DEVICE_CLASS_1 =

            createBtClass(BluetoothClass.Device.AUDIO_VIDEO_HEADPHONES);
@@ -337,6 +342,7 @@ public class CsipDeviceManagerTest { 
    }



    @Test

    @DisableFlags(Flags.FLAG_ENABLE_LE_AUDIO_SHARING)

    public void addMemberDevicesIntoMainDevice_preferredDeviceIsMainAndTwoMain_returnTrue() {

        // Condition: The preferredDevice is main and there is another main device in top list

        // Expected Result: return true and there is the preferredDevice in top list
@@ -346,7 +352,6 @@ public class CsipDeviceManagerTest { 
        mCachedDevices.add(preferredDevice);

        mCachedDevices.add(mCachedDevice2);

        mCachedDevices.add(mCachedDevice3);

        mSetFlagsRule.disableFlags(Flags.FLAG_ENABLE_LE_AUDIO_SHARING);



        assertThat(mCsipDeviceManager.addMemberDevicesIntoMainDevice(GROUP1, preferredDevice))

                .isTrue();
@@ -359,6 +364,7 @@ public class CsipDeviceManagerTest { 
    }



    @Test

    @EnableFlags({Flags.FLAG_ENABLE_LE_AUDIO_SHARING, Flags.FLAG_ENABLE_TEMPORARY_BOND_DEVICES_UI})

    public void

            addMemberDevicesIntoMainDevice_preferredDeviceIsMainAndTwoMain_workProfile_doNothing() {

        // Condition: The preferredDevice is main and there is another main device in top list
@@ -369,7 +375,6 @@ public class CsipDeviceManagerTest { 
        mCachedDevices.add(preferredDevice);

        mCachedDevices.add(mCachedDevice2);

        mCachedDevices.add(mCachedDevice3);

        mSetFlagsRule.enableFlags(Flags.FLAG_ENABLE_LE_AUDIO_SHARING);

        when(mBroadcast.isEnabled(null)).thenReturn(true);

        BluetoothLeBroadcastMetadata metadata = Mockito.mock(BluetoothLeBroadcastMetadata.class);

        when(mBroadcast.getLatestBluetoothLeBroadcastMetadata()).thenReturn(metadata);
@@ -377,6 +382,8 @@ public class CsipDeviceManagerTest { 
                BluetoothLeBroadcastReceiveState.class);

        when(state.getBisSyncState()).thenReturn(ImmutableList.of(1L));

        when(mAssistant.getAllSources(mDevice2)).thenReturn(ImmutableList.of(state));

        when(mDevice2.getMetadata(METADATA_FAST_PAIR_CUSTOMIZED_FIELDS))

                .thenReturn(TEMP_BOND_METADATA.getBytes());

        when(mContext.getSystemService(UserManager.class)).thenReturn(mUserManager);

        when(mUserManager.isManagedProfile()).thenReturn(true);
@@ -387,10 +394,13 @@ public class CsipDeviceManagerTest { 
        assertThat(mCachedDevices.contains(mCachedDevice3)).isTrue();

        assertThat(preferredDevice.getMemberDevice()).contains(mCachedDevice2);

        verify(mAssistant, never()).addSource(mDevice1, metadata, /* isGroupOp= */ false);

        verify(mDevice1, never()).setMetadata(METADATA_FAST_PAIR_CUSTOMIZED_FIELDS,

                TEMP_BOND_METADATA.getBytes());

    }



    @Test

    public void addMemberDevicesIntoMainDevice_preferredDeviceIsMainAndTwoMain_syncSource() {

    @EnableFlags({Flags.FLAG_ENABLE_LE_AUDIO_SHARING, Flags.FLAG_ENABLE_TEMPORARY_BOND_DEVICES_UI})

    public void addMemberDevicesIntoMainDevice_preferredDeviceIsMainAndTwoMain_syncState() {

        // Condition: The preferredDevice is main and there is another main device in top list

        // Expected Result: return true and there is the preferredDevice in top list

        CachedBluetoothDevice preferredDevice = mCachedDevice1;
@@ -399,7 +409,6 @@ public class CsipDeviceManagerTest { 
        mCachedDevices.add(preferredDevice);

        mCachedDevices.add(mCachedDevice2);

        mCachedDevices.add(mCachedDevice3);

        mSetFlagsRule.enableFlags(Flags.FLAG_ENABLE_LE_AUDIO_SHARING);

        when(mBroadcast.isEnabled(null)).thenReturn(true);

        BluetoothLeBroadcastMetadata metadata = Mockito.mock(BluetoothLeBroadcastMetadata.class);

        when(mBroadcast.getLatestBluetoothLeBroadcastMetadata()).thenReturn(metadata);
@@ -407,6 +416,8 @@ public class CsipDeviceManagerTest { 
                BluetoothLeBroadcastReceiveState.class);

        when(state.getBisSyncState()).thenReturn(ImmutableList.of(1L));

        when(mAssistant.getAllSources(mDevice2)).thenReturn(ImmutableList.of(state));

        when(mDevice2.getMetadata(METADATA_FAST_PAIR_CUSTOMIZED_FIELDS))

                .thenReturn(TEMP_BOND_METADATA.getBytes());



        assertThat(mCsipDeviceManager.addMemberDevicesIntoMainDevice(GROUP1, preferredDevice))

                .isTrue();
@@ -415,6 +426,8 @@ public class CsipDeviceManagerTest { 
        assertThat(mCachedDevices.contains(mCachedDevice3)).isTrue();

        assertThat(preferredDevice.getMemberDevice()).contains(mCachedDevice2);

        verify(mAssistant).addSource(mDevice1, metadata, /* isGroupOp= */ false);

        verify(mDevice1).setMetadata(METADATA_FAST_PAIR_CUSTOMIZED_FIELDS,

                TEMP_BOND_METADATA.getBytes());

    }



    @Test
@@ -436,13 +449,13 @@ public class CsipDeviceManagerTest { 
    }



    @Test

    @EnableFlags({Flags.FLAG_ENABLE_LE_AUDIO_SHARING, Flags.FLAG_ENABLE_TEMPORARY_BOND_DEVICES_UI})

    public void addMemberDevicesIntoMainDevice_preferredDeviceIsMemberAndTwoMain_returnTrue() {

        // Condition: The preferredDevice is member and there are two main device in top list

        // Expected Result: return true and there is the preferredDevice in top list

        CachedBluetoothDevice preferredDevice = mCachedDevice2;

        BluetoothDevice expectedMainBluetoothDevice = preferredDevice.getDevice();

        mCachedDevice3.setGroupId(GROUP1);

        mSetFlagsRule.enableFlags(Flags.FLAG_ENABLE_LE_AUDIO_SHARING);

        when(mBroadcast.isEnabled(null)).thenReturn(false);



        assertThat(mCsipDeviceManager.addMemberDevicesIntoMainDevice(GROUP1, preferredDevice))
@@ -457,16 +470,20 @@ public class CsipDeviceManagerTest { 
        assertThat(mCachedDevice1.getDevice()).isEqualTo(expectedMainBluetoothDevice);

        verify(mAssistant, never()).addSource(any(BluetoothDevice.class),

                any(BluetoothLeBroadcastMetadata.class), anyBoolean());

        verify(mDevice2, never()).setMetadata(METADATA_FAST_PAIR_CUSTOMIZED_FIELDS,

                TEMP_BOND_METADATA.getBytes());

        verify(mDevice3, never()).setMetadata(METADATA_FAST_PAIR_CUSTOMIZED_FIELDS,

                TEMP_BOND_METADATA.getBytes());

    }



    @Test

    public void addMemberDevicesIntoMainDevice_preferredDeviceIsMemberAndTwoMain_syncSource() {

    @EnableFlags({Flags.FLAG_ENABLE_LE_AUDIO_SHARING, Flags.FLAG_ENABLE_TEMPORARY_BOND_DEVICES_UI})

    public void addMemberDevicesIntoMainDevice_preferredDeviceIsMemberAndTwoMain_syncState() {

        // Condition: The preferredDevice is member and there are two main device in top list

        // Expected Result: return true and there is the preferredDevice in top list

        CachedBluetoothDevice preferredDevice = mCachedDevice2;

        BluetoothDevice expectedMainBluetoothDevice = preferredDevice.getDevice();

        mCachedDevice3.setGroupId(GROUP1);

        mSetFlagsRule.enableFlags(Flags.FLAG_ENABLE_LE_AUDIO_SHARING);

        when(mBroadcast.isEnabled(null)).thenReturn(true);

        BluetoothLeBroadcastMetadata metadata = Mockito.mock(BluetoothLeBroadcastMetadata.class);

        when(mBroadcast.getLatestBluetoothLeBroadcastMetadata()).thenReturn(metadata);
@@ -474,6 +491,8 @@ public class CsipDeviceManagerTest { 
                BluetoothLeBroadcastReceiveState.class);

        when(state.getBisSyncState()).thenReturn(ImmutableList.of(1L));

        when(mAssistant.getAllSources(mDevice1)).thenReturn(ImmutableList.of(state));

        when(mDevice1.getMetadata(METADATA_FAST_PAIR_CUSTOMIZED_FIELDS))

                .thenReturn(TEMP_BOND_METADATA.getBytes());



        assertThat(mCsipDeviceManager.addMemberDevicesIntoMainDevice(GROUP1, preferredDevice))

                .isTrue();
@@ -488,6 +507,10 @@ public class CsipDeviceManagerTest { 
        assertThat(mCachedDevice1.getDevice()).isEqualTo(expectedMainBluetoothDevice);

        verify(mAssistant).addSource(mDevice2, metadata, /* isGroupOp= */ false);

        verify(mAssistant).addSource(mDevice3, metadata, /* isGroupOp= */ false);

        verify(mDevice2).setMetadata(METADATA_FAST_PAIR_CUSTOMIZED_FIELDS,

                TEMP_BOND_METADATA.getBytes());

        verify(mDevice3).setMetadata(METADATA_FAST_PAIR_CUSTOMIZED_FIELDS,

                TEMP_BOND_METADATA.getBytes());

    }



    @Test