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

Switch WebvttParserTest to use partial-output method

Browse Source 
PiperOrigin-RevId: 566598735
This commit is contained in:
ibaker 2023-09-19 05:16:33 -07:00 committed by Copybara-Service
parent 92a3f3a8cd
commit f35e7bdc6a
1 changed files with 47 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
libraries/extractor/src/test/java/androidx/media3/extractor/text/webvtt/WebvttParserTest.java
View File

@ -27,6 +27,7 @@ import androidx.media3.common.text.TextAnnotation;
import androidx.media3.common.util.Assertions; import androidx.media3.common.util.Assertions;
import androidx.media3.common.util.ColorParser; import androidx.media3.common.util.ColorParser;
import androidx.media3.extractor.text.CuesWithTiming; import androidx.media3.extractor.text.CuesWithTiming;
import androidx.media3.extractor.text.SubtitleParser;
import androidx.media3.test.utils.TestUtil; import androidx.media3.test.utils.TestUtil;
import androidx.test.core.app.ApplicationProvider; import androidx.test.core.app.ApplicationProvider;
import androidx.test.ext.junit.runners.AndroidJUnit4; import androidx.test.ext.junit.runners.AndroidJUnit4;
@ -76,7 +77,11 @@ public class WebvttParserTest {
public void parseEmpty() throws IOException { public void parseEmpty() throws IOException {
WebvttParser parser = new WebvttParser(); WebvttParser parser = new WebvttParser();
byte[] bytes = TestUtil.getByteArray(ApplicationProvider.getApplicationContext(), EMPTY_FILE); byte[] bytes = TestUtil.getByteArray(ApplicationProvider.getApplicationContext(), EMPTY_FILE);
assertThrows(IllegalArgumentException.class, () -> parser.parse(bytes)); assertThrows(
IllegalArgumentException.class,
() ->
parser.parse(
bytes, SubtitleParser.OutputOptions.allCues(), unusedCuesWithTiming -> {}));
} }
@Test @Test
@ -96,6 +101,38 @@ public class WebvttParserTest {
assertThat(secondCue.text.toString()).isEqualTo("This is the second subtitle."); assertThat(secondCue.text.toString()).isEqualTo("This is the second subtitle.");
} }
@Test
public void parseTypical_withStartTime() throws Exception {
ImmutableList<CuesWithTiming> allCues =
getCuesForTestAsset(TYPICAL_FILE, SubtitleParser.OutputOptions.onlyCuesAfter(2_000_000));
assertThat(allCues).hasSize(1);
assertThat(allCues.get(0).startTimeUs).isEqualTo(2_345_000L);
assertThat(allCues.get(0).durationUs).isEqualTo(3_456_000L - 2_345_000L);
Cue secondCue = Iterables.getOnlyElement(allCues.get(0).cues);
assertThat(secondCue.text.toString()).isEqualTo("This is the second subtitle.");
}
@Test
public void parseTypical_withStartTimeAndRemainingCues() throws Exception {
ImmutableList<CuesWithTiming> allCues =
getCuesForTestAsset(
TYPICAL_FILE, SubtitleParser.OutputOptions.cuesAfterThenRemainingCuesBefore(2_000_000));
assertThat(allCues).hasSize(2);
assertThat(allCues.get(0).startTimeUs).isEqualTo(2_345_000L);
assertThat(allCues.get(0).durationUs).isEqualTo(3_456_000L - 2_345_000L);
Cue secondCue = Iterables.getOnlyElement(allCues.get(0).cues);
assertThat(secondCue.text.toString()).isEqualTo("This is the second subtitle.");
assertThat(allCues.get(1).startTimeUs).isEqualTo(0L);
assertThat(allCues.get(1).durationUs).isEqualTo(1_234_000L);
Cue firstCue = Iterables.getOnlyElement(allCues.get(1).cues);
assertThat(firstCue.text.toString()).isEqualTo("This is the first subtitle.");
}
@Test @Test
public void parseWithBom() throws Exception { public void parseWithBom() throws Exception {
List<CuesWithTiming> allCues = getCuesForTestAsset(WITH_BOM); List<CuesWithTiming> allCues = getCuesForTestAsset(WITH_BOM);
@ -558,10 +595,17 @@ public class WebvttParserTest {
"Combine ".length(), "Combine 0004".length()); "Combine ".length(), "Combine 0004".length());
} }
private List<CuesWithTiming> getCuesForTestAsset(String asset) throws IOException { private ImmutableList<CuesWithTiming> getCuesForTestAsset(String asset) throws IOException {
return getCuesForTestAsset(asset, SubtitleParser.OutputOptions.allCues());
}
private ImmutableList<CuesWithTiming> getCuesForTestAsset(
String asset, SubtitleParser.OutputOptions outputOptions) throws IOException {
WebvttParser parser = new WebvttParser(); WebvttParser parser = new WebvttParser();
byte[] bytes = TestUtil.getByteArray(ApplicationProvider.getApplicationContext(), asset); byte[] bytes = TestUtil.getByteArray(ApplicationProvider.getApplicationContext(), asset);
return parser.parse(bytes); ImmutableList.Builder<CuesWithTiming> result = ImmutableList.builder();
parser.parse(bytes, outputOptions, /* output= */ result::add);
return result.build();
} }
private Spanned getUniqueSpanTextAt(CuesWithTiming cuesWithTiming) { private Spanned getUniqueSpanTextAt(CuesWithTiming cuesWithTiming) {