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

Commit ee4cf685 authored by Matías Hernández's avatar Matías Hernández Committed by Automerger Merge Worker
Browse files

Fixes to HeadsUpManagerPhoneTest & friends am: d0cd9756 

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



Change-Id: Icda239706c7a0a215710d225b03c23a67aa0320d
Signed-off-by: default avatarAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
parents 4d6eb9b0 d0cd9756

packages/SystemUI/tests/src/com/android/systemui/statusbar/AlertingNotificationManagerTest.java

+30 −28
Original line number Diff line number Diff line
@@ -73,9 +73,7 @@ public class AlertingNotificationManagerTest extends SysuiTestCase { 
    // Number of notifications to use in tests requiring multiple notifications

    private static final int TEST_NUM_NOTIFICATIONS = 4;

    protected static final int TEST_TIMEOUT_TIME = 15000;

    protected final Runnable TEST_TIMEOUT_RUNNABLE = () -> mTimedOut = true;



    private AlertingNotificationManager mAlertingNotificationManager;

    protected final Runnable mTestTimeoutRunnable = () -> mTimedOut = true;



    protected NotificationEntry mEntry;

    protected Handler mTestHandler;
@@ -84,7 +82,7 @@ public class AlertingNotificationManagerTest extends SysuiTestCase { 


    @Mock protected ExpandableNotificationRow mRow;



    private final class TestableAlertingNotificationManager extends AlertingNotificationManager {

    private static class TestableAlertingNotificationManager extends AlertingNotificationManager {

        private AlertEntry mLastCreatedEntry;



        private TestableAlertingNotificationManager(Handler handler) {
@@ -112,8 +110,8 @@ public class AlertingNotificationManagerTest extends SysuiTestCase { 
        }

    }



    protected AlertingNotificationManager createAlertingNotificationManager(Handler handler) {

        return new TestableAlertingNotificationManager(handler);

    protected AlertingNotificationManager createAlertingNotificationManager() {

        return new TestableAlertingNotificationManager(mTestHandler);

    }



    protected StatusBarNotification createNewSbn(int id, Notification n) {
@@ -169,8 +167,6 @@ public class AlertingNotificationManagerTest extends SysuiTestCase { 
                .setSbn(mSbn)

                .build();

        mEntry.setRow(mRow);



        mAlertingNotificationManager = createAlertingNotificationManager(mTestHandler);

    }



    @After
@@ -180,68 +176,74 @@ public class AlertingNotificationManagerTest extends SysuiTestCase { 


    @Test

    public void testShowNotification_addsEntry() {

        mAlertingNotificationManager.showNotification(mEntry);

        AlertingNotificationManager alm = createAlertingNotificationManager();



        assertTrue(mAlertingNotificationManager.isAlerting(mEntry.getKey()));

        assertTrue(mAlertingNotificationManager.hasNotifications());

        assertEquals(mEntry, mAlertingNotificationManager.getEntry(mEntry.getKey()));

        alm.showNotification(mEntry);



        assertTrue(alm.isAlerting(mEntry.getKey()));

        assertTrue(alm.hasNotifications());

        assertEquals(mEntry, alm.getEntry(mEntry.getKey()));

    }



    @Test

    public void testShowNotification_autoDismisses() {

        mAlertingNotificationManager.showNotification(mEntry);

        mTestHandler.postDelayed(TEST_TIMEOUT_RUNNABLE, TEST_TIMEOUT_TIME);

        AlertingNotificationManager alm = createAlertingNotificationManager();



        alm.showNotification(mEntry);

        mTestHandler.postDelayed(mTestTimeoutRunnable, TEST_TIMEOUT_TIME);



        // Wait for remove runnable and then process it immediately

        TestableLooper.get(this).processMessages(1);



        assertFalse("Test timed out", mTimedOut);

        assertFalse(mAlertingNotificationManager.isAlerting(mEntry.getKey()));

        assertFalse(alm.isAlerting(mEntry.getKey()));

    }



    @Test

    public void testRemoveNotification_removeDeferred() {

        mAlertingNotificationManager.showNotification(mEntry);

        AlertingNotificationManager alm = createAlertingNotificationManager();

        alm.showNotification(mEntry);



        // Try to remove but defer, since the notification has not been shown long enough.

        mAlertingNotificationManager.removeNotification(

                mEntry.getKey(), false /* releaseImmediately */);

        alm.removeNotification(mEntry.getKey(), false /* releaseImmediately */);



        assertTrue(mAlertingNotificationManager.isAlerting(mEntry.getKey()));

        assertTrue(alm.isAlerting(mEntry.getKey()));

    }



    @Test

    public void testRemoveNotification_forceRemove() {

        mAlertingNotificationManager.showNotification(mEntry);

        AlertingNotificationManager alm = createAlertingNotificationManager();

        alm.showNotification(mEntry);



        // Remove forcibly with releaseImmediately = true.

        mAlertingNotificationManager.removeNotification(

                mEntry.getKey(), true /* releaseImmediately */);

        alm.removeNotification(mEntry.getKey(), true /* releaseImmediately */);



        assertFalse(mAlertingNotificationManager.isAlerting(mEntry.getKey()));

        assertFalse(alm.isAlerting(mEntry.getKey()));

    }



    @Test

    public void testReleaseAllImmediately() {

        AlertingNotificationManager alm = createAlertingNotificationManager();

        for (int i = 0; i < TEST_NUM_NOTIFICATIONS; i++) {

            StatusBarNotification sbn = createNewNotification(i);

            NotificationEntry entry = new NotificationEntryBuilder()

                    .setSbn(sbn)

                    .build();

            entry.setRow(mRow);

            mAlertingNotificationManager.showNotification(entry);

            alm.showNotification(entry);

        }



        mAlertingNotificationManager.releaseAllImmediately();

        alm.releaseAllImmediately();



        assertEquals(0, mAlertingNotificationManager.getAllEntries().count());

        assertEquals(0, alm.getAllEntries().count());

    }



    @Test

    public void testCanRemoveImmediately_notShownLongEnough() {

        mAlertingNotificationManager.showNotification(mEntry);

        AlertingNotificationManager alm = createAlertingNotificationManager();

        alm.showNotification(mEntry);



        // The entry has just been added so we should not remove immediately.

        assertFalse(mAlertingNotificationManager.canRemoveImmediately(mEntry.getKey()));

        assertFalse(alm.canRemoveImmediately(mEntry.getKey()));

    }

}

packages/SystemUI/tests/src/com/android/systemui/statusbar/phone/HeadsUpManagerPhoneTest.java

+9 −4
Original line number Diff line number Diff line
@@ -32,6 +32,7 @@ import android.testing.TestableLooper; 
import androidx.test.filters.SmallTest;



import com.android.internal.logging.UiEventLogger;

import com.android.systemui.R;

import com.android.systemui.plugins.statusbar.StatusBarStateController;

import com.android.systemui.shade.ShadeExpansionStateManager;

import com.android.systemui.statusbar.AlertingNotificationManager;
@@ -107,11 +108,13 @@ public class HeadsUpManagerPhoneTest extends AlertingNotificationManagerTest { 
        }

    }



    @Override

    protected AlertingNotificationManager createAlertingNotificationManager() {

        return mHeadsUpManager;

    }



    @Before

    @Override

    public void setUp() {

        AccessibilityManagerWrapper accessibilityMgr =

                mDependency.injectMockDependency(AccessibilityManagerWrapper.class);
@@ -119,8 +122,10 @@ public class HeadsUpManagerPhoneTest extends AlertingNotificationManagerTest { 
                .thenReturn(TEST_AUTO_DISMISS_TIME);

        when(mVSProvider.isReorderingAllowed()).thenReturn(true);

        mDependency.injectMockDependency(NotificationShadeWindowController.class);

        super.setUp();

        mContext.getOrCreateTestableResources().addOverride(

                R.integer.ambient_notification_extension_time, 500);



        super.setUp();

        mHeadsUpManager = new TestableHeadsUpManagerPhone(

                mContext,

                mHeadsUpManagerLogger,
@@ -137,8 +142,9 @@ public class HeadsUpManagerPhoneTest extends AlertingNotificationManagerTest { 
    }



    @After

    @Override

    public void tearDown() {

        mTestHandler.removeCallbacksAndMessages(null);

        super.tearDown();

    }



    @Test
@@ -184,7 +190,6 @@ public class HeadsUpManagerPhoneTest extends AlertingNotificationManagerTest { 
        assertTrue(mHeadsUpManager.canRemoveImmediately(mEntry.getKey()));

    }





    @Test

    public void testExtendHeadsUp() {

        mHeadsUpManager.showNotification(mEntry);
@@ -192,7 +197,7 @@ public class HeadsUpManagerPhoneTest extends AlertingNotificationManagerTest { 
                () -> mLivesPastNormalTime = mHeadsUpManager.isAlerting(mEntry.getKey());

        mTestHandler.postDelayed(pastNormalTimeRunnable,

                TEST_AUTO_DISMISS_TIME + mHeadsUpManager.mExtensionTime / 2);

        mTestHandler.postDelayed(TEST_TIMEOUT_RUNNABLE, TEST_TIMEOUT_TIME);

        mTestHandler.postDelayed(mTestTimeoutRunnable, TEST_TIMEOUT_TIME);



        mHeadsUpManager.extendHeadsUp();

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

+5 −2
Original line number Diff line number Diff line
@@ -89,11 +89,13 @@ public class HeadsUpManagerTest extends AlertingNotificationManagerTest { 
        }

    }



    @Override

    protected AlertingNotificationManager createAlertingNotificationManager() {

        return mHeadsUpManager;

    }



    @Before

    @Override

    public void setUp() {

        initMocks(this);

        when(mEntry.getSbn()).thenReturn(mSbn);
@@ -105,8 +107,9 @@ public class HeadsUpManagerTest extends AlertingNotificationManagerTest { 
    }



    @After

    @Override

    public void tearDown() {

        mTestHandler.removeCallbacksAndMessages(null);

        super.tearDown();

    }



    @Test
@@ -173,7 +176,7 @@ public class HeadsUpManagerTest extends AlertingNotificationManagerTest { 
                () -> mLivesPastNormalTime = mHeadsUpManager.isAlerting(mEntry.getKey());

        mTestHandler.postDelayed(pastNormalTimeRunnable,

                        (TEST_A11Y_AUTO_DISMISS_TIME + TEST_AUTO_DISMISS_TIME) / 2);

        mTestHandler.postDelayed(TEST_TIMEOUT_RUNNABLE, TEST_A11Y_TIMEOUT_TIME);

        mTestHandler.postDelayed(mTestTimeoutRunnable, TEST_A11Y_TIMEOUT_TIME);



        TestableLooper.get(this).processMessages(2);