/*
* ====================================================================
*
* Frame2 Open Source License
*
* Copyright (c) 2004-2007 Megatome Technologies. All rights
* reserved.
*
* 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. The end-user documentation included with the redistribution, if
* any, must include the following acknowlegement:
* "This product includes software developed by
* Megatome Technologies."
* Alternately, this acknowlegement may appear in the software itself,
* if and wherever such third-party acknowlegements normally appear.
*
* 4. The names "The Frame2 Project", and "Frame2",
* must not be used to endorse or promote products derived
* from this software without prior written permission. For written
* permission, please contact iamthechad@sourceforge.net.
*
* 5. Products derived from this software may not be called "Frame2"
* nor may "Frame2" appear in their names without prior written
* permission of Megatome Technologies.
*
* THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED 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 MEGATOME TECHNOLOGIES OR
* ITS 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.
* ====================================================================
*/
package org.megatome.frame2.front;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import javax.servlet.ServletException;
import org.junit.Test;
import org.megatome.frame2.Globals;
import servletunit.frame2.Frame2HttpServletRequestSimulator;
import servletunit.frame2.MockFrame2TestCase;
import HTTPClient.Codecs;
import HTTPClient.NVPair;
public class TestMissingFileUpload extends MockFrame2TestCase {
private void postNegativeHttpReqProc() {
HttpFrontController servlet = initializeServlet();
try {
servlet.doPost(getRequest(), getResponse());
fail("Failed to catch expected exception"); //$NON-NLS-1$
} catch (Exception expected) {
// expected
}
}
private HttpFrontController initializeServlet() {
HttpFrontController servlet = getServlet();
try {
servlet.init();
} catch (ServletException e) {
fail("Unexpected ServletException: " + e.getMessage()); //$NON-NLS-1$
}
Configuration config = getServlet().getConfiguration();
assertNotNull(config);
return servlet;
}
@Test
public void testPrerequisites() {
try {
Class.forName("org.apache.commons.fileupload.DiskFileUpload"); //$NON-NLS-1$
fail("Did not catch expected ClassDef exception"); //$NON-NLS-1$
} catch (ClassNotFoundException expected) {
// This is the expected result
} catch (Exception e) {
fail("Unexpected Exception"); //$NON-NLS-1$
}
}
@Test
public void testUploadWithoutLibrary() throws Exception {
sendContextInitializedEvent(Globals.CONFIG_FILE,
"org/megatome/frame2/front/test-config.xml"); //$NON-NLS-1$
Map<String, String> parms = new HashMap<String, String>();
Map<String, String> files = new HashMap<String, String>();
String fileName = "/WEB-INF/frame2-config.xml"; //$NON-NLS-1$
parms.put("parm1", "value1"); //$NON-NLS-1$ //$NON-NLS-2$
files.put("fileparm", fileName); //$NON-NLS-1$
prepareMultipartUpload(new Object[] { parms }, new Object[] { files },
"/eventFileUpload"); //$NON-NLS-1$
postNegativeHttpReqProc();
}
@Test
public void testMapRequestToEvent() throws Exception {
Configuration config = new Configuration(
"org/megatome/frame2/front/test-config.xml"); //$NON-NLS-1$
addRequestParameter("parm1", "value1"); //$NON-NLS-1$ //$NON-NLS-2$
addRequestParameter("parm2", "value2"); //$NON-NLS-1$ //$NON-NLS-2$
setServletPath("/event1"); //$NON-NLS-1$
HttpRequestProcessor request = createHelper(config);
Event1 event = (Event1) request.getEvent();
assertTrue(request.mapRequestToEvent(event, true));
assertEquals("value1", event.getParm1()); //$NON-NLS-1$
assertEquals("value2", event.getParm2()); //$NON-NLS-1$
}
@SuppressWarnings("unchecked")
private void prepareMultipartUpload(Object[] formParms, Object[] fileParms,
String eventName) throws Exception {
NVPair[] hdrs = new NVPair[1];
Frame2HttpServletRequestSimulator request = (Frame2HttpServletRequestSimulator) getRequest();
NVPair[] opts;
opts = new NVPair[formParms.length];
for (int i = 0; i < formParms.length; i++) {
Map<String, String> parm = (HashMap<String, String>) formParms[i];
for (Entry<String, String> entry : parm.entrySet()) {
String key = entry.getKey();
String value = entry.getValue();
opts[i] = new NVPair(key, value);
}
}
NVPair[] file;
file = new NVPair[fileParms.length];
for (int i = 0; i < fileParms.length; i++) {
Map<String, String> parm = (HashMap<String, String>) fileParms[i];
for (Entry<String, String> entry : parm.entrySet()) {
String key = entry.getKey();
String value = entry.getValue();
URL configFile = getContext().getResource(value);
file[i] = new NVPair(key, configFile.getFile());
}
}
try {
byte[] data = Codecs.mpFormDataEncode(opts, file, hdrs);
String rawData = new String(data);
System.out.println(rawData);
InputStream is = new ByteArrayInputStream(data);
request.setDataInputStream(data.length, is);
request.setContentType(hdrs[0].getValue());
request.setHeader(HttpRequestProcessor.CONTENT_TYPE, hdrs[0]
.getValue());
is.close();
} catch (IOException e) {
fail();
}
setServletPath(eventName);
}
HttpRequestProcessor createHelper(Configuration config) {
return (HttpRequestProcessor) RequestProcessorFactory.instance(config,
getContext(), getRequest(), getResponse());
}
}