Extract YouTube search suggestions

2020-03-20 16:11:28 +01:00
parent c852b13d5a
commit dc29d87962
2 changed files with 38 additions and 17 deletions
--- a/extractor/src/test/java/org/schabi/newpipe/extractor/services/youtube/search/YoutubeSearchExtractorMusicTest.java
+++ b/extractor/src/test/java/org/schabi/newpipe/extractor/services/youtube/search/YoutubeSearchExtractorMusicTest.java
@@ -76,4 +76,14 @@ public class YoutubeSearchExtractorMusicTest extends YoutubeSearchExtractorBaseT
        assertEquals("json", queryPairs.get("alt"));
        assertTrue(queryPairs.get("key").length() > 5);
    }
+
+    @Test
+    public void testSuggestions() throws Exception {
+        YoutubeSearchExtractor newExtractor = (YoutubeSearchExtractor) YouTube.getSearchExtractor("megaman x3",
+                asList(YoutubeSearchQueryHandlerFactory.MUSIC_SONGS), null);
+        newExtractor.fetchPage();
+
+        assertTrue(newExtractor.getInitialPage().getItems().size() > 10);
+        assertEquals("mega man x3", newExtractor.getSearchSuggestion());
+    }
 }