X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FFreenetInterfaceTest.java;h=c753740791c0c593a9b949f5760ed2d8ba8b878b;hb=bf767ac0b8e02d82ca0253741f1c83ea10faef12;hp=aa261f54b193f49bf17f30dbc68fa3a58f97c873;hpb=fdffe49b10613e1d9caacdf62ab99bca06edf3e1;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 aa261f5..c753740 100644
--- a/src/test/java/net/pterodactylus/sone/core/FreenetInterfaceTest.java
+++ b/src/test/java/net/pterodactylus/sone/core/FreenetInterfaceTest.java
@@ -38,7 +38,7 @@ import net.pterodactylus.sone.core.event.ImageInsertFailedEvent;
 import net.pterodactylus.sone.core.event.ImageInsertFinishedEvent;
 import net.pterodactylus.sone.core.event.ImageInsertStartedEvent;
 import net.pterodactylus.sone.data.Image;
-import net.pterodactylus.sone.data.ImageImpl;
+import net.pterodactylus.sone.data.impl.ImageImpl;
 import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.sone.data.TemporaryImage;
 import net.pterodactylus.sone.freenet.StringBucket;
@@ -223,6 +223,42 @@ public class FreenetInterfaceTest {
 	}
 
 	@Test
+	public void registeringAnActiveUskWillSubscribeToItCorrectly() {
+		FreenetURI freenetUri = createRandom(randomSource, "test-0").getURI().uskForSSK();
+		final USKCallback uskCallback = mock(USKCallback.class);
+		freenetInterface.registerActiveUsk(freenetUri, uskCallback);
+		verify(uskManager).subscribe(any(USK.class), eq(uskCallback), eq(true), any(RequestClient.class));
+	}
+
+	@Test
+	public void registeringAnInactiveUskWillSubscribeToItCorrectly() {
+		FreenetURI freenetUri = createRandom(randomSource, "test-0").getURI().uskForSSK();
+		final USKCallback uskCallback = mock(USKCallback.class);
+		freenetInterface.registerPassiveUsk(freenetUri, uskCallback);
+		verify(uskManager).subscribe(any(USK.class), eq(uskCallback), eq(false), any(RequestClient.class));
+	}
+
+	@Test
+	public void registeringAnActiveNonUskWillNotSubscribeToAUsk()
+	throws MalformedURLException {
+		FreenetURI freenetUri = createRandom(randomSource, "test-0").getURI();
+	    freenetInterface.registerActiveUsk(freenetUri, null);
+		verify(uskManager, never()).subscribe(any(USK.class),
+				any(USKCallback.class), anyBoolean(),
+				eq((RequestClient) highLevelSimpleClient));
+	}
+
+	@Test
+	public void registeringAnInactiveNonUskWillNotSubscribeToAUsk()
+	throws MalformedURLException {
+		FreenetURI freenetUri = createRandom(randomSource, "test-0").getURI();
+	    freenetInterface.registerPassiveUsk(freenetUri, null);
+		verify(uskManager, never()).subscribe(any(USK.class),
+				any(USKCallback.class), anyBoolean(),
+				eq((RequestClient) highLevelSimpleClient));
+	}
+
+	@Test
 	public void unregisteringANotRegisteredUskDoesNothing() {
 		FreenetURI freenetURI = createRandom(randomSource, "test-0").getURI().uskForSSK();
 		freenetInterface.unregisterUsk(freenetURI);