X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2FTestAlbumBuilder.java;h=ea211187e86d16cb0841e7dcdff3a8bd76a65b47;hp=9890a70f349f72bc56d8c9dbc60ed3860f4aceaa;hb=7b55e0be6a3283e43a9bbab98f82aebdd948eb33;hpb=f229fe41f708d2b275c20ceb9aba5993761218a3

diff --git a/src/test/java/net/pterodactylus/sone/TestAlbumBuilder.java b/src/test/java/net/pterodactylus/sone/TestAlbumBuilder.java
index 9890a70..ea21118 100644
--- a/src/test/java/net/pterodactylus/sone/TestAlbumBuilder.java
+++ b/src/test/java/net/pterodactylus/sone/TestAlbumBuilder.java
@@ -1,7 +1,7 @@
 package net.pterodactylus.sone;
 
 import static java.util.UUID.randomUUID;
-import static org.mockito.Matchers.any;
+import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.Mockito.doAnswer;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
@@ -46,17 +46,6 @@ public class TestAlbumBuilder implements AlbumBuilder {
 				return description;
 			}
 		});
-		when(album.getAlbumImage()).thenAnswer(new Answer<Image>() {
-			@Override
-			public Image answer(InvocationOnMock invocation) {
-				if (imageId == null) {
-					return null;
-				}
-				Image image = mock(Image.class);
-				when(image.getId()).thenReturn(imageId);
-				return image;
-			}
-		});
 		when(album.getAlbums()).thenReturn(albums);
 		when(album.getImages()).thenReturn(images);
 		doAnswer(new Answer<Void>() {
@@ -101,12 +90,6 @@ public class TestAlbumBuilder implements AlbumBuilder {
 			}
 
 			@Override
-			public Modifier setAlbumImage(String imageId) {
-				TestAlbumBuilder.this.imageId = imageId;
-				return this;
-			}
-
-			@Override
 			public Album update() throws IllegalStateException {
 				return album;
 			}