api-themoviedb/src/test/java/com/omertron/themoviedbapi/methods/TmdbSeasonsTest.java

/*
 *      Copyright (c) 2004-2015 Stuart Boston
 *
 *      This file is part of TheMovieDB API.
 *
 *      TheMovieDB API is free software: you can redistribute it and/or modify
 *      it under the terms of the GNU General Public License as published by
 *      the Free Software Foundation, either version 3 of the License, or
 *      any later version.
 *
 *      TheMovieDB API is distributed in the hope that it will be useful,
 *      but WITHOUT ANY WARRANTY; without even the implied warranty of
 *      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *      GNU General Public License for more details.
 *
 *      You should have received a copy of the GNU General Public License
 *      along with TheMovieDB API.  If not, see <http://www.gnu.org/licenses/>.
 *
 */
package com.omertron.themoviedbapi.methods;

import com.omertron.themoviedbapi.AbstractTests;
import com.omertron.themoviedbapi.ArtworkResults;
import com.omertron.themoviedbapi.MovieDbException;
import com.omertron.themoviedbapi.TestID;
import com.omertron.themoviedbapi.TestSuite;
import com.omertron.themoviedbapi.enumeration.ArtworkType;
import com.omertron.themoviedbapi.model.artwork.Artwork;
import com.omertron.themoviedbapi.model.media.MediaCreditCast;
import com.omertron.themoviedbapi.model.media.MediaCreditList;
import com.omertron.themoviedbapi.model.media.MediaState;
import com.omertron.themoviedbapi.model.media.Video;
import com.omertron.themoviedbapi.model.person.ExternalID;
import com.omertron.themoviedbapi.model.tv.TVSeasonInfo;
import com.omertron.themoviedbapi.results.ResultList;
import java.util.ArrayList;
import java.util.List;
import org.junit.After;
import org.junit.AfterClass;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;

/**
 *
 * @author Stuart.Boston
 */
public class TmdbSeasonsTest extends AbstractTests {

    private static TmdbSeasons instance;
    private static final List<TestID> TV_IDS = new ArrayList<TestID>();

    public TmdbSeasonsTest() {
    }

    @BeforeClass
    public static void setUpClass() throws MovieDbException {
        doConfiguration();
        instance = new TmdbSeasons(getApiKey(), getHttpTools());

        TV_IDS.add(new TestID("The Walking Dead", "tt1520211", 1402, "Andrew Lincoln"));
        TV_IDS.add(new TestID("Supernatural", "tt0460681", 1622, "Misha Collins"));
        TV_IDS.add(new TestID("The Big Bang Theory", "tt0898266", 1418, "Kaley Cuoco"));
    }

    @AfterClass
    public static void tearDownClass() {
    }

    @Before
    public void setUp() {
    }

    @After
    public void tearDown() {
    }

    /**
     * Test of getSeasonInfo method, of class TmdbSeasons.
     *
     * @throws com.omertron.themoviedbapi.MovieDbException
     */
    @Test
    public void testGetSeasonInfo() throws MovieDbException {
        LOG.info("getSeasonInfo");

        int seasonNumber = 1;
        String language = LANGUAGE_DEFAULT;
        String[] appendToResponse = null;

        for (TestID test : TV_IDS) {
            TVSeasonInfo result = instance.getSeasonInfo(test.getTmdb(), seasonNumber, language, appendToResponse);
            TestSuite.test(result);
        }
    }

    /**
     * Test of getSeasonChanges method, of class TmdbSeasons.
     *
     * @throws com.omertron.themoviedbapi.MovieDbException
     */
    @Test
    public void testGetSeasonChanges() throws MovieDbException {
        LOG.info("getSeasonChanges");
        // This is too empherial to test
    }

    /**
     * Test of getSeasonAccountState method, of class TmdbSeasons.
     *
     * @throws com.omertron.themoviedbapi.MovieDbException
     */
    @Test
    public void testGetSeasonAccountState() throws MovieDbException {
        LOG.info("getSeasonAccountState");

        for (TestID test : TV_IDS) {
            MediaState result = instance.getSeasonAccountState(test.getTmdb(), getSessionId());
            assertNotNull("Null result", result);
            assertTrue("Invalid rating", result.getRated() > -2f);
        }
    }

    /**
     * Test of getSeasonCredits method, of class TmdbSeasons.
     *
     * @throws com.omertron.themoviedbapi.MovieDbException
     */
    @Test
    public void testGetSeasonCredits() throws MovieDbException {
        LOG.info("getSeasonCredits");

        int seasonNumber = 0;

        for (TestID test : TV_IDS) {
            MediaCreditList result = instance.getSeasonCredits(test.getTmdb(), seasonNumber);
            assertNotNull(result);
            TestSuite.test(result.getCast());
            TestSuite.test(result.getCrew());
            TestSuite.test(result.getGuestStars());

            boolean found = false;
            for (MediaCreditCast p : result.getCast()) {
                if (test.getOther().equals(p.getName())) {
                    found = true;
                    break;
                }
            }
            assertTrue(test.getOther() + " not found in cast!", found);
        }
    }

    /**
     * Test of getSeasonExternalID method, of class TmdbSeasons.
     *
     * @throws com.omertron.themoviedbapi.MovieDbException
     */
    @Test
    public void testGetSeasonExternalID() throws MovieDbException {
        LOG.info("getSeasonExternalID");

        int seasonNumber = 0;
        String language = LANGUAGE_DEFAULT;

        for (TestID test : TV_IDS) {
            ExternalID result = instance.getSeasonExternalID(test.getTmdb(), seasonNumber, language);
            assertEquals("Wrong IMDB ID", test.getImdb(), result.getImdbId());
        }
    }

    /**
     * Test of getSeasonImages method, of class TmdbSeasons.
     *
     * @throws com.omertron.themoviedbapi.MovieDbException
     */
    @Test
    public void testGetSeasonImages() throws MovieDbException {
        LOG.info("getSeasonImages");

        int seasonNumber = 1;
        String language = LANGUAGE_DEFAULT;
        String[] includeImageLanguage = null;

        ArtworkResults results = new ArtworkResults();

        for (TestID test : TV_IDS) {
            ResultList<Artwork> result = instance.getSeasonImages(test.getTmdb(), seasonNumber, language, includeImageLanguage);
            TestSuite.test(result);
            for (Artwork artwork : result.getResults()) {
                results.found(artwork.getArtworkType());
            }

            // We should only have posters
            results.validateResults(ArtworkType.POSTER);
        }
    }

    /**
     * Test of getSeasonVideos method, of class TmdbSeasons.
     *
     * @throws com.omertron.themoviedbapi.MovieDbException
     */
    @Test
    public void testGetSeasonVideos() throws MovieDbException {
        LOG.info("getSeasonVideos");

        int seasonNumber = 0;
        String language = LANGUAGE_DEFAULT;
        boolean found = false;

        for (TestID test : TV_IDS) {
            ResultList<Video> result = instance.getSeasonVideos(test.getTmdb(), seasonNumber, language);
            found = found || !result.isEmpty();
        }

        assertTrue("No videos", found);
    }

}