KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > springframework > mock > web > MockMultipartHttpServletRequest


1 /*
2  * Copyright 2002-2006 the original author or authors.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package org.springframework.mock.web;
18
19 import java.util.Collections JavaDoc;
20 import java.util.Iterator JavaDoc;
21 import java.util.Map JavaDoc;
22
23 import org.springframework.core.CollectionFactory;
24 import org.springframework.util.Assert;
25 import org.springframework.web.multipart.MultipartFile;
26 import org.springframework.web.multipart.MultipartHttpServletRequest;
27
28 /**
29  * Mock implementation of the
30  * {@link org.springframework.web.multipart.MultipartHttpServletRequest} interface.
31  *
32  * <p>Useful for testing application controllers that access multipart uploads.
33  * The {@link MockMultipartFile} can be used to populate these mock requests
34  * with files.
35  *
36  * @author Juergen Hoeller
37  * @author Eric Crampton
38  * @since 2.0
39  * @see MockMultipartFile
40  */
41 public class MockMultipartHttpServletRequest extends MockHttpServletRequest implements MultipartHttpServletRequest {
42
43     private final Map JavaDoc multipartFiles = CollectionFactory.createLinkedMapIfPossible(4);
44
45
46     /**
47      * Add a file to this request. The parameter name from the multipart
48      * form is taken from the {@link MultipartFile#getName()}.
49      * @param file multipart file to be added
50      */
51     public void addFile(MultipartFile file) {
52         Assert.notNull(file, "MultipartFile must not be null");
53         this.multipartFiles.put(file.getName(), file);
54     }
55
56     public Iterator JavaDoc getFileNames() {
57         return getFileMap().keySet().iterator();
58     }
59
60     public MultipartFile getFile(String JavaDoc name) {
61         return (MultipartFile) this.multipartFiles.get(name);
62     }
63
64     public Map JavaDoc getFileMap() {
65         return Collections.unmodifiableMap(this.multipartFiles);
66     }
67
68 }
69
Popular Tags