MediaControllerWithMediaSessionCompatTest

Fix a race condition making test flaky.

PiperOrigin-RevId: 480104467
(cherry picked from commit cb3a6725)

libraries/test_session_current/src/androidTest/java/androidx/media3/session/MediaControllerWithMediaSessionCompatTest.java

@@ -168,6 +168,7 @@ public class MediaControllerWithMediaSessionCompatTest {
   @Test
   @Test
   public void disconnected_byControllerReleaseRightAfterCreated() throws Exception {
   public void disconnected_byControllerReleaseRightAfterCreated() throws Exception {
     CountDownLatch latch = new CountDownLatch(1);
     CountDownLatch latch = new CountDownLatch(1);
+    AtomicReference<Exception> exception = new AtomicReference<>();
     MediaController controller =
     MediaController controller =
         controllerTestRule.createController(
         controllerTestRule.createController(
             session.getSessionToken(),
             session.getSessionToken(),
@@ -177,8 +178,16 @@ public class MediaControllerWithMediaSessionCompatTest {
                 latch.countDown();
                 latch.countDown();
               }
               }
             },
             },
-            /* controllerCreationListener= */ MediaController::release);
+            /* controllerCreationListener= */ mediaController -> {
+              // We must release the controller on the app thread.
+              try {
+                threadTestRule.getHandler().postAndSync(() -> mediaController.release());
+              } catch (Exception e) {
+                exception.set(e);
+              }
+            });
     assertThat(latch.await(TIMEOUT_MS, MILLISECONDS)).isTrue();
     assertThat(latch.await(TIMEOUT_MS, MILLISECONDS)).isTrue();
+    assertThat(exception.get()).isNull();
     assertThat(controller.isConnected()).isFalse();
     assertThat(controller.isConnected()).isFalse();
   }
   }