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

Add test for seek map duration correction in MP3 index seeker

Browse Source 
PiperOrigin-RevId: 295543916
This commit is contained in:
kimvde 2020-02-17 10:32:55 +00:00 committed by Ian Baker
parent 74443f47f0
commit b1b6c8dfea
1 changed files with 22 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
library/extractor/src/test/java/com/google/android/exoplayer2/extractor/mp3/IndexSeekerTest.java
View File

@ -16,6 +16,7 @@
package com.google.android.exoplayer2.extractor.mp3;
import static com.google.android.exoplayer2.extractor.mp3.Mp3Extractor.FLAG_ENABLE_INDEX_SEEKING;
import static com.google.android.exoplayer2.testutil.TestUtil.extractAllSamplesFromFile;
import static com.google.common.truth.Truth.assertThat;
import android.net.Uri;
@ -39,7 +40,8 @@ import org.junit.runner.RunWith;
@RunWith(AndroidJUnit4.class)
public class IndexSeekerTest {
private static final String TEST_FILE = "mp3/bear-vbr-xing-header.mp3";
private static final String TEST_FILE_NO_SEEK_TABLE = "mp3/bear-vbr-no-seek-table.mp3";
private static final int TEST_FILE_NO_SEEK_TABLE_DURATION = 2_808_000;
private Mp3Extractor extractor;
private FakeExtractorOutput extractorOutput;
@ -55,18 +57,28 @@ public class IndexSeekerTest {
}
@Test
public void mp3ExtractorReads_returnSeekableSeekMap() throws Exception {
Uri fileUri = TestUtil.buildAssetUri(TEST_FILE);
public void mp3ExtractorReads_returnsSeekableSeekMap() throws Exception {
Uri fileUri = TestUtil.buildAssetUri(TEST_FILE_NO_SEEK_TABLE);
SeekMap seekMap = TestUtil.extractSeekMap(extractor, extractorOutput, dataSource, fileUri);
assertThat(seekMap.getDurationUs()).isEqualTo(2_808_000);
assertThat(seekMap.isSeekable()).isTrue();
}
@Test
public void mp3ExtractorReads_correctsInexactDuration() throws Exception {
FakeExtractorOutput extractorOutput =
TestUtil.extractAllSamplesFromFile(
extractor, ApplicationProvider.getApplicationContext(), TEST_FILE_NO_SEEK_TABLE);
SeekMap seekMap = extractorOutput.seekMap;
assertThat(seekMap.getDurationUs()).isEqualTo(TEST_FILE_NO_SEEK_TABLE_DURATION);
}
@Test
public void seeking_handlesSeekToZero() throws Exception {
String fileName = TEST_FILE;
String fileName = TEST_FILE_NO_SEEK_TABLE;
Uri fileUri = TestUtil.buildAssetUri(fileName);
SeekMap seekMap = TestUtil.extractSeekMap(extractor, extractorOutput, dataSource, fileUri);
FakeTrackOutput trackOutput = extractorOutput.trackOutputs.get(0);
@ -84,12 +96,12 @@ public class IndexSeekerTest {
@Test
public void seeking_handlesSeekToEof() throws Exception {
String fileName = TEST_FILE;
String fileName = TEST_FILE_NO_SEEK_TABLE;
Uri fileUri = TestUtil.buildAssetUri(fileName);
SeekMap seekMap = TestUtil.extractSeekMap(extractor, extractorOutput, dataSource, fileUri);
FakeTrackOutput trackOutput = extractorOutput.trackOutputs.get(0);
long targetSeekTimeUs = seekMap.getDurationUs();
long targetSeekTimeUs = TEST_FILE_NO_SEEK_TABLE_DURATION;
int extractedFrameIndex =
TestUtil.seekToTimeUs(
extractor, seekMap, targetSeekTimeUs, dataSource, trackOutput, fileUri);
@ -102,7 +114,7 @@ public class IndexSeekerTest {
@Test
public void seeking_handlesSeekingBackward() throws Exception {
String fileName = TEST_FILE;
String fileName = TEST_FILE_NO_SEEK_TABLE;
Uri fileUri = TestUtil.buildAssetUri(fileName);
SeekMap seekMap = TestUtil.extractSeekMap(extractor, extractorOutput, dataSource, fileUri);
FakeTrackOutput trackOutput = extractorOutput.trackOutputs.get(0);
@ -122,7 +134,7 @@ public class IndexSeekerTest {
@Test
public void seeking_handlesSeekingForward() throws Exception {
String fileName = TEST_FILE;
String fileName = TEST_FILE_NO_SEEK_TABLE;
Uri fileUri = TestUtil.buildAssetUri(fileName);
SeekMap seekMap = TestUtil.extractSeekMap(extractor, extractorOutput, dataSource, fileUri);
FakeTrackOutput trackOutput = extractorOutput.trackOutputs.get(0);
@ -171,7 +183,7 @@ public class IndexSeekerTest {
private static FakeTrackOutput getExpectedTrackOutput(String fileName)
throws IOException, InterruptedException {
return TestUtil.extractAllSamplesFromFile(
return extractAllSamplesFromFile(
new Mp3Extractor(FLAG_ENABLE_INDEX_SEEKING),
ApplicationProvider.getApplicationContext(),
fileName)