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