Session tests: Add missing CountdownLatch.await()

This helps deflake this test.

PiperOrigin-RevId: 567288892
This commit is contained in:
ibaker 2023-09-21 06:27:22 -07:00 committed by Copybara-Service
parent b4d03a9515
commit fc91ee1d7b

View File

@ -616,7 +616,7 @@ public class MediaControllerCompatPlaybackStateCompatActionsWithMediaSessionTest
/* availableCommands= */ new Player.Commands.Builder() /* availableCommands= */ new Player.Commands.Builder()
.add(Player.COMMAND_SEEK_TO_NEXT) .add(Player.COMMAND_SEEK_TO_NEXT)
.build(), .build(),
/* excludedCommand= */ new Player.Commands.Builder() /* excludedCommands= */ new Player.Commands.Builder()
.add(Player.COMMAND_SEEK_TO_NEXT_MEDIA_ITEM) .add(Player.COMMAND_SEEK_TO_NEXT_MEDIA_ITEM)
.build()); .build());
MediaSession mediaSession = createMediaSession(player); MediaSession mediaSession = createMediaSession(player);
@ -1515,6 +1515,7 @@ public class MediaControllerCompatPlaybackStateCompatActionsWithMediaSessionTest
getInstrumentation().runOnMainSync(() -> mediaSession.setCustomLayout(customLayout)); getInstrumentation().runOnMainSync(() -> mediaSession.setCustomLayout(customLayout));
assertThat(latch.await(TIMEOUT_MS, MILLISECONDS)).isTrue();
assertThat(reportedCustomLayout.get()) assertThat(reportedCustomLayout.get())
.containsExactly( .containsExactly(
customLayout.get(0).copyWithIsEnabled(true), customLayout.get(0).copyWithIsEnabled(true),