    1   /*
    2    *  Licensed to the Apache Software Foundation (ASF) under one
    3    *  or more contributor license agreements.  See the NOTICE file
    4    *  distributed with this work for additional information
    5    *  regarding copyright ownership.  The ASF licenses this file
    6    *  to you under the Apache License, Version 2.0 (the
    7    *  "License"); you may not use this file except in compliance
    8    *  with the License.  You may obtain a copy of the License at
    9    *
    10   *    http://www.apache.org/licenses/LICENSE-2.0
    11   *
    12   *  Unless required by applicable law or agreed to in writing,
    13   *  software distributed under the License is distributed on an
    14   *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
    15   *  KIND, either express or implied.  See the License for the
    16   *  specific language governing permissions and limitations
    17   *  under the License.
    18   *
    19   */
    20  package org.apache.mina.filter.support;
    21
    22  import java.io.IOException;
    23  import java.nio.charset.Charset;
    24
    25  import org.apache.mina.common.ByteBuffer;
    26
    27  import junit.framework.TestCase;
    28
    29  /**
    30   * @author The Apache Directory Project (mina-dev@directory.apache.org)
    31   * @version $Rev: 555855 $, $Date: 2007-07-13 12:19:00 +0900 (Fri, 13 Jul 2007) $
    32   */
    33  public class ZlibTest extends TestCase {
    34      private Zlib deflater = null;
    35
    36      private Zlib inflater = null;
    37
    38      protected void setUp() throws Exception {
    39          deflater = new Zlib(Zlib.COMPRESSION_MAX, Zlib.MODE_DEFLATER);
    40          inflater = new Zlib(Zlib.COMPRESSION_MAX, Zlib.MODE_INFLATER);
    41      }
    42
    43      public void testCompression() throws Exception {
    44          String strInput = "";
    45
    46          // increase the count to as many as required to generate a long
    47          // string for input
    48          for (int i = 0; i < 10; i++) {
    49              strInput += "The quick brown fox jumps over the lazy dog.  ";
    50          }
    51          ByteBuffer byteInput = ByteBuffer.wrap(strInput.getBytes("UTF8"));
    52
    53          // increase the count to have the compression and decompression
    54          // done using the same instance of Zlib
    55          for (int i = 0; i < 5; i++) {
    56              ByteBuffer byteCompressed = deflater.deflate(byteInput);
    57              ByteBuffer byteUncompressed = inflater.inflate(byteCompressed);
    58              String strOutput = byteUncompressed.getString(Charset.forName(
    59                      "UTF8").newDecoder());
    60              assertTrue(strOutput.equals(strInput));
    61          }
    62      }
    63
    64      public void testCorruptedData() throws Exception {
    65          String strInput = "Hello World";
    66          ByteBuffer byteInput = ByteBuffer.wrap(strInput.getBytes("UTF8"));
    67
    68          ByteBuffer byteCompressed = deflater.deflate(byteInput);
    69          // change the contents to something else. Since this doesn't check
    70          // for integrity, it wont throw an exception
    71          byteCompressed.put(5, (byte) 0xa);
    72          ByteBuffer byteUncompressed = inflater.inflate(byteCompressed);
    73          String strOutput = byteUncompressed.getString(Charset.forName("UTF8")
    74                  .newDecoder());
    75          assertFalse(strOutput.equals(strInput));
    76      }
    77
    78      public void testCorruptedHeader() throws Exception {
    79          String strInput = "Hello World";
    80          ByteBuffer byteInput = ByteBuffer.wrap(strInput.getBytes("UTF8"));
    81
    82          ByteBuffer byteCompressed = deflater.deflate(byteInput);
    83          // write a bad value into the zlib header. Make sure that
    84          // the decompression fails
    85          byteCompressed.put(0, (byte) 0xca);
    86          try {
    87              inflater.inflate(byteCompressed);
    88          } catch (IOException e) {
    89              assertTrue(true);
    90              return;
    91          }
    92          assertTrue(false);
    93      }
    94
    95      public void testFragments() throws Exception {
    96          String strInput = "";
    97          for (int i = 0; i < 10; i++) {
    98              strInput += "The quick brown fox jumps over the lazy dog.  ";
    99          }
    100         ByteBuffer byteInput = ByteBuffer.wrap(strInput.getBytes("UTF8"));
    101         ByteBuffer byteCompressed = null;
    102
    103         for (int i = 0; i < 5; i++) {
    104             byteCompressed = deflater.deflate(byteInput);
    105             if (i == 0) {
    106                 // decompress the first compressed output since it contains
    107                 // the zlib header, which will not be generated for further
    108                 // compressions done with the same instance
    109                 ByteBuffer byteUncompressed = inflater.inflate(byteCompressed);
    110                 String strOutput = byteUncompressed.getString(Charset.forName(
    111                         "UTF8").newDecoder());
    112                 assertTrue(strOutput.equals(strInput));
    113             }
    114         }
    115         // check if the last compressed data block can be decompressed
    116         // successfully.
    117         ByteBuffer byteUncompressed = inflater.inflate(byteCompressed);
    118         String strOutput = byteUncompressed.getString(Charset.forName("UTF8")
    119                 .newDecoder());
    120         assertTrue(strOutput.equals(strInput));
    121     }
    122 }