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

Commit 8ba2137e authored by Ze Li's avatar Ze Li
Browse files

Fix some old tests failing due to flag progress 

Test: com.android.settingslib.bluetooth.CachedBluetoothDeviceTest
Flag: com.android.settingslib.flags.adopt_primary_group_management_api_v2
Change-Id: I9fa5cc90c2e9b261d97455df5c3722ba9148500a
parent 15a6ed45

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

+53 −2
Original line number Diff line number Diff line
@@ -2326,7 +2326,57 @@ public class CachedBluetoothDeviceTest { 
    }



    @Test

    public void getConnectionSummary_GuestDeviceBroadcastPrimary_activeDevice_returnActive() {

    @EnableFlags(com.android.settingslib.flags.Flags.FLAG_ADOPT_PRIMARY_GROUP_MANAGEMENT_API_V2)

    public void

            getConnectionSummary_adoptAPI_guestDeviceBroadcastPrimary_activeDevice_returnActive() {

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

        when(mCachedDevice.getDevice()).thenReturn(mDevice);

        when(mLeAudioProfile.getBroadcastToUnicastFallbackGroup()).thenReturn(

                BluetoothCsipSetCoordinator.GROUP_ID_INVALID);

        when(mDevice.getMetadata(METADATA_FAST_PAIR_CUSTOMIZED_FIELDS))

                .thenReturn(TEMP_BOND_METADATA.getBytes());



        List<Long> bisSyncState = new ArrayList<>();

        bisSyncState.add(1L);

        when(mLeBroadcastReceiveState.getBisSyncState()).thenReturn(bisSyncState);

        List<BluetoothLeBroadcastReceiveState> sourceList = new ArrayList<>();

        sourceList.add(mLeBroadcastReceiveState);

        when(mAssistant.getAllSources(any())).thenReturn(sourceList);



        when(mCachedDevice.getGroupId()).thenReturn(1);

        when(mCachedDevice.isActiveDevice(BluetoothProfile.LE_AUDIO)).thenReturn(true);



        assertThat(mCachedDevice.getConnectionSummary(false))

                .isEqualTo(mContext.getString(R.string.bluetooth_active_no_battery_level));

    }



    @Test

    @EnableFlags(com.android.settingslib.flags.Flags.FLAG_ADOPT_PRIMARY_GROUP_MANAGEMENT_API_V2)

    public void

            getConnectionSummary_adoptAPI_guestDeviceNotPrimary_activeDevice_returnGuestMedia() {

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

        when(mCachedDevice.getDevice()).thenReturn(mDevice);

        when(mLeAudioProfile.getBroadcastToUnicastFallbackGroup()).thenReturn(1);

        when(mDevice.getMetadata(METADATA_FAST_PAIR_CUSTOMIZED_FIELDS))

                .thenReturn(TEMP_BOND_METADATA.getBytes());



        List<Long> bisSyncState = new ArrayList<>();

        bisSyncState.add(1L);

        when(mLeBroadcastReceiveState.getBisSyncState()).thenReturn(bisSyncState);

        List<BluetoothLeBroadcastReceiveState> sourceList = new ArrayList<>();

        sourceList.add(mLeBroadcastReceiveState);

        when(mAssistant.getAllSources(any())).thenReturn(sourceList);



        when(mCachedDevice.getGroupId()).thenReturn(BluetoothCsipSetCoordinator.GROUP_ID_INVALID);



        assertThat(mCachedDevice.getConnectionSummary(false))

                .isEqualTo(

                        mContext.getString(R.string.bluetooth_guest_media_only_no_battery_level));

    }



    @Test

    @DisableFlags(com.android.settingslib.flags.Flags.FLAG_ADOPT_PRIMARY_GROUP_MANAGEMENT_API_V2)

    public void getConnectionSummary_guestDeviceBroadcastPrimary_activeDevice_returnActive() {

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

        when(mCachedDevice.getDevice()).thenReturn(mDevice);

        Settings.Secure.putInt(
@@ -2351,7 +2401,8 @@ public class CachedBluetoothDeviceTest { 
    }



    @Test

    public void getConnectionSummary_GuestDeviceBroadcastSecondary_activeDevice_returnGuestMedia() {

    @DisableFlags(com.android.settingslib.flags.Flags.FLAG_ADOPT_PRIMARY_GROUP_MANAGEMENT_API_V2)

    public void getConnectionSummary_guestDeviceBroadcastSecondary_activeDevice_returnGuestMedia() {

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

        when(mCachedDevice.getDevice()).thenReturn(mDevice);

        Settings.Secure.putInt(