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

Commit dedd662b authored by Bryce Lee's avatar Bryce Lee Committed by Automerger Merge Worker
Browse files

Merge "Check condition presence before updating callbacks." into tm-qpr-dev am: dfc5118d

parents eaa80956 dfc5118d
Loading
Loading
Loading
Loading
+9 −1
Original line number Diff line number Diff line
@@ -92,7 +92,15 @@ public class Monitor {
    }

    private void updateConditionMetState(Condition condition) {
        mConditions.get(condition).stream().forEach(token -> mSubscriptions.get(token).update());
        final ArraySet<Subscription.Token> subscriptions = mConditions.get(condition);

        // It's possible the condition was removed between the time the callback occurred and
        // update was executed on the main thread.
        if (subscriptions == null) {
            return;
        }

        subscriptions.stream().forEach(token -> mSubscriptions.get(token).update());
    }

    /**
+18 −0
Original line number Diff line number Diff line
@@ -159,6 +159,24 @@ public class ConditionMonitorTest extends SysuiTestCase {
        Mockito.clearInvocations(callback);
    }

    // Ensure that updating a callback that is removed doesn't result in an exception due to the
    // absence of the condition.
    @Test
    public void testUpdateRemovedCallback() {
        final Monitor.Callback callback1 =
                mock(Monitor.Callback.class);
        final Monitor.Subscription.Token subscription1 =
                mConditionMonitor.addSubscription(getDefaultBuilder(callback1).build());
        ArgumentCaptor<Condition.Callback> monitorCallback =
                ArgumentCaptor.forClass(Condition.Callback.class);
        mExecutor.runAllReady();
        verify(mCondition1).addCallback(monitorCallback.capture());
        // This will execute first before the handler for onConditionChanged.
        mConditionMonitor.removeSubscription(subscription1);
        monitorCallback.getValue().onConditionChanged(mCondition1);
        mExecutor.runAllReady();
    }

    @Test
    public void addCallback_addFirstCallback_addCallbackToAllConditions() {
        final Monitor.Callback callback1 =