NewPipeExtractor/extractor/src/test/java/org/schabi/newpipe/extractor/services/bandcamp/BandcampSearchExtractorTest.java

// Created by Fynn Godau 2019, licensed GNU GPL version 3 or later

package org.schabi.newpipe.extractor.services.bandcamp;

import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import org.schabi.newpipe.downloader.DownloaderTestImpl;
import org.schabi.newpipe.extractor.*;
import org.schabi.newpipe.extractor.exceptions.ExtractionException;
import org.schabi.newpipe.extractor.playlist.PlaylistInfoItem;
import org.schabi.newpipe.extractor.search.SearchExtractor;
import org.schabi.newpipe.extractor.services.DefaultSearchExtractorTest;
import org.schabi.newpipe.extractor.services.bandcamp.extractors.BandcampSearchExtractor;
import org.schabi.newpipe.extractor.stream.StreamInfoItem;

import javax.annotation.Nullable;
import java.io.IOException;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.schabi.newpipe.extractor.ServiceList.Bandcamp;

/**
 * Test for {@link BandcampSearchExtractor}
 */
public class BandcampSearchExtractorTest {

    @BeforeAll
    public static void setUp() {
        NewPipe.init(DownloaderTestImpl.getInstance());

    }

    /**
     * Tests whether searching bandcamp for "best friend's basement" returns
     * the accordingly named song by Zach Benson
     */
    @Test
    public void testStreamSearch() throws ExtractionException, IOException {
        final SearchExtractor extractor = Bandcamp.getSearchExtractor("best friend's basement");

        final ListExtractor.InfoItemsPage<InfoItem> page = extractor.getInitialPage();
        final StreamInfoItem bestFriendsBasement = (StreamInfoItem) page.getItems().get(0);

        // The track by Zach Benson should be the first result, no?
        assertEquals("Best Friend's Basement", bestFriendsBasement.getName());
        assertEquals("Zach Benson", bestFriendsBasement.getUploaderName());
        assertTrue(bestFriendsBasement.getThumbnailUrl().endsWith(".jpg"));
        assertTrue(bestFriendsBasement.getThumbnailUrl().contains("f4.bcbits.com/img/"));
        assertEquals(InfoItem.InfoType.STREAM, bestFriendsBasement.getInfoType());
    }

    /**
     * Tests whether searching bandcamp for "C418" returns the artist's profile
     */
    @Test
    public void testChannelSearch() throws ExtractionException, IOException {
        final SearchExtractor extractor = Bandcamp.getSearchExtractor("C418");
        final InfoItem c418 = extractor.getInitialPage()
                .getItems().get(0);

        // C418's artist profile should be the first result, no?
        assertEquals("C418", c418.getName());
        assertTrue(c418.getThumbnailUrl().endsWith(".jpg"));
        assertTrue(c418.getThumbnailUrl().contains("f4.bcbits.com/img/"));
        assertEquals("https://c418.bandcamp.com", c418.getUrl());

    }

    /**
     * Tests whether searching bandcamp for "minecraft volume alpha" returns the corresponding album
     */
    @Test
    public void testAlbumSearch() throws ExtractionException, IOException {
        final SearchExtractor extractor = Bandcamp.getSearchExtractor("minecraft volume alpha");
        InfoItem minecraft = extractor.getInitialPage()
                .getItems().get(0);

        // Minecraft volume alpha should be the first result, no?
        assertEquals("Minecraft - Volume Alpha", minecraft.getName());
        assertTrue(minecraft.getThumbnailUrl().endsWith(".jpg"));
        assertTrue(minecraft.getThumbnailUrl().contains("f4.bcbits.com/img/"));
        assertEquals("https://c418.bandcamp.com/album/minecraft-volume-alpha", minecraft.getUrl());

        // Verify that playlist tracks counts get extracted correctly
        assertEquals(24, ((PlaylistInfoItem) minecraft).getStreamCount());

    }

    /**
     * Tests searches with multiple pages
     */
    @Test
    public void testMultiplePages() throws ExtractionException, IOException {
        // A query practically guaranteed to have the maximum amount of pages
        final SearchExtractor extractor = Bandcamp.getSearchExtractor("e");

        final Page page2 = extractor.getInitialPage().getNextPage();
        assertEquals("https://bandcamp.com/search?q=e&page=2", page2.getUrl());

        final Page page3 = extractor.getPage(page2).getNextPage();
        assertEquals("https://bandcamp.com/search?q=e&page=3", page3.getUrl());
    }

    public static class DefaultTest extends DefaultSearchExtractorTest {
        private static SearchExtractor extractor;
        private static final String QUERY = "noise";

        @BeforeAll
        public static void setUp() throws Exception {
            NewPipe.init(DownloaderTestImpl.getInstance());
            extractor = Bandcamp.getSearchExtractor(QUERY);
            extractor.fetchPage();
        }

        @Override public SearchExtractor extractor() { return extractor; }
        @Override public StreamingService expectedService() { return Bandcamp; }
        @Override public String expectedName() { return QUERY; }
        @Override public String expectedId() { return QUERY; }
        @Override public String expectedUrlContains() { return "bandcamp.com/search?q=" + QUERY; }
        @Override public String expectedOriginalUrlContains() { return "bandcamp.com/search?q=" + QUERY; }
        @Override public String expectedSearchString() { return QUERY; }
        @Nullable @Override public String expectedSearchSuggestion() { return null; }
    }
}