Skip to content

Commit 09544ce

Browse files
committed
Fix tests
1 parent 644cc6c commit 09544ce

1 file changed

Lines changed: 12 additions & 2 deletions

File tree

extractor/src/test/java/org/schabi/newpipe/extractor/services/youtube/search/YoutubeSearchExtractorTest.java

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,10 @@
3030
import org.schabi.newpipe.extractor.stream.StreamInfoItem;
3131

3232
import java.io.IOException;
33+
import java.io.UnsupportedEncodingException;
3334
import java.net.MalformedURLException;
3435
import java.net.URL;
36+
import java.net.URLEncoder;
3537
import java.util.Collections;
3638
import java.util.List;
3739

@@ -391,12 +393,20 @@ public static void setUp() throws Exception {
391393
extractor.fetchPage();
392394
}
393395

396+
private String getUrlEncodedQuery() {
397+
try {
398+
return URLEncoder.encode(QUERY, "UTF-8");
399+
} catch (UnsupportedEncodingException e) {
400+
throw new RuntimeException(e);
401+
}
402+
}
403+
394404
@Override public SearchExtractor extractor() { return extractor; }
395405
@Override public StreamingService expectedService() { return YouTube; }
396406
@Override public String expectedName() { return QUERY; }
397407
@Override public String expectedId() { return QUERY; }
398-
@Override public String expectedUrlContains() { return "youtube.com/results?search_query=" + QUERY; }
399-
@Override public String expectedOriginalUrlContains() { return "youtube.com/results?search_query=" + QUERY; }
408+
@Override public String expectedUrlContains() { return "youtube.com/results?search_query=" + getUrlEncodedQuery(); }
409+
@Override public String expectedOriginalUrlContains() { return "youtube.com/results?search_query=" + getUrlEncodedQuery(); }
400410
@Override public String expectedSearchString() { return QUERY; }
401411
@Nullable @Override public String expectedSearchSuggestion() { return null; }
402412
@Override public InfoItem.InfoType expectedInfoItemType() { return InfoItem.InfoType.STREAM; }

0 commit comments

Comments
 (0)