X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FFreenetInterfaceTest.java;h=d2a11217cbb5f0ef880a2c8f6203770d2a30cf3c;hb=f67d0cbb1bc328c0072fe5f795ffb31cf1b987bb;hp=f4217eb228691215c628944827f08829734faa0c;hpb=fa017bd4ce41d111cde531a4a7b1dc0320c471d3;p=Sone.git

diff --git a/src/test/java/net/pterodactylus/sone/core/FreenetInterfaceTest.java b/src/test/java/net/pterodactylus/sone/core/FreenetInterfaceTest.java
index f4217eb..d2a1121 100644
--- a/src/test/java/net/pterodactylus/sone/core/FreenetInterfaceTest.java
+++ b/src/test/java/net/pterodactylus/sone/core/FreenetInterfaceTest.java
@@ -93,6 +93,7 @@ public class FreenetInterfaceTest {
 	private final Sone sone = mock(Sone.class);
 	private final ArgumentCaptor<USKCallback> callbackCaptor = forClass(USKCallback.class);
 	private final SoneDownloader soneDownloader = mock(SoneDownloader.class);
+	private final SoneUpdater soneUpdater = mock(SoneUpdater.class);
 	private final Image image = mock(Image.class);
 	private InsertToken insertToken;
 
@@ -290,32 +291,9 @@ public class FreenetInterfaceTest {
 
 	@Test
 	public void callbackForRegisteredSoneWithHigherEditionTriggersDownload() throws InterruptedException {
-		freenetInterface.registerUsk(sone, soneDownloader);
-		final CountDownLatch downloadTriggered = new CountDownLatch(1);
-		doAnswer(new Answer<Void>() {
-			@Override
-			public Void answer(InvocationOnMock invocation) throws Throwable {
-				downloadTriggered.countDown();
-				return null;
-			}
-		}).when(soneDownloader).fetchSone(sone);
+		freenetInterface.registerUsk(sone, soneUpdater);
 		callbackCaptor.getValue().onFoundEdition(1, null, null, null, false, (short) 0, null, false, false);
-		assertThat(downloadTriggered.await(1, SECONDS), is(true));
-	}
-
-	@Test
-	public void callbackForRegisteredSoneWithTheSameEditionDoesNotTriggerDownload() throws InterruptedException {
-		freenetInterface.registerUsk(sone, soneDownloader);
-		final CountDownLatch downloadTriggered = new CountDownLatch(1);
-		doAnswer(new Answer<Void>() {
-			@Override
-			public Void answer(InvocationOnMock invocation) throws Throwable {
-				downloadTriggered.countDown();
-				return null;
-			}
-		}).when(soneDownloader).fetchSone(sone);
-		callbackCaptor.getValue().onFoundEdition(0, null, null, null, false, (short) 0, null, false, false);
-		assertThat(downloadTriggered.await(1, SECONDS), is(false));
+		verify(soneUpdater).updateSone(1);
 	}
 
 	@Test