Spring?MVC中的Controller進(jìn)行單元測試的實(shí)現(xiàn)

更新時間：2022年02月17日 08:58:31 作者：nuccch

本文主要介紹了如何對Spring?MVC中的Controller進(jìn)行單元測試的實(shí)現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下

對Controller進(jìn)行單元測試是Spring框架原生就支持的能力，它可以模擬HTTP客戶端發(fā)起對服務(wù)地址的請求，可以不用借助于諸如Postman這樣的外部工具就能完成對接口的測試。
具體來講，是由Spring框架中的spring-test模塊提供的實(shí)現(xiàn)，詳見MockMvc。

如下將詳細(xì)闡述如何使用MockMvc測試框架實(shí)現(xiàn)對“Spring Controller”進(jìn)行單元測試，基于Spring Boot開發(fā)框架進(jìn)行驗(yàn)證。
添加測試框架依賴：

<!-- Spring框架 -->
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-web</artifactId>
</dependency>
!<-- Spring測試框架 -->
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-test</artifactId>
    <scope>test</scope>
</dependency>
<!-- 文件操作工具 -->
<dependency>
    <groupId>commons-io</groupId>
    <artifactId>commons-io</artifactId>
    <version>2.11.0</version>
</dependency>

導(dǎo)入靜態(tài)工具方法

為了便于在編寫測試用例時直接調(diào)用測試框架自帶的靜態(tài)方法，首先需要導(dǎo)入這些靜態(tài)工具方法。
需要導(dǎo)入的靜態(tài)方法如下：

import static org.springframework.test.web.servlet.setup.MockMvcBuilders.*;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.*;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.*;
import static org.springframework.test.web.servlet.result.MockMvcResultHandlers.*;
import static org.springframework.test.web.servlet.setup.SharedHttpSessionConfigurer.*;

初始化MockMvc

初始化MockMvc有2種方式：
方式1：明確指定需要測試的“Controller”類進(jìn)行配置
方式2：基于Spring容器進(jìn)行配置，包含了Spring MVC環(huán)境和所有“Controller”類，通常使用這種方式。

@SpringBootTest
public class TestControllerTest {

    MockMvc mockMvc;

    // 初始化MockMvc
    @BeforeEach
    void setUp(WebApplicationContext wac) {
        // 方式1：明確指定需要測試的“Controller”類
        this.mockMvc = MockMvcBuilders.standaloneSetup(new TestController()).build();

        // 方式2：基于Spring容器進(jìn)行配置，包含了Spring MVC環(huán)境和所有“Controller”類。
        this.mockMvc = MockMvcBuilders.webAppContextSetup(wac).build();
    }
}

另外，還可以對MockMvc進(jìn)行全局配置。

// 全局配置MockMvc
this.mockMvc = MockMvcBuilders.webAppContextSetup(wac)
        .defaultRequest(get("/").accept(MediaType.APPLICATION_JSON)) // 默認(rèn)請求路徑
        .apply(sharedHttpSession()) // 配置session
        .alwaysExpect(status().isOk()) // 預(yù)期響應(yīng)狀態(tài)碼
        .alwaysExpect(content().contentType("application/json;charset=UTF-8")) // 預(yù)期內(nèi)容類型
        .build();

執(zhí)行測試

MockMvc支持對常見的HTTP方法，如：GET，POST，PUT，DELETE等，甚至還支持文件上傳請求。

測試GET接口

// 訪問GET接口：不帶參數(shù)
@Test
public void testSimpleGet() throws Exception {
    MvcResult result = this.mockMvc.perform(get("/test/simple/get")
            .accept(MediaType.APPLICATION_JSON)) // 接受JSON格式響應(yīng)消息
            .andReturn(); // 獲取返回結(jié)果
    Assertions.assertEquals("OK", result.getResponse().getContentAsString());
}

// 訪問GET接口：帶URL參數(shù)
@Test
public void testParamGet() throws Exception {
    int id = 10;
    // 方式1：在URI模板中指定參數(shù)
    //MvcResult result = this.mockMvc.perform(get("/test/param/get?id={id}", id).accept(MediaType.APPLICATION_JSON)).andReturn();

    // 方式2：通過param()方法指定參數(shù)
    //MvcResult result = this.mockMvc.perform(get("/test/param/get").param("id", String.valueOf(id)).accept(MediaType.APPLICATION_JSON)).andReturn();

    // 方式3：通過queryParam()方法指定參數(shù)
    MvcResult result = this.mockMvc.perform(get("/test/param/get").queryParam("id", String.valueOf(id)).accept(MediaType.APPLICATION_JSON)).andReturn();
    Assertions.assertEquals("OK: " + id, result.getResponse().getContentAsString());
}

測試POST接口

// 傳遞表單參數(shù)
@Test
public void testSimplePost() throws Exception {
    int id = 10;

    // 調(diào)用param()方法傳遞參數(shù)
    MvcResult result = this.mockMvc.perform(post("/test/simple/post")
            .param("id", String.valueOf(id))
            .contentType(MediaType.APPLICATION_FORM_URLENCODED)
            .accept(MediaType.APPLICATION_JSON))
            .andReturn();
    Assertions.assertEquals("{\"id\":10}", result.getResponse().getContentAsString());
}

// 傳遞JSON參數(shù)
@Test
public void testSimplePostJson() throws Exception {
    // 調(diào)用content()方法傳遞json字符串參數(shù)
    Subject subject = new Subject();
    subject.setId(10);
    String content = JSON.toJSONString(subject);
    MvcResult result = this.mockMvc.perform(post("/test/simple/post/json")
            .content(content)
            .contentType(MediaType.APPLICATION_JSON)
            .accept(MediaType.APPLICATION_JSON))
            .andReturn();
    Assertions.assertEquals("{\"id\":10}", result.getResponse().getContentAsString());
}

測試文件上傳

@Test
public void testFileUploadSingle() throws Exception {
    File file = new File("C:\\Users\\xxx\\Downloads\\test.jpg");
    String fileName = FilenameUtils.getName(file.getName());
    byte[] bytes = FileUtils.readFileToByteArray(file);
    MockMultipartFile mockMultipartFile = new MockMultipartFile("file", fileName, MediaType.MULTIPART_FORM_DATA_VALUE, bytes);
    this.mockMvc.perform(multipart("/test/upload/single").file(mockMultipartFile))
                .andExpect(status().isOk())
                .andExpect(content().string("OK"))
                .andDo(print());
}

定義預(yù)期結(jié)果

斷言響應(yīng)結(jié)果時，有2種方式：
1.使用JUnit提供的Assert斷言工具判斷返回結(jié)果，這是一種非常普遍和常見的方式
2.在MockMvc框架中可以通過andExpect()方法定義一個或多個預(yù)期結(jié)果，當(dāng)其中一個期望結(jié)果斷言失敗時，就不會斷言其他期望值了

// 使用Junit斷言工具判斷返回結(jié)果是否符合預(yù)期
@Test
public void testAssertResult() throws Exception {
    MvcResult result = this.mockMvc.perform(get("/test/simple/get").accept(MediaType.APPLICATION_JSON)).andDo(print()).andReturn();
    Assert.assertEquals("OK", result.getResponse().getContentAsString());
}

// 在MockMvc框架中定義預(yù)期結(jié)果
@Test
public void testExpectations() throws Exception {
    this.mockMvc.perform(get("/test/simple/get").accept(MediaType.APPLICATION_JSON))
            .andExpect(status().isOk())        // 預(yù)期響應(yīng)狀態(tài)碼為200
            .andExpect(content().string("OK")) // 預(yù)期返回值為字符串“OK”
            .andDo(print());
}

相比于使用Junit的斷言工具判斷返回結(jié)果，在MockMvc框架中直接定義預(yù)期結(jié)果進(jìn)行斷言檢查更加簡潔。

寫在最后

使用Spring提供的測試框架MockMvc可以非常方便地實(shí)現(xiàn)對HTTP服務(wù)接口進(jìn)行單元測試，不要把基礎(chǔ)的功能驗(yàn)證工作都交給測試童鞋，應(yīng)該通過單元測試來保證代碼迭代的穩(wěn)定性。

到此這篇關(guān)于Spring MVC中的Controller進(jìn)行單元測試的實(shí)現(xiàn)的文章就介紹到這了,更多相關(guān)Spring MVC Controller單元測試內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: