Fix flaky unit tests

PiperOrigin-RevId: 290910992

library/core/src/test/java/com/google/android/exoplayer2/mediacodec/DedicatedThreadAsyncMediaCodecAdapterTest.java

@@ -290,11 +290,14 @@ public class DedicatedThreadAsyncMediaCodecAdapterTest {
     AtomicInteger onCodecStartCount = new AtomicInteger(0);
     AtomicInteger onCodecStartCount = new AtomicInteger(0);
     adapter.setCodecStartRunnable(() -> onCodecStartCount.incrementAndGet());
     adapter.setCodecStartRunnable(() -> onCodecStartCount.incrementAndGet());
     adapter.start();
     adapter.start();
+    // Grab reference to Looper before shutting down the adapter otherwise handlerThread.getLooper()
+    // might return null.
+    Looper looper = handlerThread.getLooper();
     adapter.flush();
     adapter.flush();
     adapter.shutdown();
     adapter.shutdown();
 
 
     // Wait until all tasks have been handled.
     // Wait until all tasks have been handled.
-    Shadows.shadowOf(handlerThread.getLooper()).idle();
+    Shadows.shadowOf(looper).idle();
     // Only adapter.start() calls onCodecStart.
     // Only adapter.start() calls onCodecStart.
     assertThat(onCodecStartCount.get()).isEqualTo(1);
     assertThat(onCodecStartCount.get()).isEqualTo(1);
   }
   }

library/core/src/test/java/com/google/android/exoplayer2/mediacodec/MultiLockAsyncMediaCodecAdapterTest.java

@@ -290,10 +290,13 @@ public class MultiLockAsyncMediaCodecAdapterTest {
     AtomicInteger codecStartCalls = new AtomicInteger(0);
     AtomicInteger codecStartCalls = new AtomicInteger(0);
     adapter.setCodecStartRunnable(() -> codecStartCalls.incrementAndGet());
     adapter.setCodecStartRunnable(() -> codecStartCalls.incrementAndGet());
     adapter.start();
     adapter.start();
+    // Grab reference to Looper before shutting down the adapter otherwise handlerThread.getLooper()
+    // might return null.
+    Looper looper = handlerThread.getLooper();
     adapter.flush();
     adapter.flush();
     adapter.shutdown();
     adapter.shutdown();
 
 
-    Shadows.shadowOf(handlerThread.getLooper()).idle();
+    Shadows.shadowOf(looper).idle();
     // Only adapter.start() called codec#start()
     // Only adapter.start() called codec#start()
     assertThat(codecStartCalls.get()).isEqualTo(1);
     assertThat(codecStartCalls.get()).isEqualTo(1);
   }
   }