* fall back to year field if exact release date in unavailable
This commit is contained in:
parent
d1b80b825c
commit
7305b0f939
|
@ -231,6 +231,11 @@ public class IMDbClient implements MovieIdentificationService {
|
|||
Date released = Date.parse(data.get("released"), "dd MMM yyyy");
|
||||
if (released != null) {
|
||||
fields.put(MovieProperty.release_date, released.format("yyyy-MM-dd"));
|
||||
} else {
|
||||
Date year = Date.parse(data.get("year"), "yyyy");
|
||||
if (year != null) {
|
||||
fields.put(MovieProperty.release_date, year.format("yyyy-MM-dd"));
|
||||
}
|
||||
}
|
||||
|
||||
List<String> genres = new ArrayList<String>();
|
||||
|
|
|
@ -1,165 +1,158 @@
|
|||
|
||||
package net.sourceforge.filebot.web;
|
||||
|
||||
|
||||
import static org.junit.Assert.*;
|
||||
import static org.junit.Assert.assertEquals;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.Locale;
|
||||
|
||||
import net.sourceforge.filebot.web.TMDbClient.MovieInfo;
|
||||
|
||||
import org.junit.Test;
|
||||
|
||||
|
||||
public class IMDbClientTest {
|
||||
|
||||
|
||||
private final IMDbClient imdb = new IMDbClient();
|
||||
|
||||
|
||||
|
||||
@Test
|
||||
public void searchMovie1() throws Exception {
|
||||
List<Movie> results = imdb.searchMovie("Avatar", null);
|
||||
Movie movie = results.get(0);
|
||||
|
||||
|
||||
assertEquals("Avatar", movie.getName());
|
||||
assertEquals(2009, movie.getYear());
|
||||
assertEquals(499549, movie.getImdbId(), 0);
|
||||
}
|
||||
|
||||
|
||||
|
||||
@Test
|
||||
public void searchMovie2() throws Exception {
|
||||
List<Movie> results = imdb.searchMovie("The Illusionist", null);
|
||||
Movie movie = results.get(0);
|
||||
|
||||
|
||||
assertEquals("The Illusionist", movie.getName());
|
||||
assertEquals(2006, movie.getYear());
|
||||
assertEquals(443543, movie.getImdbId(), 0);
|
||||
}
|
||||
|
||||
|
||||
|
||||
@Test
|
||||
public void searchMovie3() throws Exception {
|
||||
List<Movie> results = imdb.searchMovie("Amélie", null);
|
||||
Movie movie = results.get(0);
|
||||
|
||||
|
||||
assertEquals("Amélie", movie.getName());
|
||||
assertEquals(2001, movie.getYear());
|
||||
assertEquals(211915, movie.getImdbId(), 0);
|
||||
}
|
||||
|
||||
|
||||
|
||||
@Test
|
||||
public void searchMovie4() throws Exception {
|
||||
List<Movie> results = imdb.searchMovie("Heat", null);
|
||||
Movie movie = results.get(0);
|
||||
|
||||
|
||||
assertEquals("Heat", movie.getName());
|
||||
assertEquals(1995, movie.getYear());
|
||||
assertEquals(113277, movie.getImdbId(), 0);
|
||||
}
|
||||
|
||||
|
||||
|
||||
@Test
|
||||
public void searchMovie5() throws Exception {
|
||||
List<Movie> results = imdb.searchMovie("Det sjunde inseglet", null);
|
||||
Movie movie = results.get(0);
|
||||
|
||||
|
||||
assertEquals("The Seventh Seal", movie.getName());
|
||||
assertEquals(1957, movie.getYear());
|
||||
assertEquals(50976, movie.getImdbId(), 0);
|
||||
}
|
||||
|
||||
|
||||
|
||||
@Test
|
||||
public void searchMovie6() throws Exception {
|
||||
List<Movie> results = imdb.searchMovie("Drive 2011", null);
|
||||
Movie movie = results.get(0);
|
||||
|
||||
|
||||
assertEquals("Drive", movie.getName());
|
||||
assertEquals(2011, movie.getYear());
|
||||
assertEquals(780504, movie.getImdbId(), 0);
|
||||
}
|
||||
|
||||
|
||||
|
||||
@Test
|
||||
public void searchMovieRedirect() throws Exception {
|
||||
List<Movie> results = imdb.searchMovie("(500) Days of Summer (2009)", null);
|
||||
|
||||
|
||||
Movie movie = results.get(0);
|
||||
|
||||
|
||||
assertEquals("(500) Days of Summer", movie.getName());
|
||||
assertEquals(2009, movie.getYear());
|
||||
assertEquals(1022603, movie.getImdbId(), 0);
|
||||
}
|
||||
|
||||
|
||||
|
||||
@Test
|
||||
public void getMovieDescriptor1() throws Exception {
|
||||
Movie movie = imdb.getMovieDescriptor(499549, null);
|
||||
|
||||
|
||||
assertEquals("Avatar", movie.getName());
|
||||
assertEquals(2009, movie.getYear());
|
||||
assertEquals(499549, movie.getImdbId(), 0);
|
||||
}
|
||||
|
||||
|
||||
|
||||
@Test
|
||||
public void getMovieDescriptor2() throws Exception {
|
||||
Movie movie = imdb.getMovieDescriptor(211915, null);
|
||||
|
||||
|
||||
assertEquals("Amélie", movie.getName());
|
||||
assertEquals(2001, movie.getYear());
|
||||
assertEquals(211915, movie.getImdbId(), 0);
|
||||
}
|
||||
|
||||
|
||||
|
||||
@Test
|
||||
public void getMovieDescriptor3() throws Exception {
|
||||
Movie movie = imdb.getMovieDescriptor(75610, null);
|
||||
|
||||
|
||||
assertEquals("21", movie.getName());
|
||||
assertEquals(1977, movie.getYear());
|
||||
assertEquals(75610, movie.getImdbId(), 0);
|
||||
}
|
||||
|
||||
|
||||
|
||||
@Test
|
||||
public void getMovieDescriptor4() throws Exception {
|
||||
Movie movie = imdb.getMovieDescriptor(369702, null);
|
||||
|
||||
|
||||
assertEquals("The Sea Inside", movie.getName());
|
||||
assertEquals(2004, movie.getYear());
|
||||
assertEquals(369702, movie.getImdbId(), 0);
|
||||
}
|
||||
|
||||
|
||||
|
||||
@Test
|
||||
public void getMovieDescriptor5() throws Exception {
|
||||
Movie movie = imdb.getMovieDescriptor(1020960, null);
|
||||
|
||||
|
||||
assertEquals("God, the Universe and Everything Else", movie.getName());
|
||||
assertEquals(1988, movie.getYear());
|
||||
assertEquals(1020960, movie.getImdbId(), 0);
|
||||
}
|
||||
|
||||
|
||||
|
||||
@Test
|
||||
public void getAkaMovieDescriptor() throws Exception {
|
||||
Movie movie = imdb.getMovieDescriptor(106559, Locale.ENGLISH);
|
||||
|
||||
|
||||
assertEquals("Ching Se", movie.getName());
|
||||
assertEquals(1993, movie.getYear());
|
||||
assertEquals(106559, movie.getImdbId(), 0);
|
||||
}
|
||||
|
||||
|
||||
|
||||
@Test
|
||||
public void getAkaMovieDescriptorExtra() throws Exception {
|
||||
Movie movie = imdb.getMovieDescriptor(470761, Locale.ENGLISH);
|
||||
|
||||
|
||||
assertEquals("First Born", movie.getName());
|
||||
assertEquals(2007, movie.getYear());
|
||||
assertEquals(470761, movie.getImdbId(), 0);
|
||||
}
|
||||
|
||||
|
||||
@Test
|
||||
public void getImdbApiMovieInfoReleasedNA() throws Exception {
|
||||
MovieInfo movie = imdb.getImdbApiMovieInfo(new Movie(null, -1, 1287357, -1));
|
||||
assertEquals("Sommersonntag", movie.getName());
|
||||
assertEquals(2008, movie.getReleased().getYear());
|
||||
assertEquals("2008-01-01", movie.getReleased().toString());
|
||||
}
|
||||
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue