diff --git a/quickstep/tests/multivalentScreenshotTests/src/com/android/launcher3/taskbar/bubbles/BubbleBarViewScreenshotTest.kt b/quickstep/tests/multivalentScreenshotTests/src/com/android/launcher3/taskbar/bubbles/BubbleBarViewScreenshotTest.kt index e4b8069b7f..82a76250c2 100644 --- a/quickstep/tests/multivalentScreenshotTests/src/com/android/launcher3/taskbar/bubbles/BubbleBarViewScreenshotTest.kt +++ b/quickstep/tests/multivalentScreenshotTests/src/com/android/launcher3/taskbar/bubbles/BubbleBarViewScreenshotTest.kt @@ -97,6 +97,8 @@ class BubbleBarViewScreenshotTest(emulationSpec: DeviceEmulationSpec) { fun bubbleBarView_expanded_threeBubbles() { // if we're still expanding, wait with taking a screenshot val shouldWait: (ComponentActivity, View) -> Boolean = { _, _ -> bubbleBarView.isExpanding } + // increase the frame limit to allow the animation to end before taking the screenshot + screenshotRule.frameLimit = 50 screenshotRule.screenshotTest( "bubbleBarView_expanded_threeBubbles", checkView = shouldWait,