Update account methods
Move to their own class Updated HttpClient for POST and DELETEmaster
Stuart Boston
11 years ago
parent
e736aee48a
commit
47ed93b69f
@ -0,0 +1,275 @@
|
|||||||
|
/*
|
||||||
|
* 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;
|
||||||
|
|
||||||
|
import com.omertron.themoviedbapi.model.Account;
|
||||||
|
import com.omertron.themoviedbapi.model.MovieDb;
|
||||||
|
import com.omertron.themoviedbapi.model.MovieDbList;
|
||||||
|
import com.omertron.themoviedbapi.model.StatusCode;
|
||||||
|
import com.omertron.themoviedbapi.model.TokenAuthorisation;
|
||||||
|
import com.omertron.themoviedbapi.model.TokenSession;
|
||||||
|
import java.io.File;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Properties;
|
||||||
|
import java.util.Random;
|
||||||
|
import org.junit.After;
|
||||||
|
import org.junit.AfterClass;
|
||||||
|
import org.junit.Assert;
|
||||||
|
import static org.junit.Assert.assertEquals;
|
||||||
|
import static org.junit.Assert.assertFalse;
|
||||||
|
import static org.junit.Assert.assertNotNull;
|
||||||
|
import static org.junit.Assert.assertTrue;
|
||||||
|
import static org.junit.Assert.fail;
|
||||||
|
import org.junit.Before;
|
||||||
|
import org.junit.BeforeClass;
|
||||||
|
import org.junit.Ignore;
|
||||||
|
import org.junit.Test;
|
||||||
|
import org.slf4j.Logger;
|
||||||
|
import org.slf4j.LoggerFactory;
|
||||||
|
|
||||||
|
public class TestAccounts {
|
||||||
|
|
||||||
|
private static final Logger LOG = LoggerFactory.getLogger(TestAccounts.class);
|
||||||
|
// API Key
|
||||||
|
private static final String PROP_FIlENAME = "testing.properties";
|
||||||
|
private static String API_KEY;
|
||||||
|
private static String ACCOUNT_USERNAME;
|
||||||
|
private static String ACCOUNT_PASSWORD;
|
||||||
|
private static TheMovieDbApi tmdb;
|
||||||
|
private static TokenSession tokenSession = null;
|
||||||
|
private static Account account = null;
|
||||||
|
|
||||||
|
public TestAccounts() throws MovieDbException {
|
||||||
|
}
|
||||||
|
|
||||||
|
@BeforeClass
|
||||||
|
public static void setUpClass() throws MovieDbException {
|
||||||
|
TestLogger.Configure();
|
||||||
|
|
||||||
|
Properties props = new Properties();
|
||||||
|
File f = new File(PROP_FIlENAME);
|
||||||
|
if (f.exists()) {
|
||||||
|
LOG.info("Loading properties from '{}'", PROP_FIlENAME);
|
||||||
|
TestLogger.loadProperties(props, f);
|
||||||
|
|
||||||
|
API_KEY = props.getProperty("API_Key");
|
||||||
|
ACCOUNT_USERNAME = props.getProperty("Username");
|
||||||
|
ACCOUNT_PASSWORD = props.getProperty("Password");
|
||||||
|
} else {
|
||||||
|
// Properties file is created in the main test class
|
||||||
|
fail("Failed to get key information from properties file '" + PROP_FIlENAME + "'");
|
||||||
|
}
|
||||||
|
|
||||||
|
tmdb = new TheMovieDbApi(API_KEY);
|
||||||
|
}
|
||||||
|
|
||||||
|
@AfterClass
|
||||||
|
public static void tearDownClass() {
|
||||||
|
}
|
||||||
|
|
||||||
|
@Before
|
||||||
|
public void setUp() throws MovieDbException {
|
||||||
|
if (tokenSession == null) {
|
||||||
|
testSessionCreation();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (account == null) {
|
||||||
|
testAccount();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@After
|
||||||
|
public void tearDown() {
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Test and create a session token for the rest of the tests
|
||||||
|
*
|
||||||
|
* @throws MovieDbException
|
||||||
|
*/
|
||||||
|
@Test
|
||||||
|
public void testSessionCreation() throws MovieDbException {
|
||||||
|
LOG.info("Test and create a session token for the rest of the tests");
|
||||||
|
|
||||||
|
// 1: Create a request token
|
||||||
|
TokenAuthorisation token = tmdb.getAuthorisationToken();
|
||||||
|
assertFalse("Token (auth) is null", token == null);
|
||||||
|
assertTrue("Token (auth) is not valid", token.getSuccess());
|
||||||
|
LOG.info("Token (auth): {}", token.toString());
|
||||||
|
|
||||||
|
// 2b; Get user permission
|
||||||
|
token = tmdb.getSessionTokenLogin(token, ACCOUNT_USERNAME, ACCOUNT_PASSWORD);
|
||||||
|
assertFalse("Token (login) is null", token == null);
|
||||||
|
assertTrue("Token (login) is not valid", token.getSuccess());
|
||||||
|
LOG.info("Token (login): {}", token.toString());
|
||||||
|
|
||||||
|
// 3: Create the sessions ID
|
||||||
|
tokenSession = tmdb.getSessionToken(token);
|
||||||
|
assertFalse("Session token is null", tokenSession == null);
|
||||||
|
assertTrue("Session token is not valid", tokenSession.getSuccess());
|
||||||
|
LOG.info("Session token: {}", tokenSession.toString());
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Test the account information
|
||||||
|
*
|
||||||
|
* @throws MovieDbException
|
||||||
|
*/
|
||||||
|
@Test
|
||||||
|
public void testAccount() throws MovieDbException {
|
||||||
|
LOG.info("Using Session ID '{}' for test", tokenSession.getSessionId());
|
||||||
|
account = tmdb.getAccount(tokenSession.getSessionId());
|
||||||
|
|
||||||
|
LOG.info("Account: {}", account);
|
||||||
|
|
||||||
|
// Make sure properties are extracted correctly
|
||||||
|
assertEquals("Wrong username!", ACCOUNT_USERNAME, account.getUserName());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testWatchList() throws MovieDbException {
|
||||||
|
// If the static account is null, get it
|
||||||
|
if (account == null) {
|
||||||
|
account = tmdb.getAccount(tokenSession.getSessionId());
|
||||||
|
}
|
||||||
|
|
||||||
|
// make sure it's empty (because it's just a test account
|
||||||
|
Assert.assertTrue(tmdb.getWatchList(tokenSession.getSessionId(), account.getId()).isEmpty());
|
||||||
|
|
||||||
|
// add a movie
|
||||||
|
tmdb.addToWatchList(tokenSession.getSessionId(), account.getId(), 550);
|
||||||
|
|
||||||
|
List<MovieDb> watchList = tmdb.getWatchList(tokenSession.getSessionId(), account.getId());
|
||||||
|
assertNotNull("Empty watch list returned", watchList);
|
||||||
|
assertEquals("Watchlist wrong size", 1, watchList.size());
|
||||||
|
|
||||||
|
// clean up again
|
||||||
|
tmdb.removeFromWatchList(tokenSession.getSessionId(), account.getId(), 550);
|
||||||
|
|
||||||
|
Assert.assertTrue(tmdb.getWatchList(tokenSession.getSessionId(), account.getId()).isEmpty());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testFavorites() throws MovieDbException {
|
||||||
|
// If the static account is null, get it
|
||||||
|
if (account == null) {
|
||||||
|
account = tmdb.getAccount(tokenSession.getSessionId());
|
||||||
|
}
|
||||||
|
|
||||||
|
// make sure it's empty (because it's just a test account
|
||||||
|
Assert.assertTrue(tmdb.getFavoriteMovies(tokenSession.getSessionId(), account.getId()).isEmpty());
|
||||||
|
|
||||||
|
// add a movie
|
||||||
|
tmdb.changeFavoriteStatus(tokenSession.getSessionId(), account.getId(), 550, true);
|
||||||
|
|
||||||
|
List<MovieDb> watchList = tmdb.getFavoriteMovies(tokenSession.getSessionId(), account.getId());
|
||||||
|
assertNotNull("Empty watch list returned", watchList);
|
||||||
|
assertEquals("Watchlist wrong size", 1, watchList.size());
|
||||||
|
|
||||||
|
// clean up again
|
||||||
|
tmdb.changeFavoriteStatus(tokenSession.getSessionId(), account.getId(), 550, false);
|
||||||
|
|
||||||
|
Assert.assertTrue(tmdb.getFavoriteMovies(tokenSession.getSessionId(), account.getId()).isEmpty());
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Test of getSessionToken method, of class TheMovieDbApi.
|
||||||
|
*
|
||||||
|
* TODO: Cannot be tested without a HTTP authorisation:
|
||||||
|
* http://help.themoviedb.org/kb/api/user-authentication
|
||||||
|
*
|
||||||
|
* @throws MovieDbException
|
||||||
|
*/
|
||||||
|
@Ignore("Tested in setup")
|
||||||
|
public void testGetSessionToken() throws MovieDbException {
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Test of getGuestSessionToken method, of class TheMovieDbApi.
|
||||||
|
*
|
||||||
|
* @throws MovieDbException
|
||||||
|
*/
|
||||||
|
@Test
|
||||||
|
public void testGetGuestSessionToken() throws MovieDbException {
|
||||||
|
LOG.info("getGuestSessionToken");
|
||||||
|
TokenSession result = tmdb.getGuestSessionToken();
|
||||||
|
|
||||||
|
assertTrue("Failed to get guest session", result.getSuccess());
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Test of postMovieRating method, of class TheMovieDbApi.
|
||||||
|
*
|
||||||
|
* TODO: Cannot be tested without a HTTP authorisation:
|
||||||
|
* http://help.themoviedb.org/kb/api/user-authentication
|
||||||
|
*
|
||||||
|
* @throws MovieDbException
|
||||||
|
*/
|
||||||
|
@Test
|
||||||
|
public void testMovieRating() throws MovieDbException {
|
||||||
|
LOG.info("postMovieRating");
|
||||||
|
|
||||||
|
Integer movieID = 68724;
|
||||||
|
Integer rating = new Random().nextInt(10) + 1;
|
||||||
|
|
||||||
|
boolean wasPosted = tmdb.postMovieRating(tokenSession.getSessionId(), movieID, rating);
|
||||||
|
assertTrue("Rating was not added", wasPosted);
|
||||||
|
|
||||||
|
// get all rated movies
|
||||||
|
List<MovieDb> ratedMovies = tmdb.getRatedMovies(tokenSession.getSessionId(), account.getId());
|
||||||
|
assertTrue("No rated movies", ratedMovies.size() > 0);
|
||||||
|
|
||||||
|
// make sure that we find the movie and it is rated correctly
|
||||||
|
boolean foundMovie = false;
|
||||||
|
for (MovieDb movie : ratedMovies) {
|
||||||
|
if (movie.getId() == movieID) {
|
||||||
|
assertEquals("Incorrect movie rating", movie.getUserRating(), (float) rating, 0);
|
||||||
|
foundMovie = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
assertTrue(foundMovie);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testMovieLists() throws MovieDbException {
|
||||||
|
Integer movieID = 68724;
|
||||||
|
|
||||||
|
// use a random name to avoid that we clash we leftovers of incomplete test runs
|
||||||
|
String name = "test list " + new Random().nextInt(100);
|
||||||
|
|
||||||
|
// create the list
|
||||||
|
String listId = tmdb.createList(tokenSession.getSessionId(), name, "api testing only");
|
||||||
|
|
||||||
|
// add a movie, and test that it is on the list now
|
||||||
|
tmdb.addMovieToList(tokenSession.getSessionId(), listId, movieID);
|
||||||
|
MovieDbList list = tmdb.getList(listId);
|
||||||
|
assertNotNull("Movie list returned was null", list);
|
||||||
|
assertEquals("Unexpected number of items returned", 1, list.getItemCount());
|
||||||
|
assertEquals((int) movieID, list.getItems().get(0).getId());
|
||||||
|
|
||||||
|
// now remove the movie
|
||||||
|
tmdb.removeMovieFromList(tokenSession.getSessionId(), listId, movieID);
|
||||||
|
assertEquals(tmdb.getList(listId).getItemCount(), 0);
|
||||||
|
|
||||||
|
// delete the test list
|
||||||
|
StatusCode statusCode = tmdb.deleteMovieList(tokenSession.getSessionId(), listId);
|
||||||
|
assertEquals(statusCode.getStatusCode(), 13);
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue