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