diff options
| -rw-r--r-- | packages/SystemUI/tests/src/com/android/systemui/communal/CommunalManagerUpdaterTest.java | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/packages/SystemUI/tests/src/com/android/systemui/communal/CommunalManagerUpdaterTest.java b/packages/SystemUI/tests/src/com/android/systemui/communal/CommunalManagerUpdaterTest.java index fb8efa97a7cb..abc20994749a 100644 --- a/packages/SystemUI/tests/src/com/android/systemui/communal/CommunalManagerUpdaterTest.java +++ b/packages/SystemUI/tests/src/com/android/systemui/communal/CommunalManagerUpdaterTest.java @@ -16,12 +16,14 @@ package com.android.systemui.communal; +import static org.mockito.ArgumentMatchers.any; +import static org.mockito.Mockito.clearInvocations; +import static org.mockito.Mockito.doAnswer; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.verify; import android.app.communal.CommunalManager; import android.testing.AndroidTestingRunner; -import android.testing.TestableLooper; import androidx.test.filters.SmallTest; @@ -36,7 +38,6 @@ import org.mockito.MockitoAnnotations; @SmallTest @RunWith(AndroidTestingRunner.class) -@TestableLooper.RunWithLooper public class CommunalManagerUpdaterTest extends SysuiTestCase { private CommunalSourceMonitor mMonitor; @Mock @@ -49,9 +50,16 @@ public class CommunalManagerUpdaterTest extends SysuiTestCase { MockitoAnnotations.initMocks(this); mContext.addMockSystemService(CommunalManager.class, mCommunalManager); + doAnswer(invocation -> { + final CommunalConditionsMonitor.Callback callback = invocation.getArgument(0); + callback.onConditionsChanged(true); + return null; + }).when(mCommunalConditionsMonitor).addCallback(any()); + mMonitor = new CommunalSourceMonitor(mCommunalConditionsMonitor); final CommunalManagerUpdater updater = new CommunalManagerUpdater(mContext, mMonitor); updater.start(); + clearInvocations(mCommunalManager); } @Test |