~testplan-team/testplan/source-collection

<a name="14" href="#14">14</a> * <a href="http://www.apache.org/licenses/LICENSE-2.0" target="alexandria_uri">http://www.apache.org/licenses/LICENSE-2.0</a>

<a name="15" href="#15">15</a> *

<a name="16" href="#16">16</a> * Unless required by applicable law or agreed to in writing, software

<a name="17" href="#17">17</a> * distributed under the License is distributed on an "AS IS" BASIS,

<a name="18" href="#18">18</a> * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

<a name="19" href="#19">19</a> * See the License for the specific language governing permissions and

<a name="20" href="#20">20</a> * limitations under the License.

<a name="21" href="#21">21</a> * ====================================================================

<a name="22" href="#22">22</a> *

<a name="23" href="#23">23</a> * This software consists of voluntary contributions made by many

<a name="24" href="#24">24</a> * individuals on behalf of the Apache Software Foundation. For more

<a name="25" href="#25">25</a> * information on the Apache Software Foundation, please see

<a name="26" href="#26">26</a> * <<a href="http://www.apache.org/" target="alexandria_uri">http://www.apache.org/</a>>.

<a name="27" href="#27">27</a> *

<a name="28" href="#28">28</a> * [Additional notices, if required by prior licensing conditions]

<a name="29" href="#29">29</a> *

<a name="30" href="#30">30</a> */

<a name="32" href="#32">32</a> package org.apache.commons.httpclient;

<a name="34" href="#34">34</a> import java.io.IOException;

<a name="35" href="#35">35</a> import java.io.InputStream;

<a name="36" href="#36">36</a> import java.io.OutputStream;

<a name="37" href="#37">37</a> import java.net.InetAddress;

<a name="38" href="#38">38</a> import java.net.Socket;

<a name="39" href="#39">39</a> import java.net.UnknownHostException;

<a name="41" href="#41">41</a> import junit.framework.Test;

<a name="42" href="#42">42</a> import junit.framework.TestSuite;

<a name="44" href="#44">44</a> import org.apache.commons.httpclient.methods.GetMethod;

<a name="45" href="#45">45</a> import org.apache.commons.httpclient.params.HttpConnectionParams;

<a name="46" href="#46">46</a> import org.apache.commons.httpclient.protocol.Protocol;

<a name="47" href="#47">47</a> import org.apache.commons.httpclient.protocol.ProtocolSocketFactory;

<a name="48" href="#48">48</a> import org.apache.commons.httpclient.protocol.ControllerThreadSocketFactory;

<a name="50" href="#50">50</a> /**

<a name="51" href="#51">51</a> *

<a name="52" href="#52">52</a> * Unit tests for {@link HttpConnection}.

<a name="53" href="#53">53</a> *

<a name="54" href="#54">54</a> * @author Sean C. Sullivan

<a name="55" href="#55">55</a> *

<a name="56" href="#56">56</a> * @version $Id: TestHttpConnection.java 480424 2006-11-29 05:56:49Z bayard $

<a name="57" href="#57">57</a> *

<a name="58" href="#58">58</a> */

<a name="59" href="#59">59</a> public class <a href="../../../../org/apache/commons/httpclient/TestHttpConnection.html">TestHttpConnection</a> extends <a href="../../../../org/apache/commons/httpclient/HttpClientTestBase.html">HttpClientTestBase</a> {

<a name="61" href="#61">61</a> // ------------------------------------------------------------ Constructor

<a name="62" href="#62">62</a> public <a href="../../../../org/apache/commons/httpclient/TestHttpConnection.html">TestHttpConnection</a>(String testName) throws Exception {

<a name="63" href="#63">63</a> super(testName);

<a name="64" href="#64">64</a> }

<a name="66" href="#66">66</a> // ------------------------------------------------------------------- Main

<a name="67" href="#67">67</a> public static void main(String args[]) {

<a name="68" href="#68">68</a> String[] testCaseName = { TestHttpConnection.class.getName() };

<a name="69" href="#69">69</a> junit.textui.TestRunner.main(testCaseName);

<a name="70" href="#70">70</a> }

<a name="72" href="#72">72</a> // ------------------------------------------------------- TestCase Methods

<a name="74" href="#74">74</a> public static Test suite() {

<a name="75" href="#75">75</a> return new TestSuite(TestHttpConnection.class);

<a name="76" href="#76">76</a> }

<a name="79" href="#79">79</a> // ----------------------------------------------------------- Test Methods

<a name="81" href="#81">81</a> public void testConstructThenClose() {

<a name="82" href="#82">82</a> this.server.setHttpService(new <a href="../../../../org/apache/commons/httpclient/EchoService.html">EchoService</a>());

<a name="83" href="#83">83</a> HttpConnection conn = new HttpConnection(

<a name="84" href="#84">84</a> this.server.getLocalAddress(), this.server.getLocalPort());

<a name="85" href="#85">85</a> conn.close();

<a name="86" href="#86">86</a> assertTrue(!conn.isOpen());

<a name="87" href="#87">87</a> }

<a name="89" href="#89">89</a> public void testConnTimeoutRelease() {

<a name="90" href="#90">90</a> this.server.setHttpService(new <a href="../../../../org/apache/commons/httpclient/EchoService.html">EchoService</a>());

100

<a name="91" href="#91">91</a> // create a custom protocol that will delay for 500 milliseconds

101

<a name="92" href="#92">92</a> Protocol testProtocol = new Protocol(

102

<a name="93" href="#93">93</a> "timeout",

103

<a name="94" href="#94">94</a> new DelayedProtocolSocketFactory(

104

<a name="95" href="#95">95</a> 500,

105

<a name="96" href="#96">96</a> Protocol.getProtocol("http").getSocketFactory()

106

<a name="97" href="#97">97</a> ),

107

<a name="98" href="#98">98</a> this.server.getLocalPort()

108

<a name="99" href="#99">99</a> );

109

110

<a name="101" href="#101">101</a> <a href="../../../../org/apache/commons/httpclient/NoHostHttpConnectionManager.html">NoHostHttpConnectionManager</a> connectionManager = new <a href="../../../../org/apache/commons/httpclient/NoHostHttpConnectionManager.html">NoHostHttpConnectionManager</a>();

111

<a name="102" href="#102">102</a> connectionManager.setConnection(

112

<a name="103" href="#103">103</a> new HttpConnection(

113

<a name="104" href="#104">104</a> this.server.getLocalAddress(), this.server.getLocalPort(), testProtocol));

114

<a name="105" href="#105">105</a> this.client.setHttpConnectionManager(connectionManager);

115

<a name="106" href="#106">106</a> client.getHostConfiguration().setHost(

116

<a name="107" href="#107">107</a> this.server.getLocalAddress(), this.server.getLocalPort(), testProtocol);

117

<a name="108" href="#108">108</a> client.getHttpConnectionManager().getParams().setConnectionTimeout(1);

118

119

<a name="110" href="#110">110</a> try {

120

<a name="111" href="#111">111</a> GetMethod get = new GetMethod();

121

<a name="112" href="#112">112</a> client.executeMethod(get);

122

<a name="113" href="#113">113</a> fail("Should have timed out");

123

<a name="114" href="#114">114</a> } catch(IOException e) {

124

<a name="115" href="#115">115</a> /* should fail */

125

<a name="116" href="#116">116</a> assertTrue(e instanceof ConnectTimeoutException);

126

<a name="117" href="#117">117</a> assertTrue(connectionManager.isConnectionReleased());

127

<a name="118" href="#118">118</a> }

128

<a name="119" href="#119">119</a> }

129

130

131

<a name="122" href="#122">122</a> public void testConnTimeout() {

132

133

<a name="124" href="#124">124</a> // create a custom protocol that will delay for 500 milliseconds

134

<a name="125" href="#125">125</a> Protocol testProtocol = new Protocol(

135

<a name="126" href="#126">126</a> "timeout",

136

<a name="127" href="#127">127</a> new DelayedProtocolSocketFactory(

137

<a name="128" href="#128">128</a> 500,

138

<a name="129" href="#129">129</a> Protocol.getProtocol("http").getSocketFactory()

139

<a name="130" href="#130">130</a> ),

140

<a name="131" href="#131">131</a> this.server.getLocalPort()

141

<a name="132" href="#132">132</a> );

142

143

<a name="134" href="#134">134</a> HttpConnection conn = new HttpConnection(

144

<a name="135" href="#135">135</a> this.server.getLocalAddress(), this.server.getLocalPort(), testProtocol);

145

<a name="136" href="#136">136</a> // 1 ms is short enough to make this fail

146

<a name="137" href="#137">137</a> conn.getParams().setConnectionTimeout(1);

147

<a name="138" href="#138">138</a> try {

148

<a name="139" href="#139">139</a> conn.open();

149

<a name="140" href="#140">140</a> fail("Should have timed out");

150

<a name="141" href="#141">141</a> } catch(IOException e) {

151

<a name="142" href="#142">142</a> assertTrue(e instanceof ConnectTimeoutException);

152

<a name="143" href="#143">143</a> /* should fail */

153

<a name="144" href="#144">144</a> }

154

<a name="145" href="#145">145</a> }

155

156

<a name="147" href="#147">147</a> public void testForIllegalStateExceptions() {

157

<a name="148" href="#148">148</a> HttpConnection conn = new HttpConnection(

158

<a name="149" href="#149">149</a> this.server.getLocalAddress(), this.server.getLocalPort());

159

<a name="150" href="#150">150</a> try {

160

<a name="151" href="#151">151</a> OutputStream out = conn.getRequestOutputStream();

161

<a name="152" href="#152">152</a> fail("getRequestOutputStream did not throw the expected exception");

162

<a name="153" href="#153">153</a> }

163

<a name="154" href="#154">154</a> catch (IllegalStateException expected) {

164

<a name="155" href="#155">155</a> // this exception is expected

165

<a name="156" href="#156">156</a> }

166

<a name="157" href="#157">157</a> catch (IOException ex) {

167

<a name="158" href="#158">158</a> fail("getRequestOutputStream did not throw the expected exception");

168

<a name="159" href="#159">159</a> }

169

170

<a name="161" href="#161">161</a> try {

171

<a name="162" href="#162">162</a> OutputStream out = new ChunkedOutputStream(conn.getRequestOutputStream());

172

<a name="163" href="#163">163</a> fail("getRequestOutputStream(true) did not throw the expected exception");

173

<a name="164" href="#164">164</a> }

174

<a name="165" href="#165">165</a> catch (IllegalStateException expected) {

175

<a name="166" href="#166">166</a> // this exception is expected

176

<a name="167" href="#167">167</a> }

177

<a name="168" href="#168">168</a> catch (IOException ex) {

178

<a name="169" href="#169">169</a> fail("getRequestOutputStream(true) did not throw the expected exception");

179

<a name="170" href="#170">170</a> }

180

181

<a name="172" href="#172">172</a> try {

182

<a name="173" href="#173">173</a> InputStream in = conn.getResponseInputStream();

183

<a name="174" href="#174">174</a> fail("getResponseInputStream() did not throw the expected exception");

184

<a name="175" href="#175">175</a> }

185

<a name="176" href="#176">176</a> catch (IllegalStateException expected) {

186

<a name="177" href="#177">177</a> // this exception is expected

187

<a name="178" href="#178">178</a> }

188

<a name="179" href="#179">179</a> catch (IOException ex) {

189

<a name="180" href="#180">180</a> fail("getResponseInputStream() did not throw the expected exception");

190

<a name="181" href="#181">181</a> }

191

192

<a name="183" href="#183">183</a> }

193

194

<a name="185" href="#185">185</a> /**

195

<a name="186" href="#186">186</a> * A ProtocolSocketFactory that delays before creating a socket.

196

<a name="187" href="#187">187</a> */

197

<a name="188" href="#188">188</a> class DelayedProtocolSocketFactory implements ProtocolSocketFactory {

198

199

<a name="190" href="#190">190</a> private int delay;

200

<a name="191" href="#191">191</a> private ProtocolSocketFactory realFactory;

201

202

<a name="193" href="#193">193</a> public DelayedProtocolSocketFactory(int delay, ProtocolSocketFactory realFactory) {

203

<a name="194" href="#194">194</a> this.delay = delay;

204

<a name="195" href="#195">195</a> this.realFactory = realFactory;

205

<a name="196" href="#196">196</a> }

206

207

<a name="198" href="#198">198</a> public Socket createSocket(

208

<a name="199" href="#199">199</a> String host,

209

<a name="200" href="#200">200</a> int port,

210

<a name="201" href="#201">201</a> InetAddress localAddress,

211

<a name="202" href="#202">202</a> int localPort

212

<a name="203" href="#203">203</a> ) throws IOException, UnknownHostException {

213

214

<a name="205" href="#205">205</a> synchronized (this) {

215

<a name="206" href="#206">206</a> try {

216

<a name="207" href="#207">207</a> this.wait(delay);

217

<a name="208" href="#208">208</a> } catch (InterruptedException e) {}

218

<a name="209" href="#209">209</a> }

219

<a name="210" href="#210">210</a> return realFactory.createSocket(host, port, localAddress, localPort);

220

<a name="211" href="#211">211</a> }

221

222

<a name="213" href="#213">213</a> public Socket createSocket(

223

<a name="214" href="#214">214</a> final String host,

224

<a name="215" href="#215">215</a> final int port,

225

<a name="216" href="#216">216</a> final InetAddress localAddress,

226

<a name="217" href="#217">217</a> final int localPort,

227

<a name="218" href="#218">218</a> final HttpConnectionParams params

228

<a name="219" href="#219">219</a> ) throws IOException, UnknownHostException {

229

230

<a name="221" href="#221">221</a> if (params == null) {

231

<a name="222" href="#222">222</a> throw new IllegalArgumentException("Parameters may not be null");

232

<a name="223" href="#223">223</a> }

233

<a name="224" href="#224">224</a> int timeout = params.getConnectionTimeout();

234

<a name="225" href="#225">225</a> ControllerThreadSocketFactory.SocketTask task = new ControllerThreadSocketFactory.SocketTask() {

235

<a name="226" href="#226">226</a> public void doit() throws IOException {

236

<a name="227" href="#227">227</a> synchronized (this) {

237

<a name="228" href="#228">228</a> try {

238

<a name="229" href="#229">229</a> this.wait(delay);

239

<a name="230" href="#230">230</a> } catch (InterruptedException e) {}

240

<a name="231" href="#231">231</a> }

241

<a name="232" href="#232">232</a> setSocket(realFactory.createSocket(host, port, localAddress, localPort));

242

<a name="233" href="#233">233</a> }

243

<a name="234" href="#234">234</a> };

244

<a name="235" href="#235">235</a> return ControllerThreadSocketFactory.createSocket(task, timeout);

245

<a name="236" href="#236">236</a> }

246

247

<a name="238" href="#238">238</a> public Socket createSocket(String host, int port)

248

<a name="239" href="#239">239</a> throws IOException, UnknownHostException {

249

<a name="240" href="#240">240</a> synchronized (this) {

250

<a name="241" href="#241">241</a> try {

251

<a name="242" href="#242">242</a> this.wait(delay);

252

<a name="243" href="#243">243</a> } catch (InterruptedException e) {}

253

<a name="244" href="#244">244</a> }

254

<a name="245" href="#245">245</a> return realFactory.createSocket(host, port);

255

<a name="246" href="#246">246</a> }

256

257

<a name="248" href="#248">248</a> }

258

259

<a name="250" href="#250">250</a> }

260

261

</pre>

262

<hr/><div id="footer">This page was automatically generated by <a href="http://maven.apache.org/">Maven</a></div></body>

263

</html>

264

Older »