1. 程式人生 > >SpringBootTest使用Mock測試檔案上傳

SpringBootTest使用Mock測試檔案上傳

@SpringBootTest
@RunWith(SpringJUnit4ClassRunner.class)
public class OssControllerTest {
@Autowired
private WebApplicationContext wac;
private MockMvc mvc;
@Before
public void setupMockMvc(){
mvc = MockMvcBuilders.webAppContextSetup(wac).build(); //初始化MockMvc物件
}
@Test
public void uploadFilePublic() throws 
Exception { File file = new File("C:\\Users\\Administrator\\Desktop\\xxxx.jpg"); //檔案之外的引數 String key = OssUtils.createFileKey(file.getName(), null); MockMultipartFile firstFile = new MockMultipartFile("file", "xxxx.jpg", MediaType.TEXT_PLAIN_VALUE, new FileInputStream(file)); mvc.perform(MockMvcRequestBuilders
.fileUpload("/ossService/uploadFilePublic") .file(firstFile)//檔案 .param("key", key))//引數 .andExpect(MockMvcResultMatchers.status().isOk()); LogUtils.info("檔案的key為:"+key); } }

參考部落格:http://tobato.iteye.com/blog/2315174