//======================================================================== //Copyright 2006 Mort Bay Consulting Pty. Ltd. //------------------------------------------------------------------------ //Licensed under the Apache License, Version 2.0 (the "License"); //you may not use this file except in compliance with the License. //You may obtain a copy of the License at //http://www.apache.org/licenses/LICENSE-2.0 //Unless required by applicable law or agreed to in writing, software //distributed under the License is distributed on an "AS IS" BASIS, //WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. //See the License for the specific language governing permissions and //limitations under the License. //======================================================================== package org.mortbay.jetty; import java.io.File; import java.io.FileNotFoundException; import java.io.IOException; import java.net.HttpURLConnection; import java.net.MalformedURLException; import java.net.URL; import junit.framework.TestCase; import org.mortbay.jetty.Connector; import org.mortbay.jetty.Handler; import org.mortbay.jetty.HttpHeaders; import org.mortbay.jetty.MimeTypes; import org.mortbay.jetty.NCSARequestLog; import org.mortbay.jetty.Server; import org.mortbay.jetty.handler.ContextHandlerCollection; import org.mortbay.jetty.handler.DefaultHandler; import org.mortbay.jetty.handler.HandlerCollection; import org.mortbay.jetty.handler.RequestLogHandler; import org.mortbay.jetty.nio.SelectChannelConnector; import org.mortbay.jetty.security.HashUserRealm; import org.mortbay.jetty.security.UserRealm; import org.mortbay.jetty.webapp.WebAppContext; import org.mortbay.thread.BoundedThreadPool; import org.mortbay.util.IO; public class WebAppTest extends TestCase { Server server = new Server(); BoundedThreadPool threadPool = new BoundedThreadPool(); Connector connector=new SelectChannelConnector(); HandlerCollection handlers = new HandlerCollection(); ContextHandlerCollection contexts = new ContextHandlerCollection(); HashUserRealm userRealm = new HashUserRealm(); RequestLogHandler requestLogHandler = new RequestLogHandler(); WebAppContext context; protected void setUp() throws Exception { File dir = new File(".").getAbsoluteFile(); while (!new File(dir,"webapps").exists()) { dir=dir.getParentFile(); } threadPool.setMaxThreads(100); server.setThreadPool(threadPool); server.setConnectors(new Connector[]{connector}); handlers.setHandlers(new Handler[]{contexts,new DefaultHandler(),requestLogHandler}); server.setHandler(handlers); context=new WebAppContext(contexts,dir.getAbsolutePath()+"/webapps/test","/test"); userRealm.setName("Test Realm"); userRealm.setConfig(dir.getAbsolutePath()+"/etc/realm.properties"); server.setUserRealms(new UserRealm[]{userRealm}); File file = File.createTempFile("test",".log"); NCSARequestLog requestLog = new NCSARequestLog(file.getAbsolutePath()); requestLog.setExtended(false); requestLogHandler.setRequestLog(requestLog); server.setSendServerVersion(true); server.start(); Thread.sleep(1000); } protected void tearDown() throws Exception { Thread.sleep(100); server.stop(); Thread.sleep(100); } public void testDoGet() throws Exception { URL url = null; url=new URL("http://127.0.0.1:"+connector.getLocalPort()+"/test/dump/info?query=foo"); assertTrue(IO.toString(url.openStream()).startsWith("")); url=new URL("http://127.0.0.1:"+connector.getLocalPort()+"/"); try{IO.toString(url.openStream()); assertTrue(false); } catch(FileNotFoundException e) { assertTrue(true); } url=new URL("http://127.0.0.1:"+connector.getLocalPort()+"/test"); IO.toString(url.openStream()); url=new URL("http://127.0.0.1:"+connector.getLocalPort()+"/test/"); String s1=IO.toString(url.openStream()); url=new URL("http://127.0.0.1:"+connector.getLocalPort()+"/test/index.html"); String s2=IO.toString(url.openStream()); assertEquals(s1,s2); url=new URL("http://127.0.0.1:"+connector.getLocalPort()+"/test/d.txt"); assertTrue(IO.toString(url.openStream()).startsWith("0000")); url=new URL("http://127.0.0.1:"+connector.getLocalPort()+"/test/data.txt"); String result = IO.toString(url.openStream()); if (result.endsWith("\r\n")) { //windows result = result.substring(0,result.length() - 2); } else if (result.endsWith("\n")) { //*nix result = result.substring(0,result.length() - 1); } else { //Error: Unexpected end of stream data encountered assertTrue(false); } assertTrue(result.endsWith("9999 3333333333333333333333333333333333333333333333333333333")); url=new URL("http://127.0.0.1:"+connector.getLocalPort()+"/test/dispatch/forward/dump/info?query=foo"); assertTrue(IO.toString(url.openStream()).startsWith("")); url=new URL("http://127.0.0.1:"+connector.getLocalPort()+"/test/dispatch/includeW/dump/info?query=foo"); assertTrue(IO.toString(url.openStream()).startsWith("

")); url=new URL("http://127.0.0.1:"+connector.getLocalPort()+"/test/dispatch/includeS/dump/info?query=foo"); assertTrue(IO.toString(url.openStream()).startsWith("

")); url=new URL("http://127.0.0.1:"+connector.getLocalPort()+"/test/dump/info?continue=1000"); assertTrue(IO.toString(url.openStream()).startsWith("")); url=new URL("http://127.0.0.1:"+connector.getLocalPort()+"/test/dump/info?lines=100"); HttpURLConnection connection = (HttpURLConnection) url.openConnection(); connection.addRequestProperty("accept-encoding","gzip"); connection.connect(); assertEquals("gzip",connection.getHeaderField("Content-Encoding")); } public void testRequestListener() throws Exception { URL url = null; String result; url=new URL("http://127.0.0.1:"+connector.getLocalPort()+"/test/dump/info/"); result=IO.toString(url.openStream()); assertTrue(result.startsWith("")); assertTrue(result.indexOf("requestInitialized")>0); assertTrue(result.indexOf("'/test'")>0); } public void testSecurity() throws Exception { URL url = null; String[] hidden = { "/WEB-INF/web.xml", "/WEB-INF/wEb.xml", "/WEB-INF/web.xml%00", "/WEB-INF/web.xml\00", "/WEB-INF/web.xml\u0000", "/WEB-INF//web.xml", "//WEB-INF/web.xml", "/WEB-INF//web.xml", "//WEB-INF//web.xml", "//auth/file.txt" }; String[] forbidden = { "/auth/", "/auth/file.txt", "/auth//file.txt", }; String[] ok = { "/auth/relax.txt", }; for (int i=0;i=0); } } for (int i=0;i")); assertTrue(s0.indexOf("POST")>0); assertTrue(s0.indexOf("abcd: 1234")>0); } public void testWebInfAccess() throws Exception { assertNotFound("WEB-INF/foo"); assertNotFound("web-inf"); assertNotFound("web-inf/"); assertNotFound("./web-inf/"); assertNotFound("web-inf/jetty-web.xml"); assertNotFound("Web-Inf/web.xml"); assertNotFound("./WEB-INF/web.xml"); assertNotFound("META-INF"); assertNotFound("meta-inf/manifest.mf"); assertNotFound("Meta-Inf/foo"); assertFound("index.html"); } public void testUnavailable() throws Exception { URL url = null; url=new URL("http://127.0.0.1:"+connector.getLocalPort()+"/test/dump/info?query=foo"); assertTrue(IO.toString(url.openStream()).startsWith("")); assertTrue(context.getServletHandler().isAvailable()); url=new URL("http://127.0.0.1:"+connector.getLocalPort()+"/test/dump/ex3/2"); try{IO.toString(url.openStream());} catch(IOException e){} assertFalse(context.getServletHandler().isAvailable()); Thread.sleep(5000); assertTrue(context.getServletHandler().isAvailable()); } private void assertNotFound(String resource) throws MalformedURLException, IOException { try { getResource(resource); } catch (FileNotFoundException e) { return; } fail("Expected 404 for resource: " + resource); } private void assertFound(String resource) throws MalformedURLException, IOException { try { getResource(resource); } catch (FileNotFoundException e) { fail("Expected 200 for resource: " + resource); } // Pass return; } private void getResource(String resource) throws MalformedURLException, IOException { URL url; url = new URL("http://127.0.0.1:" + connector.getLocalPort() + "/test/" + resource); url.openStream(); } }