2014-01-16 71 views
4

我试图用Spring MVC 3.2创建一个RESTful控制器端点来上传文件以及该文件的元数据映射。这样定义的:使用MockMVC测试Spring MVC请求参数映射和MultipartFile

@Controller 
@RequestMapping("/file") 
public class FileServiceController { 
    @RequestMapping(value="/upload", method=RequestMethod.POST) 
    @ResponseBody 
    public void upload(@RequestParam MultipartFile file, 
         @RequestParam String fileType, 
         @RequestParam(value="metadata") Map<String, List<String>> metadata) { 
     // TODO: stuff with things 
    } 
} 

这我然后试图测试是这样的:

@RunWith(SpringJUnit4ClassRunner.class) 
@ContextConfiguration(classes = { AppConfig.class }) 
@WebAppConfiguration 
public class ProductIngestServiceControllerTest { 

    @Autowired 
    private WebApplicationContext wac; 
    private MockMvc mockMvc; 
    @Mock 
    private FileServiceController controller; 

    @Before 
    public void setup() { 
     mockMvc = MockMvcBuilders.standaloneSetup(controller).build(); 
    } 

    @Test 
    public void testUpload() { 
     MockMultipartFile mockFile = new MockMultipartFile("file", "This is a test file".getBytes()); 
     Map<String, List<String>> metadata = new HashMap<>(); 
     metadata.put("custom0", Arrays.asList("test1", "test2")); 
     metadata.put("custom1", Arrays.asList("test3", "test4", "test5")); 
     metadata.put("custom2", Arrays.asList("test6")); 

     Gson gson = new Gson(); 
     String mapStr = gson.toJson(metadata); 

     mockMvc.perform(fileUpload(baseURL + "/test") 
      .file(mockFile) 
      .param("fileType", "test_type") 
      .param("metadata", mapStr)) 
      .andDo(print()) 
      .andExpect(status().isOk()) 
      .andReturn(); 
    } 
} 

我期待在地图包含只是“customX”的条目,但它包含了所有的请求参数相反(除了MultipartFile ...)。

如果我尝试:

public void upload(@RequestParam MultiValueMap parameters) { 
    ... 
} 

的MultipartFile不包括在该地图无论是。

我该怎么做呢?唯一的解决办法我能想出的时刻是只使用一个字符串的元数据RequestParam和做类似:

ObjectMapper mapper = new ObjectMapper(); 
JsonFactory factory = mapper.getJsonFactory(); 

JsonParser jp = factory.createJsonParser(testMap); 
Map<String, List<String>> map = jp.readValueAs(Map.class); 

但感觉很脏。

任何帮助将不胜感激。谢谢!

+0

因此,您无法获得MultipartFile或存在元数据中不应存在的值的问题? – phantastes

回答

0

使用MockHttpServletRequest。

@Before 
public void before() { 
    MockitoAnnotations.initMocks(this); 
    this.mockMvc = MockMvcBuilders.webAppContextSetup(this.wac) 
      .dispatchOptions(true).build(); 
    MockHttpServletRequest request;= new MockHttpServletRequest(); 
    Map<String, List<String>> metadata=request.getParameter("metadata"); 
}