Java tutorial
/** * This file Copyright (c) 2003-2012 Magnolia International * Ltd. (http://www.magnolia-cms.com). All rights reserved. * * * This file is dual-licensed under both the Magnolia * Network Agreement and the GNU General Public License. * You may elect to use one or the other of these licenses. * * This file is distributed in the hope that it will be * useful, but AS-IS and WITHOUT ANY WARRANTY; without even the * implied warranty of MERCHANTABILITY or FITNESS FOR A * PARTICULAR PURPOSE, TITLE, or NONINFRINGEMENT. * Redistribution, except as permitted by whichever of the GPL * or MNA you select, is prohibited. * * 1. For the GPL license (GPL), you can redistribute and/or * modify this file under the terms of the GNU General * Public License, Version 3, as published by the Free Software * Foundation. You should have received a copy of the GNU * General Public License, Version 3 along with this program; * if not, write to the Free Software Foundation, Inc., 51 * Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. * * 2. For the Magnolia Network Agreement (MNA), this file * and the accompanying materials are made available under the * terms of the MNA which accompanies this distribution, and * is available at http://www.magnolia-cms.com/mna.html * * Any modifications to this file must keep this entire header * intact. * */ package info.magnolia.cms.filters; import static org.easymock.EasyMock.*; import static org.junit.Assert.*; import info.magnolia.cms.beans.runtime.Document; import info.magnolia.cms.beans.runtime.MultipartForm; import info.magnolia.cms.core.SystemProperty; import info.magnolia.context.MgnlContext; import info.magnolia.context.WebContext; import info.magnolia.test.ComponentsTestUtil; import info.magnolia.test.MgnlTestCase; import java.io.ByteArrayInputStream; import java.io.ByteArrayOutputStream; import java.io.File; import java.io.IOException; import java.io.InputStream; import javax.servlet.Filter; import javax.servlet.FilterChain; import javax.servlet.ServletInputStream; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import org.apache.commons.httpclient.methods.PostMethod; import org.apache.commons.httpclient.methods.multipart.FilePart; import org.apache.commons.httpclient.methods.multipart.MultipartRequestEntity; import org.apache.commons.httpclient.methods.multipart.Part; import org.apache.commons.httpclient.methods.multipart.StringPart; import org.easymock.EasyMock; import org.easymock.IAnswer; import org.junit.After; import org.junit.Before; import org.junit.Test; /** * @version $Id$ */ public class MultipartRequestFilterTest extends MgnlTestCase { private File testFile; private HttpServletRequest req; private HttpServletResponse res; private FilterChain filterChain; private WebContext webCtx; @Override @Before public void setUp() throws Exception { super.setUp(); ComponentsTestUtil.setImplementation(WebContainerResources.class, WebContainerResourcesImpl.class); SystemProperty.setProperty(SystemProperty.MAGNOLIA_UPLOAD_TMPDIR, System.getProperty("java.io.tmpdir")); SystemProperty.setProperty("info.magnolia.cms.util.UnicodeNormalizer$Normalizer", "info.magnolia.cms.util.UnicodeNormalizer$AutoDetectNormalizer"); SystemProperty.setProperty("magnolia.utf8.enabled", "true"); testFile = new File("pom.xml"); assertTrue(testFile.getAbsolutePath() + " can't be found.", testFile.exists()); req = createMock(HttpServletRequest.class); expect(req.getAttribute(EasyMock.<String>anyObject())).andReturn(null).anyTimes(); res = createNiceMock(HttpServletResponse.class); filterChain = createNiceMock(FilterChain.class); webCtx = createStrictMock(WebContext.class); MgnlContext.setInstance(webCtx); } @Override @After public void tearDown() throws Exception { MgnlContext.setInstance(null); super.tearDown(); } @Test public void testFilterCOS() throws Throwable { webCtx.push(isA(MultipartRequestWrapper.class), eq(res)); doTest(new CosMultipartRequestFilter(), "text/xml"); } @Test public void testFilterCommonsFileUpload() throws Throwable { webCtx.push(isA(MultipartRequestWrapper.class), eq(res)); doTest(new MultipartRequestFilter(), "text/xml; charset=UTF-8"); } public void doTest(Filter filter, final String expectedDocumentType) throws Throwable { final MultipartRequestEntity multipart = newMultipartRequestEntity(); final ByteArrayOutputStream output = new ByteArrayOutputStream(); multipart.writeRequest(output); final byte[] bytes = output.toByteArray(); final ByteArrayInputStream delegateStream = new ByteArrayInputStream(bytes); final ServletInputStream servletInputStream = new ServletInputStream() { @Override public int read() throws IOException { return delegateStream.read(); } }; req.setAttribute(isA(String.class), isA(Boolean.class)); expect(req.getContentType()).andReturn(multipart.getContentType()).anyTimes(); expect(req.getHeader("Content-Type")).andReturn(multipart.getContentType()).anyTimes(); expect(req.getCharacterEncoding()).andReturn("UTF-8").anyTimes(); expect(req.getQueryString()).andReturn("").anyTimes(); expect(req.getContentLength()).andReturn(Integer.valueOf((int) multipart.getContentLength())).anyTimes(); expect(req.getInputStream()).andReturn(servletInputStream); req.setAttribute(eq(MultipartForm.REQUEST_ATTRIBUTE_NAME), isA(MultipartForm.class)); expectLastCall().andAnswer(new IAnswer<Object>() { @Override public Object answer() throws Throwable { final Object[] args = getCurrentArguments(); checkMultipartForm((MultipartForm) args[1], expectedDocumentType); return null; } }); webCtx.pop(); replay(req, res, filterChain, webCtx); filter.doFilter(req, res, filterChain); verify(req, res, filterChain, webCtx); } private void checkMultipartForm(MultipartForm form, String expectedDocumentType) throws IOException { assertNotNull("MultipartForm request attribute expected", form); assertEquals(3, form.getParameters().size()); assertEquals("value1", form.getParameter("param1")); assertEquals("", form.getParameter("param2")); String[] value3 = form.getParameterValues("param3"); assertNotNull("multi-value parameter has not been parsed", value3); assertEquals(2, value3.length); assertEquals(1, form.getDocuments().size()); Document document = form.getDocument("document"); assertNotNull("expected non-null Document", document); assertEquals("document", document.getAtomName()); assertEquals("xml", document.getExtension()); assertEquals("pom", document.getFileName()); assertEquals("pom.xml", document.getFileNameWithExtension()); assertEquals(testFile.length(), document.getLength()); assertEquals(expectedDocumentType, document.getType()); assertTrue(document.getType().startsWith("text/xml")); File documentFile = document.getFile(); assertTrue(documentFile.exists()); assertTrue(documentFile.canRead()); InputStream stream1 = document.getStream(); assertEquals(testFile.length(), stream1.available()); assertEquals(testFile.length(), stream1.skip(testFile.length())); assertEquals(0, stream1.available()); documentFile.deleteOnExit(); } private MultipartRequestEntity newMultipartRequestEntity() throws Exception { PostMethod method = new PostMethod(); Part[] parts = { new StringPart("param1", "value1", "UTF-8"), new StringPart("param2", "", "UTF-8"), new StringPart("param3", "value3a", "UTF-8"), new StringPart("param3", "value3b", "UTF-8"), new FilePart("document", testFile, "text/xml", "UTF-8") }; return new MultipartRequestEntity(parts, method.getParams()); } }