X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2FWebPageTest.java;h=6f99879bf6d7f445ed82d0ad064372dabee02b50;hp=df000b859433152e1eb60a11b1807b10b9cd01d3;hb=9acbc5bdec4ccb752e0856a501568b0bb6161579;hpb=23202a30c41448d317a34ef87210bc236030ff89 diff --git a/src/test/java/net/pterodactylus/sone/web/WebPageTest.java b/src/test/java/net/pterodactylus/sone/web/WebPageTest.java index df000b8..6f99879 100644 --- a/src/test/java/net/pterodactylus/sone/web/WebPageTest.java +++ b/src/test/java/net/pterodactylus/sone/web/WebPageTest.java @@ -1,5 +1,6 @@ package net.pterodactylus.sone.web; +import static net.pterodactylus.sone.test.GuiceKt.supply; import static net.pterodactylus.sone.web.WebTestUtils.redirectsTo; import static org.junit.Assert.fail; import static org.mockito.ArgumentMatchers.anyBoolean; @@ -12,6 +13,7 @@ import static org.mockito.Mockito.when; import java.io.ByteArrayOutputStream; import java.io.IOException; +import java.io.InputStream; import java.io.PipedInputStream; import java.io.PipedOutputStream; import java.net.URI; @@ -47,15 +49,20 @@ import net.pterodactylus.util.web.Response; import freenet.clients.http.ToadletContext; import freenet.l10n.BaseL10n; +import freenet.support.SimpleReadOnlyArrayBucket; +import freenet.support.api.Bucket; import freenet.support.api.HTTPRequest; +import freenet.support.api.HTTPUploadedFile; +import freenet.support.io.NullBucket; import com.google.common.base.Optional; import com.google.common.collect.ArrayListMultimap; -import com.google.common.collect.HashMultimap; import com.google.common.collect.ListMultimap; import com.google.common.collect.Multimap; import com.google.common.eventbus.EventBus; import com.google.common.io.ByteStreams; +import com.google.inject.Guice; +import com.google.inject.Injector; import org.junit.Before; import org.junit.Rule; import org.junit.rules.ExpectedException; @@ -83,8 +90,11 @@ public abstract class WebPageTest { protected final TemplateContext templateContext = new TemplateContext(); protected final HTTPRequest httpRequest = mock(HTTPRequest.class); - protected final Multimap requestParameters = HashMultimap.create(); + protected final Multimap requestParameters = ArrayListMultimap.create(); protected final Map requestHeaders = new HashMap<>(); + private final Map uploadedFilesNames = new HashMap<>(); + private final Map uploadedFilesContentTypes = new HashMap<>(); + private final Map uploadedFilesSources = new HashMap<>(); protected final FreenetRequest freenetRequest = mock(FreenetRequest.class); private final PipedOutputStream responseOutputStream = new PipedOutputStream(); private final PipedInputStream responseInputStream; @@ -96,6 +106,11 @@ public abstract class WebPageTest { protected final List localSones = new ArrayList<>(); private final ListMultimap postReplies = ArrayListMultimap.create(); + protected final Injector injector = Guice.createInjector( + supply(WebInterface.class).byInstance(webInterface), + supply(Template.class).byInstance(template) + ); + protected WebPageTest() { try { responseInputStream = new PipedInputStream(responseOutputStream); @@ -149,6 +164,13 @@ public abstract class WebPageTest { return requestParameters.containsKey(parameter) ? requestParameters.get(parameter).iterator().next() : invocation.getArgument(1); } }); + when(httpRequest.isParameterSet(anyString())).thenAnswer(new Answer() { + @Override + public Boolean answer(InvocationOnMock invocation) throws Throwable { + return requestParameters.containsKey(invocation.getArgument(0)) && + requestParameters.get(invocation.getArgument(0)).iterator().next() != null; + } + }); when(httpRequest.isPartSet(anyString())).thenAnswer(new Answer() { @Override public Boolean answer(InvocationOnMock invocation) throws Throwable { @@ -168,6 +190,36 @@ public abstract class WebPageTest { return requestHeaders.get(invocation.getArgument(0).toLowerCase()); } }); + when(httpRequest.getUploadedFile(anyString())).thenAnswer(new Answer() { + @Override + public HTTPUploadedFile answer(InvocationOnMock invocation) throws Throwable { + final String name = invocation.getArgument(0); + if (!uploadedFilesSources.containsKey(name)) { + return null; + } + return new HTTPUploadedFile() { + @Override + public String getContentType() { + return uploadedFilesContentTypes.get(name); + } + + @Override + public Bucket getData() { + try (InputStream inputStream = getClass().getResourceAsStream(uploadedFilesSources.get(name))) { + byte[] bytes = ByteStreams.toByteArray(inputStream); + return new SimpleReadOnlyArrayBucket(bytes, 0, bytes.length); + } catch (IOException ioe1) { + return new NullBucket(); + } + } + + @Override + public String getFilename() { + return uploadedFilesNames.get(name); + } + }; + } + }); } @Before @@ -191,6 +243,7 @@ public abstract class WebPageTest { } }); when(core.getPost(anyString())).thenReturn(Optional.absent()); + when(core.getPostReply(anyString())).thenReturn(Optional.absent()); when(core.getReplies(anyString())).thenAnswer(new Answer>() { @Override public List answer(InvocationOnMock invocation) throws Throwable { @@ -204,14 +257,24 @@ public abstract class WebPageTest { } @Before + public void setupL10n() { + when(l10n.getString(anyString())).thenAnswer(new Answer() { + @Override + public String answer(InvocationOnMock invocation) throws Throwable { + return invocation.getArgument(0); + } + }); + } + + @Before public final void setupIdentityManager() { when(core.getIdentityManager().getAllOwnIdentities()).thenReturn(ownIdentities); } @Before public final void setupWebInterface() { - when(webInterface.getCurrentSone(toadletContext)).thenReturn(currentSone); - when(webInterface.getCurrentSone(eq(toadletContext), anyBoolean())).thenReturn(currentSone); + when(webInterface.getCurrentSoneCreatingSession(toadletContext)).thenReturn(currentSone); + when(webInterface.getCurrentSoneWithoutCreatingSession(toadletContext)).thenReturn(currentSone); when(webInterface.getNotification(anyString())).thenReturn(Optional.absent()); when(webInterface.getNotifications(currentSone)).thenReturn(new ArrayList()); } @@ -226,8 +289,8 @@ public abstract class WebPageTest { } protected void unsetCurrentSone() { - when(webInterface.getCurrentSone(toadletContext)).thenReturn(null); - when(webInterface.getCurrentSone(eq(toadletContext), anyBoolean())).thenReturn(null); + when(webInterface.getCurrentSoneCreatingSession(toadletContext)).thenReturn(null); + when(webInterface.getCurrentSoneWithoutCreatingSession(toadletContext)).thenReturn(null); } protected void request(String uri, Method method) { @@ -285,6 +348,12 @@ public abstract class WebPageTest { when(core.getTemporaryImage(eq(imageId))).thenReturn(temporaryImage); } + protected void addUploadedFile(@Nonnull String name, @Nonnull String filename, @Nonnull String contentType, @Nonnull String resource) { + uploadedFilesNames.put(name, filename); + uploadedFilesContentTypes.put(name, contentType); + uploadedFilesSources.put(name, resource); + } + protected byte[] getResponseBytes() throws IOException { response.getContent().close(); try (ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) { @@ -302,7 +371,7 @@ public abstract class WebPageTest { getPage().handleRequest(freenetRequest, templateContext); } - protected void verifyRedirect(String target, Runnable verification) throws RedirectException { + protected void verifyRedirect(String target, Runnable verification) throws RedirectException { expectedException.expect(redirectsTo(target)); try { getPage().handleRequest(freenetRequest, templateContext);