Use TrackGroup instead of TrackSelection in FakeAdaptiveDataSet.

This was a bug as the data set should contain sample data for all available
tracks. Also added a unit test.

-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=163806579
This commit is contained in:
tonihei 2017-08-01 02:18:48 -07:00 committed by Oliver Woodman
parent 08faafb4c1
commit 1151104e2f
2 changed files with 12 additions and 12 deletions

View file

@ -17,11 +17,11 @@ package com.google.android.exoplayer2.testutil;
import com.google.android.exoplayer2.C;
import com.google.android.exoplayer2.Format;
import com.google.android.exoplayer2.trackselection.TrackSelection;
import com.google.android.exoplayer2.source.TrackGroup;
/**
* Fake data set emulating the data of an adaptive media source.
* It provides chunk data for all {@link Format}s in the given {@link TrackSelection}.
* It provides chunk data for all {@link Format}s in the given {@link TrackGroup}.
*/
public final class FakeAdaptiveDataSet extends FakeDataSet {
@ -36,8 +36,8 @@ public final class FakeAdaptiveDataSet extends FakeDataSet {
this.chunkDurationUs = chunkDurationUs;
}
public FakeAdaptiveDataSet createDataSet(TrackSelection trackSelection, long mediaDurationUs) {
return new FakeAdaptiveDataSet(trackSelection, mediaDurationUs, chunkDurationUs);
public FakeAdaptiveDataSet createDataSet(TrackGroup trackGroup, long mediaDurationUs) {
return new FakeAdaptiveDataSet(trackGroup, mediaDurationUs, chunkDurationUs);
}
}
@ -46,15 +46,14 @@ public final class FakeAdaptiveDataSet extends FakeDataSet {
private final long chunkDurationUs;
private final long lastChunkDurationUs;
public FakeAdaptiveDataSet(TrackSelection trackSelection, long mediaDurationUs,
long chunkDurationUs) {
public FakeAdaptiveDataSet(TrackGroup trackGroup, long mediaDurationUs, long chunkDurationUs) {
this.chunkDurationUs = chunkDurationUs;
int selectionCount = trackSelection.length();
int trackCount = trackGroup.length;
long lastChunkDurationUs = mediaDurationUs % chunkDurationUs;
int fullChunks = (int) (mediaDurationUs / chunkDurationUs);
for (int i = 0; i < selectionCount; i++) {
for (int i = 0; i < trackCount; i++) {
String uri = getUri(i);
Format format = trackSelection.getFormat(i);
Format format = trackGroup.getFormat(i);
int chunkLength = (int) (format.bitrate * chunkDurationUs / (8 * C.MICROS_PER_SECOND));
FakeData newData = this.newData(uri);
for (int j = 0; j < fullChunks; j++) {
@ -74,8 +73,8 @@ public final class FakeAdaptiveDataSet extends FakeDataSet {
return chunkCount;
}
public String getUri(int trackSelectionIndex) {
return "fake://adaptive.media/" + Integer.toString(trackSelectionIndex);
public String getUri(int trackIndex) {
return "fake://adaptive.media/" + trackIndex;
}
public long getChunkDuration(int chunkIndex) {

View file

@ -50,7 +50,8 @@ public final class FakeChunkSource implements ChunkSource {
}
public FakeChunkSource createChunkSource(TrackSelection trackSelection, long durationUs) {
FakeAdaptiveDataSet dataSet = dataSetFactory.createDataSet(trackSelection, durationUs);
FakeAdaptiveDataSet dataSet =
dataSetFactory.createDataSet(trackSelection.getTrackGroup(), durationUs);
dataSourceFactory.setFakeDataSet(dataSet);
DataSource dataSource = dataSourceFactory.createDataSource();
return new FakeChunkSource(trackSelection, dataSource, dataSet);