From 1ace5fa95b73aba590effa450fbb818d73cf2899 Mon Sep 17 00:00:00 2001
From: tonihei <tonihei@google.com>
Date: Fri, 26 Jan 2018 08:17:18 -0800
Subject: [PATCH] Make FakeMediaSource reusable.

GitHub:#3498

-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=183392095
---
 .../google/android/exoplayer2/testutil/FakeMediaSource.java  | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/testutils/src/main/java/com/google/android/exoplayer2/testutil/FakeMediaSource.java b/testutils/src/main/java/com/google/android/exoplayer2/testutil/FakeMediaSource.java
index f2478845aa..17613ce519 100644
--- a/testutils/src/main/java/com/google/android/exoplayer2/testutil/FakeMediaSource.java
+++ b/testutils/src/main/java/com/google/android/exoplayer2/testutil/FakeMediaSource.java
@@ -78,6 +78,7 @@ public class FakeMediaSource implements MediaSource {
       ExoPlayer player, boolean isTopLevelSource, Listener listener) {
     assertThat(preparedSource).isFalse();
     preparedSource = true;
+    releasedSource = false;
     this.listener = listener;
     sourceInfoRefreshHandler = new Handler();
     if (timeline != null) {
@@ -116,6 +117,10 @@ public class FakeMediaSource implements MediaSource {
     assertThat(releasedSource).isFalse();
     assertThat(activeMediaPeriods.isEmpty()).isTrue();
     releasedSource = true;
+    preparedSource = false;
+    sourceInfoRefreshHandler.removeCallbacksAndMessages(null);
+    sourceInfoRefreshHandler = null;
+    listener = null;
   }
 
   /**