blob: 69d817d6465bd07820359a2d6af22bad722c5018 [file] [log] [blame]
<!DOCTYPE HTML>
<html lang="fr">
<head>
<title>Source code</title>
<link rel="stylesheet" type="text/css" href="../../../../../../stylesheet.css" title="Style">
</head>
<body>
<main role="main">
<div class="sourceContainer">
<pre><span class="sourceLineNo">001</span><a id="line.1">/*</a>
<span class="sourceLineNo">002</span><a id="line.2"> * Licensed to the Apache Software Foundation (ASF) under one</a>
<span class="sourceLineNo">003</span><a id="line.3"> * or more contributor license agreements. See the NOTICE file</a>
<span class="sourceLineNo">004</span><a id="line.4"> * distributed with this work for additional information</a>
<span class="sourceLineNo">005</span><a id="line.5"> * regarding copyright ownership. The ASF licenses this file</a>
<span class="sourceLineNo">006</span><a id="line.6"> * to you under the Apache License, Version 2.0 (the</a>
<span class="sourceLineNo">007</span><a id="line.7"> * "License"); you may not use this file except in compliance</a>
<span class="sourceLineNo">008</span><a id="line.8"> * with the License. You may obtain a copy of the License at</a>
<span class="sourceLineNo">009</span><a id="line.9"> *</a>
<span class="sourceLineNo">010</span><a id="line.10"> * http://www.apache.org/licenses/LICENSE-2.0</a>
<span class="sourceLineNo">011</span><a id="line.11"> *</a>
<span class="sourceLineNo">012</span><a id="line.12"> * Unless required by applicable law or agreed to in writing,</a>
<span class="sourceLineNo">013</span><a id="line.13"> * software distributed under the License is distributed on an</a>
<span class="sourceLineNo">014</span><a id="line.14"> * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY</a>
<span class="sourceLineNo">015</span><a id="line.15"> * KIND, either express or implied. See the License for the</a>
<span class="sourceLineNo">016</span><a id="line.16"> * specific language governing permissions and limitations</a>
<span class="sourceLineNo">017</span><a id="line.17"> * under the License.</a>
<span class="sourceLineNo">018</span><a id="line.18"> */</a>
<span class="sourceLineNo">019</span><a id="line.19">package org.apache.shiro.testing.web;</a>
<span class="sourceLineNo">020</span><a id="line.20"></a>
<span class="sourceLineNo">021</span><a id="line.21">import org.apache.shiro.codec.Base64;</a>
<span class="sourceLineNo">022</span><a id="line.22"></a>
<span class="sourceLineNo">023</span><a id="line.23">import com.gargoylesoftware.htmlunit.WebClient;</a>
<span class="sourceLineNo">024</span><a id="line.24">import com.github.mjeanroy.junit.servers.jetty.EmbeddedJetty;</a>
<span class="sourceLineNo">025</span><a id="line.25">import com.github.mjeanroy.junit.servers.jetty.EmbeddedJettyConfiguration;</a>
<span class="sourceLineNo">026</span><a id="line.26">import org.eclipse.jetty.annotations.AnnotationConfiguration;</a>
<span class="sourceLineNo">027</span><a id="line.27">import org.eclipse.jetty.http.HttpVersion;</a>
<span class="sourceLineNo">028</span><a id="line.28">import org.eclipse.jetty.server.HttpConfiguration;</a>
<span class="sourceLineNo">029</span><a id="line.29">import org.eclipse.jetty.server.HttpConnectionFactory;</a>
<span class="sourceLineNo">030</span><a id="line.30">import org.eclipse.jetty.server.SecureRequestCustomizer;</a>
<span class="sourceLineNo">031</span><a id="line.31">import org.eclipse.jetty.server.Server;</a>
<span class="sourceLineNo">032</span><a id="line.32">import org.eclipse.jetty.server.ServerConnector;</a>
<span class="sourceLineNo">033</span><a id="line.33">import org.eclipse.jetty.server.SslConnectionFactory;</a>
<span class="sourceLineNo">034</span><a id="line.34">import org.eclipse.jetty.util.resource.FileResource;</a>
<span class="sourceLineNo">035</span><a id="line.35">import org.eclipse.jetty.util.ssl.SslContextFactory;</a>
<span class="sourceLineNo">036</span><a id="line.36">import org.eclipse.jetty.webapp.Configuration;</a>
<span class="sourceLineNo">037</span><a id="line.37">import org.eclipse.jetty.webapp.FragmentConfiguration;</a>
<span class="sourceLineNo">038</span><a id="line.38">import org.eclipse.jetty.webapp.JettyWebXmlConfiguration;</a>
<span class="sourceLineNo">039</span><a id="line.39">import org.eclipse.jetty.webapp.MetaInfConfiguration;</a>
<span class="sourceLineNo">040</span><a id="line.40">import org.eclipse.jetty.webapp.WebAppContext;</a>
<span class="sourceLineNo">041</span><a id="line.41">import org.eclipse.jetty.webapp.WebInfConfiguration;</a>
<span class="sourceLineNo">042</span><a id="line.42">import org.eclipse.jetty.webapp.WebXmlConfiguration;</a>
<span class="sourceLineNo">043</span><a id="line.43">import org.junit.AfterClass;</a>
<span class="sourceLineNo">044</span><a id="line.44">import org.junit.Before;</a>
<span class="sourceLineNo">045</span><a id="line.45">import org.junit.BeforeClass;</a>
<span class="sourceLineNo">046</span><a id="line.46"></a>
<span class="sourceLineNo">047</span><a id="line.47">import java.io.File;</a>
<span class="sourceLineNo">048</span><a id="line.48">import java.io.FilenameFilter;</a>
<span class="sourceLineNo">049</span><a id="line.49">import java.io.IOException;</a>
<span class="sourceLineNo">050</span><a id="line.50">import java.io.UnsupportedEncodingException;</a>
<span class="sourceLineNo">051</span><a id="line.51">import java.net.ServerSocket;</a>
<span class="sourceLineNo">052</span><a id="line.52">import java.net.URL;</a>
<span class="sourceLineNo">053</span><a id="line.53">import java.nio.file.Files;</a>
<span class="sourceLineNo">054</span><a id="line.54">import java.nio.file.Path;</a>
<span class="sourceLineNo">055</span><a id="line.55">import java.nio.file.StandardCopyOption;</a>
<span class="sourceLineNo">056</span><a id="line.56"></a>
<span class="sourceLineNo">057</span><a id="line.57">import static com.github.mjeanroy.junit.servers.commons.Strings.isNotBlank;</a>
<span class="sourceLineNo">058</span><a id="line.58">import static org.eclipse.jetty.util.resource.Resource.newResource;</a>
<span class="sourceLineNo">059</span><a id="line.59">import static org.junit.Assert.assertEquals;</a>
<span class="sourceLineNo">060</span><a id="line.60">import static org.junit.Assert.assertTrue;</a>
<span class="sourceLineNo">061</span><a id="line.61"></a>
<span class="sourceLineNo">062</span><a id="line.62">public abstract class AbstractContainerIT {</a>
<span class="sourceLineNo">063</span><a id="line.63"></a>
<span class="sourceLineNo">064</span><a id="line.64"> protected static EmbeddedJetty jetty;</a>
<span class="sourceLineNo">065</span><a id="line.65"></a>
<span class="sourceLineNo">066</span><a id="line.66"> protected static int tlsPort;</a>
<span class="sourceLineNo">067</span><a id="line.67"></a>
<span class="sourceLineNo">068</span><a id="line.68"> protected final WebClient webClient = new WebClient();</a>
<span class="sourceLineNo">069</span><a id="line.69"></a>
<span class="sourceLineNo">070</span><a id="line.70"> protected static final File TEST_KEYSTORE_PATH = setupKeyStore();</a>
<span class="sourceLineNo">071</span><a id="line.71"> protected static final String TEST_KEYSTORE_PASSWORD = "password";</a>
<span class="sourceLineNo">072</span><a id="line.72"></a>
<span class="sourceLineNo">073</span><a id="line.73"> @BeforeClass</a>
<span class="sourceLineNo">074</span><a id="line.74"> public static void startContainer() throws Exception {</a>
<span class="sourceLineNo">075</span><a id="line.75"></a>
<span class="sourceLineNo">076</span><a id="line.76"> EmbeddedJettyConfiguration config = EmbeddedJettyConfiguration.builder()</a>
<span class="sourceLineNo">077</span><a id="line.77"> .withWebapp(getWarDir())</a>
<span class="sourceLineNo">078</span><a id="line.78"> .build();</a>
<span class="sourceLineNo">079</span><a id="line.79"></a>
<span class="sourceLineNo">080</span><a id="line.80"> jetty = new EmbeddedJetty(config) {</a>
<span class="sourceLineNo">081</span><a id="line.81"></a>
<span class="sourceLineNo">082</span><a id="line.82"> /**</a>
<span class="sourceLineNo">083</span><a id="line.83"> * Overriding with contents of this pull request, to make fragment scanning work.</a>
<span class="sourceLineNo">084</span><a id="line.84"> * https://github.com/mjeanroy/junit-servers/pull/3</a>
<span class="sourceLineNo">085</span><a id="line.85"> */</a>
<span class="sourceLineNo">086</span><a id="line.86"> protected WebAppContext createdWebAppContext() throws Exception {</a>
<span class="sourceLineNo">087</span><a id="line.87"> final String path = configuration.getPath();</a>
<span class="sourceLineNo">088</span><a id="line.88"> final String webapp = configuration.getWebapp();</a>
<span class="sourceLineNo">089</span><a id="line.89"> final String classpath = configuration.getClasspath();</a>
<span class="sourceLineNo">090</span><a id="line.90"></a>
<span class="sourceLineNo">091</span><a id="line.91"> WebAppContext ctx = new WebAppContext();</a>
<span class="sourceLineNo">092</span><a id="line.92"> ctx.setClassLoader(Thread.currentThread().getContextClassLoader());</a>
<span class="sourceLineNo">093</span><a id="line.93"> ctx.setContextPath(path);</a>
<span class="sourceLineNo">094</span><a id="line.94"></a>
<span class="sourceLineNo">095</span><a id="line.95"> // Useful for WebXmlConfiguration</a>
<span class="sourceLineNo">096</span><a id="line.96"> ctx.setBaseResource(newResource(webapp));</a>
<span class="sourceLineNo">097</span><a id="line.97"></a>
<span class="sourceLineNo">098</span><a id="line.98"> ctx.setConfigurations(new Configuration[]{</a>
<span class="sourceLineNo">099</span><a id="line.99"> new WebInfConfiguration(),</a>
<span class="sourceLineNo">100</span><a id="line.100"> new WebXmlConfiguration(),</a>
<span class="sourceLineNo">101</span><a id="line.101"> new AnnotationConfiguration(),</a>
<span class="sourceLineNo">102</span><a id="line.102"> new JettyWebXmlConfiguration(),</a>
<span class="sourceLineNo">103</span><a id="line.103"> new MetaInfConfiguration(),</a>
<span class="sourceLineNo">104</span><a id="line.104"> new FragmentConfiguration(),</a>
<span class="sourceLineNo">105</span><a id="line.105"> });</a>
<span class="sourceLineNo">106</span><a id="line.106"></a>
<span class="sourceLineNo">107</span><a id="line.107"> if (isNotBlank(classpath)) {</a>
<span class="sourceLineNo">108</span><a id="line.108"> // Fix to scan Spring WebApplicationInitializer</a>
<span class="sourceLineNo">109</span><a id="line.109"> // This will add compiled classes to jetty classpath</a>
<span class="sourceLineNo">110</span><a id="line.110"> // See: http://stackoverflow.com/questions/13222071/spring-3-1-webapplicationinitializer-embedded-jetty-8-annotationconfiguration</a>
<span class="sourceLineNo">111</span><a id="line.111"> // And more precisely: http://stackoverflow.com/a/18449506/1215828</a>
<span class="sourceLineNo">112</span><a id="line.112"> File classes = new File(classpath);</a>
<span class="sourceLineNo">113</span><a id="line.113"> FileResource containerResources = new FileResource(classes.toURI());</a>
<span class="sourceLineNo">114</span><a id="line.114"> ctx.getMetaData().addContainerResource(containerResources);</a>
<span class="sourceLineNo">115</span><a id="line.115"> }</a>
<span class="sourceLineNo">116</span><a id="line.116"></a>
<span class="sourceLineNo">117</span><a id="line.117"> Server server = getDelegate();</a>
<span class="sourceLineNo">118</span><a id="line.118"></a>
<span class="sourceLineNo">119</span><a id="line.119"> // web app</a>
<span class="sourceLineNo">120</span><a id="line.120"> ctx.setParentLoaderPriority(true);</a>
<span class="sourceLineNo">121</span><a id="line.121"> ctx.setWar(webapp);</a>
<span class="sourceLineNo">122</span><a id="line.122"> ctx.setServer(server);</a>
<span class="sourceLineNo">123</span><a id="line.123"></a>
<span class="sourceLineNo">124</span><a id="line.124"> // Add server context</a>
<span class="sourceLineNo">125</span><a id="line.125"> server.setHandler(ctx);</a>
<span class="sourceLineNo">126</span><a id="line.126"></a>
<span class="sourceLineNo">127</span><a id="line.127"> return ctx;</a>
<span class="sourceLineNo">128</span><a id="line.128"> }</a>
<span class="sourceLineNo">129</span><a id="line.129"> };</a>
<span class="sourceLineNo">130</span><a id="line.130"></a>
<span class="sourceLineNo">131</span><a id="line.131"> Server server = jetty.getDelegate();</a>
<span class="sourceLineNo">132</span><a id="line.132"></a>
<span class="sourceLineNo">133</span><a id="line.133"> // TLS</a>
<span class="sourceLineNo">134</span><a id="line.134"> tlsPort = getFreePort();</a>
<span class="sourceLineNo">135</span><a id="line.135"></a>
<span class="sourceLineNo">136</span><a id="line.136"> final SslContextFactory sslContextFactory = new SslContextFactory.Server();</a>
<span class="sourceLineNo">137</span><a id="line.137"> sslContextFactory.setKeyStorePath(TEST_KEYSTORE_PATH.getAbsolutePath());</a>
<span class="sourceLineNo">138</span><a id="line.138"> sslContextFactory.setKeyStorePassword(TEST_KEYSTORE_PASSWORD);</a>
<span class="sourceLineNo">139</span><a id="line.139"> sslContextFactory.setKeyManagerPassword(TEST_KEYSTORE_PASSWORD);</a>
<span class="sourceLineNo">140</span><a id="line.140"></a>
<span class="sourceLineNo">141</span><a id="line.141"> HttpConfiguration https = new HttpConfiguration();</a>
<span class="sourceLineNo">142</span><a id="line.142"> https.addCustomizer(new SecureRequestCustomizer());</a>
<span class="sourceLineNo">143</span><a id="line.143"></a>
<span class="sourceLineNo">144</span><a id="line.144"> final ServerConnector httpsConnector = new ServerConnector(</a>
<span class="sourceLineNo">145</span><a id="line.145"> server,</a>
<span class="sourceLineNo">146</span><a id="line.146"> new SslConnectionFactory(sslContextFactory, HttpVersion.HTTP_1_1.asString()),</a>
<span class="sourceLineNo">147</span><a id="line.147"> new HttpConnectionFactory(https));</a>
<span class="sourceLineNo">148</span><a id="line.148"> httpsConnector.setPort(tlsPort);</a>
<span class="sourceLineNo">149</span><a id="line.149"> server.addConnector(httpsConnector);</a>
<span class="sourceLineNo">150</span><a id="line.150"></a>
<span class="sourceLineNo">151</span><a id="line.151"> jetty.start();</a>
<span class="sourceLineNo">152</span><a id="line.152"></a>
<span class="sourceLineNo">153</span><a id="line.153"> assertTrue(jetty.isStarted());</a>
<span class="sourceLineNo">154</span><a id="line.154"> }</a>
<span class="sourceLineNo">155</span><a id="line.155"></a>
<span class="sourceLineNo">156</span><a id="line.156"> protected static String getBaseUri() {</a>
<span class="sourceLineNo">157</span><a id="line.157"> return "http://localhost:" + jetty.getPort() + "/";</a>
<span class="sourceLineNo">158</span><a id="line.158"> }</a>
<span class="sourceLineNo">159</span><a id="line.159"></a>
<span class="sourceLineNo">160</span><a id="line.160"> protected static String getTlsBaseUri() {</a>
<span class="sourceLineNo">161</span><a id="line.161"> return "https://localhost:" + tlsPort + "/";</a>
<span class="sourceLineNo">162</span><a id="line.162"> }</a>
<span class="sourceLineNo">163</span><a id="line.163"></a>
<span class="sourceLineNo">164</span><a id="line.164"> protected static String getWarDir() {</a>
<span class="sourceLineNo">165</span><a id="line.165"> File[] warFiles = new File("target").listFiles(new FilenameFilter() {</a>
<span class="sourceLineNo">166</span><a id="line.166"> @Override</a>
<span class="sourceLineNo">167</span><a id="line.167"> public boolean accept(File dir, String name) {</a>
<span class="sourceLineNo">168</span><a id="line.168"> return name.endsWith(".war");</a>
<span class="sourceLineNo">169</span><a id="line.169"> }</a>
<span class="sourceLineNo">170</span><a id="line.170"> });</a>
<span class="sourceLineNo">171</span><a id="line.171"></a>
<span class="sourceLineNo">172</span><a id="line.172"> assertEquals("Expected only one war file in target directory, run 'mvn clean' and try again", 1, warFiles.length);</a>
<span class="sourceLineNo">173</span><a id="line.173"></a>
<span class="sourceLineNo">174</span><a id="line.174"> return warFiles[0].getAbsolutePath().replaceFirst("\\.war$", "");</a>
<span class="sourceLineNo">175</span><a id="line.175"> }</a>
<span class="sourceLineNo">176</span><a id="line.176"></a>
<span class="sourceLineNo">177</span><a id="line.177"> protected static String getBasicAuthorizationHeaderValue(String username, String password) throws UnsupportedEncodingException {</a>
<span class="sourceLineNo">178</span><a id="line.178"> String authorizationHeader = username + ":" + password;</a>
<span class="sourceLineNo">179</span><a id="line.179"> byte[] valueBytes;</a>
<span class="sourceLineNo">180</span><a id="line.180"> valueBytes = authorizationHeader.getBytes("UTF-8");</a>
<span class="sourceLineNo">181</span><a id="line.181"> authorizationHeader = new String(Base64.encode(valueBytes));</a>
<span class="sourceLineNo">182</span><a id="line.182"> return "Basic " + authorizationHeader;</a>
<span class="sourceLineNo">183</span><a id="line.183"> }</a>
<span class="sourceLineNo">184</span><a id="line.184"></a>
<span class="sourceLineNo">185</span><a id="line.185"> @Before</a>
<span class="sourceLineNo">186</span><a id="line.186"> public void beforeTest() {</a>
<span class="sourceLineNo">187</span><a id="line.187"> webClient.getOptions().setThrowExceptionOnFailingStatusCode(true);</a>
<span class="sourceLineNo">188</span><a id="line.188"> }</a>
<span class="sourceLineNo">189</span><a id="line.189"></a>
<span class="sourceLineNo">190</span><a id="line.190"> @AfterClass</a>
<span class="sourceLineNo">191</span><a id="line.191"> public static void stopContainer() {</a>
<span class="sourceLineNo">192</span><a id="line.192"> if (jetty != null) {</a>
<span class="sourceLineNo">193</span><a id="line.193"> jetty.stop();</a>
<span class="sourceLineNo">194</span><a id="line.194"> }</a>
<span class="sourceLineNo">195</span><a id="line.195"> }</a>
<span class="sourceLineNo">196</span><a id="line.196"></a>
<span class="sourceLineNo">197</span><a id="line.197"> private static int getFreePort() {</a>
<span class="sourceLineNo">198</span><a id="line.198"> try (ServerSocket socket = new ServerSocket(0)) {</a>
<span class="sourceLineNo">199</span><a id="line.199"> return socket.getLocalPort();</a>
<span class="sourceLineNo">200</span><a id="line.200"> } catch (IOException e) {</a>
<span class="sourceLineNo">201</span><a id="line.201"> throw new IllegalStateException("Failed to allocate free port", e);</a>
<span class="sourceLineNo">202</span><a id="line.202"> }</a>
<span class="sourceLineNo">203</span><a id="line.203"> }</a>
<span class="sourceLineNo">204</span><a id="line.204"></a>
<span class="sourceLineNo">205</span><a id="line.205"> // Dealing with a keystore is NOT fun, it's easier to script one with the keytool</a>
<span class="sourceLineNo">206</span><a id="line.206"> // see src/main/resources/createKeyStore.sh for more info</a>
<span class="sourceLineNo">207</span><a id="line.207"> private static File setupKeyStore() {</a>
<span class="sourceLineNo">208</span><a id="line.208"> try {</a>
<span class="sourceLineNo">209</span><a id="line.209"> Path outKeyStoreFile = File.createTempFile("test-keystore", "jks").toPath();</a>
<span class="sourceLineNo">210</span><a id="line.210"> URL keyStoreResource = Thread.currentThread().getContextClassLoader().getResource("test-keystore.jks");</a>
<span class="sourceLineNo">211</span><a id="line.211"> Files.copy(keyStoreResource.openStream(), outKeyStoreFile, StandardCopyOption.REPLACE_EXISTING);</a>
<span class="sourceLineNo">212</span><a id="line.212"> File keyStoreFile = outKeyStoreFile.toFile();</a>
<span class="sourceLineNo">213</span><a id="line.213"></a>
<span class="sourceLineNo">214</span><a id="line.214"> // clients will pick up the ssl keystore this way, so just set SSL properties</a>
<span class="sourceLineNo">215</span><a id="line.215"> System.setProperty("javax.net.ssl.trustStore", keyStoreFile.getAbsolutePath());</a>
<span class="sourceLineNo">216</span><a id="line.216"> System.setProperty("javax.net.ssl.trustStorePassword", TEST_KEYSTORE_PASSWORD);</a>
<span class="sourceLineNo">217</span><a id="line.217"> return keyStoreFile;</a>
<span class="sourceLineNo">218</span><a id="line.218"> } catch (IOException e) {</a>
<span class="sourceLineNo">219</span><a id="line.219"> throw new IllegalStateException("Failed to create test keystore", e);</a>
<span class="sourceLineNo">220</span><a id="line.220"> }</a>
<span class="sourceLineNo">221</span><a id="line.221"> }</a>
<span class="sourceLineNo">222</span><a id="line.222">}</a>
</pre>
</div>
</main>
</body>
</html>