Open Source Repository

Home /spring/spring-test-3.0.5 | Repository Home



org/springframework/mock/web/MockMultipartHttpServletRequest.java
/*
 * Copyright 2002-2010 the original author or authors.
 *
 * Licensed 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.
 */

package org.springframework.mock.web;

import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.springframework.util.Assert;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;

/**
 * Mock implementation of the
 {@link org.springframework.web.multipart.MultipartHttpServletRequest} interface.
 *
 <p>Useful for testing application controllers that access multipart uploads.
 * The {@link MockMultipartFile} can be used to populate these mock requests
 * with files.
 *
 @author Juergen Hoeller
 @author Eric Crampton
 @author Arjen Poutsma
 @since 2.0
 @see MockMultipartFile
 */
public class MockMultipartHttpServletRequest extends MockHttpServletRequest implements MultipartHttpServletRequest {

  private final MultiValueMap<String, MultipartFile> multipartFiles =
      new LinkedMultiValueMap<String, MultipartFile>();


  public MockMultipartHttpServletRequest() {
    setMethod("POST");
    setContentType("multipart/form-data");
  }


  /**
   * Add a file to this request. The parameter name from the multipart
   * form is taken from the {@link MultipartFile#getName()}.
   @param file multipart file to be added
   */
  public void addFile(MultipartFile file) {
    Assert.notNull(file, "MultipartFile must not be null");
    this.multipartFiles.add(file.getName(), file);
  }

  public Iterator<String> getFileNames() {
    return this.multipartFiles.keySet().iterator();
  }

  public MultipartFile getFile(String name) {
    return this.multipartFiles.getFirst(name);
  }

  public List<MultipartFile> getFiles(String name) {
    List<MultipartFile> multipartFiles = this.multipartFiles.get(name);
    if (multipartFiles != null) {
      return multipartFiles;
    }
    else {
      return Collections.emptyList();
    }
  }

  public Map<String, MultipartFile> getFileMap() {
    return this.multipartFiles.toSingleValueMap();
  }

  public MultiValueMap<String, MultipartFile> getMultiFileMap() {
    return new LinkedMultiValueMap<String, MultipartFile>(this.multipartFiles);
  }

}