SpringMVC框架如何與Junit整合看這個就夠瞭

Posted on by

系統環境

軟件 版本
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個依賴，其中：

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

編寫單元測試基類

@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;
    
   //獲取系統上下文 
    protected WebApplicationContext getCtx() {
        return this.context;
    }
 //初始化方法
    @Before
    public void setup() throws SQLException {
       this.mvc = MockMvcBuilders.webAppContextSetup(this.context).build();
    }
 //封裝返回結果
    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來運行測試用例。
  • @WebAppConfiguration 註解表明使用的ApplicationContext將是WebApplicationContext類型的；value屬性指定web應用的根；同時，他必須與 @ContextConfiguration一同使用。
  • @ContextConfiguration註解的作用是導入我們需要的配置文件或者類，這裡指定瞭SpringMVC的配置文件
  • @ActiveProfiles 註解是指定單元測試連接的環境。
  • @TransactionConfiguration 註解是用與管理spring事務的。
  • MockHttpServletRequest 類為瞭模擬HttpServletRequest 對象，可以通過它來設置請求頭，這在需要鑒權的接口測試中十分實用。
  • MockMvc 實現瞭對Http請求的模擬，能夠直接使用網絡的形式，轉換到Controller的調用，這樣可以使得測試速度快，不依賴網絡環境，而且提供瞭一套驗證的工具，這樣可以使得請求的驗證統一而且很方便。

getMockResultStr 方法是封裝瞭公共的測試方法以及響應參數。 其中： 

MvcResult mvcResult = this.mockMvc
                    .perform(requestBuilder)
                    .andExpect(status().isOk())
                    .andDo(MockMvcResultHandlers.print())
                    .andReturn();
  • perform方法是執行一個RequestBuilder請求，調用Controller的業務處理邏輯。
  • andExpect 方法是添加執行完成後的斷言，添加ResultMatcher驗證規則，驗證控制器執行完成後結果是否正確，此處是驗證狀態是否是200
  • andDo 方法是添加一個結果處理器，比如再此處是andDo(MockMvcResultHandlers.print())輸出整個響應結果信息，可以在調試的時候使用。
  • andReturn：方法是執行完成後返回相應的結果。

MockMvcRequestBuilders

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

返回結果的content-type是application/json的請求的單元測試

get請求測試

get請求的測試就是調用MockMvcRequestBuilders的get方法， 在該方法中傳入接口地址。

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

post請求測試

post請求的測試就是調用MockMvcRequestBuilders的post方法， 在該方法中傳入接口地址。

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

文件上傳測試

文件上傳的測試就是調用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);

發送一個json格式的數據到Controller層 

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

返回結果是視圖請求的單元測試

那麼，問題來瞭，如果一個接口的返回結果是視圖，那麼這個接口的單元測試該如何寫呢？比如下面這個接口：

  @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("角標編輯頁面返回成功");
       return "manage/superScript/edit";

   }

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

   @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()獲取視圖的名稱。

直接測試一個service的方法

上面的那些測試針對的都是Controller層的接口，那麼如果要測試一個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();
    }

被測試的MallMailJob類也可以通過@Autowired 註解直接註入進Spring IOC容器中。拿到被測試類的實例之後就可以直接調用被測試的方法，是不是灰常的簡單呀。

總結

本文詳細介紹在SpringMVC任何使用Junit框架。首先介紹瞭如何引入依賴，接著介紹瞭編寫一個測試基類，並且對其中涉及的各個註解做瞭一個詳細說明。最後就是羅列瞭各種測試情況。

到此這篇關於SpringMVC框架如何與Junit整合看這個就夠瞭的文章就介紹到這瞭,更多相關SpringMVC整合Junit內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：