ChoreographerTest: deflake testPostFrameCallbackWithoutDelay
Bug: 421119185
Test: presubmit
Flag: EXEMPT: bugfix
Change-Id: I70403b5656022cd7ac006801d63179cab445b02a
diff --git a/tests/surfacecontrol/src/android/view/surfacecontrol/cts/ChoreographerTest.java b/tests/surfacecontrol/src/android/view/surfacecontrol/cts/ChoreographerTest.java
index 4176803..d4bf0a4 100644
--- a/tests/surfacecontrol/src/android/view/surfacecontrol/cts/ChoreographerTest.java
+++ b/tests/surfacecontrol/src/android/view/surfacecontrol/cts/ChoreographerTest.java
@@ -220,7 +220,8 @@
ArgumentCaptor<Long> frameTimeNanosCaptor2 = ArgumentCaptor.forClass(Long.class);
verify(addedFrameCallback1, timeout(NOMINAL_VSYNC_PERIOD * 10).times(1))
.doFrame(frameTimeNanosCaptor1.capture());
- verify(addedFrameCallback2, times(1)).doFrame(frameTimeNanosCaptor2.capture());
+ verify(addedFrameCallback2, timeout(NOMINAL_VSYNC_PERIOD).times(1))
+ .doFrame(frameTimeNanosCaptor2.capture());
verifyNoMoreInteractions(removedFrameCallback);
assertTimeDeltaLessThan(frameTimeNanosCaptor1.getValue() - postTimeNanos,
@@ -237,7 +238,8 @@
verify(addedFrameCallback1, timeout(NOMINAL_VSYNC_PERIOD * 10).times(2))
.doFrame(frameTimeNanosCaptor1.capture());
- verify(addedFrameCallback2, times(1)).doFrame(frameTimeNanosCaptor2.capture());
+ verify(addedFrameCallback2, timeout(NOMINAL_VSYNC_PERIOD).times(1))
+ .doFrame(frameTimeNanosCaptor2.capture());
verifyNoMoreInteractions(removedFrameCallback);
assertTimeDeltaLessThan(frameTimeNanosCaptor1.getAllValues().get(1) - postTimeNanos,
NOMINAL_VSYNC_PERIOD * 10 * NANOS_PER_MS);