X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2Fimpl%2FDefaultImageBuilderTest.java;h=2f007356766f471b31bc33b0869470600a1dee02;hb=8230a34cdb7e2ee5f796008f66e99a45160f5d1d;hp=c1368188000a097218435a6104568e4e4076c6d1;hpb=4e61d9d4c916ff7d314baa2923087203aec3842d;p=Sone.git

diff --git a/src/test/java/net/pterodactylus/sone/data/impl/DefaultImageBuilderTest.java b/src/test/java/net/pterodactylus/sone/data/impl/DefaultImageBuilderTest.java
index c136818..2f00735 100644
--- a/src/test/java/net/pterodactylus/sone/data/impl/DefaultImageBuilderTest.java
+++ b/src/test/java/net/pterodactylus/sone/data/impl/DefaultImageBuilderTest.java
@@ -22,7 +22,6 @@ import static org.hamcrest.CoreMatchers.notNullValue;
 import static org.hamcrest.MatcherAssert.assertThat;
 import static org.mockito.Mockito.mock;
 
-import net.pterodactylus.sone.data.Album;
 import net.pterodactylus.sone.data.Image;
 import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.sone.database.ImageBuilder;
@@ -44,7 +43,7 @@ public class DefaultImageBuilderTest {
 	private static final int HEIGHT = 270;
 
 	private final Sone sone = mock(Sone.class);
-	private final Album album = mock(Album.class);
+	private final DefaultAlbum album = mock(DefaultAlbum.class);
 	private final ImageBuilder imageBuilder = new DefaultImageBuilder(sone, album);
 
 	@Test
@@ -61,7 +60,6 @@ public class DefaultImageBuilderTest {
 
 	@Test
 	public void testImageCreationWithRandomId() {
-		Sone sone = mock(Sone.class);
 		Image image = imageBuilder.randomId().created(CREATION_TIME).at(KEY).sized(WIDTH, HEIGHT).build();
 		assertThat(image, CoreMatchers.notNullValue());
 		assertThat(image.getId(), notNullValue());