| package fi.iki.elonen; |
| |
| import org.junit.Test; |
| |
| import java.io.BufferedReader; |
| import java.io.FileReader; |
| import java.util.ArrayList; |
| import java.util.Arrays; |
| import java.util.List; |
| import java.util.UUID; |
| |
| import static junit.framework.Assert.assertEquals; |
| |
| public class HttpPostRequestTest extends HttpServerTest { |
| |
| public static final String CONTENT_LENGTH = "Content-Length: "; |
| public static final String FIELD = "caption"; |
| public static final String VALUE = "Summer vacation"; |
| public static final String FIELD2 = "location"; |
| public static final String VALUE2 = "Grand Canyon"; |
| public static final String POST_RAW_CONTENT_FILE_ENTRY = "postData"; |
| public static final String VALUE_TEST_SIMPLE_RAW_DATA_WITH_AMPHASIS = "Test raw data & Result value"; |
| |
| @Test |
| public void testSimpleRawPostData() throws Exception { |
| String header = "POST " + URI + " HTTP/1.1\n"; |
| String content = VALUE_TEST_SIMPLE_RAW_DATA_WITH_AMPHASIS + "\n"; |
| int size = content.length() + header.length(); |
| int contentLengthHeaderValueSize = String.valueOf(size).length(); |
| int contentLength = size + contentLengthHeaderValueSize + CONTENT_LENGTH.length(); |
| String input = header + CONTENT_LENGTH + (contentLength+4) + "\r\n\r\n" + content; |
| invokeServer(input); |
| assertEquals(0, testServer.parms.size()); |
| assertEquals(1, testServer.files.size()); |
| assertEquals(VALUE_TEST_SIMPLE_RAW_DATA_WITH_AMPHASIS, testServer.files.get(POST_RAW_CONTENT_FILE_ENTRY)); |
| } |
| |
| @Test |
| public void testSimplePostWithSingleMultipartFormField() throws Exception { |
| String divider = UUID.randomUUID().toString(); |
| String header = "POST " + URI + " HTTP/1.1\nContent-Type: " + |
| "multipart/form-data; boundary=" + divider + "\n"; |
| String content = "--" + divider + "\n" + |
| "Content-Disposition: form-data; name=\""+FIELD+"\"\n" + |
| "\n" + |
| VALUE +"\n" + |
| "--" + divider + "--\n"; |
| int size = content.length() + header.length(); |
| int contentLengthHeaderValueSize = String.valueOf(size).length(); |
| int contentLength = size + contentLengthHeaderValueSize + CONTENT_LENGTH.length(); |
| String input = header + CONTENT_LENGTH + (contentLength+4) + "\r\n\r\n" + content; |
| invokeServer(input); |
| |
| assertEquals(1, testServer.parms.size()); |
| assertEquals(VALUE, testServer.parms.get(FIELD)); |
| } |
| |
| @Test |
| public void testPostWithMultipleMultipartFormFields() throws Exception { |
| String divider = UUID.randomUUID().toString(); |
| String header = "POST " + URI + " HTTP/1.1\nContent-Type: " + |
| "multipart/form-data; boundary=" + divider + "\n"; |
| String content = "--" + divider + "\n" + |
| "Content-Disposition: form-data; name=\""+FIELD+"\"\n" + |
| "\n" + |
| VALUE +"\n" +"--" + divider + "\n" + |
| "Content-Disposition: form-data; name=\""+FIELD2+"\"\n" + |
| "\n" + |
| VALUE2 +"\n" + |
| "--" + divider + "--\n"; |
| int size = content.length() + header.length(); |
| int contentLengthHeaderValueSize = String.valueOf(size).length(); |
| int contentLength = size + contentLengthHeaderValueSize + CONTENT_LENGTH.length(); |
| String input = header + CONTENT_LENGTH + (contentLength+4) + "\r\n\r\n" + content; |
| invokeServer(input); |
| |
| assertEquals(2, testServer.parms.size()); |
| assertEquals(VALUE, testServer.parms.get(FIELD)); |
| assertEquals(VALUE2, testServer.parms.get(FIELD2)); |
| } |
| |
| @Test |
| public void testPostWithMultipleMultipartFormFieldsWhereContentTypeWasSeparatedByComma() throws Exception { |
| String divider = UUID.randomUUID().toString(); |
| String header = "POST " + URI + " HTTP/1.1\nContent-Type: " + |
| "multipart/form-data, boundary=" + divider + "\n"; |
| String content = "--" + divider + "\n" + |
| "Content-Disposition: form-data; name=\""+FIELD+"\"\n" + |
| "\n" + |
| VALUE +"\n" +"--" + divider + "\n" + |
| "Content-Disposition: form-data; name=\""+FIELD2+"\"\n" + |
| "\n" + |
| VALUE2 +"\n" + |
| "--" + divider + "--\n"; |
| int size = content.length() + header.length(); |
| int contentLengthHeaderValueSize = String.valueOf(size).length(); |
| int contentLength = size + contentLengthHeaderValueSize + CONTENT_LENGTH.length(); |
| String input = header + CONTENT_LENGTH + (contentLength+4) + "\r\n\r\n" + content; |
| invokeServer(input); |
| |
| assertEquals(2, testServer.parms.size()); |
| assertEquals(VALUE, testServer.parms.get(FIELD)); |
| assertEquals(VALUE2, testServer.parms.get(FIELD2)); |
| } |
| |
| @Test |
| public void testPostWithMultipartFormUpload() throws Exception { |
| String filename = "GrandCanyon.txt"; |
| String fileContent = VALUE; |
| String input = preparePostWithMultipartForm(filename, fileContent); |
| |
| invokeServer(input); |
| |
| assertEquals(1, testServer.parms.size()); |
| BufferedReader reader = new BufferedReader(new FileReader(testServer.files.get(FIELD))); |
| List<String> lines = readLinesFromFile(reader); |
| assertLinesOfText(new String[]{fileContent}, lines); |
| } |
| |
| @Test |
| public void testPostWithMultipartFormUploadFilenameHasSpaces() throws Exception { |
| String fileNameWithSpace = "Grand Canyon.txt"; |
| String fileContent = VALUE; |
| String input = preparePostWithMultipartForm(fileNameWithSpace, fileContent); |
| |
| invokeServer(input); |
| |
| String fileNameAfter = new ArrayList<String>(testServer.parms.values()).get(0); |
| |
| assertEquals(fileNameWithSpace, fileNameAfter); |
| } |
| |
| /** |
| * contains common preparation steps for testing POST with Multipart Form |
| * @param fileName Name of file to be uploaded |
| * @param fileContent Content of file to be uploaded |
| * @return input String with POST request complete information including header, length and content |
| */ |
| private String preparePostWithMultipartForm(String fileName, String fileContent) { |
| String divider = UUID.randomUUID().toString(); |
| String header = "POST " + URI + " HTTP/1.1\nContent-Type: " + |
| "multipart/form-data, boundary=" + divider + "\n"; |
| String content = "--" + divider + "\n" + |
| "Content-Disposition: form-data; name=\""+FIELD+"\"; filename=\""+fileName+"\"\n" + |
| "Content-Type: image/jpeg\r\n"+ |
| "\r\n" + |
| fileContent +"\r\n" + |
| "--" + divider + "--\n"; |
| int size = content.length() + header.length(); |
| int contentLengthHeaderValueSize = String.valueOf(size).length(); |
| int contentLength = size + contentLengthHeaderValueSize + CONTENT_LENGTH.length(); |
| String input = header + CONTENT_LENGTH + (contentLength+5) + "\r\n\r\n" + content; |
| |
| return input; |
| } |
| |
| } |