前言

在日常開發(fā)的過程中，對自己的代碼進(jìn)行單元測試是個非常重要的過程，一方面可以最小范圍的針對一個方法進(jìn)行測試，提高測試的簡便性以及測試的成本，不用啟動這個項(xiàng)目。另一方面，做好單元測試能降低代碼的BUG率。本篇文章主要是為了總結(jié)一下如何優(yōu)雅的在Springboot項(xiàng)目中使用單元測試去測試功能。

一、SpringBoot單元測試的使用

1.1 引入依賴

<dependency>
   <groupId>org.springframework.boot</groupId>
   <artifactId>spring-boot-starter-test</artifactId>
   <scope>test</scope>
</dependency>

1.2 創(chuàng)建單元測試類

如果使用的開發(fā)工具為IntelliJ IDEA，點(diǎn)擊進(jìn)入方法，鼠標(biāo)右鍵

點(diǎn)擊Generate然后選擇Test

選擇好之后點(diǎn)擊Ok就創(chuàng)建好一個測試類了。

然后在測試類上添加@SpringBootTest注解，需要測試的方法上已經(jīng)有@Test注解了 。點(diǎn)擊方法左側(cè)的三角形即可運(yùn)行單元測試方法。

二、Spring Boot使用Mockito進(jìn)行單元測試

Mockito可以模擬一個類或者方法，使用Mockito進(jìn)行單元測試的話就可以只關(guān)注這一個待測試的方法而不用去啟動整個項(xiàng)目。項(xiàng)目依賴很多環(huán)境，比如中間件、數(shù)據(jù)庫等，如果使用第一種方法進(jìn)行測試的話，則這些環(huán)境都要準(zhǔn)備好。

2.1 Mockito中經(jīng)常使用的注解以及注解的作用

• @Mock：用于創(chuàng)建被mock的對象實(shí)例。添加了@Mock注解的對象中的方法都需要mock出來，而不是調(diào)用對象本來的方法
• @Spy：創(chuàng)建保留原對象中的方法的對象。與@Mock注解不同的是，@Spy注解會保留原對象的行為，除了被特別標(biāo)記的方法，其他的方法都會執(zhí)行原有的代碼
• @InjectMocks：用于創(chuàng)建需要注入被Mock對象的類實(shí)例。例如：Service中注入了一個Dao，需要測試的Service中的方法使用了Dao，這個Dao上面添加了@Mock注解。則測試類中的Service就需要添加@InjectMocks注解。
• @Captor：用于捕獲方法調(diào)用的參數(shù)，方便進(jìn)行進(jìn)一步的斷言和校驗(yàn)
• @MockBean：用于創(chuàng)建Spring Bean的Mock對象，主要用于集成測試。在進(jìn)行集成測試時，有時需要使用Spring容器中的Bean進(jìn)行測試，但是又不希望與其他服務(wù)產(chǎn)生依賴關(guān)系。這時可以使用@MockBean注解，創(chuàng)建一個Spring Bean的Mock對象。
• @MockitoSettings：用于設(shè)置Mockito框架的全局設(shè)置。在進(jìn)行單元測試時，有時需要設(shè)置Mockito框架的一些全局設(shè)置，例如默認(rèn)的返回值等。這時可以使用@MockitoSettings注解來設(shè)置這些全局配置。

了解完了Mockito常使用的一些注解之后，下面就開始對各種情況的Mock

2.2 使用Mockito測試類中的方法

@SpringBootTest
public class ProductImageServiceImplMockito {
 
    @Mock
    private ProductImageMapper productImageMapper;
 
    @InjectMocks
    private ProductImageServiceImpl productImageService;
 
 
    @BeforeEach
    public void setup() {
        MockitoAnnotations.openMocks(this);
    }
 
 
    @Test
    public void testGet() {
        ProductImage productImage = new ProductImage();
        productImage.setId(1l);
        productImage.setImageUrl("mockUrl");
        // mock方法的邏輯
        when(productImageMapper.selectById(1l)).thenReturn(productImage);
        ProductImage image = productImageService.getByImageId(1l);
        assertEquals("mockUrl", image.getImageUrl());
    }
}

在Mapper上面添加了@Mock注解，則Mapper中的方法都是mock的，這里mock了selectById方法。

2.3 使用Mockito測試Controller層的方法

// Controller層代碼
@RestController
@RequestMapping("/test")
public class ProductImageController {
 
    @Autowired
    private ProductImageServiceImpl productImageService;
 
    @GetMapping("/productImage/{id}")
    public ProductImage getProductById(@PathVariable("id") Long id) {
        return productImageService.getByImageId(id);
    }
}
 
// 測試方法代碼
@WebMvcTest(ProductImageController.class)
public class ProductImageServiceImplMockitoV2 {
 
    @MockBean
    private ProductImageServiceImpl productImageService;
 
    @Autowired
    private MockMvc mockMvc;
 
    @Test
    public void test() throws Exception {
        ProductImage productImage = new ProductImage();
        productImage.setId(1l);
        productImage.setImageUrl("mockUrl");
        when(productImageService.getByImageId(1l)).thenReturn(productImage);
        MvcResult mvcResult = mockMvc.perform(MockMvcRequestBuilders.get("/test/productImage/1"))
                .andExpect(status().isOk())
                .andReturn();
        String contentAsString = mvcResult.getResponse().getContentAsString();
    }
}

直接模擬發(fā)送http請求到Controller的API接口，并調(diào)用@MockBean中mock出來的方法

2.4 mock測試其它場景

還有很多場景，這里就不一一列舉了，直接參考文章《在Spring Boot環(huán)境中使用Mockito進(jìn)行單元測試》

總結(jié)

本文介紹了一些單元測試的方法，在日常開發(fā)中應(yīng)該避免使用main方法測試的方式進(jìn)行測試，因?yàn)閙ain方法既無法模擬項(xiàng)目的環(huán)境，而且又受靜態(tài)方法的影響只能調(diào)用靜態(tài)方法。還有一些其它的測試工具，錄入yapi、easymock等也可以進(jìn)行測試使用。

以上就是SpringBoot項(xiàng)目創(chuàng)建單元測試的流程步驟的詳細(xì)內(nèi)容，更多關(guān)于SpringBoot創(chuàng)建單元測試的資料請關(guān)注腳本之家其它相關(guān)文章！

最新評論