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

Commit e594252b authored by TreeHugger Robot's avatar TreeHugger Robot Committed by Automerger Merge Worker
Browse files

Merge "Copy keys to iterate over so we don't concurrently modify the map."... 

Merge "Copy keys to iterate over so we don't concurrently modify the map." into rvc-dev am: 5a7298f6 am: 936dc2cf

Original change: https://googleplex-android-review.googlesource.com/c/platform/frameworks/base/+/12001034

Change-Id: I902ba4184269e8ba1ab2a444ea056207ebf8ca04
parents afc7a382 936dc2cf

services/core/java/com/android/server/notification/ShortcutHelper.java

+6 −1
Original line number Diff line number Diff line
@@ -37,7 +37,9 @@ import java.util.ArrayList; 
import java.util.Arrays;

import java.util.Collections;

import java.util.HashMap;

import java.util.HashSet;

import java.util.List;

import java.util.Set;



/**

 * Helper for querying shortcuts.
@@ -249,8 +251,11 @@ public class ShortcutHelper { 
                if (!TextUtils.isEmpty(shortcutId)) {

                    packageBubbles.remove(shortcutId);

                } else {

                    // Copy the shortcut IDs to avoid a concurrent modification exception.

                    final Set<String> shortcutIds = new HashSet<>(packageBubbles.keySet());



                    // Check if there was a matching entry

                    for (String pkgShortcutId : packageBubbles.keySet()) {

                    for (String pkgShortcutId : shortcutIds) {

                        String entryKey = packageBubbles.get(pkgShortcutId);

                        if (r.getKey().equals(entryKey)) {

                            // No longer has shortcut id so remove it

services/tests/uiservicestests/src/com/android/server/notification/ShortcutHelperTest.java

+35 −7
Original line number Diff line number Diff line
@@ -45,6 +45,7 @@ import org.junit.Test; 
import org.junit.runner.RunWith;

import org.mockito.ArgumentCaptor;

import org.mockito.Mock;

import org.mockito.Mockito;

import org.mockito.MockitoAnnotations;



import java.util.ArrayList;
@@ -85,14 +86,19 @@ public class ShortcutHelperTest extends UiServiceTestCase { 


        mShortcutHelper = new ShortcutHelper(

                mLauncherApps, mShortcutListener, mShortcutServiceInternal);

        when(mNr.getKey()).thenReturn(KEY);

        when(mNr.getSbn()).thenReturn(mSbn);

        when(mSbn.getPackageName()).thenReturn(PKG);

        when(mNr.getNotification()).thenReturn(mNotif);

        when(mNr.getShortcutInfo()).thenReturn(mShortcutInfo);

        when(mShortcutInfo.getId()).thenReturn(SHORTCUT_ID);

        when(mNotif.getBubbleMetadata()).thenReturn(mBubbleMetadata);

        when(mBubbleMetadata.getShortcutId()).thenReturn(SHORTCUT_ID);



        setUpMockNotificationRecord(mNr, KEY);

    }



    private void setUpMockNotificationRecord(NotificationRecord mockRecord, String key) {

        when(mockRecord.getKey()).thenReturn(key);

        when(mockRecord.getSbn()).thenReturn(mSbn);

        when(mockRecord.getNotification()).thenReturn(mNotif);

        when(mockRecord.getShortcutInfo()).thenReturn(mShortcutInfo);

    }



    private LauncherApps.Callback addShortcutBubbleAndVerifyListener() {
@@ -159,9 +165,31 @@ public class ShortcutHelperTest extends UiServiceTestCase { 
        // First set it up to listen

        addShortcutBubbleAndVerifyListener();



        // Clear out shortcutId

        when(mNr.getShortcutInfo()).thenReturn(null);

        mShortcutHelper.maybeListenForShortcutChangesForBubbles(mNr,

        NotificationRecord validMock1 = Mockito.mock(NotificationRecord.class);

        setUpMockNotificationRecord(validMock1, "KEY1");



        NotificationRecord validMock2 = Mockito.mock(NotificationRecord.class);

        setUpMockNotificationRecord(validMock2, "KEY2");



        NotificationRecord validMock3 = Mockito.mock(NotificationRecord.class);

        setUpMockNotificationRecord(validMock3, "KEY3");



        mShortcutHelper.maybeListenForShortcutChangesForBubbles(validMock1,

                false /* removed */,

                null /* handler */);



        mShortcutHelper.maybeListenForShortcutChangesForBubbles(validMock2,

                false /* removed */,

                null /* handler */);



        mShortcutHelper.maybeListenForShortcutChangesForBubbles(validMock3,

                false /* removed */,

                null /* handler */);



        // Clear out shortcutId of the bubble in the middle, to double check that we don't hit a

        // concurrent modification exception (removing the last bubble would sidestep that check).

        when(validMock2.getShortcutInfo()).thenReturn(null);

        mShortcutHelper.maybeListenForShortcutChangesForBubbles(validMock2,

                false /* removed */,

                null /* handler */);