001 /** 002 * Licensed to the Apache Software Foundation (ASF) under one 003 * or more contributor license agreements. See the NOTICE file 004 * distributed with this work for additional information 005 * regarding copyright ownership. The ASF licenses this file 006 * to you under the Apache License, Version 2.0 (the 007 * "License"); you may not use this file except in compliance 008 * with the License. You may obtain a copy of the License at 009 * 010 * http://www.apache.org/licenses/LICENSE-2.0 011 * 012 * Unless required by applicable law or agreed to in writing, software 013 * distributed under the License is distributed on an "AS IS" BASIS, 014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 015 * See the License for the specific language governing permissions and 016 * limitations under the License. 017 */ 018 package org.apache.oozie.test; 019 020 import org.mortbay.jetty.Server; 021 import org.mortbay.jetty.servlet.FilterHolder; 022 import org.mortbay.jetty.servlet.ServletHolder; 023 import org.mortbay.jetty.servlet.Context; 024 025 import java.net.InetAddress; 026 import java.net.ServerSocket; 027 028 /** 029 * An embedded servlet container for testing purposes. <p/> It provides reduced functionality, it supports only 030 * Servlets. <p/> The servlet container is started in a free port. 031 */ 032 public class EmbeddedServletContainer { 033 private Server server; 034 private String host = null; 035 private int port = -1; 036 private String contextPath; 037 Context context; 038 039 /** 040 * Create a servlet container. 041 * 042 * @param contextPath context path for the servlet, it must not be prefixed or append with "/", for the default 043 * context use "" 044 */ 045 public EmbeddedServletContainer(String contextPath) { 046 this.contextPath = contextPath; 047 server = new Server(0); 048 context = new Context(); 049 context.setContextPath("/" + contextPath); 050 server.setHandler(context); 051 } 052 053 /** 054 * Add a servlet to the container. 055 * 056 * @param servletPath servlet path for the servlet, it should be prefixed with '/", it may contain a wild card at 057 * the end. 058 * @param servletClass servlet class 059 */ 060 public void addServletEndpoint(String servletPath, Class servletClass) { 061 context.addServlet(new ServletHolder(servletClass), servletPath); 062 } 063 064 /** 065 * Add a filter to the container. 066 * 067 * @param filterPath path for the filter, it should be prefixed with '/", it may contain a wild card at 068 * the end. 069 * @param filterClass servlet class 070 */ 071 public void addFilter(String filterPath, Class filterClass) { 072 context.addFilter(new FilterHolder(filterClass), filterPath, 0); 073 } 074 075 /** 076 * Start the servlet container. <p/> The container starts on a free port. 077 * 078 * @throws Exception thrown if the container could not start. 079 */ 080 public void start() throws Exception { 081 host = InetAddress.getLocalHost().getHostName(); 082 ServerSocket ss = new ServerSocket(0); 083 port = ss.getLocalPort(); 084 ss.close(); 085 server.getConnectors()[0].setHost(host); 086 server.getConnectors()[0].setPort(port); 087 server.start(); 088 System.out.println("Running embedded servlet container at: http://" + host + ":" + port); 089 } 090 091 /** 092 * Return the hostname the servlet container is bound to. 093 * 094 * @return the hostname. 095 */ 096 public String getHost() { 097 return host; 098 } 099 100 /** 101 * Return the port number the servlet container is bound to. 102 * 103 * @return the port number. 104 */ 105 public int getPort() { 106 return port; 107 } 108 109 /** 110 * Return the full URL (including protocol, host, port, context path, servlet path) for the context path. 111 * 112 * @return URL to the context path. 113 */ 114 public String getContextURL() { 115 return "http://" + host + ":" + port + "/" + contextPath; 116 } 117 118 /** 119 * Return the full URL (including protocol, host, port, context path, servlet path) for a servlet path. 120 * 121 * @param servletPath the path which will be expanded to a full URL. 122 * @return URL to the servlet. 123 */ 124 public String getServletURL(String servletPath) { 125 String path = servletPath; 126 if (path.endsWith("*")) { 127 path = path.substring(0, path.length() - 1); 128 } 129 return getContextURL() + path; 130 } 131 132 /** 133 * Stop the servlet container. 134 */ 135 public void stop() { 136 try { 137 server.stop(); 138 } 139 catch (Exception e) { 140 // ignore exception 141 } 142 143 try { 144 server.destroy(); 145 } 146 catch (Exception e) { 147 // ignore exception 148 } 149 150 host = null; 151 port = -1; 152 } 153 154 }