Source code: org/apache/http/io/TestIdentityOutputStream.java

   /*
    * $HeadURL: https://svn.apache.org/repos/asf/jakarta/httpcomponents/httpcore/tags/4.0-alpha2/src/test/org/apache/http/io/TestIdentityOutputStream.java $
    * $Revision: 321483 $
    * $Date: 2005-10-15 22:32:14 +0200 (Sat, 15 Oct 2005) $
    * ====================================================================
    *
    *  Copyright 2002-2004 The Apache Software Foundation
    *
    *  Licensed under the Apache License, Version 2.0 (the "License");
   *  you may not use this file except in compliance with the License.
   *  You may obtain a copy of the License at
   *
   *      http://www.apache.org/licenses/LICENSE-2.0
   *
   *  Unless required by applicable law or agreed to in writing, software
   *  distributed under the License is distributed on an "AS IS" BASIS,
   *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   *  See the License for the specific language governing permissions and
   *  limitations under the License.
   * ====================================================================
   *
   * This software consists of voluntary contributions made by many
   * individuals on behalf of the Apache Software Foundation.  For more
   * information on the Apache Software Foundation, please see
   * <http://www.apache.org/>.
   *
   */
  
  package org.apache.http.io;
  
  import java.io.ByteArrayOutputStream;
  import java.io.IOException;
  import java.io.OutputStream;
  
  import org.apache.http.mockup.HttpDataTransmitterMockup;
  
  import junit.framework.Test;
  import junit.framework.TestCase;
  import junit.framework.TestSuite;
  
  public class TestIdentityOutputStream extends TestCase {
  
      public TestIdentityOutputStream(String testName) {
          super(testName);
      }
  
      // ------------------------------------------------------- TestCase Methods
  
      public static Test suite() {
          return new TestSuite(TestIdentityOutputStream.class);
      }
  
      // ------------------------------------------------------------------- Main
      public static void main(String args[]) {
          String[] testCaseName = { TestIdentityOutputStream.class.getName() };
          junit.textui.TestRunner.main(testCaseName);
      }
  
      public void testConstructors() throws Exception {
          new IdentityOutputStream(new HttpDataTransmitterMockup());
          try {
              new IdentityOutputStream(null);
              fail("IllegalArgumentException should have been thrown");
          } catch (IllegalArgumentException ex) {
              // expected
          }
      }
  
      public void testBasics() throws Exception {
        ByteArrayOutputStream buffer = new ByteArrayOutputStream();
        HttpDataTransmitterMockup datatransmitter = new HttpDataTransmitterMockup(buffer);
        OutputStream out = new IdentityOutputStream(datatransmitter);
  
          byte[] tmp = new byte[10];
          out.write(tmp, 0, 10);
          out.write(tmp);
          out.write(1);
          out.flush();
          out.close();
          byte[] data = datatransmitter.getData();
          assertEquals(21, data.length);
      }
  
      public void testClose() throws Exception {
        ByteArrayOutputStream buffer = new ByteArrayOutputStream();
        HttpDataTransmitterMockup datatransmitter = new HttpDataTransmitterMockup(buffer);
        OutputStream out = new IdentityOutputStream(datatransmitter);
        out.close();
        out.close();
          byte[] tmp = new byte[10];
          try {
            out.write(tmp);
              fail("IOException should have been thrown");
          } catch (IOException ex) {
              // expected
          }
          try {
              out.write(1);
              fail("IOException should have been thrown");
         } catch (IOException ex) {
             // expected
         }
     }
     
 }