| package fi.iki.elonen; |
| |
| /* |
| * #%L |
| * NanoHttpd-apache file upload integration |
| * %% |
| * Copyright (C) 2012 - 2015 nanohttpd |
| * %% |
| * Redistribution and use in source and binary forms, with or without modification, |
| * are permitted provided that the following conditions are met: |
| * |
| * 1. Redistributions of source code must retain the above copyright notice, this |
| * list of conditions and the following disclaimer. |
| * |
| * 2. Redistributions in binary form must reproduce the above copyright notice, |
| * this list of conditions and the following disclaimer in the documentation |
| * and/or other materials provided with the distribution. |
| * |
| * 3. Neither the name of the nanohttpd nor the names of its contributors |
| * may be used to endorse or promote products derived from this software without |
| * specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND |
| * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED |
| * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. |
| * IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, |
| * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, |
| * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF |
| * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE |
| * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED |
| * OF THE POSSIBILITY OF SUCH DAMAGE. |
| * #L% |
| */ |
| |
| import java.io.File; |
| import java.io.IOException; |
| import java.io.InputStream; |
| import java.io.OutputStream; |
| import java.net.InetAddress; |
| import java.util.Arrays; |
| import java.util.HashMap; |
| import java.util.List; |
| import java.util.Map; |
| |
| import org.apache.commons.fileupload.FileItem; |
| import org.apache.commons.fileupload.FileItemIterator; |
| import org.apache.commons.fileupload.FileItemStream; |
| import org.apache.commons.fileupload.FileUploadException; |
| import org.apache.commons.fileupload.disk.DiskFileItemFactory; |
| import org.apache.commons.fileupload.util.Streams; |
| import org.apache.http.HttpEntity; |
| import org.apache.http.HttpResponse; |
| import org.apache.http.client.ClientProtocolException; |
| import org.apache.http.client.methods.CloseableHttpResponse; |
| import org.apache.http.client.methods.HttpPost; |
| import org.apache.http.client.methods.HttpTrace; |
| import org.apache.http.entity.ContentType; |
| import org.apache.http.entity.mime.HttpMultipartMode; |
| import org.apache.http.entity.mime.MultipartEntityBuilder; |
| import org.apache.http.entity.mime.content.FileBody; |
| import org.apache.http.entity.mime.content.StringBody; |
| import org.apache.http.impl.client.CloseableHttpClient; |
| import org.apache.http.impl.client.HttpClients; |
| import org.junit.After; |
| import org.junit.Assert; |
| import org.junit.Before; |
| import org.junit.FixMethodOrder; |
| import org.junit.Test; |
| import org.junit.internal.runners.statements.Fail; |
| |
| import fi.iki.elonen.NanoHTTPD.Response.Status; |
| |
| /** |
| * very strange but if the file upload is the first request the test fails. |
| * |
| * @author ritchieGitHub |
| */ |
| @FixMethodOrder |
| public class TestNanoFileUpLoad { |
| |
| protected TestServer testServer; |
| |
| public static class TestServer extends NanoHTTPD { |
| |
| public Response response = newFixedLengthResponse(""); |
| |
| public String uri; |
| |
| public Method method; |
| |
| public Map<String, String> header; |
| |
| public Map<String, String> parms; |
| |
| public Map<String, List<FileItem>> files; |
| |
| public Map<String, List<String>> decodedParamters; |
| |
| public Map<String, List<String>> decodedParamtersFromParameter; |
| |
| public String queryParameterString; |
| |
| public TestServer() { |
| super(8192); |
| uploader = new NanoFileUpload(new DiskFileItemFactory()); |
| } |
| |
| public HTTPSession createSession(TempFileManager tempFileManager, InputStream inputStream, OutputStream outputStream) { |
| return new HTTPSession(tempFileManager, inputStream, outputStream); |
| } |
| |
| public HTTPSession createSession(TempFileManager tempFileManager, InputStream inputStream, OutputStream outputStream, InetAddress inetAddress) { |
| return new HTTPSession(tempFileManager, inputStream, outputStream, inetAddress); |
| } |
| |
| NanoFileUpload uploader; |
| |
| @Override |
| public Response serve(IHTTPSession session) { |
| |
| this.uri = session.getUri(); |
| this.method = session.getMethod(); |
| this.header = session.getHeaders(); |
| this.parms = session.getParms(); |
| if (NanoFileUpload.isMultipartContent(session)) { |
| try { |
| if ("/uploadFile1".equals(this.uri)) { |
| session.getHeaders().put("content-length", "AA"); |
| files = uploader.parseParameterMap(session); |
| } |
| if ("/uploadFile2".equals(this.uri)) { |
| files = new HashMap<String, List<FileItem>>(); |
| List<FileItem> parseRequest = uploader.parseRequest(session); |
| files.put(parseRequest.get(0).getFieldName(), parseRequest); |
| } |
| if ("/uploadFile3".equals(this.uri)) { |
| files = new HashMap<String, List<FileItem>>(); |
| FileItemIterator iter = uploader.getItemIterator(session); |
| while (iter.hasNext()) { |
| FileItemStream item = iter.next(); |
| final String fileName = item.getName(); |
| FileItem fileItem = uploader.getFileItemFactory().createItem(item.getFieldName(), item.getContentType(), item.isFormField(), fileName); |
| files.put(fileItem.getFieldName(), Arrays.asList(new FileItem[]{ |
| fileItem |
| })); |
| try { |
| Streams.copy(item.openStream(), fileItem.getOutputStream(), true); |
| } catch (Exception e) { |
| } |
| fileItem.setHeaders(item.getHeaders()); |
| } |
| } |
| } catch (Exception e) { |
| this.response.setStatus(Status.INTERNAL_ERROR); |
| e.printStackTrace(); |
| } |
| } |
| this.queryParameterString = session.getQueryParameterString(); |
| this.decodedParamtersFromParameter = decodeParameters(this.queryParameterString); |
| this.decodedParamters = decodeParameters(session.getQueryParameterString()); |
| return this.response; |
| } |
| |
| } |
| |
| @Test |
| public void testNormalRequest() throws Exception { |
| CloseableHttpClient httpclient = HttpClients.createDefault(); |
| HttpTrace httphead = new HttpTrace("http://localhost:8192/index.html"); |
| CloseableHttpResponse response = httpclient.execute(httphead); |
| Assert.assertEquals(200, response.getStatusLine().getStatusCode()); |
| response.close(); |
| } |
| |
| @Test |
| public void testPostWithMultipartFormUpload1() throws Exception { |
| CloseableHttpClient httpclient = HttpClients.createDefault(); |
| String textFileName = "src/test/java/fi/iki/elonen/TestNanoFileUpLoad.java"; |
| HttpPost post = new HttpPost("http://localhost:8192/uploadFile1"); |
| |
| executeUpload(httpclient, textFileName, post); |
| FileItem file = this.testServer.files.get("upfile").get(0); |
| Assert.assertEquals(file.getSize(), new File(textFileName).length()); |
| } |
| |
| @Test |
| public void testPostWithMultipartFormUpload2() throws Exception { |
| CloseableHttpClient httpclient = HttpClients.createDefault(); |
| String textFileName = "src/test/java/fi/iki/elonen/TestNanoFileUpLoad.java"; |
| HttpPost post = new HttpPost("http://localhost:8192/uploadFile2"); |
| |
| executeUpload(httpclient, textFileName, post); |
| FileItem file = this.testServer.files.get("upfile").get(0); |
| Assert.assertEquals(file.getSize(), new File(textFileName).length()); |
| } |
| |
| @Test |
| public void testPostWithMultipartFormUpload3() throws Exception { |
| CloseableHttpClient httpclient = HttpClients.createDefault(); |
| String textFileName = "src/test/java/fi/iki/elonen/TestNanoFileUpLoad.java"; |
| HttpPost post = new HttpPost("http://localhost:8192/uploadFile3"); |
| |
| executeUpload(httpclient, textFileName, post); |
| FileItem file = this.testServer.files.get("upfile").get(0); |
| Assert.assertEquals(file.getSize(), new File(textFileName).length()); |
| } |
| |
| private void executeUpload(CloseableHttpClient httpclient, String textFileName, HttpPost post) throws IOException, ClientProtocolException { |
| FileBody fileBody = new FileBody(new File(textFileName), ContentType.DEFAULT_BINARY); |
| StringBody stringBody1 = new StringBody("Message 1", ContentType.MULTIPART_FORM_DATA); |
| |
| MultipartEntityBuilder builder = MultipartEntityBuilder.create(); |
| builder.setMode(HttpMultipartMode.BROWSER_COMPATIBLE); |
| builder.addPart("upfile", fileBody); |
| builder.addPart("text1", stringBody1); |
| HttpEntity entity = builder.build(); |
| // |
| post.setEntity(entity); |
| HttpResponse response = httpclient.execute(post); |
| Assert.assertEquals(200, response.getStatusLine().getStatusCode()); |
| } |
| |
| @Before |
| public void setUp() throws IOException { |
| this.testServer = new TestServer(); |
| this.testServer.start(); |
| try { |
| long start = System.currentTimeMillis(); |
| Thread.sleep(100L); |
| while (!this.testServer.wasStarted()) { |
| Thread.sleep(100L); |
| if (System.currentTimeMillis() - start > 2000) { |
| Assert.fail("could not start server"); |
| } |
| } |
| } catch (InterruptedException e) { |
| } |
| } |
| |
| @After |
| public void tearDown() { |
| this.testServer.stop(); |
| } |
| |
| } |