323 lines
12 KiB
Java
323 lines
12 KiB
Java
|
package org.schabi.newpipe.extractor.services.youtube;
|
||
|
|
|
||
|
|
import static org.hamcrest.CoreMatchers.containsString;
|
||
|
|
import static org.hamcrest.CoreMatchers.startsWith;
|
||
|
|
import static org.junit.Assert.*;
|
||
|
|
import static org.schabi.newpipe.extractor.ExtractorAsserts.assertIsSecureUrl;
|
||
|
|
import static org.schabi.newpipe.extractor.ServiceList.YouTube;
|
||
|
|
|
||
|
|
import org.junit.BeforeClass;
|
||
|
|
import org.junit.Test;
|
||
|
|
import org.schabi.newpipe.DownloaderTestImpl;
|
||
|
|
import org.schabi.newpipe.extractor.ListExtractor;
|
||
|
|
import org.schabi.newpipe.extractor.ListExtractor.InfoItemsPage;
|
||
|
|
import org.schabi.newpipe.extractor.NewPipe;
|
||
|
|
import org.schabi.newpipe.extractor.exceptions.ExtractionException;
|
||
|
|
import org.schabi.newpipe.extractor.services.youtube.extractors.YoutubeMixPlaylistExtractor;
|
||
|
|
import org.schabi.newpipe.extractor.stream.StreamInfoItem;
|
||
|
|
|
||
|
|
public class YoutubeMixPlaylistExtractorTest {
|
||
|
|
|
||
|
|
private static YoutubeMixPlaylistExtractor extractor;
|
||
|
|
private static String videoId = "_AzeUSL9lZc";
|
||
|
|
private static String videoTitle = "Most Beautiful And Emotional Piano: Anime Music Shigatsu wa Kimi no Uso OST IMO";
|
||
|
|
|
||
|
|
public static class Mix {
|
||
|
|
|
||
|
|
@BeforeClass
|
||
|
|
public static void setUp() throws Exception {
|
||
|
|
NewPipe.init(DownloaderTestImpl.getInstance());
|
||
|
|
extractor = (YoutubeMixPlaylistExtractor) YouTube
|
||
|
|
.getPlaylistExtractor(
|
||
|
|
"https://www.youtube.com/watch?v=" + videoId + "&list=RD" + videoId);
|
||
|
|
extractor.fetchPage();
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getServiceId() {
|
||
|
|
assertEquals(YouTube.getServiceId(), extractor.getServiceId());
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getName() throws Exception {
|
||
|
|
String name = extractor.getName();
|
||
|
|
assertThat(name, startsWith("Mix"));
|
||
|
|
assertThat(name, containsString(videoTitle));
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getThumbnailUrl() throws Exception {
|
||
|
|
final String thumbnailUrl = extractor.getThumbnailUrl();
|
||
|
|
assertIsSecureUrl(thumbnailUrl);
|
||
|
|
assertThat(thumbnailUrl, containsString("yt"));
|
||
|
|
assertThat(thumbnailUrl, containsString(videoId));
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getNextPageUrl() throws Exception {
|
||
|
|
final String nextPageUrl = extractor.getNextPageUrl();
|
||
|
|
assertIsSecureUrl(nextPageUrl);
|
||
|
|
assertThat(nextPageUrl, containsString("list=RD" + videoId));
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getInitialPage() throws Exception {
|
||
|
|
InfoItemsPage<StreamInfoItem> streams = extractor.getInitialPage();
|
||
|
|
assertFalse(streams.getItems().isEmpty());
|
||
|
|
assertTrue(streams.hasNextPage());
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getPage() throws Exception {
|
||
|
|
InfoItemsPage<StreamInfoItem> streams = extractor.getPage(extractor.getNextPageUrl());
|
||
|
|
assertFalse(streams.getItems().isEmpty());
|
||
|
|
assertTrue(streams.hasNextPage());
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getPageMultipleTimes() throws Exception {
|
||
|
|
InfoItemsPage<StreamInfoItem> streams = extractor.getPage(extractor.getNextPageUrl());
|
||
|
|
|
||
|
|
//Should work infinitely, but for testing purposes only 3 times
|
||
|
|
for (int i = 0; i < 3; i++) {
|
||
|
|
assertTrue(streams.hasNextPage());
|
||
|
|
assertFalse(streams.getItems().isEmpty());
|
||
|
|
|
||
|
|
streams = extractor.getPage(streams.getNextPageUrl());
|
||
|
|
}
|
||
|
|
assertTrue(streams.hasNextPage());
|
||
|
|
assertFalse(streams.getItems().isEmpty());
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getStreamCount() throws Exception {
|
||
|
|
assertEquals(ListExtractor.ITEM_COUNT_INFINITE, extractor.getStreamCount());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public static class MixWithIndex {
|
||
|
|
|
||
|
|
private static String index = "&index=13";
|
||
|
|
private static String videoIdNumber13 = "qHtzO49SDmk";
|
||
|
|
|
||
|
|
@BeforeClass
|
||
|
|
public static void setUp() throws Exception {
|
||
|
|
NewPipe.init(DownloaderTestImpl.getInstance());
|
||
|
|
extractor = (YoutubeMixPlaylistExtractor) YouTube
|
||
|
|
.getPlaylistExtractor(
|
||
|
|
"https://www.youtube.com/watch?v=" + videoIdNumber13 + "&list=RD" + videoId
|
||
|
|
+ index);
|
||
|
|
extractor.fetchPage();
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getName() throws Exception {
|
||
|
|
String name = extractor.getName();
|
||
|
|
assertThat(name, startsWith("Mix"));
|
||
|
|
assertThat(name, containsString(videoTitle));
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getThumbnailUrl() throws Exception {
|
||
|
|
final String thumbnailUrl = extractor.getThumbnailUrl();
|
||
|
|
assertIsSecureUrl(thumbnailUrl);
|
||
|
|
assertThat(thumbnailUrl, containsString("yt"));
|
||
|
|
assertThat(thumbnailUrl, containsString(videoId));
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getNextPageUrl() throws Exception {
|
||
|
|
final String nextPageUrl = extractor.getNextPageUrl();
|
||
|
|
assertIsSecureUrl(nextPageUrl);
|
||
|
|
assertThat(nextPageUrl, containsString("list=RD" + videoId));
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getInitialPage() throws Exception {
|
||
|
|
InfoItemsPage<StreamInfoItem> streams = extractor.getInitialPage();
|
||
|
|
assertFalse(streams.getItems().isEmpty());
|
||
|
|
assertTrue(streams.hasNextPage());
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getPage() throws Exception {
|
||
|
|
InfoItemsPage<StreamInfoItem> streams = extractor.getPage(extractor.getNextPageUrl());
|
||
|
|
assertFalse(streams.getItems().isEmpty());
|
||
|
|
assertTrue(streams.hasNextPage());
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getPageMultipleTimes() throws Exception {
|
||
|
|
InfoItemsPage<StreamInfoItem> streams = extractor.getPage(extractor.getNextPageUrl());
|
||
|
|
|
||
|
|
//Should work infinitely, but for testing purposes only 3 times
|
||
|
|
for (int i = 0; i < 3; i++) {
|
||
|
|
assertTrue(streams.hasNextPage());
|
||
|
|
assertFalse(streams.getItems().isEmpty());
|
||
|
|
|
||
|
|
streams = extractor.getPage(streams.getNextPageUrl());
|
||
|
|
}
|
||
|
|
assertTrue(streams.hasNextPage());
|
||
|
|
assertFalse(streams.getItems().isEmpty());
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getStreamCount() {
|
||
|
|
assertEquals(ListExtractor.ITEM_COUNT_INFINITE, extractor.getStreamCount());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public static class MyMix {
|
||
|
|
|
||
|
|
@BeforeClass
|
||
|
|
public static void setUp() throws Exception {
|
||
|
|
NewPipe.init(DownloaderTestImpl.getInstance());
|
||
|
|
extractor = (YoutubeMixPlaylistExtractor) YouTube
|
||
|
|
.getPlaylistExtractor(
|
||
|
|
"https://www.youtube.com/watch?v=" + videoId + "&list=RDMM" + videoId);
|
||
|
|
extractor.fetchPage();
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getServiceId() {
|
||
|
|
assertEquals(YouTube.getServiceId(), extractor.getServiceId());
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getName() throws Exception {
|
||
|
|
String name = extractor.getName();
|
||
|
|
assertEquals("My Mix", name);
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getThumbnailUrl() throws Exception {
|
||
|
|
final String thumbnailUrl = extractor.getThumbnailUrl();
|
||
|
|
assertIsSecureUrl(thumbnailUrl);
|
||
|
|
assertThat(thumbnailUrl, startsWith("https://i.ytimg.com/vi/_AzeUSL9lZc"));
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getNextPageUrl() throws Exception {
|
||
|
|
final String nextPageUrl = extractor.getNextPageUrl();
|
||
|
|
assertIsSecureUrl(nextPageUrl);
|
||
|
|
assertThat(nextPageUrl, containsString("list=RDMM" + videoId));
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getInitialPage() throws Exception {
|
||
|
|
InfoItemsPage<StreamInfoItem> streams = extractor.getInitialPage();
|
||
|
|
assertFalse(streams.getItems().isEmpty());
|
||
|
|
assertTrue(streams.hasNextPage());
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getPage() throws Exception {
|
||
|
|
InfoItemsPage<StreamInfoItem> streams = extractor.getPage(extractor.getNextPageUrl());
|
||
|
|
assertFalse(streams.getItems().isEmpty());
|
||
|
|
assertTrue(streams.hasNextPage());
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getPageMultipleTimes() throws Exception {
|
||
|
|
InfoItemsPage<StreamInfoItem> streams = extractor.getPage(extractor.getNextPageUrl());
|
||
|
|
|
||
|
|
//Should work infinitely, but for testing purposes only 3 times
|
||
|
|
for (int i = 0; i < 3; i++) {
|
||
|
|
assertTrue(streams.hasNextPage());
|
||
|
|
assertFalse(streams.getItems().isEmpty());
|
||
|
|
|
||
|
|
streams = extractor.getPage(streams.getNextPageUrl());
|
||
|
|
}
|
||
|
|
assertTrue(streams.hasNextPage());
|
||
|
|
assertFalse(streams.getItems().isEmpty());
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getStreamCount() throws Exception {
|
||
|
|
assertEquals(ListExtractor.ITEM_COUNT_INFINITE, extractor.getStreamCount());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public static class Invalid {
|
||
|
|
|
||
|
|
@BeforeClass
|
||
|
|
public static void setUp() {
|
||
|
|
NewPipe.init(DownloaderTestImpl.getInstance());
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test(expected = ExtractionException.class)
|
||
|
|
public void getPageEmptyUrl() throws Exception {
|
||
|
|
extractor = (YoutubeMixPlaylistExtractor) YouTube
|
||
|
|
.getPlaylistExtractor(
|
||
|
|
"https://www.youtube.com/watch?v=" + videoId + "&list=RD" + videoId);
|
||
|
|
extractor.fetchPage();
|
||
|
|
extractor.getPage("");
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test(expected = NullPointerException.class)
|
||
|
|
public void invalidVideoId() throws Exception {
|
||
|
|
extractor = (YoutubeMixPlaylistExtractor) YouTube
|
||
|
|
.getPlaylistExtractor(
|
||
|
|
"https://www.youtube.com/watch?v=" + "abcde" + "&list=RD" + "abcde");
|
||
|
|
extractor.fetchPage();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public static class ChannelMix {
|
||
|
|
|
||
|
|
private static String channelId = "UCXuqSBlHAE6Xw-yeJA0Tunw";
|
||
|
|
private static String videoIdOfChannel = "mnk6gnOBYIo";
|
||
|
|
private static String channelTitle = "Linus Tech Tips";
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
@BeforeClass
|
||
|
|
public static void setUp() throws Exception {
|
||
|
|
NewPipe.init(DownloaderTestImpl.getInstance());
|
||
|
|
extractor = (YoutubeMixPlaylistExtractor) YouTube
|
||
|
|
.getPlaylistExtractor(
|
||
|
|
"https://www.youtube.com/watch?v=" + videoIdOfChannel + "&list=RDCM" + channelId);
|
||
|
|
extractor.fetchPage();
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getName() throws Exception {
|
||
|
|
String name = extractor.getName();
|
||
|
|
assertThat(name, startsWith("Mix"));
|
||
|
|
assertThat(name, containsString(channelTitle));
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getThumbnailUrl() throws Exception {
|
||
|
|
final String thumbnailUrl = extractor.getThumbnailUrl();
|
||
|
|
assertIsSecureUrl(thumbnailUrl);
|
||
|
|
assertThat(thumbnailUrl, containsString("yt"));
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getNextPageUrl() throws Exception {
|
||
|
|
final String nextPageUrl = extractor.getNextPageUrl();
|
||
|
|
assertIsSecureUrl(nextPageUrl);
|
||
|
|
assertThat(nextPageUrl, containsString("list=RDCM" + channelId));
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getInitialPage() throws Exception {
|
||
|
|
InfoItemsPage<StreamInfoItem> streams = extractor.getInitialPage();
|
||
|
|
assertFalse(streams.getItems().isEmpty());
|
||
|
|
assertTrue(streams.hasNextPage());
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getPage() throws Exception {
|
||
|
|
InfoItemsPage<StreamInfoItem> streams = extractor.getPage(extractor.getNextPageUrl());
|
||
|
|
assertFalse(streams.getItems().isEmpty());
|
||
|
|
assertTrue(streams.hasNextPage());
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void getStreamCount() throws Exception {
|
||
|
|
assertEquals(ListExtractor.ITEM_COUNT_INFINITE, extractor.getStreamCount());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|