Skip to content

Commit 7f0269c

Browse files
committed
[YouTube] Edit YoutubeStreamExtractorDefaultTest.AudioTrackLanguage to test audio locale property
The Hindi audio track language presence test has been changed from audio track label to audio locale.
1 parent 034f82d commit 7f0269c

1 file changed

Lines changed: 12 additions & 8 deletions

File tree

extractor/src/test/java/org/schabi/newpipe/extractor/services/youtube/stream/YoutubeStreamExtractorDefaultTest.java

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -49,13 +49,16 @@
4949
import org.schabi.newpipe.extractor.stream.StreamExtractor;
5050
import org.schabi.newpipe.extractor.stream.StreamSegment;
5151
import org.schabi.newpipe.extractor.stream.StreamType;
52+
import org.schabi.newpipe.extractor.utils.LocaleCompat;
5253

5354
import java.io.IOException;
5455
import java.net.MalformedURLException;
5556
import java.net.URL;
5657
import java.util.Arrays;
5758
import java.util.Collections;
5859
import java.util.List;
60+
import java.util.Locale;
61+
import java.util.Objects;
5962

6063
import javax.annotation.Nullable;
6164

@@ -547,19 +550,20 @@ public static void setUp() throws Exception {
547550

548551
@Test
549552
void testCheckAudioStreams() throws Exception {
550-
assertFalse(extractor.getAudioStreams().isEmpty());
553+
final List<AudioStream> audioStreams = extractor.getAudioStreams();
554+
assertFalse(audioStreams.isEmpty());
551555

552-
for (final AudioStream audioStream : extractor.getAudioStreams()) {
553-
assertNotNull(audioStream.getAudioTrackName());
556+
for (final AudioStream stream : audioStreams) {
557+
assertNotNull(stream.getAudioTrackName());
554558
}
555559

556-
assertTrue(extractor.getAudioStreams()
557-
.stream()
560+
assertTrue(audioStreams.stream()
558561
.anyMatch(audioStream -> "English".equals(audioStream.getAudioTrackName())));
559562

560-
assertTrue(extractor.getAudioStreams()
561-
.stream()
562-
.anyMatch(audioStream -> "Hindi".equals(audioStream.getAudioTrackName())));
563+
final Locale hindiLocale = LocaleCompat.forLanguageTag("hi");
564+
assertTrue(audioStreams.stream()
565+
.anyMatch(audioStream ->
566+
Objects.equals(audioStream.getAudioLocale(), hindiLocale)));
563567
}
564568
}
565569

0 commit comments

Comments
 (0)