SpringMVC框架如何與Junit整合看這個(gè)就夠了

更新時(shí)間：2021年05月06日 10:47:25 作者：碼農(nóng)飛哥

這篇文章主要介紹了SpringMVC框架如何與Junit整合看這個(gè)就夠了，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

系統(tǒng)環(huán)境

軟件	版本
spring-webmvc	4.3.6.RELEASE
spring-test	4.3.6.RELEASE
junit	4.12

引入依賴

 <dependency>
                <groupId>junit</groupId>
                <artifactId>junit</artifactId>
                <version>4.12</version>
                <scope>test</scope>
            </dependency>
     <dependency>
                <groupId>org.springframework</groupId>
                <artifactId>spring-test</artifactId>
                <version>${spring.version}</version>
                <scope>test</scope>
            </dependency>

如上，引入了4個(gè)依賴，其中：

junit 是Junit單元測(cè)試的基礎(chǔ)依賴，它提供了@RunWith和@Test等單元測(cè)試必備的注解。<scope>test</scope> 則指明了該依賴的作用范圍是test階段。
spring-test依賴可以理解成是spring框架與junit框架的橋梁。提供了@WebAppConfiguration等注解。

編寫單元測(cè)試基類

@RunWith(SpringJUnit4ClassRunner.class)
@WebAppConfiguration
@ContextConfiguration("file:src/test/java/spring-config.xml")
@ActiveProfiles("dev")
@TransactionConfiguration(transactionManager = "transactionManager", defaultRollback = true)
public abstract class BaseControllerTest  {
    
    protected MockMvc mvc;
    
    @Autowired
    private WebApplicationContext context;
    
   //獲取系統(tǒng)上下文 
    protected WebApplicationContext getCtx() {
        return this.context;
    }
 //初始化方法
    @Before
    public void setup() throws SQLException {
       this.mvc = MockMvcBuilders.webAppContextSetup(this.context).build();
    }
 //封裝返回結(jié)果
    protected String getMockResultStr(MockHttpServletRequestBuilder requestBuilder) {
        try {
            MvcResult mvcResult = this.mockMvc
                    .perform(requestBuilder)
                    .andExpect(status().isOk())
                    .andDo(MockMvcResultHandlers.print())
                    .andReturn();
            MockHttpServletResponse response = mvcResult.getResponse();
            response.setContentType(MediaType.APPLICATION_JSON_UTF8_VALUE);
            return response.getContentAsString();
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }
}

代碼說明

@RunWith(SpringJUnit4ClassRunner.class) 注解表明使用SpringJUnit4ClassRunner來運(yùn)行測(cè)試用例。
@WebAppConfiguration 注解表明使用的ApplicationContext將是WebApplicationContext類型的；value屬性指定web應(yīng)用的根；同時(shí)，他必須與 @ContextConfiguration一同使用。
@ContextConfiguration注解的作用是導(dǎo)入我們需要的配置文件或者類，這里指定了SpringMVC的配置文件
@ActiveProfiles 注解是指定單元測(cè)試連接的環(huán)境。
@TransactionConfiguration 注解是用與管理spring事務(wù)的。
MockHttpServletRequest 類為了模擬HttpServletRequest 對(duì)象，可以通過它來設(shè)置請(qǐng)求頭，這在需要鑒權(quán)的接口測(cè)試中十分實(shí)用。
MockMvc 實(shí)現(xiàn)了對(duì)Http請(qǐng)求的模擬，能夠直接使用網(wǎng)絡(luò)的形式，轉(zhuǎn)換到Controller的調(diào)用，這樣可以使得測(cè)試速度快，不依賴網(wǎng)絡(luò)環(huán)境，而且提供了一套驗(yàn)證的工具，這樣可以使得請(qǐng)求的驗(yàn)證統(tǒng)一而且很方便。

getMockResultStr 方法是封裝了公共的測(cè)試方法以及響應(yīng)參數(shù)。其中：

MvcResult mvcResult = this.mockMvc
                    .perform(requestBuilder)
                    .andExpect(status().isOk())
                    .andDo(MockMvcResultHandlers.print())
                    .andReturn();

perform方法是執(zhí)行一個(gè)RequestBuilder請(qǐng)求，調(diào)用Controller的業(yè)務(wù)處理邏輯。
andExpect 方法是添加執(zhí)行完成后的斷言，添加ResultMatcher驗(yàn)證規(guī)則，驗(yàn)證控制器執(zhí)行完成后結(jié)果是否正確，此處是驗(yàn)證狀態(tài)是否是200
andDo 方法是添加一個(gè)結(jié)果處理器，比如再此處是andDo(MockMvcResultHandlers.print())輸出整個(gè)響應(yīng)結(jié)果信息，可以在調(diào)試的時(shí)候使用。
andReturn：方法是執(zhí)行完成后返回相應(yīng)的結(jié)果。

MockMvcRequestBuilders

RequestBuilder是用來構(gòu)建請(qǐng)求的。其提供了一個(gè)方法buildRequest(ServletContext servletContext) 用于構(gòu)建MockHttpServletRequest；其主要有兩個(gè)子類MockHttpServletRequestBuilder和MockMultipartHttpServletRequestBuilder（文件上傳時(shí)使用），即用來mock客戶端請(qǐng)求需要的所有數(shù)據(jù)。
主要API有MockHttpServletRequestBuilder.get(String urlTemplate, Object… urlVariables)：根據(jù)uri模板和uri變量值得到一個(gè)GET請(qǐng)求方式的RequestBuilder，如果在controller的方法中method選擇的是RequestMethod.GET，那在controllerTest中對(duì)應(yīng)就要使用MockMvcRequestBuilders.get。
MockHttpServletRequestBuilder.post(String urlTemplate, Object… urlVariables)：同get類似，但是是post方法，通過param 方法來設(shè)置請(qǐng)求參數(shù)。

返回結(jié)果的content-type是application/json的請(qǐng)求的單元測(cè)試

get請(qǐng)求測(cè)試

get請(qǐng)求的測(cè)試就是調(diào)用MockMvcRequestBuilders的get方法，在該方法中傳入接口地址。

  MockHttpServletRequestBuilder requestBuilder = MockMvcRequestBuilders.get("/v1/works/lineup/set")
                .param("id", "123")
                .param("first_flag", String.valueOf(true));
      String result = getMockResultStr(requestBuilder);

post請(qǐng)求測(cè)試

post請(qǐng)求的測(cè)試就是調(diào)用MockMvcRequestBuilders的post方法，在該方法中傳入接口地址。

 MockHttpServletRequestBuilder requestBuilder = MockMvcRequestBuilders.post("/v1/blockly/copy")
                .param("id", "1211");
        success(requestBuilder);

文件上傳測(cè)試

文件上傳的測(cè)試就是調(diào)用MockMvcRequestBuilders的multipart方法，在該方法中傳入接口地址。

  File file = new File(RestArmWorksControllerTest.class.getClassLoader().getResource("20201102172053.png").toURI());
        MockMultipartFile mockMultipartFile = new MockMultipartFile("file", "20201102172053.png", "", new FileInputStream(file));
        MockMultipartHttpServletRequestBuilder servletRequestBuilder = MockMvcRequestBuilders.multipart("/creation/fileUpload.do")
                .file(mockMultipartFile);
        String result = getMockResultStr(servletRequestBuilder);

發(fā)送一個(gè)json格式的數(shù)據(jù)到Controller層

String requestJson = JSONObject.toJSONString(folderInfo);
      MockHttpServletRequestBuilder requestBuilder = MockMvcRequestBuilders.post("http://v1/works/save").contentType(MediaType.APPLICATION_JSON).content(requestJson)；

返回結(jié)果是視圖請(qǐng)求的單元測(cè)試

那么，問題來了，如果一個(gè)接口的返回結(jié)果是視圖，那么這個(gè)接口的單元測(cè)試該如何寫呢？比如下面這個(gè)接口：

  @RequestMapping(value = "/edit")
   public String edit(ModelMap model, HttpServletRequest request, HttpServletResponse response) {
       Long scriptId = null;
       if (StringUtils.isNotBlank(request.getParameter("scriptId"))) {
           scriptId = Long.parseLong(request.getParameter("scriptId"));
       }
       if (scriptId != null) {
           MallSuperscript mallSuperscript = superScriptService.getScriptById(scriptId);
           model.addAttribute("superScript", mallSuperscript);
       }
       model.addAttribute("scriptTypeMap", ScriptType.toMap());
       logger.info("角標(biāo)編輯頁面返回成功");
       return "manage/superScript/edit";

   }

這個(gè)接口返回的是一個(gè)視圖，就是名稱為edit.jsp頁面。它的測(cè)試方法可以像下面這樣寫。

   @Test
    public void testEdit() throws Exception {

        MvcResult mvcResult = mvc
                .perform(
                        get("/manage/tab2/superScript/edit").sessionAttr("adminid", "1111")
                                .sessionAttr("platformid", 1001L).param("scriptId", "1111"))
                .andExpect(status().isOk()).andDo(print()).andReturn();
       
        assertEquals("manage/superScript/edit", mvcResult.getModelAndView().getViewName());
    }

通過mvcResult.getModelAndView().getViewName()獲取視圖的名稱。

直接測(cè)試一個(gè)service的方法

上面的那些測(cè)試針對(duì)的都是Controller層的接口，那么如果要測(cè)試一個(gè)service層的方法，該如何處理呢？

public class MallMailJobTest extends BaseControllerTest {

    private MallMailJob mallMailJob;

    @Override
    @Before
    public void setup() throws SQLException {
        super.setup();
        mallMailJob = getCtx().getBean(MallMailJob.class);
    }

    /**
     * 
     * @throws Exception
     */
    @Test
    public void testExecute() throws Exception {
        mallMailJob.execute();
    }

被測(cè)試的MallMailJob類也可以通過@Autowired 注解直接注入進(jìn)Spring IOC容器中。拿到被測(cè)試類的實(shí)例之后就可以直接調(diào)用被測(cè)試的方法，是不是灰常的簡(jiǎn)單呀。