loliball/media3
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add two helpful methods to FakeMediaSource for testing

Browse Source 
PiperOrigin-RevId: 566636545
This commit is contained in:
tianyifeng 2023-09-19 08:12:06 -07:00 committed by Copybara-Service
parent 39c1ac7f3a
commit ceac1f458f
1 changed files with 25 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
libraries/test_utils/src/main/java/androidx/media3/test/utils/FakeMediaSource.java
View File

@ -49,6 +49,7 @@ import androidx.media3.exoplayer.source.TrackGroupArray;
import androidx.media3.exoplayer.upstream.Allocator;
import androidx.media3.test.utils.FakeMediaPeriod.TrackDataFactory;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Iterables;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
@ -106,6 +107,7 @@ public class FakeMediaSource extends BaseMediaSource {
private boolean releasedSource;
@Nullable private Handler sourceInfoRefreshHandler;
@Nullable private TransferListener transferListener;
private boolean periodDefersOnPreparedCallback;
/** Creates a {@link FakeMediaSource} with a default {@link FakeTimeline}. */
public FakeMediaSource() {
@ -359,6 +361,27 @@ public class FakeMediaSource extends BaseMediaSource {
return createdMediaPeriods;
}
/**
* Sets whether the created {@link FakeMediaPeriod period} should defer to call {@link
* MediaPeriod.Callback#onPrepared(MediaPeriod)}. If set to true, {@link
* MediaPeriod.Callback#onPrepared(MediaPeriod)} should be called only after {@link
* FakeMediaPeriod#setPreparationComplete()} has been called, otherwise the preparation completes
* immediately.
*/
public void setPeriodDefersOnPreparedCallback(boolean periodDefersOnPreparedCallback) {
this.periodDefersOnPreparedCallback = periodDefersOnPreparedCallback;
}
/**
* Returns the last created active {@link MediaPeriod}.
*
* <p>Must only be called if the source has created at least one period and it hasn't been
* released.
*/
public MediaPeriod getLastCreatedActiveMediaPeriod() {
return Iterables.getLast(activeMediaPeriods);
}
/**
* Creates a {@link MediaPeriod} for this media source.
*
@ -372,7 +395,7 @@ public class FakeMediaSource extends BaseMediaSource {
* events.
* @param transferListener The transfer listener which should be informed of any data transfers.
* May be null if no listener is available.
* @return A new {@link FakeMediaPeriod}.
* @return A new {@link MediaPeriod}.
*/
@RequiresNonNull("this.timeline")
protected MediaPeriod createMediaPeriod(
@ -395,7 +418,7 @@ public class FakeMediaSource extends BaseMediaSource {
mediaSourceEventDispatcher,
drmSessionManager,
drmEventDispatcher,
/* deferOnPrepared= */ false);
periodDefersOnPreparedCallback);
}
/**