View Javadoc

    1   /*
    2    * Licensed to the Apache Software Foundation (ASF) under one or more
    3    * contributor license agreements.  See the NOTICE file distributed with
    4    * this work for additional information regarding copyright ownership.
    5    * The ASF licenses this file to You under the Apache License, Version 2.0
    6    * (the "License"); you may not use this file except in compliance with
    7    * the License.  You may obtain a copy of the License at
    8    *
    9    *      http://www.apache.org/licenses/LICENSE-2.0
    10   *
    11   * Unless required by applicable law or agreed to in writing, software
    12   * distributed under the License is distributed on an "AS IS" BASIS,
    13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    14   * See the License for the specific language governing permissions and
    15   * limitations under the License.
    16   */
    17  package org.apache.commons.fileupload2.core;
    18
    19  import static org.junit.jupiter.api.Assertions.assertNotNull;
    20  import static org.junit.jupiter.api.Assertions.assertThrows;
    21
    22  import java.io.ByteArrayInputStream;
    23  import java.io.InputStream;
    24
    25  import org.junit.jupiter.api.Test;
    26
    27  /**
    28   * Unit tests {@link MultipartInput}.
    29   */
    30  public class MultipartStreamTest {
    31
    32      static private final String BOUNDARY_TEXT = "myboundary";
    33
    34      @Test
    35      public void testSmallBuffer() {
    36          final var strData = "foobar";
    37          final var contents = strData.getBytes();
    38          final InputStream input = new ByteArrayInputStream(contents);
    39          final var boundary = BOUNDARY_TEXT.getBytes();
    40          final var iBufSize = 1;
    41          assertThrows(IllegalArgumentException.class, () -> MultipartInput.builder().setInputStream(input).setBoundary(boundary).setBufferSize(iBufSize)
    42                  .setProgressNotifier(new MultipartInput.ProgressNotifier(null, contents.length)).get());
    43      }
    44
    45      @Test
    46      public void testThreeParamConstructor() throws Exception {
    47          final var strData = "foobar";
    48          final var contents = strData.getBytes();
    49          final InputStream input = new ByteArrayInputStream(contents);
    50          final var boundary = BOUNDARY_TEXT.getBytes();
    51          final var iBufSize = boundary.length + MultipartInput.BOUNDARY_PREFIX.length + 1;
    52          final var ms = MultipartInput.builder().setInputStream(input).setBoundary(boundary).setBufferSize(iBufSize)
    53                  .setProgressNotifier(new MultipartInput.ProgressNotifier(null, contents.length)).get();
    54          assertNotNull(ms);
    55      }
    56
    57      @Test
    58      public void testTwoParamConstructor() throws Exception {
    59          final var strData = "foobar";
    60          final var contents = strData.getBytes();
    61          final InputStream input = new ByteArrayInputStream(contents);
    62          final var boundary = BOUNDARY_TEXT.getBytes();
    63          final var ms = MultipartInput.builder().setInputStream(input).setBoundary(boundary)
    64                  .setProgressNotifier(new MultipartInput.ProgressNotifier(null, contents.length)).get();
    65          assertNotNull(ms);
    66      }
    67
    68  }