X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FSoneDownloaderTest.java;h=21d01f63db36110a87e255044a3b6e341e57c44a;hb=f9ec36b55750b14c7aec195e7f3ff092a570003d;hp=413b8fff1e3f5795815d1213f406d361a022f24c;hpb=94088d808dfca4bf5464ee4f3b099175fb4c058a;p=Sone.git

diff --git a/src/test/java/net/pterodactylus/sone/core/SoneDownloaderTest.java b/src/test/java/net/pterodactylus/sone/core/SoneDownloaderTest.java
index 413b8ff..21d01f6 100644
--- a/src/test/java/net/pterodactylus/sone/core/SoneDownloaderTest.java
+++ b/src/test/java/net/pterodactylus/sone/core/SoneDownloaderTest.java
@@ -6,6 +6,7 @@ import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 
 import net.pterodactylus.sone.core.SoneDownloader.FetchSone;
+import net.pterodactylus.sone.core.SoneDownloader.FetchSoneWithUri;
 import net.pterodactylus.sone.data.Sone;
 
 import freenet.keys.FreenetURI;
@@ -20,14 +21,23 @@ import org.junit.Test;
 public class SoneDownloaderTest {
 
 	@Test
-	public void fetchSoneDownloadsSone() {
+	public void fetchSoneWithUriDownloadsSoneWithUri() {
 		SoneDownloader soneDownloader = mock(SoneDownloader.class);
 		Sone sone = mock(Sone.class);
 		FreenetURI soneUri = mock(FreenetURI.class);
 		when(sone.getRequestUri()).thenReturn(soneUri);
+		FetchSoneWithUri fetchSoneWithUri = soneDownloader.new FetchSoneWithUri(sone);
+		fetchSoneWithUri.run();
+		verify(soneDownloader).fetchSone(eq(sone), eq(soneUri));
+	}
+
+	@Test
+	public void fetchSoneDownloadsSone() {
+		SoneDownloader soneDownloader = mock(SoneDownloader.class);
+		Sone sone = mock(Sone.class);
 		FetchSone fetchSone = soneDownloader.new FetchSone(sone);
 		fetchSone.run();
-		verify(soneDownloader).fetchSone(eq(sone), eq(soneUri));
+		verify(soneDownloader).fetchSone(eq(sone));
 	}
 
 }