X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FSoneDownloaderTest.java;fp=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FSoneDownloaderTest.java;h=28879bf50e58cbc418ef749eafaba0f43bc6282e;hp=c910233bc08186621380364810b292515dc7b37d;hb=210684b4bc499e298a0d0abeddd4008cdeb406bc;hpb=2b47186b72e30460a6710f95a76e4a99c305909a

diff --git a/src/test/java/net/pterodactylus/sone/core/SoneDownloaderTest.java b/src/test/java/net/pterodactylus/sone/core/SoneDownloaderTest.java
index c910233..28879bf 100644
--- a/src/test/java/net/pterodactylus/sone/core/SoneDownloaderTest.java
+++ b/src/test/java/net/pterodactylus/sone/core/SoneDownloaderTest.java
@@ -49,6 +49,8 @@ import net.pterodactylus.sone.database.AlbumBuilder;
 import net.pterodactylus.sone.database.ImageBuilder;
 import net.pterodactylus.sone.database.PostBuilder;
 import net.pterodactylus.sone.database.PostReplyBuilder;
+import net.pterodactylus.sone.database.SoneBuilder;
+import net.pterodactylus.sone.database.memory.MemorySoneBuilder;
 import net.pterodactylus.sone.freenet.wot.Identity;
 
 import freenet.client.ClientMetadata;
@@ -78,7 +80,7 @@ public class SoneDownloaderTest {
 	private final FreenetInterface freenetInterface = mock(FreenetInterface.class);
 	private final SoneDownloaderImpl soneDownloader = new SoneDownloaderImpl(core, freenetInterface);
 	private final FreenetURI requestUri = mock(FreenetURI.class);
-	private final Sone sone = mock(Sone.class);
+	private Sone sone = mock(Sone.class);
 	private final PostBuilder postBuilder = mock(PostBuilder.class);
 	private final List<Post> createdPosts = new ArrayList<Post>();
 	private Post post = mock(Post.class);
@@ -96,6 +98,7 @@ public class SoneDownloaderTest {
 
 	@Before
 	public void setupSone() {
+		Sone sone = SoneDownloaderTest.this.sone;
 		Identity identity = mock(Identity.class);
 		when(identity.getId()).thenReturn("identity");
 		when(sone.getId()).thenReturn("identity");
@@ -118,6 +121,16 @@ public class SoneDownloaderTest {
 	}
 
 	@Before
+	public void setupSoneBuilder() {
+		when(core.soneBuilder()).thenAnswer(new Answer<SoneBuilder>() {
+			@Override
+			public SoneBuilder answer(InvocationOnMock invocation) {
+				return new MemorySoneBuilder();
+			}
+		});
+	}
+
+	@Before
 	public void setupPost() {
 		when(post.getRecipientId()).thenReturn(Optional.<String>absent());
 	}
@@ -921,7 +934,7 @@ public class SoneDownloaderTest {
 	@Test
 	public void exceptionWhileFetchingSoneWillNotUpdateTheCore() throws IOException {
 		final Fetched fetchResult = createFetchResult(requestUri, getClass().getResourceAsStream("sone-parser-no-payload.xml"));
-		when(sone.getId()).thenThrow(NullPointerException.class);
+		when(core.soneBuilder()).thenReturn(null);
 		when(freenetInterface.fetchUri(requestUri)).thenReturn(fetchResult);
 		soneDownloader.fetchSoneAction(sone).run();
 		verify(core, never()).updateSone(any(Sone.class));