func TestDeleteMovie(t *testing.T) {
lib, err := newMockLibrary()
defer lib.cleanup()
if err != nil {
t.Fatalf("expected no error, got %q", err)
}
m, err := lib.mockMovie("movieTest")
if err != nil {
t.Fatalf("expected no error, got %q", err)
}
// Add the movie to the library
if err := lib.Add(m, mockLogEntry); err != nil {
t.Fatalf("failed to add the movie: %q", err)
}
// Count the movies in the index
ids := lib.MovieIDs()
movieCount := len(ids)
if movieCount != 1 {
t.Fatalf("the library should contains 1 movie instead of %d", movieCount)
}
// Delete the movie from the library
if err := lib.Delete(m, mockLogEntry); err != nil {
t.Fatalf("failed to add the movie: %q", err)
}
// Count the movies in the index
ids = lib.MovieIDs()
movieCount = len(ids)
if movieCount != 0 {
t.Fatalf("the library should contains 0 movie instead of %d", movieCount)
}
// Ensure the movie folder has been deleted
if exists(lib.getMovieDir(m)) {
t.Fatal("the movie directory should have been deleted", err)
}
// Rebuild the index, it should remain empty
if err := lib.RebuildIndex(mockLogEntry); err != nil {
t.Fatalf("expected no error, got %q", err)
}
// Count the movies in the index
ids = lib.MovieIDs()
movieCount = len(ids)
if movieCount != 0 {
t.Fatalf("the library should contains 0 movie instead of %d", movieCount)
}
}
func TestAddMovie(t *testing.T) {
lib, err := newMockLibrary()
defer lib.cleanup()
if err != nil {
t.Fatalf("expected no error, got %q", err)
}
m, err := lib.mockMovie("movieTest.mp4")
if err != nil {
t.Fatalf("expected no error, got %q", err)
}
oldMoviePath := m.Path
// Add the movie to the library
if err := lib.Add(m, mockLogEntry); err != nil {
t.Fatalf("failed to add the movie: %q", err)
}
// Check the new file location
expectedPath := filepath.Join(lib.tmpDir, "movies/Movie tt12345 (2000)/movieTest.mp4")
if m.Path != expectedPath {
t.Errorf("file location, expected %q got %q", expectedPath, m.Path)
}
// Check that the old path is a symlink that point to the movie's new path
gotNewPath, err := filepath.EvalSymlinks(oldMoviePath)
if err != nil {
t.Fatalf("expected no error, got %q", err)
}
if gotNewPath != m.Path {
t.Errorf("invalid symlink, expected %q got %q", m.Path, gotNewPath)
}
// Get the same mock movie again
m, err = lib.mockMovie("movieTest.mp4")
if err != nil {
t.Fatalf("expected no error, got %q", err)
}
// Add the movie again, this should replace the old file
if err := lib.Add(m, mockLogEntry); err != nil {
t.Fatalf("failed to add the movie again: %q", err)
}
// Check the content of the downloaded files
for _, imgPath := range []string{
m.MovieFanartPath(),
m.MovieThumbPath(),
} {
content, err := ioutil.ReadFile(imgPath)
if err != nil {
t.Fatalf("failed to add the movie: %q", err)
}
// The mock content comes from the httptest server
if string(content) != "mockContent" {
t.Error("invalid image content")
}
}
// Check the movie index
expectedIDs := []string{m.ImdbID}
gotIDs := lib.MovieIDs()
if !reflect.DeepEqual(gotIDs, expectedIDs) {
t.Errorf("invalid ids, expected %+v got %+v", expectedIDs, gotIDs)
}
// Ensure the library has the movie
hasMovie, err := lib.HasVideo(m)
if err != nil {
t.Fatalf("expected no error, got %q", err)
}
if !hasMovie {
t.Fatal("the movie should be in the index")
}
// Get the movie from the lib
movieFromLib, err := lib.GetMovie(m.ImdbID)
if err != nil {
t.Fatalf("expected no error, got %q", err)
}
if !reflect.DeepEqual(m, movieFromLib) {
t.Errorf("invalid movie from lib, expected %+v got %+v", m, movieFromLib)
}
// Rebuild the index, the movie should be found and added to the index
if err := lib.RebuildIndex(mockLogEntry); err != nil {
t.Fatalf("expected no error, got %q", err)
}
// Check the movie index again
gotIDs = lib.MovieIDs()
if !reflect.DeepEqual(gotIDs, expectedIDs) {
t.Errorf("invalid ids, expected %+v got %+v", expectedIDs, gotIDs)
}
}
