X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2FTestAlbumBuilder.java;h=a499019cec787370c2517ebdc71c38168fce9f12;hp=9890a70f349f72bc56d8c9dbc60ed3860f4aceaa;hb=91b8268946412b35bcce5b4406106ecd4cb58e3c;hpb=3a86d8894d306039e8d523763420350997fafa2e

diff --git a/src/test/java/net/pterodactylus/sone/TestAlbumBuilder.java b/src/test/java/net/pterodactylus/sone/TestAlbumBuilder.java
index 9890a70..a499019 100644
--- a/src/test/java/net/pterodactylus/sone/TestAlbumBuilder.java
+++ b/src/test/java/net/pterodactylus/sone/TestAlbumBuilder.java
@@ -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;
 			}