X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FImageInserterTest.java;fp=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FImageInserterTest.java;h=0912391776c933a4170d7da97e8ed96c1e2d9c4d;hb=746a59d03f62acf17083065011c6048c4d2ddd6f;hp=0000000000000000000000000000000000000000;hpb=916a38a274bcddc755c06fe21bdbc41d494b1825;p=Sone.git

diff --git a/src/test/java/net/pterodactylus/sone/core/ImageInserterTest.java b/src/test/java/net/pterodactylus/sone/core/ImageInserterTest.java
new file mode 100644
index 0000000..0912391
--- /dev/null
+++ b/src/test/java/net/pterodactylus/sone/core/ImageInserterTest.java
@@ -0,0 +1,58 @@
+package net.pterodactylus.sone.core;
+
+import static org.mockito.Matchers.any;
+import static org.mockito.Matchers.eq;
+import static org.mockito.Mockito.doThrow;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.never;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+import net.pterodactylus.sone.core.FreenetInterface.InsertToken;
+import net.pterodactylus.sone.data.Image;
+import net.pterodactylus.sone.data.TemporaryImage;
+
+import com.google.common.base.Function;
+import org.junit.Test;
+
+/**
+ * Unit test for {@link ImageInserter}.
+ *
+ * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+ */
+public class ImageInserterTest {
+
+	private final TemporaryImage temporaryImage = when(mock(TemporaryImage.class).getId()).thenReturn("image-id").getMock();
+	private final Image image = when(mock(Image.class).getId()).thenReturn("image-id").getMock();
+	private final FreenetInterface freenetInterface = mock(FreenetInterface.class);
+	private final InsertToken insertToken = mock(InsertToken.class);
+	private final Function<Image, InsertToken> insertTokenSupplier = when(mock(Function.class).apply(any(Image.class))).thenReturn(insertToken).getMock();
+	private final ImageInserter imageInserter = new ImageInserter(freenetInterface, insertTokenSupplier);
+
+	@Test
+	public void inserterInsertsImage() throws SoneException {
+		imageInserter.insertImage(temporaryImage, image);
+		verify(freenetInterface).insertImage(eq(temporaryImage), eq(image), any(InsertToken.class));
+	}
+
+	@Test
+	public void exceptionWhenInsertingImageIsIgnored() throws SoneException {
+		doThrow(SoneException.class).when(freenetInterface).insertImage(eq(temporaryImage), eq(image), any(InsertToken.class));
+		imageInserter.insertImage(temporaryImage, image);
+		verify(freenetInterface).insertImage(eq(temporaryImage), eq(image), any(InsertToken.class));
+	}
+
+	@Test
+	public void cancellingImageInsertThatIsNotRunningDoesNothing() {
+		imageInserter.cancelImageInsert(image);
+		verify(insertToken, never()).cancel();
+	}
+
+	@Test
+	public void cancellingImage() {
+		imageInserter.insertImage(temporaryImage, image);
+		imageInserter.cancelImageInsert(image);
+		verify(insertToken).cancel();
+	}
+
+}