~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/entity/TestHttpEntityWrapper.java $

 * $Revision: 744517 $

 * $Date: 2009-02-14 17:39:33 +0100 (Sat, 14 Feb 2009) $

 * 

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

 * 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.entity;

 

import java.io.ByteArrayOutputStream;

 

import junit.framework.Test;

import junit.framework.TestCase;

import junit.framework.TestSuite;

 

import org.apache.http.protocol.HTTP;

 

/**

 * Unit tests for {@link HttpEntityWrapper}.

 *

 */

public class TestHttpEntityWrapper extends TestCase {

 

    public TestHttpEntityWrapper(String testName) {

        super(testName);

    }

 

    public static void main(String args[]) {

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

        junit.textui.TestRunner.main(testCaseName);

    }

 

    public static Test suite() {

        return new TestSuite(TestHttpEntityWrapper.class);

    }

 

    public void testBasics() throws Exception {

        String s = "Message content";

        StringEntity httpentity = new StringEntity(s, HTTP.ISO_8859_1);

        httpentity.setContentType(HTTP.PLAIN_TEXT_TYPE);

        httpentity.setContentEncoding(HTTP.IDENTITY_CODING);

        HttpEntityWrapper wrapped = new HttpEntityWrapper(httpentity); 

        

        assertEquals(httpentity.getContentLength(), wrapped.getContentLength());

        assertEquals(httpentity.getContentType(), wrapped.getContentType());

        assertEquals(httpentity.getContentEncoding(), wrapped.getContentEncoding());

        assertEquals(httpentity.isChunked(), wrapped.isChunked());

        assertEquals(httpentity.isRepeatable(), wrapped.isRepeatable());

        assertEquals(httpentity.isStreaming(), wrapped.isStreaming());

        assertNotNull(wrapped.getContent());

    }

 

    public void testIllegalConstructor() throws Exception {

        try {

            new HttpEntityWrapper(null);

            fail("IllegalArgumentException should have been thrown");

        } catch (IllegalArgumentException ex) {

            // expected

        }

    }

 

    public void testWriteTo() throws Exception {

        String s = "Message content";

        byte[] bytes = s.getBytes(HTTP.ISO_8859_1);

        StringEntity httpentity = new StringEntity(s);

        HttpEntityWrapper wrapped = new HttpEntityWrapper(httpentity); 

        

        ByteArrayOutputStream out = new ByteArrayOutputStream();

        wrapped.writeTo(out);

        byte[] bytes2 = out.toByteArray();

        assertNotNull(bytes2);

        assertEquals(bytes.length, bytes2.length);

        for (int i = 0; i < bytes.length; i++) {

            assertEquals(bytes[i], bytes2[i]);

        }

 

        out = new ByteArrayOutputStream();

        wrapped.writeTo(out);

        bytes2 = out.toByteArray();

        assertNotNull(bytes2);

        assertEquals(bytes.length, bytes2.length);

        for (int i = 0; i < bytes.length; i++) {

            assertEquals(bytes[i], bytes2[i]);

        }

        

        try {

            wrapped.writeTo(null);

            fail("IllegalArgumentException should have been thrown");

        } catch (IllegalArgumentException ex) {

            // expected

        }

    }

 

    public void testConsumeContent() throws Exception {

        String s = "Message content";

        StringEntity httpentity = new StringEntity(s);

        HttpEntityWrapper wrapped = new HttpEntityWrapper(httpentity);

        wrapped.consumeContent();

        wrapped.consumeContent();

    }

    

}