X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2FWebPageTest.java;fp=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2FWebPageTest.java;h=d8b9462b3db794a0acc7dcde1d2c22ad9fb6a9d4;hb=c2c776f038ec312447ab64504efa88b1f3533b9a;hp=a6f7aa35eecd329d4d7be730bcc47702ac385a79;hpb=721e99302c43d08fd2a63174273de28791c04c8a;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 a6f7aa3..d8b9462 100644
--- a/src/test/java/net/pterodactylus/sone/web/WebPageTest.java
+++ b/src/test/java/net/pterodactylus/sone/web/WebPageTest.java
@@ -12,6 +12,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,11 +48,14 @@ 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;
@@ -85,6 +89,9 @@ public abstract class WebPageTest {
 	protected final HTTPRequest httpRequest = mock(HTTPRequest.class);
 	protected final Multimap<String, String> requestParameters = ArrayListMultimap.create();
 	protected final Map<String, String> requestHeaders = new HashMap<>();
+	private final Map<String, String> uploadedFilesNames = new HashMap<>();
+	private final Map<String, String> uploadedFilesContentTypes = new HashMap<>();
+	private final Map<String, String> uploadedFilesSources = new HashMap<>();
 	protected final FreenetRequest freenetRequest = mock(FreenetRequest.class);
 	private final PipedOutputStream responseOutputStream = new PipedOutputStream();
 	private final PipedInputStream responseInputStream;
@@ -168,6 +175,36 @@ public abstract class WebPageTest {
 				return requestHeaders.get(invocation.<String>getArgument(0).toLowerCase());
 			}
 		});
+		when(httpRequest.getUploadedFile(anyString())).thenAnswer(new Answer<HTTPUploadedFile>() {
+			@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
@@ -204,6 +241,16 @@ public abstract class WebPageTest {
 	}
 
 	@Before
+	public void setupL10n() {
+		when(l10n.getString(anyString())).thenAnswer(new Answer<String>() {
+			@Override
+			public String answer(InvocationOnMock invocation) throws Throwable {
+				return invocation.getArgument(0);
+			}
+		});
+	}
+
+	@Before
 	public final void setupIdentityManager() {
 		when(core.getIdentityManager().getAllOwnIdentities()).thenReturn(ownIdentities);
 	}
@@ -285,6 +332,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()) {