Remove CountDownLatch from MockPlayer

The MockPlayer has a single CountDownLatch field and multiple boolean
flags that track if a player method was called. Upon calling the methods
the latch count. Tests set the latch count to match exactly with the
number of expected player interactions then block the test thread until
the latch reaches zero and assert the respective method flags are true.

This is subject to false positives. If the underneath implementation
changes and call more player method, then the test thread will unblock
as soon as a certain number of interactions is performed, which may be
less than what the test expected originally. However, the test may stil
pass if the player thread had enough time to update the expected method
flag.

This change removes the single CountDownLatch and the boolean flags and
instead it adds APIs to query the MockPlayer if a method has been called
and await until a method is called. Internally, the MockPlayer has a
ConditionVariable per method.

PiperOrigin-RevId: 432399077
(cherry picked from commit 45d51216)

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

@@ -181,7 +181,6 @@ public class MediaSessionAndControllerTest {
     MockPlayer player =
         new MockPlayer.Builder()
             .setApplicationLooper(threadTestRule.getHandler().getLooper())
-            .setLatchCount(1)
             .build();
     MediaSession session =
         sessionTestRule.ensureReleaseAfterTest(
@@ -190,8 +189,7 @@ public class MediaSessionAndControllerTest {
 
     threadTestRule.getHandler().postAndSync(controller::play);
 
-    assertThat(player.countDownLatch.await(TIMEOUT_MS, MILLISECONDS)).isTrue();
-    assertThat(player.playCalled).isTrue();
+    player.awaitMethodCalled(MockPlayer.METHOD_PLAY, TIMEOUT_MS);
   }
 
   @Test

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

@@ -74,7 +74,6 @@ public class MediaSessionCallbackTest {
     context = ApplicationProvider.getApplicationContext();
     player =
         new MockPlayer.Builder()
-            .setLatchCount(1)
             .setApplicationLooper(threadTestRule.getHandler().getLooper())
             .build();
   }
@@ -157,15 +156,14 @@ public class MediaSessionCallbackTest {
         controllerTestRule.createRemoteController(session.getToken());
 
     controller.prepare();
-    assertThat(player.countDownLatch.await(NO_RESPONSE_TIMEOUT_MS, MILLISECONDS)).isFalse();
-    assertThat(player.prepareCalled).isFalse();
+    Thread.sleep(NO_RESPONSE_TIMEOUT_MS);
+    assertThat(player.hasMethodBeenCalled(MockPlayer.METHOD_PREPARE)).isFalse();
     assertThat(commands).hasSize(1);
     assertThat(commands.get(0)).isEqualTo(Player.COMMAND_PREPARE);
 
     controller.play();
-    assertThat(player.countDownLatch.await(TIMEOUT_MS, MILLISECONDS)).isTrue();
-    assertThat(player.playCalled).isTrue();
-    assertThat(player.prepareCalled).isFalse();
+    player.awaitMethodCalled(MockPlayer.METHOD_PLAY, TIMEOUT_MS);
+    assertThat(player.hasMethodBeenCalled(MockPlayer.METHOD_PREPARE)).isFalse();
     assertThat(commands).hasSize(2);
     assertThat(commands.get(1)).isEqualTo(Player.COMMAND_PLAY_PAUSE);
   }

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

@@ -89,8 +89,7 @@ public class MediaSessionKeyEventTest {
     Context context = ApplicationProvider.getApplicationContext();
     audioManager = (AudioManager) context.getSystemService(Context.AUDIO_SERVICE);
     handler = threadTestRule.getHandler();
-    player =
-        new MockPlayer.Builder().setLatchCount(1).setApplicationLooper(handler.getLooper()).build();
+    player = new MockPlayer.Builder().setApplicationLooper(handler.getLooper()).build();
 
     sessionCallback = new TestSessionCallback();
     session = new MediaSession.Builder(context, player).setSessionCallback(sessionCallback).build();
@@ -143,43 +142,43 @@ public class MediaSessionKeyEventTest {
   @Test
   public void playKeyEvent() throws Exception {
     dispatchMediaKeyEvent(KeyEvent.KEYCODE_MEDIA_PLAY, false);
-    assertThat(player.countDownLatch.await(TIMEOUT_MS, MILLISECONDS)).isTrue();
-    assertThat(player.playCalled).isTrue();
+
+    player.awaitMethodCalled(MockPlayer.METHOD_PLAY, TIMEOUT_MS);
   }
 
   @Test
   public void pauseKeyEvent() throws Exception {
     dispatchMediaKeyEvent(KeyEvent.KEYCODE_MEDIA_PAUSE, false);
-    assertThat(player.countDownLatch.await(TIMEOUT_MS, MILLISECONDS)).isTrue();
-    assertThat(player.pauseCalled).isTrue();
+
+    player.awaitMethodCalled(MockPlayer.METHOD_PAUSE, TIMEOUT_MS);
   }
 
   @Test
   public void nextKeyEvent() throws Exception {
     dispatchMediaKeyEvent(KeyEvent.KEYCODE_MEDIA_NEXT, false);
-    assertThat(player.countDownLatch.await(TIMEOUT_MS, MILLISECONDS)).isTrue();
-    assertThat(player.seekToNextCalled).isTrue();
+
+    player.awaitMethodCalled(MockPlayer.METHOD_SEEK_TO_NEXT, TIMEOUT_MS);
   }
 
   @Test
   public void previousKeyEvent() throws Exception {
     dispatchMediaKeyEvent(KeyEvent.KEYCODE_MEDIA_PREVIOUS, false);
-    assertThat(player.countDownLatch.await(TIMEOUT_MS, MILLISECONDS)).isTrue();
-    assertThat(player.seekToPreviousCalled).isTrue();
+
+    player.awaitMethodCalled(MockPlayer.METHOD_SEEK_TO_PREVIOUS, TIMEOUT_MS);
   }
 
   @Test
   public void stopKeyEvent() throws Exception {
     dispatchMediaKeyEvent(KeyEvent.KEYCODE_MEDIA_STOP, false);
-    assertThat(player.countDownLatch.await(TIMEOUT_MS, MILLISECONDS)).isTrue();
-    assertThat(player.stopCalled).isTrue();
+
+    player.awaitMethodCalled(MockPlayer.METHOD_STOP, TIMEOUT_MS);
   }
 
   @Test
   public void playPauseKeyEvent_play() throws Exception {
     dispatchMediaKeyEvent(KeyEvent.KEYCODE_MEDIA_PLAY_PAUSE, false);
-    assertThat(player.countDownLatch.await(TIMEOUT_MS, MILLISECONDS)).isTrue();
-    assertThat(player.playCalled).isTrue();
+
+    player.awaitMethodCalled(MockPlayer.METHOD_PLAY, TIMEOUT_MS);
   }
 
   @Test
@@ -188,18 +187,19 @@ public class MediaSessionKeyEventTest {
         () -> {
           player.playWhenReady = true;
         });
+
     dispatchMediaKeyEvent(KeyEvent.KEYCODE_MEDIA_PLAY_PAUSE, false);
-    assertThat(player.countDownLatch.await(TIMEOUT_MS, MILLISECONDS)).isTrue();
-    assertThat(player.pauseCalled).isTrue();
+
+    player.awaitMethodCalled(MockPlayer.METHOD_PAUSE, TIMEOUT_MS);
   }
 
   @Test
   public void playPauseKeyEvent_doubleTapIsTranslatedToSkipToNext() throws Exception {
     dispatchMediaKeyEvent(KeyEvent.KEYCODE_MEDIA_PLAY_PAUSE, true);
-    assertThat(player.countDownLatch.await(TIMEOUT_MS, MILLISECONDS)).isTrue();
-    assertThat(player.seekToNextCalled).isTrue();
-    assertThat(player.playCalled).isFalse();
-    assertThat(player.pauseCalled).isFalse();
+
+    player.awaitMethodCalled(MockPlayer.METHOD_SEEK_TO_NEXT, TIMEOUT_MS);
+    assertThat(player.hasMethodBeenCalled(MockPlayer.METHOD_PLAY)).isFalse();
+    assertThat(player.hasMethodBeenCalled(MockPlayer.METHOD_PAUSE)).isFalse();
   }
 
   private static class TestSessionCallback implements MediaSession.SessionCallback {

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

@@ -101,7 +101,6 @@ public class MediaSessionPermissionTest {
       SessionCommands sessionCommands, Player.Commands playerCommands) {
     player =
         new MockPlayer.Builder()
-            .setLatchCount(1)
             .setApplicationLooper(threadTestRule.getHandler().getLooper())
             .build();
     callback =

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

@@ -78,8 +78,7 @@ public class MediaSessionTest {
   public void setUp() throws Exception {
     context = ApplicationProvider.getApplicationContext();
     handler = threadTestRule.getHandler();
-    player =
-        new MockPlayer.Builder().setLatchCount(1).setApplicationLooper(handler.getLooper()).build();
+    player = new MockPlayer.Builder().setApplicationLooper(handler.getLooper()).build();
 
     session =
         sessionTestRule.ensureReleaseAfterTest(
@@ -394,8 +393,7 @@ public class MediaSessionTest {
     long testSeekPositionMs = 1234;
     controllerCompat.getTransportControls().seekTo(testSeekPositionMs);
 
-    assertThat(player.countDownLatch.await(TIMEOUT_MS, MILLISECONDS)).isTrue();
-    assertThat(player.seekToCalled).isTrue();
+    player.awaitMethodCalled(MockPlayer.METHOD_SEEK_TO, TIMEOUT_MS);
     assertThat(player.seekPositionMs).isEqualTo(testSeekPositionMs);
   }
 

libraries/test_session_current/src/main/java/androidx/media3/session/MockMediaLibraryService.java

@@ -134,8 +134,7 @@ public class MockMediaLibraryService extends MediaLibraryService {
       return (MediaLibrarySession) onGetSessionHandler.onGetSession(controllerInfo);
     }
 
-    MockPlayer player =
-        new MockPlayer.Builder().setLatchCount(1).setApplicationLooper(handler.getLooper()).build();
+    MockPlayer player = new MockPlayer.Builder().setApplicationLooper(handler.getLooper()).build();
 
     MediaLibrarySessionCallback callback = registry.getSessionCallback();
     session =