From 746a59d03f62acf17083065011c6048c4d2ddd6f Mon Sep 17 00:00:00 2001
From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?=
 <bombe@pterodactylus.net>
Date: Fri, 1 Aug 2014 02:20:17 +0200
Subject: [PATCH] Add unit test for image inserter.

---
 .../pterodactylus/sone/core/ImageInserterTest.java | 58 ++++++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 src/test/java/net/pterodactylus/sone/core/ImageInserterTest.java

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();
+	}
+
+}
-- 
2.7.4