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.fileupload;
    18
    19  import java.io.UnsupportedEncodingException;
    20  import java.util.Arrays;
    21  import java.util.List;
    22
    23  import javax.servlet.http.HttpServletRequest;
    24
    25  import org.apache.commons.fileupload.disk.DiskFileItemFactory;
    26  import org.apache.commons.fileupload.portlet.PortletFileUpload;
    27  import org.apache.commons.fileupload.servlet.ServletFileUpload;
    28  import org.apache.commons.fileupload.servlet.ServletRequestContext;
    29
    30  /**
    31   * Test utility methods.
    32   *
    33   * @since 1.4
    34   */
    35  public class Util {
    36
    37      public static List<FileItem> parseUpload(FileUpload upload, byte[] bytes) throws FileUploadException {
    38          return parseUpload(upload, bytes, Constants.CONTENT_TYPE);
    39      }
    40
    41      public static List<FileItem> parseUpload(FileUpload upload, byte[] bytes, String contentType) throws FileUploadException {
    42          final HttpServletRequest request = new MockHttpServletRequest(bytes, contentType);
    43          List<FileItem> fileItems = upload.parseRequest(new ServletRequestContext(request));
    44          return fileItems;
    45      }
    46
    47      public static List<FileItem> parseUpload(FileUpload upload, String content)
    48          throws UnsupportedEncodingException, FileUploadException {
    49          byte[] bytes = content.getBytes("US-ASCII");
    50          return parseUpload(upload, bytes, Constants.CONTENT_TYPE);
    51      }
    52
    53      /**
    54       * Return a list of {@link FileUpload} implementations for parameterized tests.
    55       * @return a list of {@link FileUpload} implementations
    56       */
    57      public static List<FileUpload> fileUploadImplementations() {
    58          return Arrays.asList(
    59                  new ServletFileUpload(new DiskFileItemFactory()),
    60                  new PortletFileUpload(new DiskFileItemFactory()));
    61      }
    62  }