X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2FWebPageTest.java;h=16610e47420c0b1b563b05dd77e27587182b8dd7;hb=9824d45ada252d673a259029052f5ac4eecc5420;hp=b759a6d3999be20999a37f0df6ac69391352e987;hpb=8dd998006410b230392c0df06a29f95be1571b6f;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 b759a6d..16610e4 100644 --- a/src/test/java/net/pterodactylus/sone/web/WebPageTest.java +++ b/src/test/java/net/pterodactylus/sone/web/WebPageTest.java @@ -8,9 +8,14 @@ import static org.mockito.Mockito.RETURNS_DEEP_STUBS; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.when; +import java.io.ByteArrayOutputStream; +import java.io.IOException; +import java.io.PipedInputStream; +import java.io.PipedOutputStream; import java.net.URI; import java.net.URISyntaxException; import java.util.ArrayList; +import java.util.Collection; import java.util.HashMap; import java.util.HashSet; import java.util.List; @@ -27,21 +32,25 @@ import net.pterodactylus.sone.data.Image; import net.pterodactylus.sone.data.Post; 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.util.notify.Notification; import net.pterodactylus.util.template.Template; import net.pterodactylus.util.template.TemplateContext; import net.pterodactylus.util.web.Method; +import net.pterodactylus.util.web.Response; import freenet.clients.http.ToadletContext; import freenet.support.api.HTTPRequest; import com.google.common.base.Optional; import com.google.common.eventbus.EventBus; +import com.google.common.io.ByteStreams; import org.junit.Before; import org.junit.Rule; import org.junit.rules.ExpectedException; +import org.mockito.ArgumentMatchers; import org.mockito.invocation.InvocationOnMock; import org.mockito.stubbing.Answer; @@ -64,12 +73,25 @@ public abstract class WebPageTest { protected final TemplateContext templateContext = new TemplateContext(); protected final HTTPRequest httpRequest = mock(HTTPRequest.class); + protected final Map requestParameters = new HashMap<>(); protected final Map requestHeaders = new HashMap<>(); protected final FreenetRequest freenetRequest = mock(FreenetRequest.class); + private final PipedOutputStream responseOutputStream = new PipedOutputStream(); + private final PipedInputStream responseInputStream; + protected final Response response = new Response(responseOutputStream); protected final ToadletContext toadletContext = mock(ToadletContext.class); private final Set ownIdentities = new HashSet<>(); - private final List localSones = new ArrayList<>(); + private final Map sones = new HashMap<>(); + protected final List localSones = new ArrayList<>(); + + protected WebPageTest() { + try { + responseInputStream = new PipedInputStream(responseOutputStream); + } catch (IOException e) { + throw new RuntimeException(e); + } + } @Before public final void setupFreenetRequest() { @@ -78,11 +100,37 @@ public abstract class WebPageTest { when(httpRequest.getPartAsStringFailsafe(anyString(), anyInt())).thenAnswer(new Answer() { @Override public String answer(InvocationOnMock invocation) throws Throwable { - return ""; + 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())) : ""; + } + }); + 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) : ""; + } + }); + 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); + } + }); + when(httpRequest.isPartSet(anyString())).thenAnswer(new Answer() { + @Override + public Boolean answer(InvocationOnMock invocation) throws Throwable { + return requestParameters.containsKey(invocation.getArgument(0)); + } + }); + when(httpRequest.getParts()).thenAnswer(new Answer() { + @Override + public String[] answer(InvocationOnMock invocation) throws Throwable { + return requestParameters.keySet().toArray(new String[requestParameters.size()]); } }); - when(httpRequest.getParam(anyString())).thenReturn(""); - when(httpRequest.getParam(anyString(), anyString())).thenReturn(""); when(httpRequest.getHeader(anyString())).thenAnswer(new Answer() { @Override public String answer(InvocationOnMock invocation) throws Throwable { @@ -99,10 +147,23 @@ public abstract class WebPageTest { when(core.getLocalSone(anyString())).thenReturn(null); when(core.getLocalSones()).thenReturn(localSones); when(core.getSone(anyString())).thenReturn(Optional.absent()); + when(core.getSones()).thenAnswer(new Answer>() { + @Override + public Collection answer(InvocationOnMock invocation) throws Throwable { + return sones.values(); + } + }); + when(core.getSone(anyString())).thenAnswer(new Answer>() { + @Override + public Optional answer(InvocationOnMock invocation) throws Throwable { + return Optional.fromNullable(sones.get(invocation.getArgument(0))); + } + }); when(core.getPost(anyString())).thenReturn(Optional.absent()); when(core.getAlbum(anyString())).thenReturn(null); when(core.getImage(anyString())).thenReturn(null); when(core.getImage(anyString(), anyBoolean())).thenReturn(null); + when(core.getTemporaryImage(anyString())).thenReturn(null); } @Before @@ -142,16 +203,7 @@ public abstract class WebPageTest { } protected void addHttpRequestParameter(String name, final String value) { - when(httpRequest.getPartAsStringFailsafe(eq(name), anyInt())).thenAnswer(new Answer() { - @Override - public String answer(InvocationOnMock invocation) throws Throwable { - int maxLength = invocation.getArgument(1); - return value.substring(0, Math.min(maxLength, value.length())); - } - }); - when(httpRequest.getParam(eq(name))).thenReturn(value); - when(httpRequest.getParam(eq(name), anyString())).thenReturn(value); - when(httpRequest.isPartSet(eq(name))).thenReturn(value != null && !value.isEmpty()); + requestParameters.put(name, value); } protected void addPost(String postId, Post post) { @@ -159,7 +211,7 @@ public abstract class WebPageTest { } protected void addSone(String soneId, Sone sone) { - when(core.getSone(eq(soneId))).thenReturn(Optional.fromNullable(sone)); + sones.put(soneId, sone); } protected void addLocalSone(String soneId, Sone sone) { @@ -180,6 +232,18 @@ public abstract class WebPageTest { when(core.getImage(eq(imageId), anyBoolean())).thenReturn(image); } + protected void addTemporaryImage(String imageId, TemporaryImage temporaryImage) { + when(core.getTemporaryImage(eq(imageId))).thenReturn(temporaryImage); + } + + protected byte[] getResponseBytes() throws IOException { + response.getContent().close(); + try (ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) { + ByteStreams.copy(responseInputStream, outputStream); + return outputStream.toByteArray(); + } + } + protected void addNotification(String notificationId, Notification notification) { when(webInterface.getNotification(eq(notificationId))).thenReturn(Optional.of(notification)); }