blob: 5de80d79e5330335276bab9e4e99b5cbdc1c0bb0 [file] [log] [blame]
<!DOCTYPE HTML>
<html lang="en">
<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">package org.apache.archiva.redback.keys;</a>
<span class="sourceLineNo">002</span><a id="line.2"></a>
<span class="sourceLineNo">003</span><a id="line.3">/*</a>
<span class="sourceLineNo">004</span><a id="line.4"> * Copyright 2001-2006 The Apache Software Foundation.</a>
<span class="sourceLineNo">005</span><a id="line.5"> *</a>
<span class="sourceLineNo">006</span><a id="line.6"> * Licensed under the Apache License, Version 2.0 (the "License");</a>
<span class="sourceLineNo">007</span><a id="line.7"> * you may not use this file except in compliance with the License.</a>
<span class="sourceLineNo">008</span><a id="line.8"> * 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, software</a>
<span class="sourceLineNo">013</span><a id="line.13"> * distributed under the License is distributed on an "AS IS" BASIS,</a>
<span class="sourceLineNo">014</span><a id="line.14"> * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.</a>
<span class="sourceLineNo">015</span><a id="line.15"> * See the License for the specific language governing permissions and</a>
<span class="sourceLineNo">016</span><a id="line.16"> * limitations under the License.</a>
<span class="sourceLineNo">017</span><a id="line.17"> */</a>
<span class="sourceLineNo">018</span><a id="line.18"></a>
<span class="sourceLineNo">019</span><a id="line.19">import junit.framework.TestCase;</a>
<span class="sourceLineNo">020</span><a id="line.20">import org.junit.Test;</a>
<span class="sourceLineNo">021</span><a id="line.21">import org.junit.runner.RunWith;</a>
<span class="sourceLineNo">022</span><a id="line.22">import org.springframework.test.context.ContextConfiguration;</a>
<span class="sourceLineNo">023</span><a id="line.23">import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;</a>
<span class="sourceLineNo">024</span><a id="line.24"></a>
<span class="sourceLineNo">025</span><a id="line.25">import java.text.SimpleDateFormat;</a>
<span class="sourceLineNo">026</span><a id="line.26">import java.util.*;</a>
<span class="sourceLineNo">027</span><a id="line.27"></a>
<span class="sourceLineNo">028</span><a id="line.28">/**</a>
<span class="sourceLineNo">029</span><a id="line.29"> * KeyManagerTestCase</a>
<span class="sourceLineNo">030</span><a id="line.30"> *</a>
<span class="sourceLineNo">031</span><a id="line.31"> * @author &lt;a href="mailto:joakim@erdfelt.com"&gt;Joakim Erdfelt&lt;/a&gt;</a>
<span class="sourceLineNo">032</span><a id="line.32"> *</a>
<span class="sourceLineNo">033</span><a id="line.33"> */</a>
<span class="sourceLineNo">034</span><a id="line.34">@RunWith( SpringJUnit4ClassRunner.class )</a>
<span class="sourceLineNo">035</span><a id="line.35">@ContextConfiguration( locations = {"classpath*:/META-INF/spring-context.xml","classpath*:/spring-context.xml"} )</a>
<span class="sourceLineNo">036</span><a id="line.36">public class KeyManagerTestCase</a>
<span class="sourceLineNo">037</span><a id="line.37"> extends TestCase</a>
<span class="sourceLineNo">038</span><a id="line.38">{</a>
<span class="sourceLineNo">039</span><a id="line.39"> private KeyManager manager;</a>
<span class="sourceLineNo">040</span><a id="line.40"></a>
<span class="sourceLineNo">041</span><a id="line.41"> public KeyManager getKeyManager()</a>
<span class="sourceLineNo">042</span><a id="line.42"> {</a>
<span class="sourceLineNo">043</span><a id="line.43"> return manager;</a>
<span class="sourceLineNo">044</span><a id="line.44"> }</a>
<span class="sourceLineNo">045</span><a id="line.45"></a>
<span class="sourceLineNo">046</span><a id="line.46"> public void setKeyManager( KeyManager manager )</a>
<span class="sourceLineNo">047</span><a id="line.47"> {</a>
<span class="sourceLineNo">048</span><a id="line.48"> this.manager = manager;</a>
<span class="sourceLineNo">049</span><a id="line.49"> }</a>
<span class="sourceLineNo">050</span><a id="line.50"></a>
<span class="sourceLineNo">051</span><a id="line.51"> private void assertSameDates( Date expected, Date actual )</a>
<span class="sourceLineNo">052</span><a id="line.52"> {</a>
<span class="sourceLineNo">053</span><a id="line.53"> if ( ( expected == null ) &amp;&amp; ( actual != null ) )</a>
<span class="sourceLineNo">054</span><a id="line.54"> {</a>
<span class="sourceLineNo">055</span><a id="line.55"> fail( "Expected date is null, actual date [" + actual + "]." );</a>
<span class="sourceLineNo">056</span><a id="line.56"> }</a>
<span class="sourceLineNo">057</span><a id="line.57"></a>
<span class="sourceLineNo">058</span><a id="line.58"> if ( ( expected != null ) &amp;&amp; ( actual == null ) )</a>
<span class="sourceLineNo">059</span><a id="line.59"> {</a>
<span class="sourceLineNo">060</span><a id="line.60"> fail( "Expected date [" + expected + "], actual date is null." );</a>
<span class="sourceLineNo">061</span><a id="line.61"> }</a>
<span class="sourceLineNo">062</span><a id="line.62"></a>
<span class="sourceLineNo">063</span><a id="line.63"> SimpleDateFormat format = new SimpleDateFormat( "EEE, d MMM yyyy HH:mm:ss Z" );</a>
<span class="sourceLineNo">064</span><a id="line.64"> assertEquals( format.format( expected ), format.format( actual ) );</a>
<span class="sourceLineNo">065</span><a id="line.65"> }</a>
<span class="sourceLineNo">066</span><a id="line.66"></a>
<span class="sourceLineNo">067</span><a id="line.67"> @Test</a>
<span class="sourceLineNo">068</span><a id="line.68"> public void testNormal()</a>
<span class="sourceLineNo">069</span><a id="line.69"> throws KeyNotFoundException, KeyManagerException</a>
<span class="sourceLineNo">070</span><a id="line.70"> {</a>
<span class="sourceLineNo">071</span><a id="line.71"> String principal = "foo";</a>
<span class="sourceLineNo">072</span><a id="line.72"> String purpose = "Testing";</a>
<span class="sourceLineNo">073</span><a id="line.73"></a>
<span class="sourceLineNo">074</span><a id="line.74"> AuthenticationKey created = getKeyManager().createKey( principal, purpose, 15 );</a>
<span class="sourceLineNo">075</span><a id="line.75"></a>
<span class="sourceLineNo">076</span><a id="line.76"> assertNotNull( created );</a>
<span class="sourceLineNo">077</span><a id="line.77"> assertNotNull( created.getKey() );</a>
<span class="sourceLineNo">078</span><a id="line.78"> assertNotNull( created.getDateCreated() );</a>
<span class="sourceLineNo">079</span><a id="line.79"> assertNotNull( created.getDateExpires() );</a>
<span class="sourceLineNo">080</span><a id="line.80"></a>
<span class="sourceLineNo">081</span><a id="line.81"> assertEquals( principal, created.getForPrincipal() );</a>
<span class="sourceLineNo">082</span><a id="line.82"> assertEquals( purpose, created.getPurpose() );</a>
<span class="sourceLineNo">083</span><a id="line.83"></a>
<span class="sourceLineNo">084</span><a id="line.84"> Date expectedCreated = created.getDateCreated();</a>
<span class="sourceLineNo">085</span><a id="line.85"> Date expectedExpires = created.getDateExpires();</a>
<span class="sourceLineNo">086</span><a id="line.86"></a>
<span class="sourceLineNo">087</span><a id="line.87"> String expectedKey = created.getKey();</a>
<span class="sourceLineNo">088</span><a id="line.88"></a>
<span class="sourceLineNo">089</span><a id="line.89"> AuthenticationKey found = getKeyManager().findKey( expectedKey );</a>
<span class="sourceLineNo">090</span><a id="line.90"></a>
<span class="sourceLineNo">091</span><a id="line.91"> assertEquals( expectedKey, found.getKey() );</a>
<span class="sourceLineNo">092</span><a id="line.92"> assertEquals( principal, found.getForPrincipal() );</a>
<span class="sourceLineNo">093</span><a id="line.93"> assertEquals( purpose, found.getPurpose() );</a>
<span class="sourceLineNo">094</span><a id="line.94"> assertSameDates( expectedCreated, found.getDateCreated() );</a>
<span class="sourceLineNo">095</span><a id="line.95"> assertSameDates( expectedExpires, found.getDateExpires() );</a>
<span class="sourceLineNo">096</span><a id="line.96"> }</a>
<span class="sourceLineNo">097</span><a id="line.97"></a>
<span class="sourceLineNo">098</span><a id="line.98"> @Test</a>
<span class="sourceLineNo">099</span><a id="line.99"> public void testGetAllKeys()</a>
<span class="sourceLineNo">100</span><a id="line.100"> throws KeyManagerException</a>
<span class="sourceLineNo">101</span><a id="line.101"> {</a>
<span class="sourceLineNo">102</span><a id="line.102"> getKeyManager().eraseDatabase();</a>
<span class="sourceLineNo">103</span><a id="line.103"> AuthenticationKey created1 = getKeyManager().createKey( "foo", "Testing", 15 );</a>
<span class="sourceLineNo">104</span><a id="line.104"> AuthenticationKey created2 = getKeyManager().createKey( "bar", "Something", 23 );</a>
<span class="sourceLineNo">105</span><a id="line.105"></a>
<span class="sourceLineNo">106</span><a id="line.106"> assertNotNull( created1 );</a>
<span class="sourceLineNo">107</span><a id="line.107"> assertNotNull( created2 );</a>
<span class="sourceLineNo">108</span><a id="line.108"></a>
<span class="sourceLineNo">109</span><a id="line.109"> assertEquals( "foo", created1.getForPrincipal() );</a>
<span class="sourceLineNo">110</span><a id="line.110"> assertEquals( "Testing", created1.getPurpose() );</a>
<span class="sourceLineNo">111</span><a id="line.111"></a>
<span class="sourceLineNo">112</span><a id="line.112"> assertEquals( "bar", created2.getForPrincipal() );</a>
<span class="sourceLineNo">113</span><a id="line.113"> assertEquals( "Something", created2.getPurpose() );</a>
<span class="sourceLineNo">114</span><a id="line.114"></a>
<span class="sourceLineNo">115</span><a id="line.115"> System.out.println("foo key "+created1.getKey());</a>
<span class="sourceLineNo">116</span><a id="line.116"> System.out.println("bar key "+created2.getKey());</a>
<span class="sourceLineNo">117</span><a id="line.117"> List&lt;AuthenticationKey&gt; keys = new ArrayList&lt;&gt;(getKeyManager().getAllKeys());</a>
<span class="sourceLineNo">118</span><a id="line.118"> Collections.sort( keys, new Comparator&lt;AuthenticationKey&gt;()</a>
<span class="sourceLineNo">119</span><a id="line.119"> {</a>
<span class="sourceLineNo">120</span><a id="line.120"> public int compare( AuthenticationKey key1, AuthenticationKey key2 )</a>
<span class="sourceLineNo">121</span><a id="line.121"> {</a>
<span class="sourceLineNo">122</span><a id="line.122"> System.out.println("Compare "+key2.getForPrincipal()+key2.getKey()+" - "+key1.getForPrincipal()+key1.getKey());</a>
<span class="sourceLineNo">123</span><a id="line.123"> return key2.getForPrincipal().compareTo( key1.getForPrincipal() );</a>
<span class="sourceLineNo">124</span><a id="line.124"> }</a>
<span class="sourceLineNo">125</span><a id="line.125"> } );</a>
<span class="sourceLineNo">126</span><a id="line.126"></a>
<span class="sourceLineNo">127</span><a id="line.127"> AuthenticationKey found = (AuthenticationKey) keys.get( 0 );</a>
<span class="sourceLineNo">128</span><a id="line.128"> assertEquals( created1.getKey(), found.getKey() );</a>
<span class="sourceLineNo">129</span><a id="line.129"> assertEquals( "foo", found.getForPrincipal() );</a>
<span class="sourceLineNo">130</span><a id="line.130"> assertEquals( "Testing", found.getPurpose() );</a>
<span class="sourceLineNo">131</span><a id="line.131"> assertSameDates( created1.getDateCreated(), found.getDateCreated() );</a>
<span class="sourceLineNo">132</span><a id="line.132"> assertSameDates( created1.getDateExpires(), found.getDateExpires() );</a>
<span class="sourceLineNo">133</span><a id="line.133"></a>
<span class="sourceLineNo">134</span><a id="line.134"> found = (AuthenticationKey) keys.get( 1 );</a>
<span class="sourceLineNo">135</span><a id="line.135"> assertEquals( created2.getKey(), found.getKey() );</a>
<span class="sourceLineNo">136</span><a id="line.136"> assertEquals( "bar", found.getForPrincipal() );</a>
<span class="sourceLineNo">137</span><a id="line.137"> assertEquals( "Something", found.getPurpose() );</a>
<span class="sourceLineNo">138</span><a id="line.138"> assertSameDates( created2.getDateCreated(), found.getDateCreated() );</a>
<span class="sourceLineNo">139</span><a id="line.139"> assertSameDates( created2.getDateExpires(), found.getDateExpires() );</a>
<span class="sourceLineNo">140</span><a id="line.140"> }</a>
<span class="sourceLineNo">141</span><a id="line.141"></a>
<span class="sourceLineNo">142</span><a id="line.142"> @Test</a>
<span class="sourceLineNo">143</span><a id="line.143"> public void testNotThere()</a>
<span class="sourceLineNo">144</span><a id="line.144"> throws KeyManagerException</a>
<span class="sourceLineNo">145</span><a id="line.145"> {</a>
<span class="sourceLineNo">146</span><a id="line.146"> String principal = "foo";</a>
<span class="sourceLineNo">147</span><a id="line.147"> String purpose = "Testing";</a>
<span class="sourceLineNo">148</span><a id="line.148"></a>
<span class="sourceLineNo">149</span><a id="line.149"> AuthenticationKey created = getKeyManager().createKey( principal, purpose, 15 );</a>
<span class="sourceLineNo">150</span><a id="line.150"></a>
<span class="sourceLineNo">151</span><a id="line.151"> assertNotNull( created );</a>
<span class="sourceLineNo">152</span><a id="line.152"> assertNotNull( created.getKey() );</a>
<span class="sourceLineNo">153</span><a id="line.153"> assertNotNull( created.getDateCreated() );</a>
<span class="sourceLineNo">154</span><a id="line.154"> assertNotNull( created.getDateExpires() );</a>
<span class="sourceLineNo">155</span><a id="line.155"></a>
<span class="sourceLineNo">156</span><a id="line.156"> assertEquals( principal, created.getForPrincipal() );</a>
<span class="sourceLineNo">157</span><a id="line.157"> assertEquals( purpose, created.getPurpose() );</a>
<span class="sourceLineNo">158</span><a id="line.158"></a>
<span class="sourceLineNo">159</span><a id="line.159"> try</a>
<span class="sourceLineNo">160</span><a id="line.160"> {</a>
<span class="sourceLineNo">161</span><a id="line.161"> getKeyManager().findKey( "deadbeefkey" );</a>
<span class="sourceLineNo">162</span><a id="line.162"> fail( "Invalid Key should not have been found." );</a>
<span class="sourceLineNo">163</span><a id="line.163"> }</a>
<span class="sourceLineNo">164</span><a id="line.164"> catch ( KeyNotFoundException e )</a>
<span class="sourceLineNo">165</span><a id="line.165"> {</a>
<span class="sourceLineNo">166</span><a id="line.166"> // Expected path for this test.</a>
<span class="sourceLineNo">167</span><a id="line.167"> }</a>
<span class="sourceLineNo">168</span><a id="line.168"> }</a>
<span class="sourceLineNo">169</span><a id="line.169"></a>
<span class="sourceLineNo">170</span><a id="line.170"> @Test</a>
<span class="sourceLineNo">171</span><a id="line.171"> public void testExpired()</a>
<span class="sourceLineNo">172</span><a id="line.172"> throws KeyManagerException, InterruptedException</a>
<span class="sourceLineNo">173</span><a id="line.173"> {</a>
<span class="sourceLineNo">174</span><a id="line.174"> String principal = "foo";</a>
<span class="sourceLineNo">175</span><a id="line.175"> String purpose = "Testing";</a>
<span class="sourceLineNo">176</span><a id="line.176"></a>
<span class="sourceLineNo">177</span><a id="line.177"> AuthenticationKey created = getKeyManager().createKey( principal, purpose, 0 );</a>
<span class="sourceLineNo">178</span><a id="line.178"></a>
<span class="sourceLineNo">179</span><a id="line.179"> assertNotNull( created );</a>
<span class="sourceLineNo">180</span><a id="line.180"> assertNotNull( created.getKey() );</a>
<span class="sourceLineNo">181</span><a id="line.181"> assertNotNull( created.getDateCreated() );</a>
<span class="sourceLineNo">182</span><a id="line.182"> assertNotNull( created.getDateExpires() );</a>
<span class="sourceLineNo">183</span><a id="line.183"></a>
<span class="sourceLineNo">184</span><a id="line.184"> assertEquals( principal, created.getForPrincipal() );</a>
<span class="sourceLineNo">185</span><a id="line.185"> assertEquals( purpose, created.getPurpose() );</a>
<span class="sourceLineNo">186</span><a id="line.186"></a>
<span class="sourceLineNo">187</span><a id="line.187"> String expectedKey = created.getKey();</a>
<span class="sourceLineNo">188</span><a id="line.188"></a>
<span class="sourceLineNo">189</span><a id="line.189"> try</a>
<span class="sourceLineNo">190</span><a id="line.190"> {</a>
<span class="sourceLineNo">191</span><a id="line.191"> Thread.sleep( 500 ); // Sleep to let it expire</a>
<span class="sourceLineNo">192</span><a id="line.192"> getKeyManager().findKey( expectedKey );</a>
<span class="sourceLineNo">193</span><a id="line.193"> fail( "Expired Key should not have been found." );</a>
<span class="sourceLineNo">194</span><a id="line.194"> }</a>
<span class="sourceLineNo">195</span><a id="line.195"> catch ( KeyNotFoundException e )</a>
<span class="sourceLineNo">196</span><a id="line.196"> {</a>
<span class="sourceLineNo">197</span><a id="line.197"> // Expected path for this test.</a>
<span class="sourceLineNo">198</span><a id="line.198"> }</a>
<span class="sourceLineNo">199</span><a id="line.199"> }</a>
<span class="sourceLineNo">200</span><a id="line.200"></a>
<span class="sourceLineNo">201</span><a id="line.201"> @Test</a>
<span class="sourceLineNo">202</span><a id="line.202"> public void testPermanent()</a>
<span class="sourceLineNo">203</span><a id="line.203"> throws KeyManagerException</a>
<span class="sourceLineNo">204</span><a id="line.204"> {</a>
<span class="sourceLineNo">205</span><a id="line.205"> String principal = "foo";</a>
<span class="sourceLineNo">206</span><a id="line.206"> String purpose = "Testing";</a>
<span class="sourceLineNo">207</span><a id="line.207"></a>
<span class="sourceLineNo">208</span><a id="line.208"> AuthenticationKey created = getKeyManager().createKey( principal, purpose, -1 );</a>
<span class="sourceLineNo">209</span><a id="line.209"></a>
<span class="sourceLineNo">210</span><a id="line.210"> assertNotNull( created );</a>
<span class="sourceLineNo">211</span><a id="line.211"> assertNotNull( created.getKey() );</a>
<span class="sourceLineNo">212</span><a id="line.212"> assertNotNull( created.getDateCreated() );</a>
<span class="sourceLineNo">213</span><a id="line.213"> assertNull( created.getDateExpires() );</a>
<span class="sourceLineNo">214</span><a id="line.214"></a>
<span class="sourceLineNo">215</span><a id="line.215"> assertEquals( principal, created.getForPrincipal() );</a>
<span class="sourceLineNo">216</span><a id="line.216"> assertEquals( purpose, created.getPurpose() );</a>
<span class="sourceLineNo">217</span><a id="line.217"></a>
<span class="sourceLineNo">218</span><a id="line.218"> Date expectedCreated = created.getDateCreated();</a>
<span class="sourceLineNo">219</span><a id="line.219"></a>
<span class="sourceLineNo">220</span><a id="line.220"> String expectedKey = created.getKey();</a>
<span class="sourceLineNo">221</span><a id="line.221"></a>
<span class="sourceLineNo">222</span><a id="line.222"> AuthenticationKey found = getKeyManager().findKey( expectedKey );</a>
<span class="sourceLineNo">223</span><a id="line.223"></a>
<span class="sourceLineNo">224</span><a id="line.224"> assertEquals( expectedKey, found.getKey() );</a>
<span class="sourceLineNo">225</span><a id="line.225"> assertEquals( principal, found.getForPrincipal() );</a>
<span class="sourceLineNo">226</span><a id="line.226"> assertEquals( purpose, found.getPurpose() );</a>
<span class="sourceLineNo">227</span><a id="line.227"> assertSameDates( expectedCreated, found.getDateCreated() );</a>
<span class="sourceLineNo">228</span><a id="line.228"> assertNull( found.getDateExpires() );</a>
<span class="sourceLineNo">229</span><a id="line.229"> }</a>
<span class="sourceLineNo">230</span><a id="line.230">}</a>
</pre>
</div>
</main>
</body>
</html>