@Test
public void testModifyAlbumRemovePhoto() throws Exception {
// Setup
Album album = new Album(TEST_ALBUM_NAME, TEST_ALBUM_DESCRIPTION);
album.modifyUser(user);
Album createdAlbum = repository.create(album);
Photo photo = new Photo(TEST_PHOTO_NAME, TEST_PHOTO_CONTENT);
photo.setUploadTime(new Date());
createdAlbum.addToPhotos(photo);
createdAlbum = repository.modify(createdAlbum);
em.flush();
// Execute
createdAlbum.removeFromPhotos(photo);
Album modifiedAlbum = repository.modify(createdAlbum);
// Verify
em.flush();
em.clear();
Album actualAlbum = em.find(Album.class, modifiedAlbum.getAlbumId());
assertEquals(modifiedAlbum, actualAlbum);
assertFalse(actualAlbum.getPhotos().contains(photo));
assertNull(actualAlbum.getCoverPhoto());
}
@Test
public void testModifyAlbumRemoveMultiplePhotos() throws Exception {
// Setup
Album album = new Album(TEST_ALBUM_NAME, TEST_ALBUM_DESCRIPTION);
album.modifyUser(user);
Album createdAlbum = repository.create(album);
Photo photo = new Photo(TEST_PHOTO_NAME, TEST_PHOTO_CONTENT);
photo.setUploadTime(new Date());
createdAlbum.addToPhotos(photo);
Photo secondPhoto = new Photo("Photo #2", "Photo Content #2".getBytes());
secondPhoto.setUploadTime(new Date());
createdAlbum.addToPhotos(secondPhoto);
Photo thirdPhoto = new Photo("Photo #3", "Photo Content #3".getBytes());
thirdPhoto.setUploadTime(new Date());
createdAlbum.addToPhotos(thirdPhoto);
createdAlbum = repository.modify(createdAlbum);
em.flush();
// Execute
createdAlbum.removeFromPhotos(photo);
Album modifiedAlbum = repository.modify(createdAlbum);
// Verify
em.flush();
em.clear();
Album actualAlbum = em.find(Album.class, modifiedAlbum.getAlbumId());
assertEquals(modifiedAlbum, actualAlbum);
assertThat(actualAlbum.getPhotos(), hasItems(secondPhoto, thirdPhoto));
assertThat(actualAlbum.getCoverPhoto(), either(equalTo(secondPhoto)).or(equalTo(thirdPhoto)));
}
@Test
public void testModifyAlbumModifyPhoto() throws Exception {
Album album = new Album(TEST_ALBUM_NAME, TEST_ALBUM_DESCRIPTION);
album.modifyUser(user);
Album createdAlbum = repository.create(album);
Photo photo = new Photo(TEST_PHOTO_NAME, TEST_PHOTO_CONTENT);
photo.setUploadTime(new Date());
createdAlbum.addToPhotos(photo);
createdAlbum = repository.modify(createdAlbum);
em.flush();
// Execute
String modifiedTitle = "New Title";
String modifiedDescription = "New Description";
long photoId = 0;
for (Photo eachPhoto : createdAlbum.getPhotos()) {
if (eachPhoto.equals(photo)) {
photoId = eachPhoto.getPhotoId();
eachPhoto.setTitle(modifiedTitle);
eachPhoto.setDescription(modifiedDescription);
break;
}
}
Album modifiedAlbum = repository.modify(createdAlbum);
// Verify
em.flush();
em.clear();
Album actualAlbum = em.find(Album.class, modifiedAlbum.getAlbumId());
Photo actualPhoto = em.find(Photo.class, photoId);
assertEquals(modifiedAlbum, actualAlbum);
assertTrue(actualAlbum.getPhotos().contains(photo));
assertEquals(modifiedTitle, actualPhoto.getTitle());
assertEquals(modifiedDescription, actualPhoto.getDescription());
assertEquals(actualAlbum.getCoverPhoto(), actualPhoto);
assertEquals(photo, actualAlbum.getCoverPhoto());
}
@Test
public void testDeleteAlbumWithPhoto() throws Exception {
Album album = new Album(TEST_ALBUM_NAME, TEST_ALBUM_DESCRIPTION);
album.modifyUser(user);
em.persist(album);
em.flush();
Photo photo = new Photo(TEST_PHOTO_NAME, TEST_PHOTO_CONTENT);
photo.setUploadTime(new Date());
album.addToPhotos(photo);
album = repository.modify(album);
em.flush();
// Execute
long albumId = album.getAlbumId();
repository.delete(album);
// Verify
em.flush();
em.clear();
Album actualAlbum = em.find(Album.class, albumId);
assertNull(actualAlbum);
}