1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138
|
package com.meterware.servletunit;
/********************************************************************************************************************
* $Id: ConfigTest.java 678 2004-09-23 22:48:20Z russgold $
*
* Copyright (c) 2000-2001,2003, Russell Gold
*
* Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
* documentation files (the "Software"), to deal in the Software without restriction, including without limitation
* the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and
* to permit persons to whom the Software is furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all copies or substantial portions
* of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO
* THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
* CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
* DEALINGS IN THE SOFTWARE.
*
*******************************************************************************************************************/
import com.meterware.httpunit.WebClient;
import com.meterware.httpunit.WebResponse;
import java.io.IOException;
import java.io.PrintWriter;
import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletContext;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import junit.framework.Test;
import junit.framework.TestCase;
import junit.framework.TestSuite;
/**
* Tests support for the servlet configuration.
**/
public class ConfigTest extends TestCase {
public static void main(String args[]) {
junit.textui.TestRunner.run( suite() );
}
public static Test suite() {
return new TestSuite( ConfigTest.class );
}
public ConfigTest( String name ) {
super( name );
}
public void testConfigObject() throws Exception {
final String resourceName = "something/interesting";
ServletRunner sr = new ServletRunner();
sr.registerServlet( resourceName, ConfigServlet.class.getName() );
WebClient wc = sr.newClient();
WebResponse response = wc.getResponse( "http://localhost/" + resourceName );
assertNotNull( "No response received", response );
assertEquals( "content type", "text/plain", response.getContentType() );
assertEquals( "servlet name is " + ConfigServlet.class.getName(), response.getText() );
}
public void testContextAttributes() throws Exception {
final String servlet1Name = "something/interesting";
final String servlet2Name = "something/else";
ServletRunner sr = new ServletRunner();
sr.registerServlet( servlet1Name, ConfigServlet.class.getName() );
sr.registerServlet( servlet2Name, ConfigServlet.class.getName() );
ServletUnitClient wc = sr.newClient();
InvocationContext ic1 = wc.newInvocation( "http://localhost/" + servlet1Name );
ServletContext sc1 = ic1.getServlet().getServletConfig().getServletContext();
sc1.setAttribute( "sample", "found me" );
InvocationContext ic2 = wc.newInvocation( "http://localhost/" + servlet2Name );
ServletContext sc2 = ic2.getServlet().getServletConfig().getServletContext();
assertEquals( "attribute 'sample'", "found me", sc2.getAttribute( "sample") );
}
public void testFileMimeType() throws Exception {
final String servlet1Name = "something/interesting";
ServletRunner sr = new ServletRunner();
sr.registerServlet( servlet1Name, ConfigServlet.class.getName() );
ServletUnitClient wc = sr.newClient();
InvocationContext ic1 = wc.newInvocation( "http://localhost/" + servlet1Name );
ServletContext context = ic1.getServlet().getServletConfig().getServletContext();
checkMimeType( context, "sample.txt", "text/plain" );
checkMimeType( context, "sample.html", "text/html" );
checkMimeType( context, "sample.gif", "image/gif" );
}
public void testServletContextAccess() throws Exception {
ServletRunner sr = new ServletRunner();
sr.registerServlet( "SimpleServlet", ConfigServlet.class.getName() );
ServletUnitClient client = sr.newClient();
InvocationContext ic = client.newInvocation( "http://localhost/SimpleServlet" );
ServletContext context = ic.getServlet().getServletConfig().getServletContext();
assertSame( "Context from session", context, ic.getRequest().getSession().getServletContext() );
}
private void checkMimeType( ServletContext context, String fileName, String expectedMimeType ) {
assertEquals( "mime type for " + fileName, expectedMimeType, context.getMimeType( fileName ) );
}
static class ConfigServlet extends HttpServlet {
static String RESPONSE_TEXT = "the desired content\r\n";
protected void doGet( HttpServletRequest req, HttpServletResponse resp ) throws ServletException,IOException {
resp.setContentType( "text/plain" );
PrintWriter pw = resp.getWriter();
ServletConfig config = getServletConfig();
if (config == null) {
pw.print( "config object is null" );
} else {
pw.print( "servlet name is " + config.getServletName() );
}
pw.close();
}
}
}
|