X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2FWebPageTest.java;h=df000b859433152e1eb60a11b1807b10b9cd01d3;hb=23202a30c41448d317a34ef87210bc236030ff89;hp=b8b7fa6dcf636402d14a12998e1527f829c06b55;hpb=108ca51ec6cf3dc90917ce763beeb16f03d2cd25;p=Sone.git diff --git a/src/test/java/net/pterodactylus/sone/web/WebPageTest.java b/src/test/java/net/pterodactylus/sone/web/WebPageTest.java index b8b7fa6..df000b8 100644 --- a/src/test/java/net/pterodactylus/sone/web/WebPageTest.java +++ b/src/test/java/net/pterodactylus/sone/web/WebPageTest.java @@ -1,5 +1,7 @@ package net.pterodactylus.sone.web; +import static net.pterodactylus.sone.web.WebTestUtils.redirectsTo; +import static org.junit.Assert.fail; import static org.mockito.ArgumentMatchers.anyBoolean; import static org.mockito.ArgumentMatchers.anyInt; import static org.mockito.ArgumentMatchers.anyString; @@ -30,11 +32,13 @@ import net.pterodactylus.sone.core.UpdateChecker; import net.pterodactylus.sone.data.Album; import net.pterodactylus.sone.data.Image; import net.pterodactylus.sone.data.Post; +import net.pterodactylus.sone.data.PostReply; import net.pterodactylus.sone.data.Sone; import net.pterodactylus.sone.data.SoneOptions.DefaultSoneOptions; import net.pterodactylus.sone.data.TemporaryImage; import net.pterodactylus.sone.freenet.wot.OwnIdentity; import net.pterodactylus.sone.web.page.FreenetRequest; +import net.pterodactylus.sone.web.page.FreenetTemplatePage.RedirectException; import net.pterodactylus.util.notify.Notification; import net.pterodactylus.util.template.Template; import net.pterodactylus.util.template.TemplateContext; @@ -42,9 +46,14 @@ import net.pterodactylus.util.web.Method; import net.pterodactylus.util.web.Response; import freenet.clients.http.ToadletContext; +import freenet.l10n.BaseL10n; import freenet.support.api.HTTPRequest; 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 org.junit.Before; @@ -68,12 +77,13 @@ public abstract class WebPageTest { protected final WebInterface webInterface = mock(WebInterface.class, RETURNS_DEEP_STUBS); protected final EventBus eventBus = mock(EventBus.class); protected final Core core = webInterface.getCore(); + protected final BaseL10n l10n = webInterface.getL10n(); protected final Sone currentSone = mock(Sone.class); protected final TemplateContext templateContext = new TemplateContext(); protected final HTTPRequest httpRequest = mock(HTTPRequest.class); - protected final Map requestParameters = new HashMap<>(); + protected final Multimap requestParameters = HashMultimap.create(); protected final Map requestHeaders = new HashMap<>(); protected final FreenetRequest freenetRequest = mock(FreenetRequest.class); private final PipedOutputStream responseOutputStream = new PipedOutputStream(); @@ -83,7 +93,8 @@ public abstract class WebPageTest { private final Set ownIdentities = new HashSet<>(); private final Map sones = new HashMap<>(); - private final List localSones = new ArrayList<>(); + protected final List localSones = new ArrayList<>(); + private final ListMultimap postReplies = ArrayListMultimap.create(); protected WebPageTest() { try { @@ -97,32 +108,52 @@ public abstract class WebPageTest { public final void setupFreenetRequest() { when(freenetRequest.getToadletContext()).thenReturn(toadletContext); when(freenetRequest.getHttpRequest()).thenReturn(httpRequest); + when(httpRequest.getMultipleParam(anyString())).thenAnswer(new Answer() { + @Override + public String[] answer(InvocationOnMock invocation) throws Throwable { + return requestParameters.get(invocation.getArgument(0)).toArray(new String[0]); + } + }); when(httpRequest.getPartAsStringFailsafe(anyString(), anyInt())).thenAnswer(new Answer() { @Override public String answer(InvocationOnMock invocation) throws Throwable { String parameter = invocation.getArgument(0); int maxLength = invocation.getArgument(1); - return requestParameters.containsKey(parameter) ? requestParameters.get(parameter).substring(0, Math.min(maxLength, requestParameters.get(parameter).length())) : ""; + Collection values = requestParameters.get(parameter); + return requestParameters.containsKey(parameter) ? values.iterator().next().substring(0, Math.min(maxLength, values.iterator().next().length())) : ""; + } + }); + when(httpRequest.hasParameters()).thenAnswer(new Answer() { + @Override + public Boolean answer(InvocationOnMock invocation) throws Throwable { + return !requestParameters.isEmpty(); + } + }); + when(httpRequest.getParameterNames()).thenAnswer(new Answer>() { + @Override + public Collection answer(InvocationOnMock invocation) throws Throwable { + return requestParameters.keySet(); } }); when(httpRequest.getParam(anyString())).thenAnswer(new Answer() { @Override public String answer(InvocationOnMock invocation) throws Throwable { String parameter = invocation.getArgument(0); - return requestParameters.containsKey(parameter) ? requestParameters.get(parameter) : ""; + return requestParameters.containsKey(parameter) ? requestParameters.get(parameter).iterator().next() : ""; } }); when(httpRequest.getParam(anyString(), ArgumentMatchers.any())).thenAnswer(new Answer() { @Override public String answer(InvocationOnMock invocation) throws Throwable { String parameter = invocation.getArgument(0); - return requestParameters.containsKey(parameter) ? requestParameters.get(parameter) : invocation.getArgument(1); + return requestParameters.containsKey(parameter) ? requestParameters.get(parameter).iterator().next() : invocation.getArgument(1); } }); when(httpRequest.isPartSet(anyString())).thenAnswer(new Answer() { @Override public Boolean answer(InvocationOnMock invocation) throws Throwable { - return requestParameters.containsKey(invocation.getArgument(0)); + return requestParameters.containsKey(invocation.getArgument(0)) && + requestParameters.get(invocation.getArgument(0)).iterator().next() != null; } }); when(httpRequest.getParts()).thenAnswer(new Answer() { @@ -160,6 +191,12 @@ public abstract class WebPageTest { } }); when(core.getPost(anyString())).thenReturn(Optional.absent()); + when(core.getReplies(anyString())).thenAnswer(new Answer>() { + @Override + public List answer(InvocationOnMock invocation) throws Throwable { + return postReplies.get(invocation.getArgument(0)); + } + }); when(core.getAlbum(anyString())).thenReturn(null); when(core.getImage(anyString())).thenReturn(null); when(core.getImage(anyString(), anyBoolean())).thenReturn(null); @@ -184,6 +221,10 @@ public abstract class WebPageTest { when(currentSone.getOptions()).thenReturn(new DefaultSoneOptions()); } + protected SoneTemplatePage getPage() { + return null; + } + protected void unsetCurrentSone() { when(webInterface.getCurrentSone(toadletContext)).thenReturn(null); when(webInterface.getCurrentSone(eq(toadletContext), anyBoolean())).thenReturn(null); @@ -191,6 +232,7 @@ public abstract class WebPageTest { protected void request(String uri, Method method) { try { + when(httpRequest.getPath()).thenReturn(uri); when(freenetRequest.getUri()).thenReturn(new URI(uri)); } catch (URISyntaxException e) { throw new RuntimeException(e); @@ -210,6 +252,13 @@ public abstract class WebPageTest { when(core.getPost(postId)).thenReturn(Optional.fromNullable(post)); } + protected void addPostReply(String postReplyId, PostReply postReply) { + if (postReply.getPostId() != null) { + postReplies.put(postReply.getPostId(), postReply); + } + when(core.getPostReply(postReplyId)).thenReturn(Optional.fromNullable(postReply)); + } + protected void addSone(String soneId, Sone sone) { sones.put(soneId, sone); } @@ -248,4 +297,19 @@ public abstract class WebPageTest { when(webInterface.getNotification(eq(notificationId))).thenReturn(Optional.of(notification)); } + protected void verifyRedirect(String target) throws RedirectException { + expectedException.expect(redirectsTo(target)); + getPage().handleRequest(freenetRequest, templateContext); + } + + protected void verifyRedirect(String target, Runnable verification) throws RedirectException { + expectedException.expect(redirectsTo(target)); + try { + getPage().handleRequest(freenetRequest, templateContext); + fail(); + } finally { + verification.run(); + } + } + }