X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FSoneParserTest.java;h=2d8a5a0ec0b0ee9e12f309607009fd9e6e1cacd6;hb=f9b5addb5d0e7a080e9f314cf6712143f5e26060;hp=b6560f41ef1c47d858dd43183bb85864445d6b14;hpb=808a37413dbcd2f8a543f26bef5f639fccba6f4f;p=Sone.git

diff --git a/src/test/java/net/pterodactylus/sone/core/SoneParserTest.java b/src/test/java/net/pterodactylus/sone/core/SoneParserTest.java
index b6560f4..2d8a5a0 100644
--- a/src/test/java/net/pterodactylus/sone/core/SoneParserTest.java
+++ b/src/test/java/net/pterodactylus/sone/core/SoneParserTest.java
@@ -114,7 +114,7 @@ public class SoneParserTest {
 		when(core.soneBuilder()).thenAnswer(new Answer<SoneBuilder>() {
 			@Override
 			public SoneBuilder answer(InvocationOnMock invocation) {
-				return new MemorySoneBuilder();
+				return new MemorySoneBuilder(null);
 			}
 		});
 	}
@@ -243,7 +243,6 @@ public class SoneParserTest {
 	@Before
 	public void setupAlbum() {
 		final Album album = SoneParserTest.this.album;
-		when(album.getAlbumImage()).thenReturn(mock(Image.class));
 		doAnswer(new Answer<Void>() {
 			@Override
 			public Void answer(InvocationOnMock invocation) {
@@ -273,7 +272,6 @@ public class SoneParserTest {
 		final Modifier albumModifier = new Modifier() {
 			private String title = album.getTitle();
 			private String description = album.getDescription();
-			private String imageId = album.getAlbumImage().getId();
 
 			@Override
 			public Modifier setTitle(String title) {
@@ -288,18 +286,9 @@ public class SoneParserTest {
 			}
 
 			@Override
-			public Modifier setAlbumImage(String imageId) {
-				this.imageId = imageId;
-				return this;
-			}
-
-			@Override
 			public Album update() throws IllegalStateException {
 				when(album.getTitle()).thenReturn(title);
 				when(album.getDescription()).thenReturn(description);
-				Image image = mock(Image.class);
-				when(image.getId()).thenReturn(imageId);
-				when(album.getAlbumImage()).thenReturn(image);
 				return album;
 			}
 		};