X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FSoneParserTest.java;h=6986e361934b90c4e27eae067498b2159300421a;hp=dc1919598dd12c8b6f2c41f1d88d8d2054f9c3ee;hb=e4daf856661958be1f88f0d3861823df91c78b1e;hpb=6b65a32a69b0e3721c58b4501a6b58fa4989f162

diff --git a/src/test/java/net/pterodactylus/sone/core/SoneParserTest.java b/src/test/java/net/pterodactylus/sone/core/SoneParserTest.java
index dc19195..6986e36 100644
--- a/src/test/java/net/pterodactylus/sone/core/SoneParserTest.java
+++ b/src/test/java/net/pterodactylus/sone/core/SoneParserTest.java
@@ -30,6 +30,7 @@ import java.util.Set;
 import net.pterodactylus.sone.data.Album;
 import net.pterodactylus.sone.data.Album.Modifier;
 import net.pterodactylus.sone.data.Client;
+import net.pterodactylus.sone.data.IdBuilder;
 import net.pterodactylus.sone.data.Image;
 import net.pterodactylus.sone.data.Post;
 import net.pterodactylus.sone.data.PostReply;
@@ -435,8 +436,8 @@ public class SoneParserTest {
 		when(imageBuilder.withId(anyString())).thenAnswer(new Answer<ImageBuilder>() {
 			@Override
 			public ImageBuilder answer(InvocationOnMock invocation) {
-				when(image.getId()).thenReturn(
-						(String) invocation.getArguments()[0]);
+				when(image.getId()).thenReturn(new IdBuilder().buildId("identity", (String) invocation.getArguments()[0]));
+				when(image.getInternalId()).thenReturn((String) invocation.getArguments()[0]);
 				return imageBuilder;
 			}
 		});
@@ -818,7 +819,8 @@ public class SoneParserTest {
 		assertThat(sone.getRootAlbum().getAlbums(), hasSize(1));
 		assertThat(sone.getRootAlbum().getAlbums().get(0).getImages(), hasSize(1));
 		Image image = sone.getRootAlbum().getAlbums().get(0).getImages().get(0);
-		assertThat(image.getId(), is("image-id"));
+		assertThat(image.getId(), is(new IdBuilder().buildId("identity", "image-id")));
+		assertThat(image.getInternalId(), is("image-id"));
 		assertThat(image.getCreationTime(), is(1407197508000L));
 		assertThat(image.getKey(), is("KSK@GPLv3.txt"));
 		assertThat(image.getTitle(), is("image-title"));