X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FFreenetInterfaceTest.java;h=e92054ab2549ffd24d5cfc4a203532f2ab364a00;hb=40a1ac4c90c1480936c9c782727209cf23acd5e5;hp=f4217eb228691215c628944827f08829734faa0c;hpb=3568c3e220993291936f182ee1845bd795b8f513;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..e92054a 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(sone, 1);
 	}
 
 	@Test