~ubuntu-branches/ubuntu/trusty/httpcomponents-core/trusty

/*

 * $HeadURL: https://svn.apache.org/repos/asf/httpcomponents/httpcore/tags/4.0.1/httpcore/src/test/java/org/apache/http/impl/io/TestContentLengthOutputStream.java $

 * $Revision: 618017 $

 * $Date: 2008-02-03 17:42:22 +0100 (Sun, 03 Feb 2008) $

 * ====================================================================

 * Licensed to the Apache Software Foundation (ASF) under one

 * or more contributor license agreements.  See the NOTICE file

 * distributed with this work for additional information

 * regarding copyright ownership.  The ASF licenses this file

 * to you 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.impl.io;

 

import java.io.ByteArrayOutputStream;

import java.io.IOException;

import java.io.OutputStream;

 

import org.apache.http.impl.io.ContentLengthOutputStream;

import org.apache.http.mockup.SessionOutputBufferMockup;

 

import junit.framework.Test;

import junit.framework.TestCase;

import junit.framework.TestSuite;

 

public class TestContentLengthOutputStream extends TestCase {

 

    public TestContentLengthOutputStream(String testName) {

        super(testName);

    }

 

    // ------------------------------------------------------- TestCase Methods

 

    public static Test suite() {

        return new TestSuite(TestContentLengthOutputStream.class);

    }

 

    // ------------------------------------------------------------------- Main

    public static void main(String args[]) {

        String[] testCaseName = { TestContentLengthOutputStream.class.getName() };

        junit.textui.TestRunner.main(testCaseName);

    }

 

    public void testConstructors() throws Exception {

        new ContentLengthOutputStream(new SessionOutputBufferMockup(), 10L);

        try {

            new ContentLengthOutputStream(null, 10L);

            fail("IllegalArgumentException should have been thrown");

        } catch (IllegalArgumentException ex) {

            // expected

        }

        try {

            new ContentLengthOutputStream(new SessionOutputBufferMockup(), -10);

            fail("IllegalArgumentException should have been thrown");

        } catch (IllegalArgumentException ex) {

            // expected

        }

    }

 

    public void testBasics() throws Exception {

        ByteArrayOutputStream buffer = new ByteArrayOutputStream();

        SessionOutputBufferMockup datatransmitter = new SessionOutputBufferMockup(buffer);

        OutputStream out = new ContentLengthOutputStream(datatransmitter, 15L);

 

        byte[] tmp = new byte[10];

        out.write(tmp, 0, 10);

        out.write(1);

        out.write(tmp, 0, 10);

        out.write(tmp, 0, 10);

        out.write(tmp);

        out.write(1);

        out.write(2);

        out.flush();

        out.close();

        byte[] data = datatransmitter.getData();

        assertEquals(15, data.length);

    }

 

    public void testClose() throws Exception {

        ByteArrayOutputStream buffer = new ByteArrayOutputStream();

        SessionOutputBufferMockup datatransmitter = new SessionOutputBufferMockup(buffer);

        OutputStream out = new ContentLengthOutputStream(datatransmitter, 15L);

        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

        }

    }

    

}