| <!DOCTYPE HTML> |
| <html lang="de"> |
| <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.fulcrum.upload;</a> |
| <span class="sourceLineNo">002</span><a id="line.2"></a> |
| <span class="sourceLineNo">003</span><a id="line.3">import static org.junit.jupiter.api.Assertions.assertEquals;</a> |
| <span class="sourceLineNo">004</span><a id="line.4">import static org.junit.jupiter.api.Assertions.assertNotNull;</a> |
| <span class="sourceLineNo">005</span><a id="line.5">import static org.junit.jupiter.api.Assertions.assertTrue;</a> |
| <span class="sourceLineNo">006</span><a id="line.6">import static org.junit.jupiter.api.Assertions.fail;</a> |
| <span class="sourceLineNo">007</span><a id="line.7">import static org.mockito.ArgumentMatchers.any;</a> |
| <span class="sourceLineNo">008</span><a id="line.8">import static org.mockito.ArgumentMatchers.anyInt;</a> |
| <span class="sourceLineNo">009</span><a id="line.9">import static org.mockito.ArgumentMatchers.anyString;</a> |
| <span class="sourceLineNo">010</span><a id="line.10">import static org.mockito.Mockito.doAnswer;</a> |
| <span class="sourceLineNo">011</span><a id="line.11">import static org.mockito.Mockito.mock;</a> |
| <span class="sourceLineNo">012</span><a id="line.12">import static org.mockito.Mockito.when;</a> |
| <span class="sourceLineNo">013</span><a id="line.13"></a> |
| <span class="sourceLineNo">014</span><a id="line.14">import java.io.ByteArrayInputStream;</a> |
| <span class="sourceLineNo">015</span><a id="line.15"></a> |
| <span class="sourceLineNo">016</span><a id="line.16">/*</a> |
| <span class="sourceLineNo">017</span><a id="line.17"> * Licensed to the Apache Software Foundation (ASF) under one</a> |
| <span class="sourceLineNo">018</span><a id="line.18"> * or more contributor license agreements. See the NOTICE file</a> |
| <span class="sourceLineNo">019</span><a id="line.19"> * distributed with this work for additional information</a> |
| <span class="sourceLineNo">020</span><a id="line.20"> * regarding copyright ownership. The ASF licenses this file</a> |
| <span class="sourceLineNo">021</span><a id="line.21"> * to you under the Apache License, Version 2.0 (the</a> |
| <span class="sourceLineNo">022</span><a id="line.22"> * "License"); you may not use this file except in compliance</a> |
| <span class="sourceLineNo">023</span><a id="line.23"> * with the License. You may obtain a copy of the License at</a> |
| <span class="sourceLineNo">024</span><a id="line.24"> *</a> |
| <span class="sourceLineNo">025</span><a id="line.25"> * http://www.apache.org/licenses/LICENSE-2.0</a> |
| <span class="sourceLineNo">026</span><a id="line.26"> *</a> |
| <span class="sourceLineNo">027</span><a id="line.27"> * Unless required by applicable law or agreed to in writing,</a> |
| <span class="sourceLineNo">028</span><a id="line.28"> * software distributed under the License is distributed on an</a> |
| <span class="sourceLineNo">029</span><a id="line.29"> * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY</a> |
| <span class="sourceLineNo">030</span><a id="line.30"> * KIND, either express or implied. See the License for the</a> |
| <span class="sourceLineNo">031</span><a id="line.31"> * specific language governing permissions and limitations</a> |
| <span class="sourceLineNo">032</span><a id="line.32"> * under the License.</a> |
| <span class="sourceLineNo">033</span><a id="line.33"> */</a> |
| <span class="sourceLineNo">034</span><a id="line.34"></a> |
| <span class="sourceLineNo">035</span><a id="line.35"></a> |
| <span class="sourceLineNo">036</span><a id="line.36">import java.io.File;</a> |
| <span class="sourceLineNo">037</span><a id="line.37">import java.io.IOException;</a> |
| <span class="sourceLineNo">038</span><a id="line.38">import java.util.HashMap;</a> |
| <span class="sourceLineNo">039</span><a id="line.39">import java.util.List;</a> |
| <span class="sourceLineNo">040</span><a id="line.40">import java.util.Locale;</a> |
| <span class="sourceLineNo">041</span><a id="line.41">import java.util.Map;</a> |
| <span class="sourceLineNo">042</span><a id="line.42">import java.util.Vector;</a> |
| <span class="sourceLineNo">043</span><a id="line.43"></a> |
| <span class="sourceLineNo">044</span><a id="line.44">import javax.servlet.ReadListener;</a> |
| <span class="sourceLineNo">045</span><a id="line.45">import javax.servlet.ServletInputStream;</a> |
| <span class="sourceLineNo">046</span><a id="line.46">import javax.servlet.http.HttpServletRequest;</a> |
| <span class="sourceLineNo">047</span><a id="line.47">import javax.servlet.http.HttpSession;</a> |
| <span class="sourceLineNo">048</span><a id="line.48"></a> |
| <span class="sourceLineNo">049</span><a id="line.49">import org.apache.avalon.framework.component.ComponentException;</a> |
| <span class="sourceLineNo">050</span><a id="line.50">import org.apache.commons.fileupload.FileItem;</a> |
| <span class="sourceLineNo">051</span><a id="line.51">import org.apache.commons.fileupload.FileItemIterator;</a> |
| <span class="sourceLineNo">052</span><a id="line.52">import org.apache.fulcrum.testcontainer.BaseUnit5Test;</a> |
| <span class="sourceLineNo">053</span><a id="line.53">import org.junit.jupiter.api.BeforeEach;</a> |
| <span class="sourceLineNo">054</span><a id="line.54">import org.junit.jupiter.api.Test;</a> |
| <span class="sourceLineNo">055</span><a id="line.55">import org.mockito.invocation.InvocationOnMock;</a> |
| <span class="sourceLineNo">056</span><a id="line.56">import org.mockito.stubbing.Answer;</a> |
| <span class="sourceLineNo">057</span><a id="line.57">/**</a> |
| <span class="sourceLineNo">058</span><a id="line.58"> * UploadServiceTest</a> |
| <span class="sourceLineNo">059</span><a id="line.59"> *</a> |
| <span class="sourceLineNo">060</span><a id="line.60"> * @author <a href="epugh@upstate.com">Eric Pugh</a></a> |
| <span class="sourceLineNo">061</span><a id="line.61"> * @version $Id$</a> |
| <span class="sourceLineNo">062</span><a id="line.62"> */</a> |
| <span class="sourceLineNo">063</span><a id="line.63">public class UploadServiceTest extends BaseUnit5Test</a> |
| <span class="sourceLineNo">064</span><a id="line.64">{</a> |
| <span class="sourceLineNo">065</span><a id="line.65"> private UploadService uploadService = null;</a> |
| <span class="sourceLineNo">066</span><a id="line.66"></a> |
| <span class="sourceLineNo">067</span><a id="line.67"></a> |
| <span class="sourceLineNo">068</span><a id="line.68"> @BeforeEach</a> |
| <span class="sourceLineNo">069</span><a id="line.69"> public void setUp() throws Exception</a> |
| <span class="sourceLineNo">070</span><a id="line.70"> {</a> |
| <span class="sourceLineNo">071</span><a id="line.71"> try</a> |
| <span class="sourceLineNo">072</span><a id="line.72"> {</a> |
| <span class="sourceLineNo">073</span><a id="line.73"> uploadService = (UploadService) this.lookup(UploadService.ROLE);</a> |
| <span class="sourceLineNo">074</span><a id="line.74"> }</a> |
| <span class="sourceLineNo">075</span><a id="line.75"> catch (ComponentException e)</a> |
| <span class="sourceLineNo">076</span><a id="line.76"> {</a> |
| <span class="sourceLineNo">077</span><a id="line.77"> e.printStackTrace();</a> |
| <span class="sourceLineNo">078</span><a id="line.78"> fail(e.getMessage());</a> |
| <span class="sourceLineNo">079</span><a id="line.79"> }</a> |
| <span class="sourceLineNo">080</span><a id="line.80"> }</a> |
| <span class="sourceLineNo">081</span><a id="line.81"> /**</a> |
| <span class="sourceLineNo">082</span><a id="line.82"> * Simple test that verify an object can be created and deleted.</a> |
| <span class="sourceLineNo">083</span><a id="line.83"> * @throws Exception</a> |
| <span class="sourceLineNo">084</span><a id="line.84"> */</a> |
| <span class="sourceLineNo">085</span><a id="line.85"> @Test</a> |
| <span class="sourceLineNo">086</span><a id="line.86"> public void testRepositoryExists() throws Exception</a> |
| <span class="sourceLineNo">087</span><a id="line.87"> {</a> |
| <span class="sourceLineNo">088</span><a id="line.88"> File f = new File(uploadService.getRepository());</a> |
| <span class="sourceLineNo">089</span><a id="line.89"> assertTrue(f.exists());</a> |
| <span class="sourceLineNo">090</span><a id="line.90"> }</a> |
| <span class="sourceLineNo">091</span><a id="line.91"> </a> |
| <span class="sourceLineNo">092</span><a id="line.92"> // This is Apache Commons, but we want to make be really sure it runs clean in Fulcrum</a> |
| <span class="sourceLineNo">093</span><a id="line.93"> @Test</a> |
| <span class="sourceLineNo">094</span><a id="line.94"> public void testUploadEncoding() throws Exception {</a> |
| <span class="sourceLineNo">095</span><a id="line.95"> HttpServletRequest request = getMockRequest();</a> |
| <span class="sourceLineNo">096</span><a id="line.96"> when(request.getContentType()).thenReturn("multipart/form-data; boundary=boundary");</a> |
| <span class="sourceLineNo">097</span><a id="line.97"> when(request.getContentLength()).thenReturn(-1);// -1</a> |
| <span class="sourceLineNo">098</span><a id="line.98"> when(request.getMethod()).thenReturn("post");</a> |
| <span class="sourceLineNo">099</span><a id="line.99"> String testData= "Überfülle=\r\nf";</a> |
| <span class="sourceLineNo">100</span><a id="line.100"> //override default settings</a> |
| <span class="sourceLineNo">101</span><a id="line.101"> requestFormData( request, testData );</a> |
| <span class="sourceLineNo">102</span><a id="line.102"> assertTrue(uploadService.isMultipart( request ));</a> |
| <span class="sourceLineNo">103</span><a id="line.103"> List<FileItem> fil = uploadService.parseRequest( request );</a> |
| <span class="sourceLineNo">104</span><a id="line.104"> assertNotNull(fil);</a> |
| <span class="sourceLineNo">105</span><a id="line.105"> assertTrue( fil.size() >0);</a> |
| <span class="sourceLineNo">106</span><a id="line.106"> FileItem fi = fil.get( 0 );</a> |
| <span class="sourceLineNo">107</span><a id="line.107"> System.out.println( fi.getString() );</a> |
| <span class="sourceLineNo">108</span><a id="line.108"> assertEquals(15,fi.getSize());</a> |
| <span class="sourceLineNo">109</span><a id="line.109"> // default is ISO-8859-1</a> |
| <span class="sourceLineNo">110</span><a id="line.110"> assertTrue( fi.getString("UTF-8").startsWith( "Überfülle" ), "data string:'" +fi.getString("UTF-8") +"' not as expected");</a> |
| <span class="sourceLineNo">111</span><a id="line.111"> </a> |
| <span class="sourceLineNo">112</span><a id="line.112"> //reset inputstream</a> |
| <span class="sourceLineNo">113</span><a id="line.113"> requestFormData( request, testData);</a> |
| <span class="sourceLineNo">114</span><a id="line.114"> FileItemIterator fii = uploadService.getItemIterator( request );</a> |
| <span class="sourceLineNo">115</span><a id="line.115"> assertNotNull(fii);</a> |
| <span class="sourceLineNo">116</span><a id="line.116"> assertTrue( fii.hasNext());</a> |
| <span class="sourceLineNo">117</span><a id="line.117"> assertNotNull(fii.next());</a> |
| <span class="sourceLineNo">118</span><a id="line.118"> }</a> |
| <span class="sourceLineNo">119</span><a id="line.119"></a> |
| <span class="sourceLineNo">120</span><a id="line.120"> private void requestFormData( HttpServletRequest request, String data)</a> |
| <span class="sourceLineNo">121</span><a id="line.121"> throws IOException</a> |
| <span class="sourceLineNo">122</span><a id="line.122"> {</a> |
| <span class="sourceLineNo">123</span><a id="line.123"> String example ="--boundary\r\n"</a> |
| <span class="sourceLineNo">124</span><a id="line.124"> + "Content-Disposition: form-data; name=\"uploadedfile\"; filename=\"12345678.txt\"\r\n"</a> |
| <span class="sourceLineNo">125</span><a id="line.125"> + "Content-Type: text/plain\r\n"</a> |
| <span class="sourceLineNo">126</span><a id="line.126">// + "Content-Transfer-Encoding: UTF-8\r\n"</a> |
| <span class="sourceLineNo">127</span><a id="line.127"> + "\r\n"</a> |
| <span class="sourceLineNo">128</span><a id="line.128"> + data </a> |
| <span class="sourceLineNo">129</span><a id="line.129"> + "\r\n--boundary--\r\n";</a> |
| <span class="sourceLineNo">130</span><a id="line.130"> final ByteArrayInputStream is = new ByteArrayInputStream (example.getBytes());// "UTF-8"</a> |
| <span class="sourceLineNo">131</span><a id="line.131"> when(request.getInputStream()).thenReturn(new ServletInputStream() {</a> |
| <span class="sourceLineNo">132</span><a id="line.132"> @Override</a> |
| <span class="sourceLineNo">133</span><a id="line.133"> public int read() throws IOException {</a> |
| <span class="sourceLineNo">134</span><a id="line.134"> return is.read();</a> |
| <span class="sourceLineNo">135</span><a id="line.135"> }</a> |
| <span class="sourceLineNo">136</span><a id="line.136"></a> |
| <span class="sourceLineNo">137</span><a id="line.137"> @Override</a> |
| <span class="sourceLineNo">138</span><a id="line.138"> public boolean isFinished() {</a> |
| <span class="sourceLineNo">139</span><a id="line.139"> // TODO Auto-generated method stub</a> |
| <span class="sourceLineNo">140</span><a id="line.140"> return false;</a> |
| <span class="sourceLineNo">141</span><a id="line.141"> }</a> |
| <span class="sourceLineNo">142</span><a id="line.142"></a> |
| <span class="sourceLineNo">143</span><a id="line.143"> @Override</a> |
| <span class="sourceLineNo">144</span><a id="line.144"> public boolean isReady() {</a> |
| <span class="sourceLineNo">145</span><a id="line.145"> // TODO Auto-generated method stub</a> |
| <span class="sourceLineNo">146</span><a id="line.146"> return false;</a> |
| <span class="sourceLineNo">147</span><a id="line.147"> }</a> |
| <span class="sourceLineNo">148</span><a id="line.148"></a> |
| <span class="sourceLineNo">149</span><a id="line.149"> @Override</a> |
| <span class="sourceLineNo">150</span><a id="line.150"> public void setReadListener(ReadListener readListener) {</a> |
| <span class="sourceLineNo">151</span><a id="line.151"> // TODO Auto-generated method stub</a> |
| <span class="sourceLineNo">152</span><a id="line.152"> </a> |
| <span class="sourceLineNo">153</span><a id="line.153"> }</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"> </a> |
| <span class="sourceLineNo">157</span><a id="line.157"> protected Map<String,Object> attributes = new HashMap<String,Object>();</a> |
| <span class="sourceLineNo">158</span><a id="line.158"> protected int maxInactiveInterval = 0;</a> |
| <span class="sourceLineNo">159</span><a id="line.159"> // from Turbine org.apache.turbine.test.BaseTestCase, should be later in Fulcrum Testcontainer BaseUnit4Test</a> |
| <span class="sourceLineNo">160</span><a id="line.160"> protected HttpServletRequest getMockRequest()</a> |
| <span class="sourceLineNo">161</span><a id="line.161"> {</a> |
| <span class="sourceLineNo">162</span><a id="line.162"> HttpServletRequest request = mock(HttpServletRequest.class);</a> |
| <span class="sourceLineNo">163</span><a id="line.163"> HttpSession session = mock(HttpSession.class);</a> |
| <span class="sourceLineNo">164</span><a id="line.164"></a> |
| <span class="sourceLineNo">165</span><a id="line.165"> doAnswer(new Answer<Object>()</a> |
| <span class="sourceLineNo">166</span><a id="line.166"> {</a> |
| <span class="sourceLineNo">167</span><a id="line.167"> @Override</a> |
| <span class="sourceLineNo">168</span><a id="line.168"> public Object answer(InvocationOnMock invocation) throws Throwable</a> |
| <span class="sourceLineNo">169</span><a id="line.169"> {</a> |
| <span class="sourceLineNo">170</span><a id="line.170"> String key = (String) invocation.getArguments()[0];</a> |
| <span class="sourceLineNo">171</span><a id="line.171"> return attributes.get(key);</a> |
| <span class="sourceLineNo">172</span><a id="line.172"> }</a> |
| <span class="sourceLineNo">173</span><a id="line.173"> }).when(session).getAttribute(anyString());</a> |
| <span class="sourceLineNo">174</span><a id="line.174"></a> |
| <span class="sourceLineNo">175</span><a id="line.175"> doAnswer(new Answer<Object>()</a> |
| <span class="sourceLineNo">176</span><a id="line.176"> {</a> |
| <span class="sourceLineNo">177</span><a id="line.177"> @Override</a> |
| <span class="sourceLineNo">178</span><a id="line.178"> public Object answer(InvocationOnMock invocation) throws Throwable</a> |
| <span class="sourceLineNo">179</span><a id="line.179"> {</a> |
| <span class="sourceLineNo">180</span><a id="line.180"> String key = (String) invocation.getArguments()[0];</a> |
| <span class="sourceLineNo">181</span><a id="line.181"> Object value = invocation.getArguments()[1];</a> |
| <span class="sourceLineNo">182</span><a id="line.182"> attributes.put(key, value);</a> |
| <span class="sourceLineNo">183</span><a id="line.183"> return null;</a> |
| <span class="sourceLineNo">184</span><a id="line.184"> }</a> |
| <span class="sourceLineNo">185</span><a id="line.185"> }).when(session).setAttribute(anyString(), any());</a> |
| <span class="sourceLineNo">186</span><a id="line.186"></a> |
| <span class="sourceLineNo">187</span><a id="line.187"> when(session.getMaxInactiveInterval()).thenReturn(maxInactiveInterval);</a> |
| <span class="sourceLineNo">188</span><a id="line.188"></a> |
| <span class="sourceLineNo">189</span><a id="line.189"> doAnswer(new Answer<Integer>()</a> |
| <span class="sourceLineNo">190</span><a id="line.190"> {</a> |
| <span class="sourceLineNo">191</span><a id="line.191"> @Override</a> |
| <span class="sourceLineNo">192</span><a id="line.192"> public Integer answer(InvocationOnMock invocation) throws Throwable</a> |
| <span class="sourceLineNo">193</span><a id="line.193"> {</a> |
| <span class="sourceLineNo">194</span><a id="line.194"> return Integer.valueOf(maxInactiveInterval);</a> |
| <span class="sourceLineNo">195</span><a id="line.195"> }</a> |
| <span class="sourceLineNo">196</span><a id="line.196"> }).when(session).getMaxInactiveInterval();</a> |
| <span class="sourceLineNo">197</span><a id="line.197"></a> |
| <span class="sourceLineNo">198</span><a id="line.198"> doAnswer(new Answer<Object>()</a> |
| <span class="sourceLineNo">199</span><a id="line.199"> {</a> |
| <span class="sourceLineNo">200</span><a id="line.200"> @Override</a> |
| <span class="sourceLineNo">201</span><a id="line.201"> public Object answer(InvocationOnMock invocation) throws Throwable</a> |
| <span class="sourceLineNo">202</span><a id="line.202"> {</a> |
| <span class="sourceLineNo">203</span><a id="line.203"> Integer value = (Integer) invocation.getArguments()[0];</a> |
| <span class="sourceLineNo">204</span><a id="line.204"> maxInactiveInterval = value.intValue();</a> |
| <span class="sourceLineNo">205</span><a id="line.205"> return null;</a> |
| <span class="sourceLineNo">206</span><a id="line.206"> }</a> |
| <span class="sourceLineNo">207</span><a id="line.207"> }).when(session).setMaxInactiveInterval(anyInt());</a> |
| <span class="sourceLineNo">208</span><a id="line.208"></a> |
| <span class="sourceLineNo">209</span><a id="line.209"> when(session.isNew()).thenReturn(true);</a> |
| <span class="sourceLineNo">210</span><a id="line.210"> when(request.getSession()).thenReturn(session);</a> |
| <span class="sourceLineNo">211</span><a id="line.211"></a> |
| <span class="sourceLineNo">212</span><a id="line.212"> when(request.getServerName()).thenReturn("bob");</a> |
| <span class="sourceLineNo">213</span><a id="line.213"> when(request.getProtocol()).thenReturn("http");</a> |
| <span class="sourceLineNo">214</span><a id="line.214"> when(request.getScheme()).thenReturn("scheme");</a> |
| <span class="sourceLineNo">215</span><a id="line.215"> when(request.getPathInfo()).thenReturn("damn");</a> |
| <span class="sourceLineNo">216</span><a id="line.216"> when(request.getServletPath()).thenReturn("damn2");</a> |
| <span class="sourceLineNo">217</span><a id="line.217"> when(request.getContextPath()).thenReturn("wow");</a> |
| <span class="sourceLineNo">218</span><a id="line.218"> when(request.getContentType()).thenReturn("html/text");</a> |
| <span class="sourceLineNo">219</span><a id="line.219"></a> |
| <span class="sourceLineNo">220</span><a id="line.220"> when(request.getCharacterEncoding()).thenReturn("UTF-8");</a> |
| <span class="sourceLineNo">221</span><a id="line.221"> when(request.getServerPort()).thenReturn(8080);</a> |
| <span class="sourceLineNo">222</span><a id="line.222"> when(request.getLocale()).thenReturn(Locale.US);</a> |
| <span class="sourceLineNo">223</span><a id="line.223"></a> |
| <span class="sourceLineNo">224</span><a id="line.224"> when(request.getHeader("Content-type")).thenReturn("html/text");</a> |
| <span class="sourceLineNo">225</span><a id="line.225"> when(request.getHeader("Accept-Language")).thenReturn("en-US");</a> |
| <span class="sourceLineNo">226</span><a id="line.226"></a> |
| <span class="sourceLineNo">227</span><a id="line.227"> Vector<String> v = new Vector<String>();</a> |
| <span class="sourceLineNo">228</span><a id="line.228"> when(request.getParameterNames()).thenReturn(v.elements());</a> |
| <span class="sourceLineNo">229</span><a id="line.229"> return request;</a> |
| <span class="sourceLineNo">230</span><a id="line.230"> }</a> |
| <span class="sourceLineNo">231</span><a id="line.231">}</a> |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| </pre> |
| </div> |
| </main> |
| </body> |
| </html> |