<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> <html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en" lang="en"> <head> <meta http-equiv="content-type" content="text/html; charset=ISO-8859-1" /><title>TestStreams xref</title> <link type="text/css" rel="stylesheet" href="../../../../stylesheet.css" /> </head> <body> <pre> <a name="1" href="#1">1</a> <em class="comment">/*</em> <a name="2" href="#2">2</a> <em class="comment"> * $Header: /home/jerenkrantz/tmp/commons/commons-convert/cvs/home/cvs/jakarta-commons//httpclient/src/test/org/apache/commons/httpclient/TestStreams.java,v 1.19 2004/10/31 14:04:13 olegk Exp $</em> <a name="3" href="#3">3</a> <em class="comment"> * $Revision: 155418 $</em> <a name="4" href="#4">4</a> <em class="comment"> * $Date: 2005-02-26 08:01:52 -0500 (Sat, 26 Feb 2005) $</em> <a name="5" href="#5">5</a> <em class="comment"> * ====================================================================</em> <a name="6" href="#6">6</a> <em class="comment"> *</em> <a name="7" href="#7">7</a> <em class="comment"> * Copyright 2002-2004 The Apache Software Foundation</em> <a name="8" href="#8">8</a> <em class="comment"> *</em> <a name="9" href="#9">9</a> <em class="comment"> * Licensed under the Apache License, Version 2.0 (the "License");</em> <a name="10" href="#10">10</a> <em class="comment"> * you may not use this file except in compliance with the License.</em> <a name="11" href="#11">11</a> <em class="comment"> * You may obtain a copy of the License at</em> <a name="12" href="#12">12</a> <em class="comment"> *</em> <a name="13" href="#13">13</a> <em class="comment"> * <a href="http://www.apache.org/licenses/LICENSE-2.0" target="alexandria_uri">http://www.apache.org/licenses/LICENSE-2.0</a></em> <a name="14" href="#14">14</a> <em class="comment"> *</em> <a name="15" href="#15">15</a> <em class="comment"> * Unless required by applicable law or agreed to in writing, software</em> <a name="16" href="#16">16</a> <em class="comment"> * distributed under the License is distributed on an "AS IS" BASIS,</em> <a name="17" href="#17">17</a> <em class="comment"> * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.</em> <a name="18" href="#18">18</a> <em class="comment"> * See the License for the specific language governing permissions and</em> <a name="19" href="#19">19</a> <em class="comment"> * limitations under the License.</em> <a name="20" href="#20">20</a> <em class="comment"> * ====================================================================</em> <a name="21" href="#21">21</a> <em class="comment"> *</em> <a name="22" href="#22">22</a> <em class="comment"> * This software consists of voluntary contributions made by many</em> <a name="23" href="#23">23</a> <em class="comment"> * individuals on behalf of the Apache Software Foundation. For more</em> <a name="24" href="#24">24</a> <em class="comment"> * information on the Apache Software Foundation, please see</em> <a name="25" href="#25">25</a> <em class="comment"> * <<a href="http://www.apache.org/" target="alexandria_uri">http://www.apache.org/</a>>.</em> <a name="26" href="#26">26</a> <em class="comment"> *</em> <a name="27" href="#27">27</a> <em class="comment"> * [Additional notices, if required by prior licensing conditions]</em> <a name="28" href="#28">28</a> <em class="comment"> *</em> <a name="29" href="#29">29</a> <em class="comment"> */</em> <a name="30" href="#30">30</a> <a name="31" href="#31">31</a> <strong>package</strong> <a href="../../../../org/apache/commons/httpclient/package-summary.html">org.apache.commons.httpclient</a>; <a name="32" href="#32">32</a> <a name="33" href="#33">33</a> <strong>import</strong> java.io.ByteArrayInputStream; <a name="34" href="#34">34</a> <strong>import</strong> java.io.ByteArrayOutputStream; <a name="35" href="#35">35</a> <strong>import</strong> java.io.IOException; <a name="36" href="#36">36</a> <strong>import</strong> java.io.InputStream; <a name="37" href="#37">37</a> <strong>import</strong> java.io.OutputStream; <a name="38" href="#38">38</a> <a name="39" href="#39">39</a> <strong>import</strong> junit.framework.Test; <a name="40" href="#40">40</a> <strong>import</strong> junit.framework.TestCase; <a name="41" href="#41">41</a> <strong>import</strong> junit.framework.TestSuite; <a name="42" href="#42">42</a> <a name="43" href="#43">43</a> <strong>import</strong> org.apache.commons.httpclient.methods.GetMethod; <a name="44" href="#44">44</a> <strong>import</strong> org.apache.commons.httpclient.util.EncodingUtil; <a name="45" href="#45">45</a> <a name="46" href="#46">46</a> <a name="47" href="#47">47</a> <strong>public</strong> <strong>class</strong> <a href="../../../../org/apache/commons/httpclient/TestStreams.html">TestStreams</a> <strong>extends</strong> TestCase { <a name="48" href="#48">48</a> <a name="49" href="#49">49</a> <strong>private</strong> <strong>static</strong> <strong>final</strong> String CONTENT_CHARSET = <span class="string">"ISO-8859-1"</span>; <a name="50" href="#50">50</a> <a name="51" href="#51">51</a> <strong>public</strong> <a href="../../../../org/apache/commons/httpclient/TestStreams.html">TestStreams</a>(String testName) { <a name="52" href="#52">52</a> <strong>super</strong>(testName); <a name="53" href="#53">53</a> } <a name="54" href="#54">54</a> <a name="55" href="#55">55</a> <strong>public</strong> <strong>void</strong> testChunkedInputStream() throws IOException { <a name="56" href="#56">56</a> String correctInput = <span class="string">"10;key=\"value\r\nnewline\"\r\n1234567890123456\r\n5\r\n12345\r\n0\r\nFooter1: abcde\r\nFooter2: fghij\r\n"</span>; <a name="57" href="#57">57</a> String correctResult = <span class="string">"123456789012345612345"</span>; <a name="58" href="#58">58</a> HttpMethod method = <strong>new</strong> <a href="../../../../org/apache/commons/httpclient/FakeHttpMethod.html">FakeHttpMethod</a>(); <a name="59" href="#59">59</a> <a name="60" href="#60">60</a> <em class="comment">//Test for when buffer is larger than chunk size</em> <a name="61" href="#61">61</a> InputStream in = <strong>new</strong> ChunkedInputStream(<strong>new</strong> ByteArrayInputStream( <a name="62" href="#62">62</a> EncodingUtil.getBytes(correctInput, CONTENT_CHARSET)), method); <a name="63" href="#63">63</a> byte[] buffer = <strong>new</strong> byte[300]; <a name="64" href="#64">64</a> ByteArrayOutputStream out = <strong>new</strong> ByteArrayOutputStream(); <a name="65" href="#65">65</a> <strong>int</strong> len; <a name="66" href="#66">66</a> <strong>while</strong> ((len = in.read(buffer)) > 0) { <a name="67" href="#67">67</a> out.write(buffer, 0, len); <a name="68" href="#68">68</a> } <a name="69" href="#69">69</a> String result = EncodingUtil.getString(out.toByteArray(), CONTENT_CHARSET); <a name="70" href="#70">70</a> assertEquals(result, correctResult); <a name="71" href="#71">71</a> Header footer = method.getResponseFooter(<span class="string">"footer1"</span>); <a name="72" href="#72">72</a> assertEquals(footer.getValue(), <span class="string">"abcde"</span>); <a name="73" href="#73">73</a> footer = method.getResponseFooter(<span class="string">"footer2"</span>); <a name="74" href="#74">74</a> assertEquals(footer.getValue(), <span class="string">"fghij"</span>); <a name="75" href="#75">75</a> <a name="76" href="#76">76</a> method = <strong>new</strong> <a href="../../../../org/apache/commons/httpclient/FakeHttpMethod.html">FakeHttpMethod</a>(); <a name="77" href="#77">77</a> <a name="78" href="#78">78</a> <em class="comment">//Test for when buffer is smaller than chunk size.</em> <a name="79" href="#79">79</a> in = <strong>new</strong> ChunkedInputStream(<strong>new</strong> ByteArrayInputStream( <a name="80" href="#80">80</a> EncodingUtil.getBytes(correctInput, CONTENT_CHARSET)), method); <a name="81" href="#81">81</a> buffer = <strong>new</strong> byte[7]; <a name="82" href="#82">82</a> out = <strong>new</strong> ByteArrayOutputStream(); <a name="83" href="#83">83</a> <strong>while</strong> ((len = in.read(buffer)) > 0) { <a name="84" href="#84">84</a> out.write(buffer, 0, len); <a name="85" href="#85">85</a> } <a name="86" href="#86">86</a> result = EncodingUtil.getString(out.toByteArray(), CONTENT_CHARSET); <a name="87" href="#87">87</a> assertEquals(result, correctResult); <a name="88" href="#88">88</a> footer = method.getResponseFooter(<span class="string">"footer1"</span>); <a name="89" href="#89">89</a> assertEquals(footer.getValue(), <span class="string">"abcde"</span>); <a name="90" href="#90">90</a> footer = method.getResponseFooter(<span class="string">"footer2"</span>); <a name="91" href="#91">91</a> assertEquals(footer.getValue(), <span class="string">"fghij"</span>); <a name="92" href="#92">92</a> } <a name="93" href="#93">93</a> <a name="94" href="#94">94</a> <strong>public</strong> <strong>void</strong> testCorruptChunkedInputStream1() throws IOException { <a name="95" href="#95">95</a> <em class="comment">//missing \r\n at the end of the first chunk</em> <a name="96" href="#96">96</a> String corrupInput = <span class="string">"10;key=\"value\"\r\n123456789012345\r\n5\r\n12345\r\n0\r\nFooter1: abcde\r\nFooter2: fghij\r\n"</span>; <a name="97" href="#97">97</a> HttpMethod method = <strong>new</strong> <a href="../../../../org/apache/commons/httpclient/FakeHttpMethod.html">FakeHttpMethod</a>(); <a name="98" href="#98">98</a> <a name="99" href="#99">99</a> InputStream in = <strong>new</strong> ChunkedInputStream(<strong>new</strong> ByteArrayInputStream( <a name="100" href="#100">100</a> EncodingUtil.getBytes(corrupInput, CONTENT_CHARSET)), method); <a name="101" href="#101">101</a> byte[] buffer = <strong>new</strong> byte[300]; <a name="102" href="#102">102</a> ByteArrayOutputStream out = <strong>new</strong> ByteArrayOutputStream(); <a name="103" href="#103">103</a> <strong>int</strong> len; <a name="104" href="#104">104</a> <strong>try</strong> { <a name="105" href="#105">105</a> <strong>while</strong> ((len = in.read(buffer)) > 0) { <a name="106" href="#106">106</a> out.write(buffer, 0, len); <a name="107" href="#107">107</a> } <a name="108" href="#108">108</a> fail(<span class="string">"Should have thrown exception"</span>); <a name="109" href="#109">109</a> } <strong>catch</strong>(IOException e) { <a name="110" href="#110">110</a> <em class="comment">/*<em class="comment"> expected exception */</em></em> <a name="111" href="#111">111</a> } <a name="112" href="#112">112</a> } <a name="113" href="#113">113</a> <a name="114" href="#114">114</a> <strong>public</strong> <strong>void</strong> testEmptyChunkedInputStream() throws IOException { <a name="115" href="#115">115</a> String input = <span class="string">"0\r\n"</span>; <a name="116" href="#116">116</a> HttpMethod method = <strong>new</strong> <a href="../../../../org/apache/commons/httpclient/FakeHttpMethod.html">FakeHttpMethod</a>(); <a name="117" href="#117">117</a> <a name="118" href="#118">118</a> InputStream in = <strong>new</strong> ChunkedInputStream(<strong>new</strong> ByteArrayInputStream( <a name="119" href="#119">119</a> EncodingUtil.getBytes(input, CONTENT_CHARSET)), method); <a name="120" href="#120">120</a> byte[] buffer = <strong>new</strong> byte[300]; <a name="121" href="#121">121</a> ByteArrayOutputStream out = <strong>new</strong> ByteArrayOutputStream(); <a name="122" href="#122">122</a> <strong>int</strong> len; <a name="123" href="#123">123</a> <strong>while</strong> ((len = in.read(buffer)) > 0) { <a name="124" href="#124">124</a> out.write(buffer, 0, len); <a name="125" href="#125">125</a> } <a name="126" href="#126">126</a> assertEquals(0, out.size()); <a name="127" href="#127">127</a> } <a name="128" href="#128">128</a> <a name="129" href="#129">129</a> <strong>public</strong> <strong>void</strong> testContentLengthInputStream() throws IOException { <a name="130" href="#130">130</a> String correct = <span class="string">"1234567890123456"</span>; <a name="131" href="#131">131</a> InputStream in = <strong>new</strong> ContentLengthInputStream(<strong>new</strong> ByteArrayInputStream( <a name="132" href="#132">132</a> EncodingUtil.getBytes(correct, CONTENT_CHARSET)), 10L); <a name="133" href="#133">133</a> byte[] buffer = <strong>new</strong> byte[50]; <a name="134" href="#134">134</a> <strong>int</strong> len = in.read(buffer); <a name="135" href="#135">135</a> ByteArrayOutputStream out = <strong>new</strong> ByteArrayOutputStream(); <a name="136" href="#136">136</a> out.write(buffer, 0, len); <a name="137" href="#137">137</a> String result = EncodingUtil.getString(out.toByteArray(), CONTENT_CHARSET); <a name="138" href="#138">138</a> assertEquals(result, <span class="string">"1234567890"</span>); <a name="139" href="#139">139</a> } <a name="140" href="#140">140</a> <a name="141" href="#141">141</a> <strong>public</strong> <strong>void</strong> testContentLengthInputStreamSkip() throws IOException { <a name="142" href="#142">142</a> InputStream in = <strong>new</strong> ContentLengthInputStream(<strong>new</strong> ByteArrayInputStream(<strong>new</strong> byte[20]), 10L); <a name="143" href="#143">143</a> assertEquals(10, in.skip(10)); <a name="144" href="#144">144</a> assertTrue(in.read() == -1); <a name="145" href="#145">145</a> <a name="146" href="#146">146</a> in = <strong>new</strong> ContentLengthInputStream(<strong>new</strong> ByteArrayInputStream(<strong>new</strong> byte[20]), 10L); <a name="147" href="#147">147</a> in.read(); <a name="148" href="#148">148</a> assertEquals(9, in.skip(10)); <a name="149" href="#149">149</a> assertTrue(in.read() == -1); <a name="150" href="#150">150</a> <a name="151" href="#151">151</a> in = <strong>new</strong> ContentLengthInputStream(<strong>new</strong> ByteArrayInputStream(<strong>new</strong> byte[20]), 2L); <a name="152" href="#152">152</a> in.read(); <a name="153" href="#153">153</a> in.read(); <a name="154" href="#154">154</a> assertTrue(in.skip(10) <= 0); <a name="155" href="#155">155</a> assertTrue(in.read() == -1); <a name="156" href="#156">156</a> } <a name="157" href="#157">157</a> <a name="158" href="#158">158</a> <strong>public</strong> <strong>void</strong> testChunkedConsitance() throws IOException { <a name="159" href="#159">159</a> String input = <span class="string">"76126;27823abcd;:q38a-\nkjc\rk%1ad\tkh/asdui\r\njkh+?//suweb"</span>; <a name="160" href="#160">160</a> ByteArrayOutputStream buffer = <strong>new</strong> ByteArrayOutputStream(); <a name="161" href="#161">161</a> OutputStream out = <strong>new</strong> ChunkedOutputStream(buffer); <a name="162" href="#162">162</a> out.write(EncodingUtil.getBytes(input, CONTENT_CHARSET)); <a name="163" href="#163">163</a> out.close(); <a name="164" href="#164">164</a> buffer.close(); <a name="165" href="#165">165</a> InputStream in = <strong>new</strong> ChunkedInputStream(<strong>new</strong> ByteArrayInputStream(buffer.toByteArray()), <strong>new</strong> GetMethod()); <a name="166" href="#166">166</a> <a name="167" href="#167">167</a> byte[] d = <strong>new</strong> byte[10]; <a name="168" href="#168">168</a> ByteArrayOutputStream result = <strong>new</strong> ByteArrayOutputStream(); <a name="169" href="#169">169</a> <strong>int</strong> len = 0; <a name="170" href="#170">170</a> <strong>while</strong> ((len = in.read(d)) > 0) { <a name="171" href="#171">171</a> result.write(d, 0, len); <a name="172" href="#172">172</a> } <a name="173" href="#173">173</a> <a name="174" href="#174">174</a> String output = EncodingUtil.getString(result.toByteArray(), CONTENT_CHARSET); <a name="175" href="#175">175</a> assertEquals(input, output); <a name="176" href="#176">176</a> } <a name="177" href="#177">177</a> <a name="178" href="#178">178</a> <strong>public</strong> <strong>void</strong> testChunkedOutputStream() throws IOException { <a name="179" href="#179">179</a> ByteArrayOutputStream buffer = <strong>new</strong> ByteArrayOutputStream(); <a name="180" href="#180">180</a> ChunkedOutputStream out = <strong>new</strong> ChunkedOutputStream(buffer, 2); <a name="181" href="#181">181</a> out.write('1'); <a name="182" href="#182">182</a> out.write('2'); <a name="183" href="#183">183</a> out.write('3'); <a name="184" href="#184">184</a> out.write('4'); <a name="185" href="#185">185</a> out.finish(); <a name="186" href="#186">186</a> out.close(); <a name="187" href="#187">187</a> <a name="188" href="#188">188</a> byte [] rawdata = buffer.toByteArray(); <a name="189" href="#189">189</a> <a name="190" href="#190">190</a> assertEquals(19, rawdata.length); <a name="191" href="#191">191</a> assertEquals('2', rawdata[0]); <a name="192" href="#192">192</a> assertEquals('\r', rawdata[1]); <a name="193" href="#193">193</a> assertEquals('\n', rawdata[2]); <a name="194" href="#194">194</a> assertEquals('1', rawdata[3]); <a name="195" href="#195">195</a> assertEquals('2', rawdata[4]); <a name="196" href="#196">196</a> assertEquals('\r', rawdata[5]); <a name="197" href="#197">197</a> assertEquals('\n', rawdata[6]); <a name="198" href="#198">198</a> assertEquals('2', rawdata[7]); <a name="199" href="#199">199</a> assertEquals('\r', rawdata[8]); <a name="200" href="#200">200</a> assertEquals('\n', rawdata[9]); <a name="201" href="#201">201</a> assertEquals('3', rawdata[10]); <a name="202" href="#202">202</a> assertEquals('4', rawdata[11]); <a name="203" href="#203">203</a> assertEquals('\r', rawdata[12]); <a name="204" href="#204">204</a> assertEquals('\n', rawdata[13]); <a name="205" href="#205">205</a> assertEquals('0', rawdata[14]); <a name="206" href="#206">206</a> assertEquals('\r', rawdata[15]); <a name="207" href="#207">207</a> assertEquals('\n', rawdata[16]); <a name="208" href="#208">208</a> assertEquals('\r', rawdata[17]); <a name="209" href="#209">209</a> assertEquals('\n', rawdata[18]); <a name="210" href="#210">210</a> } <a name="211" href="#211">211</a> <a name="212" href="#212">212</a> <strong>public</strong> <strong>void</strong> testChunkedOutputStreamLargeChunk() throws IOException { <a name="213" href="#213">213</a> ByteArrayOutputStream buffer = <strong>new</strong> ByteArrayOutputStream(); <a name="214" href="#214">214</a> ChunkedOutputStream out = <strong>new</strong> ChunkedOutputStream(buffer, 2); <a name="215" href="#215">215</a> out.write(<strong>new</strong> byte[] {'1', '2', '3', '4'}); <a name="216" href="#216">216</a> out.finish(); <a name="217" href="#217">217</a> out.close(); <a name="218" href="#218">218</a> <a name="219" href="#219">219</a> byte [] rawdata = buffer.toByteArray(); <a name="220" href="#220">220</a> <a name="221" href="#221">221</a> assertEquals(14, rawdata.length); <a name="222" href="#222">222</a> assertEquals('4', rawdata[0]); <a name="223" href="#223">223</a> assertEquals('\r', rawdata[1]); <a name="224" href="#224">224</a> assertEquals('\n', rawdata[2]); <a name="225" href="#225">225</a> assertEquals('1', rawdata[3]); <a name="226" href="#226">226</a> assertEquals('2', rawdata[4]); <a name="227" href="#227">227</a> assertEquals('3', rawdata[5]); <a name="228" href="#228">228</a> assertEquals('4', rawdata[6]); <a name="229" href="#229">229</a> assertEquals('\r', rawdata[7]); <a name="230" href="#230">230</a> assertEquals('\n', rawdata[8]); <a name="231" href="#231">231</a> assertEquals('0', rawdata[9]); <a name="232" href="#232">232</a> assertEquals('\r', rawdata[10]); <a name="233" href="#233">233</a> assertEquals('\n', rawdata[11]); <a name="234" href="#234">234</a> assertEquals('\r', rawdata[12]); <a name="235" href="#235">235</a> assertEquals('\n', rawdata[13]); <a name="236" href="#236">236</a> } <a name="237" href="#237">237</a> <a name="238" href="#238">238</a> <strong>public</strong> <strong>void</strong> testChunkedOutputStreamSmallChunk() throws IOException { <a name="239" href="#239">239</a> ByteArrayOutputStream buffer = <strong>new</strong> ByteArrayOutputStream(); <a name="240" href="#240">240</a> ChunkedOutputStream out = <strong>new</strong> ChunkedOutputStream(buffer, 2); <a name="241" href="#241">241</a> out.write('1'); <a name="242" href="#242">242</a> out.finish(); <a name="243" href="#243">243</a> out.close(); <a name="244" href="#244">244</a> <a name="245" href="#245">245</a> byte [] rawdata = buffer.toByteArray(); <a name="246" href="#246">246</a> <a name="247" href="#247">247</a> assertEquals(11, rawdata.length); <a name="248" href="#248">248</a> assertEquals('1', rawdata[0]); <a name="249" href="#249">249</a> assertEquals('\r', rawdata[1]); <a name="250" href="#250">250</a> assertEquals('\n', rawdata[2]); <a name="251" href="#251">251</a> assertEquals('1', rawdata[3]); <a name="252" href="#252">252</a> assertEquals('\r', rawdata[4]); <a name="253" href="#253">253</a> assertEquals('\n', rawdata[5]); <a name="254" href="#254">254</a> assertEquals('0', rawdata[6]); <a name="255" href="#255">255</a> assertEquals('\r', rawdata[7]); <a name="256" href="#256">256</a> assertEquals('\n', rawdata[8]); <a name="257" href="#257">257</a> assertEquals('\r', rawdata[9]); <a name="258" href="#258">258</a> assertEquals('\n', rawdata[10]); <a name="259" href="#259">259</a> } <a name="260" href="#260">260</a> <a name="261" href="#261">261</a> <em class="comment">// ------------------------------------------------------- TestCase Methods</em> <a name="262" href="#262">262</a> <a name="263" href="#263">263</a> <strong>public</strong> <strong>static</strong> Test suite() { <a name="264" href="#264">264</a> <strong>return</strong> <strong>new</strong> TestSuite(TestStreams.<strong>class</strong>); <a name="265" href="#265">265</a> } <a name="266" href="#266">266</a> <a name="267" href="#267">267</a> <em class="comment">// ------------------------------------------------------------------- Main</em> <a name="268" href="#268">268</a> <strong>public</strong> <strong>static</strong> <strong>void</strong> main(String args[]) { <a name="269" href="#269">269</a> String[] testCaseName = { TestStreams.<strong>class</strong>.getName() }; <a name="270" href="#270">270</a> junit.textui.TestRunner.main(testCaseName); <a name="271" href="#271">271</a> } <a name="272" href="#272">272</a> } <a name="273" href="#273">273</a> </pre> <hr/><div id="footer">This page was automatically generated by <a href="http://maven.apache.org/">Maven</a></div></body> </html>