diff --git a/quickstep/tests/multivalentTests/src/com/android/quickstep/inputconsumers/NavHandleLongPressInputConsumerTest.java b/quickstep/tests/multivalentTests/src/com/android/quickstep/inputconsumers/NavHandleLongPressInputConsumerTest.java index c18f60476b..98a3607834 100644 --- a/quickstep/tests/multivalentTests/src/com/android/quickstep/inputconsumers/NavHandleLongPressInputConsumerTest.java +++ b/quickstep/tests/multivalentTests/src/com/android/quickstep/inputconsumers/NavHandleLongPressInputConsumerTest.java @@ -92,14 +92,15 @@ public class NavHandleLongPressInputConsumerTest { when(mTopTaskTracker.getCachedTopTask(anyBoolean())).thenReturn(mTaskInfo); when(mDeviceState.getSquaredTouchSlop()).thenReturn(SQUARED_TOUCH_SLOP); when(mDelegate.allowInterceptByParent()).thenReturn(true); - MAIN_EXECUTOR.getHandler().removeCallbacks(mLongPressRunnable); mLongPressTriggered.set(false); when(mNavHandleLongPressHandler.getLongPressRunnable(any())).thenReturn(mLongPressRunnable); initializeObjectUnderTest(); } @After - public void tearDown() { + public void tearDown() throws Exception { + MAIN_EXECUTOR.getHandler().removeCallbacks(mLongPressRunnable); + MAIN_EXECUTOR.submit(() -> null).get(); mContext.onDestroy(); }