Extract default Image implementation as base for all future image implementations.

[Sone.git] / src / test / java / net / pterodactylus / sone / data / impl / ImageBuilderImplTest.java

diff --git a/src/test/java/net/pterodactylus/sone/data/impl/ImageBuilderImplTest.java b/src/test/java/net/pterodactylus/sone/data/impl/ImageBuilderImplTest.java
index 503e6f1..b5d2a84 100644 (file)
--- a/src/test/java/net/pterodactylus/sone/data/impl/ImageBuilderImplTest.java
+++ b/src/test/java/net/pterodactylus/sone/data/impl/ImageBuilderImplTest.java
@@ -22,6 +22,7 @@ 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;
@@ -42,8 +43,9 @@ public class ImageBuilderImplTest {
        private static final int WIDTH = 640;
        private static final int HEIGHT = 270;
 
-       private final ImageBuilder imageBuilder = new ImageBuilderImpl();
        private final Sone sone = mock(Sone.class);
+       private final Album album = mock(Album.class);
+       private final ImageBuilder imageBuilder = new ImageBuilderImpl(album);
 
        @Test
        public void testImageCreationWithAllExplicitParameters() {