Rearrange condition blocking to ensure Handler thread loops

Without this change, any prepare() operation that requires posting to
the DummyMainThread's handler cannot complete preparation.

PiperOrigin-RevId: 233050367

testutils/src/main/java/com/google/android/exoplayer2/testutil/DummyMainThread.java

@@ -53,7 +53,7 @@ public final class DummyMainThread {
    * Runs the provided {@link Runnable} on the main thread, blocking until execution completes or
    * Runs the provided {@link Runnable} on the main thread, blocking until execution completes or
    * until timeout milliseconds have passed.
    * until timeout milliseconds have passed.
    *
    *
-   * @param timeoutMs the maximum time to wait in milliseconds.
+   * @param timeoutMs The maximum time to wait in milliseconds.
    * @param runnable The {@link Runnable} to run.
    * @param runnable The {@link Runnable} to run.
    */
    */
   public void runOnMainThread(int timeoutMs, final Runnable runnable) {
   public void runOnMainThread(int timeoutMs, final Runnable runnable) {

testutils_robolectric/src/main/java/com/google/android/exoplayer2/testutil/MediaPeriodAsserts.java

@@ -187,13 +187,14 @@ public final class MediaPeriodAsserts {
   private static TrackGroupArray getTrackGroups(MediaPeriod mediaPeriod) {
   private static TrackGroupArray getTrackGroups(MediaPeriod mediaPeriod) {
     AtomicReference<TrackGroupArray> trackGroupArray = new AtomicReference<>(null);
     AtomicReference<TrackGroupArray> trackGroupArray = new AtomicReference<>(null);
     DummyMainThread dummyMainThread = new DummyMainThread();
     DummyMainThread dummyMainThread = new DummyMainThread();
+    ConditionVariable preparedCondition = new ConditionVariable();
     dummyMainThread.runOnMainThread(
     dummyMainThread.runOnMainThread(
         () -> {
         () -> {
-          ConditionVariable preparedCondition = new ConditionVariable();
           mediaPeriod.prepare(
           mediaPeriod.prepare(
               new Callback() {
               new Callback() {
                 @Override
                 @Override
                 public void onPrepared(MediaPeriod mediaPeriod) {
                 public void onPrepared(MediaPeriod mediaPeriod) {
+                  trackGroupArray.set(mediaPeriod.getTrackGroups());
                   preparedCondition.open();
                   preparedCondition.open();
                 }
                 }
 
 
@@ -203,13 +204,12 @@ public final class MediaPeriodAsserts {
                 }
                 }
               },
               },
               /* positionUs= */ 0);
               /* positionUs= */ 0);
+        });
     try {
     try {
       preparedCondition.block();
       preparedCondition.block();
     } catch (InterruptedException e) {
     } catch (InterruptedException e) {
       // Ignore.
       // Ignore.
     }
     }
-          trackGroupArray.set(mediaPeriod.getTrackGroups());
-        });
     dummyMainThread.release();
     dummyMainThread.release();
     return trackGroupArray.get();
     return trackGroupArray.get();
   }
   }