|
7 | 7 | import org.schabi.newpipe.extractor.StreamingService; |
8 | 8 | import org.schabi.newpipe.extractor.services.DefaultStreamExtractorTest; |
9 | 9 | import org.schabi.newpipe.extractor.services.media_ccc.extractors.MediaCCCStreamExtractor; |
| 10 | +import org.schabi.newpipe.extractor.stream.AudioStream; |
10 | 11 | import org.schabi.newpipe.extractor.stream.StreamExtractor; |
11 | 12 | import org.schabi.newpipe.extractor.stream.StreamType; |
| 13 | +import org.schabi.newpipe.extractor.utils.LocaleCompat; |
12 | 14 |
|
13 | 15 | import javax.annotation.Nullable; |
14 | 16 |
|
15 | 17 | import java.util.Arrays; |
16 | 18 | import java.util.List; |
17 | 19 | import java.util.Locale; |
| 20 | +import java.util.Objects; |
18 | 21 |
|
19 | 22 | import static org.junit.jupiter.api.Assertions.assertEquals; |
| 23 | +import static org.junit.jupiter.api.Assertions.assertTrue; |
20 | 24 | import static org.schabi.newpipe.extractor.ServiceList.MediaCCC; |
21 | 25 |
|
22 | 26 | /** |
@@ -85,7 +89,11 @@ public void testVideoStreams() throws Exception { |
85 | 89 | @Test |
86 | 90 | public void testAudioStreams() throws Exception { |
87 | 91 | super.testAudioStreams(); |
88 | | - assertEquals(2, extractor.getAudioStreams().size()); |
| 92 | + final List<AudioStream> audioStreams = extractor.getAudioStreams(); |
| 93 | + assertEquals(2, audioStreams.size()); |
| 94 | + final Locale expectedLocale = LocaleCompat.forLanguageTag("deu"); |
| 95 | + assertTrue(audioStreams.stream().allMatch(audioStream -> |
| 96 | + Objects.equals(audioStream.getAudioLocale(), expectedLocale))); |
89 | 97 | } |
90 | 98 | } |
91 | 99 |
|
@@ -155,7 +163,11 @@ public void testVideoStreams() throws Exception { |
155 | 163 | @Test |
156 | 164 | public void testAudioStreams() throws Exception { |
157 | 165 | super.testAudioStreams(); |
158 | | - assertEquals(2, extractor.getAudioStreams().size()); |
| 166 | + final List<AudioStream> audioStreams = extractor.getAudioStreams(); |
| 167 | + assertEquals(2, audioStreams.size()); |
| 168 | + final Locale expectedLocale = LocaleCompat.forLanguageTag("eng"); |
| 169 | + assertTrue(audioStreams.stream().allMatch(audioStream -> |
| 170 | + Objects.equals(audioStream.getAudioLocale(), expectedLocale))); |
159 | 171 | } |
160 | 172 |
|
161 | 173 | @Override |
|
0 commit comments