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);