SpringBootControllerPost接口單元測試示例

概述

成都創新互聯是一家專注于網站設計制作、成都網站設計與策劃設計,綏陽網站建設哪家好?成都創新互聯做網站,專注于網站建設十載,網設計領域的專業建站公司;建站業務涵蓋:綏陽等地區。綏陽做網站價格咨詢:18982081108

在日常的開發中，我們一般會定義一個service層，用于實現業務邏輯，并且針對service層會有與之對應的齊全的覆蓋率高的單元測試。而對于controller層，一般不怎么做單元測試，因為主要的核心業務邏輯都在service層里，controller層只是做轉發，調用service層接口而已。但是還是建議使用單元測試簡單的將controller的方法跑一下，看看轉發和數據轉換的代碼是否能正常工作。

在Spring Boot里對controller層進行單元測試非常簡單，只需要幾個注解和一點點輔助代碼即可搞定。

依賴的包

  <dependency>
   <groupId>org.junit.jupiter</groupId>
   <artifactId>junit-jupiter-api</artifactId>
   <scope>test</scope>
  </dependency>
  <dependency>
   <groupId>org.junit.jupiter</groupId>
   <artifactId>junit-jupiter-engine</artifactId>
   <scope>test</scope>
  </dependency>
  <dependency>
   <groupId>org.springframework.boot</groupId>
   <artifactId>spring-boot-starter-test</artifactId>
   <scope>test</scope>
  </dependency>
  <dependency>
   <groupId>com.alibaba</groupId>
   <artifactId>fastjson</artifactId>
  </dependency>

使用的Spring Boot 版本

2.0.4.RELEASE

代碼

@ExtendWith(SpringExtension.class)
@SpringBootTest(webEnvironment =SpringBootTest.WebEnvironment.MOCK,classes = TestApplication.class)
@AutoConfigureMockMvc
public class UserControllerTest {
 @Autowired
 private MockMvc mockMvc;
 @MockBean
 private UserService userService;
 @Test
 @DisplayName("測試controller方法")
 void test() throws Exception {
  User param = new User();
  param.setUserId(1111);
  List<Address> addressList = new ArrayList<>();
  Address address = new Address();
  address.setName("我的地址");
  addressList.add(address);
  param.setAddressList(addressList);
  MvcResult mvcResult = mockMvc.perform(
    post("/xxx/test")
      .contentType(MediaType.APPLICATION_JSON)
      .content(JSON.toJSONString(param)))
    .andReturn();
  
  System.out.println(mvcResult.getResponse().getContentAsString());
 }
}

@RequestMapping(value = "/xxx", method = RequestMethod.POST)
public Object test(@RequestBody(required = false)User user) throws Exception {
}

如果你只是想簡單的跑一下controller層，不想真正的去執行service方法的話，需要使用@MockBean將對應的service類mock掉。

 @MockBean
 private UserService userService;

使用Spring Boot Test的時候，它需要一個ApplicationContext,我們可以在@SpringBootTest注解中使用classes屬性來指定。

@SpringBootTest(webEnvironment =SpringBootTest.WebEnvironment.MOCK,classes = TestApplication.class)

TestApplication的代碼很簡單。

@SpringBootApplication
public class TestApplication {
 public static void main(String[] args){
  SpringApplicationBuilder builder = new SpringApplicationBuilder();
  builder.environment(new StandardEnvironment());
  builder.sources(TestApplication.class);
  builder.main(TestApplication.class);
  builder.run(args);
 }
}

接下來我們只需要使用MockMvc發送post請求即可。如果controller層的post方法是帶@RequestBody注解的，可以先將入參對象轉換成JSON字符串。這里使用的是fastjson。