streamwritefiltertest.html

MINA+API 关于MINA的API

<a name="299" href="#299">299</a> 
<a name="300" href="#300">300</a>         <em class="jxr_comment">/*</em>
<a name="301" href="#301">301</a> <em class="jxr_comment">         * Replay.</em>
<a name="302" href="#302">302</a> <em class="jxr_comment">         */</em>
<a name="303" href="#303">303</a>         mockNextFilter.replay();
<a name="304" href="#304">304</a> 
<a name="305" href="#305">305</a>         filter.messageSent(nextFilter, session, <strong class="jxr_keyword">new</strong> DefaultWriteRequest(
<a name="306" href="#306">306</a>                 <strong class="jxr_keyword">new</strong> Object()));
<a name="307" href="#307">307</a>         assertEquals(0, queue.size());
<a name="308" href="#308">308</a> 
<a name="309" href="#309">309</a>         <em class="jxr_comment">/*</em>
<a name="310" href="#310">310</a> <em class="jxr_comment">         * Verify.</em>
<a name="311" href="#311">311</a> <em class="jxr_comment">         */</em>
<a name="312" href="#312">312</a>         mockNextFilter.verify();
<a name="313" href="#313">313</a>     }
<a name="314" href="#314">314</a> 
<a name="315" href="#315">315</a>     <em class="jxr_javadoccomment">/**</em>
<a name="316" href="#316">316</a> <em class="jxr_javadoccomment">     * Tests that {@link StreamWriteFilter#setWriteBufferSize(int)} checks the</em>
<a name="317" href="#317">317</a> <em class="jxr_javadoccomment">     * specified size.</em>
<a name="318" href="#318">318</a> <em class="jxr_javadoccomment">     */</em>
<a name="319" href="#319">319</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> testSetWriteBufferSize() {
<a name="320" href="#320">320</a>         StreamWriteFilter filter = <strong class="jxr_keyword">new</strong> StreamWriteFilter();
<a name="321" href="#321">321</a> 
<a name="322" href="#322">322</a>         <strong class="jxr_keyword">try</strong> {
<a name="323" href="#323">323</a>             filter.setWriteBufferSize(0);
<a name="324" href="#324">324</a>             fail(<span class="jxr_string">"0 writeBuferSize specified. IllegalArgumentException expected."</span>);
<a name="325" href="#325">325</a>         } <strong class="jxr_keyword">catch</strong> (IllegalArgumentException iae) {
<a name="326" href="#326">326</a>         }
<a name="327" href="#327">327</a> 
<a name="328" href="#328">328</a>         <strong class="jxr_keyword">try</strong> {
<a name="329" href="#329">329</a>             filter.setWriteBufferSize(-100);
<a name="330" href="#330">330</a>             fail(<span class="jxr_string">"Negative writeBuferSize specified. IllegalArgumentException expected."</span>);
<a name="331" href="#331">331</a>         } <strong class="jxr_keyword">catch</strong> (IllegalArgumentException iae) {
<a name="332" href="#332">332</a>         }
<a name="333" href="#333">333</a> 
<a name="334" href="#334">334</a>         filter.setWriteBufferSize(1);
<a name="335" href="#335">335</a>         assertEquals(1, filter.getWriteBufferSize());
<a name="336" href="#336">336</a>         filter.setWriteBufferSize(1024);
<a name="337" href="#337">337</a>         assertEquals(1024, filter.getWriteBufferSize());
<a name="338" href="#338">338</a>     }
<a name="339" href="#339">339</a> 
<a name="340" href="#340">340</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> testWriteUsingSocketTransport() <strong class="jxr_keyword">throws</strong> Exception {
<a name="341" href="#341">341</a>         NioSocketAcceptor acceptor = <strong class="jxr_keyword">new</strong> NioSocketAcceptor();
<a name="342" href="#342">342</a>         acceptor.setReuseAddress(<strong class="jxr_keyword">true</strong>);
<a name="343" href="#343">343</a>         SocketAddress address = <strong class="jxr_keyword">new</strong> InetSocketAddress(<span class="jxr_string">"localhost"</span>,
<a name="344" href="#344">344</a>                 AvailablePortFinder.getNextAvailable());
<a name="345" href="#345">345</a> 
<a name="346" href="#346">346</a>         NioSocketConnector connector = <strong class="jxr_keyword">new</strong> NioSocketConnector();
<a name="347" href="#347">347</a> 
<a name="348" href="#348">348</a>         <a href="../../../../../org/apache/mina/filter/stream/StreamWriteFilterTest.html">FixedRandomInputStream</a> stream = <strong class="jxr_keyword">new</strong> <a href="../../../../../org/apache/mina/filter/stream/StreamWriteFilterTest.html">FixedRandomInputStream</a>(
<a name="349" href="#349">349</a>                 4 * 1024 * 1024);
<a name="350" href="#350">350</a> 
<a name="351" href="#351">351</a>         <a href="../../../../../org/apache/mina/filter/stream/StreamWriteFilterTest.html">SenderHandler</a> sender = <strong class="jxr_keyword">new</strong> <a href="../../../../../org/apache/mina/filter/stream/StreamWriteFilterTest.html">SenderHandler</a>(stream);
<a name="352" href="#352">352</a>         <a href="../../../../../org/apache/mina/filter/stream/StreamWriteFilterTest.html">ReceiverHandler</a> receiver = <strong class="jxr_keyword">new</strong> <a href="../../../../../org/apache/mina/filter/stream/StreamWriteFilterTest.html">ReceiverHandler</a>(stream.size);
<a name="353" href="#353">353</a> 
<a name="354" href="#354">354</a>         acceptor.setHandler(sender);
<a name="355" href="#355">355</a>         connector.setHandler(receiver);
<a name="356" href="#356">356</a>         
<a name="357" href="#357">357</a>         acceptor.bind(address);
<a name="358" href="#358">358</a>         connector.connect(address);
<a name="359" href="#359">359</a>         sender.latch.await();
<a name="360" href="#360">360</a>         receiver.latch.await();
<a name="361" href="#361">361</a> 
<a name="362" href="#362">362</a>         acceptor.dispose();
<a name="363" href="#363">363</a> 
<a name="364" href="#364">364</a>         assertEquals(stream.bytesRead, receiver.bytesRead);
<a name="365" href="#365">365</a>         assertEquals(stream.size, receiver.bytesRead);
<a name="366" href="#366">366</a>         byte[] expectedMd5 = stream.digest.digest();
<a name="367" href="#367">367</a>         byte[] actualMd5 = receiver.digest.digest();
<a name="368" href="#368">368</a>         assertEquals(expectedMd5.length, actualMd5.length);
<a name="369" href="#369">369</a>         <strong class="jxr_keyword">for</strong> (<strong class="jxr_keyword">int</strong> i = 0; i &lt; expectedMd5.length; i++) {
<a name="370" href="#370">370</a>             assertEquals(expectedMd5[i], actualMd5[i]);
<a name="371" href="#371">371</a>         }
<a name="372" href="#372">372</a>     }
<a name="373" href="#373">373</a> 
<a name="374" href="#374">374</a>     <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">class</strong> <a href="../../../../../org/apache/mina/filter/stream/StreamWriteFilterTest.html">FixedRandomInputStream</a> <strong class="jxr_keyword">extends</strong> InputStream {
<a name="375" href="#375">375</a>         <strong class="jxr_keyword">long</strong> size;
<a name="376" href="#376">376</a> 
<a name="377" href="#377">377</a>         <strong class="jxr_keyword">long</strong> bytesRead = 0;
<a name="378" href="#378">378</a> 
<a name="379" href="#379">379</a>         Random random = <strong class="jxr_keyword">new</strong> Random();
<a name="380" href="#380">380</a> 
<a name="381" href="#381">381</a>         MessageDigest digest;
<a name="382" href="#382">382</a> 
<a name="383" href="#383">383</a>         <strong class="jxr_keyword">private</strong> <a href="../../../../../org/apache/mina/filter/stream/StreamWriteFilterTest.html">FixedRandomInputStream</a>(<strong class="jxr_keyword">long</strong> size) <strong class="jxr_keyword">throws</strong> Exception {
<a name="384" href="#384">384</a>             <strong class="jxr_keyword">this</strong>.size = size;
<a name="385" href="#385">385</a>             digest = MessageDigest.getInstance(<span class="jxr_string">"MD5"</span>);
<a name="386" href="#386">386</a>         }
<a name="387" href="#387">387</a> 
<a name="388" href="#388">388</a>         @Override
<a name="389" href="#389">389</a>         <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">int</strong> read() <strong class="jxr_keyword">throws</strong> IOException {
<a name="390" href="#390">390</a>             <strong class="jxr_keyword">if</strong> (isAllWritten()) {
<a name="391" href="#391">391</a>                 <strong class="jxr_keyword">return</strong> -1;
<a name="392" href="#392">392</a>             }
<a name="393" href="#393">393</a>             bytesRead++;
<a name="394" href="#394">394</a>             byte b = (byte) random.nextInt(255);
<a name="395" href="#395">395</a>             digest.update(b);
<a name="396" href="#396">396</a>             <strong class="jxr_keyword">return</strong> b;
<a name="397" href="#397">397</a>         }
<a name="398" href="#398">398</a> 
<a name="399" href="#399">399</a>         <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">long</strong> getBytesRead() {
<a name="400" href="#400">400</a>             <strong class="jxr_keyword">return</strong> bytesRead;
<a name="401" href="#401">401</a>         }
<a name="402" href="#402">402</a> 
<a name="403" href="#403">403</a>         <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">long</strong> getSize() {
<a name="404" href="#404">404</a>             <strong class="jxr_keyword">return</strong> size;
<a name="405" href="#405">405</a>         }
<a name="406" href="#406">406</a> 
<a name="407" href="#407">407</a>         <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">boolean</strong> isAllWritten() {
<a name="408" href="#408">408</a>             <strong class="jxr_keyword">return</strong> bytesRead &gt;= size;
<a name="409" href="#409">409</a>         }
<a name="410" href="#410">410</a>     }
<a name="411" href="#411">411</a> 
<a name="412" href="#412">412</a>     <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">class</strong> <a href="../../../../../org/apache/mina/filter/stream/StreamWriteFilterTest.html">SenderHandler</a> <strong class="jxr_keyword">extends</strong> IoHandlerAdapter {
<a name="413" href="#413">413</a>         <strong class="jxr_keyword">final</strong> CountDownLatch latch = <strong class="jxr_keyword">new</strong> CountDownLatch(1);
<a name="414" href="#414">414</a> 
<a name="415" href="#415">415</a>         InputStream inputStream;
<a name="416" href="#416">416</a> 
<a name="417" href="#417">417</a>         StreamWriteFilter streamWriteFilter = <strong class="jxr_keyword">new</strong> StreamWriteFilter();
<a name="418" href="#418">418</a> 
<a name="419" href="#419">419</a>         <strong class="jxr_keyword">private</strong> <a href="../../../../../org/apache/mina/filter/stream/StreamWriteFilterTest.html">SenderHandler</a>(InputStream inputStream) {
<a name="420" href="#420">420</a>             <strong class="jxr_keyword">this</strong>.inputStream = inputStream;
<a name="421" href="#421">421</a>         }
<a name="422" href="#422">422</a> 
<a name="423" href="#423">423</a>         @Override
<a name="424" href="#424">424</a>         <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> sessionCreated(IoSession session) <strong class="jxr_keyword">throws</strong> Exception {
<a name="425" href="#425">425</a>             <strong class="jxr_keyword">super</strong>.sessionCreated(session);
<a name="426" href="#426">426</a>             session.getFilterChain().addLast(<span class="jxr_string">"codec"</span>, streamWriteFilter);
<a name="427" href="#427">427</a>         }
<a name="428" href="#428">428</a> 
<a name="429" href="#429">429</a>         @Override
<a name="430" href="#430">430</a>         <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> sessionOpened(IoSession session) <strong class="jxr_keyword">throws</strong> Exception {
<a name="431" href="#431">431</a>             session.write(inputStream);
<a name="432" href="#432">432</a>         }
<a name="433" href="#433">433</a> 
<a name="434" href="#434">434</a>         @Override
<a name="435" href="#435">435</a>         <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> exceptionCaught(IoSession session, Throwable cause)
<a name="436" href="#436">436</a>                 <strong class="jxr_keyword">throws</strong> Exception {
<a name="437" href="#437">437</a>             latch.countDown();
<a name="438" href="#438">438</a>         }
<a name="439" href="#439">439</a> 
<a name="440" href="#440">440</a>         @Override
<a name="441" href="#441">441</a>         <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> sessionClosed(IoSession session) <strong class="jxr_keyword">throws</strong> Exception {
<a name="442" href="#442">442</a>             latch.countDown();
<a name="443" href="#443">443</a>         }
<a name="444" href="#444">444</a> 
<a name="445" href="#445">445</a>         @Override
<a name="446" href="#446">446</a>         <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> sessionIdle(IoSession session, IdleStatus status)
<a name="447" href="#447">447</a>                 <strong class="jxr_keyword">throws</strong> Exception {
<a name="448" href="#448">448</a>             latch.countDown();
<a name="449" href="#449">449</a>         }
<a name="450" href="#450">450</a> 
<a name="451" href="#451">451</a>         @Override
<a name="452" href="#452">452</a>         <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> messageSent(IoSession session, Object message)