From fc91ee1d7b07c0d37090f77d0178dacb75179fc8 Mon Sep 17 00:00:00 2001
From: ibaker <ibaker@google.com>
Date: Thu, 21 Sep 2023 06:27:22 -0700
Subject: [PATCH] Session tests: Add missing `CountdownLatch.await()`

This helps deflake this test.

PiperOrigin-RevId: 567288892
---
 ...erCompatPlaybackStateCompatActionsWithMediaSessionTest.java | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/libraries/test_session_current/src/androidTest/java/androidx/media3/session/MediaControllerCompatPlaybackStateCompatActionsWithMediaSessionTest.java b/libraries/test_session_current/src/androidTest/java/androidx/media3/session/MediaControllerCompatPlaybackStateCompatActionsWithMediaSessionTest.java
index 229d315ebc..ea18da0f88 100644
--- a/libraries/test_session_current/src/androidTest/java/androidx/media3/session/MediaControllerCompatPlaybackStateCompatActionsWithMediaSessionTest.java
+++ b/libraries/test_session_current/src/androidTest/java/androidx/media3/session/MediaControllerCompatPlaybackStateCompatActionsWithMediaSessionTest.java
@@ -616,7 +616,7 @@ public class MediaControllerCompatPlaybackStateCompatActionsWithMediaSessionTest
             /* availableCommands= */ new Player.Commands.Builder()
                 .add(Player.COMMAND_SEEK_TO_NEXT)
                 .build(),
-            /* excludedCommand= */ new Player.Commands.Builder()
+            /* excludedCommands= */ new Player.Commands.Builder()
                 .add(Player.COMMAND_SEEK_TO_NEXT_MEDIA_ITEM)
                 .build());
     MediaSession mediaSession = createMediaSession(player);
@@ -1515,6 +1515,7 @@ public class MediaControllerCompatPlaybackStateCompatActionsWithMediaSessionTest
 
     getInstrumentation().runOnMainSync(() -> mediaSession.setCustomLayout(customLayout));
 
+    assertThat(latch.await(TIMEOUT_MS, MILLISECONDS)).isTrue();
     assertThat(reportedCustomLayout.get())
         .containsExactly(
             customLayout.get(0).copyWithIsEnabled(true),