Fix FlingOnBackAnimationCallbackTest flake

Bug: 390517215
Test: FlingOnBackAnimationCallbackTest
Flag: EXEMPT test-only bugfix
Change-Id: I8bd385686ed317b95cfc18ea3582ec4ea3217000

1 files changed, 5 insertions, 3 deletions

diff --git a/packages/SystemUI/tests/src/com/android/systemui/animation/back/FlingOnBackAnimationCallbackTest.kt b/packages/SystemUI/tests/src/com/android/systemui/animation/back/FlingOnBackAnimationCallbackTest.kt
index 75a5768193cf..c81900da2581 100644
--- a/packages/SystemUI/tests/src/com/android/systemui/animation/back/FlingOnBackAnimationCallbackTest.kt
+++ b/packages/SystemUI/tests/src/com/android/systemui/animation/back/FlingOnBackAnimationCallbackTest.kt
@@ -69,9 +69,11 @@ class FlingOnBackAnimationCallbackTest : SysuiTestCase() {
         callback.onBackProgressed(backEventOf(0.6f, 32))
         assertTrue("Assert onBackProgressedCompat called", callback.backProgressedCalled)
         assertEquals("Assert interpolated progress", 0.6f, callback.progressEvent?.progress)
-        getInstrumentation().runOnMainSync { callback.onBackInvoked() }
-        // Assert that onBackInvoked is not called immediately...
-        assertFalse(callback.backInvokedCalled)
+        getInstrumentation().runOnMainSync {
+            callback.onBackInvoked()
+            // Assert that onBackInvoked is not called immediately.
+            assertFalse(callback.backInvokedCalled)
+        }
         // Instead the fling animation is played and eventually onBackInvoked is called.
         callback.backInvokedLatch.await(1000, TimeUnit.MILLISECONDS)
         assertTrue(callback.backInvokedCalled)