X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2FUploadImagePageTest.java;h=3b8258e0c8a1588e2011780fd21dbcc8723dde4c;hp=3bdf5880c6eca588caf534e794d4ef43116d8595;hb=f9ddb56aac0ca9530fdcb22a5d59c1d32c9b658a;hpb=d338aacec01efa636f96c5f8aab11b2345147691

diff --git a/src/test/java/net/pterodactylus/sone/web/UploadImagePageTest.java b/src/test/java/net/pterodactylus/sone/web/UploadImagePageTest.java
index 3bdf588..3b8258e 100644
--- a/src/test/java/net/pterodactylus/sone/web/UploadImagePageTest.java
+++ b/src/test/java/net/pterodactylus/sone/web/UploadImagePageTest.java
@@ -1,61 +1,29 @@
 package net.pterodactylus.sone.web;
 
 import static net.pterodactylus.sone.web.WebTestUtils.redirectsTo;
-import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.anyInt;
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
 
-import java.net.URI;
-
-import net.pterodactylus.sone.core.Core;
-import net.pterodactylus.sone.core.UpdateChecker;
 import net.pterodactylus.sone.data.Album;
-import net.pterodactylus.sone.data.Sone;
-import net.pterodactylus.sone.web.page.FreenetRequest;
-import net.pterodactylus.util.template.Template;
-import net.pterodactylus.util.template.TemplateContext;
 import net.pterodactylus.util.web.Method;
 
-import freenet.clients.http.ToadletContext;
-import freenet.support.api.HTTPRequest;
-
 import org.junit.Before;
-import org.junit.Rule;
 import org.junit.Test;
-import org.junit.rules.ExpectedException;
 
 /**
  * Unit test for {@link UploadImagePageTest}.
  *
  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  */
-public class UploadImagePageTest {
-
-	@Rule
-	public final ExpectedException expectedException = ExpectedException.none();
+public class UploadImagePageTest extends WebPageTest {
 
-	private final Template template = new Template();
-	private final WebInterface webInterface = mock(WebInterface.class);
 	private final UploadImagePage uploadImagePage = new UploadImagePage(template, webInterface);
 
-	private final TemplateContext templateContext = new TemplateContext();
-	private final HTTPRequest httpRequest = mock(HTTPRequest.class);
-	private final ToadletContext toadletContext = mock(ToadletContext.class);
-	private final Core core = mock(Core.class);
-	private final Sone currentSone = mock(Sone.class);
 	private final Album parentAlbum = mock(Album.class);
 
 	@Before
-	public void setupWebInterface() {
-		UpdateChecker updateChecker = mock(UpdateChecker.class);
-		when(core.getUpdateChecker()).thenReturn(updateChecker);
-		when(webInterface.getCore()).thenReturn(core);
-		when(webInterface.getCurrentSone(any(ToadletContext.class))).thenReturn(currentSone);
-	}
-
-	@Before
 	public void setupParentAlbum() {
 		when(core.getAlbum("parent-id")).thenReturn(parentAlbum);
 		when(parentAlbum.getSone()).thenReturn(currentSone);
@@ -63,11 +31,11 @@ public class UploadImagePageTest {
 
 	@Test
 	public void uploadingAnImageWithoutTitleRedirectsToEmptyImageTitlePage() throws Exception {
-		FreenetRequest request = new FreenetRequest(new URI(""), Method.POST, httpRequest, toadletContext);
+		request("", Method.POST);
 		when(httpRequest.getPartAsStringFailsafe(eq("parent"), anyInt())).thenReturn("parent-id");
 		when(httpRequest.getPartAsStringFailsafe(eq("title"), anyInt())).thenReturn("  ");
 		expectedException.expect(redirectsTo("emptyImageTitle.html"));
-		uploadImagePage.processTemplate(request, templateContext);
+		uploadImagePage.processTemplate(freenetRequest, templateContext);
 	}
 
 }