五分鐘技術(shù)趣談 | Java常用的單元測試框架介紹

作者：孟祥路，單位：中移（杭州）信息技術(shù)有限公司

本文主要講述java常用的單元測試框架，包括junit5、Mockito、SpringBootTest框架，通過講解各框架常用注解、使用樣例、注意事項，讓大家在開發(fā)過程中能夠快速、高效選擇適合自己的單元測試框架。

Part 01●??JUnit5框架?●?

1.1 Junit5介紹

Junit5需要Java 8或更高版本，和Junit4只是一個單獨的Jar包不同，目前的Junit5組成如下：JUnit5=JUnit Platform+JUnit Jupiter+JUnit Vintage

- JUnit Platform：

是Junit向測試平臺演進，提供平臺功能的模塊，通過JUnit Platform，其他的自動化測試引擎或開發(fā)人員自己定制的引擎都可以接入Junit實現(xiàn)對接和執(zhí)行

- JUnit Jupiter：

這是Junit5的核心，可以看作是承載Junit4原有功能的演進，它包含了很多豐富的新特性來使JUnit自動化測試更加方便、功能更加豐富和強大。

本系列就會重點圍繞Jupiter中的一些特性進行介紹。Jupiter本身也是一個基于Junit Platform的引擎實現(xiàn)。

- JUnit Vintage：

Junit發(fā)展了10數(shù)年，Junit3和Junit4都積累了大量的用戶，作為新一代框架，這個模塊是對JUnit3，JUnit4版本兼容的測試引擎，使舊版本Junit的自動化測試腳本也可以順暢運行在Junit5下，它也可以看作是基于Junit Platform實現(xiàn)的引擎范例。

1.2 測試實例生命周期介紹

@TestInstance(TestInstance.Lifecycle.PER_CLASS)

* PER_METHOD(默認)：JUnit在執(zhí)行每個測試方法之前創(chuàng)建每個測試類的新實例

* PER_CLASS：JUnit Jupiter在同一個測試實例上執(zhí)行所有測試方法，當使用這種模式時，每個測試類將創(chuàng)建一個新的測試實例。因此，如果您的測試方法依賴于存儲在實例變量中的狀態(tài)，則可能需要在@BeforeEach或@AfterEach方法中重置該狀態(tài)。

1.3 Junit5常用注解介紹

@BeforeAll

JUnit5@BeforeAll注釋是JUnit4中@BeforeClass注釋的替代。它用于表示應(yīng)在當前測試類中的所有測試方法之前執(zhí)行的帶的方法。

備注：@BeforeAll注釋的方法必須是靜態(tài)方法，否則會報運行時錯誤。

@BeforeEach

JUnit5@BeforeEach注釋替代了JUnit4中的@Before注釋。被注釋的方法會在當前類中的每個Test方法之前執(zhí)行。也就是說有多少個test這個方法就會執(zhí)行多少次。

備注：@BeforeEach注釋的方法一定不能是靜態(tài)方法，否則會報發(fā)運行時錯誤。

@AfterEach

JUnit5@AfterEach注釋是JUnit4中@After注釋的替換。它用于表示應(yīng)在當前類中的每個@Test方法之后執(zhí)行帶注釋的方法。

@AfterAll

JUnit5@AfterAll注釋是JUnit4中@AfterClass注釋的替換。它用于表示應(yīng)在當前測試類中的所有測試之后執(zhí)行帶注釋的方法。

備注：@AfterAll注釋的方法必須是靜態(tài)方法，否則會報運行時錯誤。

Junit5當中使用@BeforeEach替換了@Before使用 @AfterEach替換了@After

@Disabled可以用于不運行某些test的場景。

@Tag可以用于將測試分類。

JUnit Jupiter支持下列注解，用于配置測試和擴展框架。

所有核心注解位于junit-jupiter-api模塊中的org.junit.jupiter.api包中。

Part 02●??spring-boot-test框架?●

2.1 版本迭代

在JUnit4中，自定義框架通常意味著使用@RunWith注釋來指定一個自定義的運行器。使用多個運行器是有問題的。

Junit5在Spring boot 2.1.x之前，@SpringBootTest需要配合@ExtendWith(SpringExtension.class)才能正常工作的。

而在Spring boot 2.1.x之后，我們查看@SpringBootTest 的代碼會發(fā)現(xiàn)，其中已經(jīng)組合了@ExtendWith(SpringExtension.class)，因此，無需在進行該注解的使用了。

Spring Boot 2.2.0版本開始引入JUnit5作為單元測試默認庫

Junit5中包含JUnit Vintage：這個模塊是兼容JUnit3、JUnit4版本的測試引擎，使得舊版本的自動化測試也可以在JUnit5下正常運行。防止使用舊的junit4相關(guān)接口，可以進行依賴排除，如下圖：

SpringBoot 2.4以上版本移除了默認對Vintage的依賴。如果需要兼容JUnit4.x版本，需要自行引入。

SpringBootTest默認集成了以下功能：

備注：JUnit4前移注意事項

2.2 Spring Boot Test中的主要注解

從功能上講，Spring Boot Test中的注解主要分如下幾類：

2.2.1 配置類型的注解

使用@SpringBootApplication啟動測試或者生產(chǎn)代碼，被@TestComponent描述的Bean會自動被排除掉。如果不是則需要向@SpringBootApplication添加TypeExcludeFilter。

2.2.2 mock類型的注解

@MockBean和@SpyBean這兩個注解，在mockito框架中本來已經(jīng)存在，且功能基本相同。Spring Boot Test又定義一份重復(fù)的注解，目的在于使MockBean和SpyBean被ApplicationContext管理，從而方便使用。MockBean和SpyBean功能非常相似，都能模擬方法的各種行為。不同之處在于MockBean是全新的對象，跟正式對象沒有關(guān)系；而SpyBean與正式對象緊密聯(lián)系，可以模擬正式對象的部分方法，沒有被模擬的方法仍然可以運行正式代碼。

@MockBean 只能 mock 本地的代碼——或者說是自己寫的代碼，對于儲存在庫中而且又是以 Bean 的形式裝配到代碼中的類無能為力。

@SpyBean 解決了 SpringBoot 的單元測試中

@MockBean 不能 mock 庫中自動裝配的 Bean 的局限

2.2.3 自動配置類型的注解（@AutoConfigure*）

這些注解可以搭配@*Test使用，用于開啟在@*Test中未自動配置的功能。例如@SpringBootTest和@AutoConfigureMockMvc組合后，就可以注入org.springframework.test.web.servlet.MockMvc。

2.2.3.1 自動配置類型有兩種使用方式

a.在功能測試（即使用@SpringBootTest）時顯示添加。

b.一般在切片測試中被隱式使用，例如@WebMvcTest注解時，隱式添加了@AutoConfigureCache

@AutoConfigureWebMvc

@AutoConfigureMockMvc。

2.2.4 啟動測試類型的注解

所有的@*Test注解都被@BootstrapWith注解，它們可以啟動。

ApplicationContext，是測試的入口，所有的測試類必須聲明一個@*Test注解。

除了@SpringBootTest之外的注解都是用來進行切面測試的，他們會默認導(dǎo)入一些自動配置。一般情況下，推薦使用@SpringBootTest而非其它切片測試的注解，簡單有效。若某次改動僅涉及特定切片，可以考慮使用切片測試。

2.2.5 常用配置介紹

@SpringBootTest，其中包含的配置項如下：

@WebEnvironment，其中包含的配置項如下：

Part 03●? Mockito框架?●

3.1 常用的Mockito方法

3.2 Mockito參數(shù)適配方法

Mockito.anyString()Mockito.anyInt()Mockito.anyLong()Mockito.anyDouble()Mockito.anyObject()   表示任何對象Mockito.any(clazz)     表示任何屬于clazz的對象Mockito.anyCollection()Mockito.anyCollectionOf(clazz)Mockito.anyList(Map, set)Mockito.anyListOf(clazz)

注：Mockito.anyString() 不能匹配到 null 參數(shù)，如果還需要匹配 null，可以使用 Mockito.any()。

Part 04●?單元測試樣例?●

4.1 Mock redis、kafka方法

方法1：

@SpringBootTest通過@Resource引入對象測試，需要依賴redis環(huán)境（會啟動spring boot 容器）

方法2：

//聲明變量private  AsyncService asyncService;//需要mock的對象private StringRedisTemplate stringRedisTemplate;//創(chuàng)建要測試對象asyncService = new AsyncServiceImpl();//mock對象(也可以使用@Mock注解方式)StringRedisTemplate stringRedisTemplate =mock(StringRedisTemplate.class,Mockito.RETURNS_DEEP_STUBS);KafkaProducer kafkaProducer = mock(KafkaProducer.class);//屬性賦值ReflectionTestUtils.setField(asyncService,"stringRedisTemplate", stringRedisTemplate);ReflectionTestUtils.setField(asyncService,"kafkaProducer", kafkaProducer);@Test@DisplayName("mock redis、kafka 測試")public void redisTest() {

when(stringRedisTemplate.opsForValue().get(anyString())).thenReturn("2222");

   Assertions.assertTrue(asyncService.testRedis("真實方法調(diào)用"));}

4.2 Spring Security模擬登錄方法

<dependency> 

 <groupId>org.springframework.security</groupId>    <artifactId>spring-security-test</artifactId>     <version>5.6.5</version>     <scope>test</scope> </dependency> 

注解：@WithMockUser(roles = "ROOT", username = "root")

4.3 遠程接口調(diào)用方法（Controller入口測試）

MockMVC的基本步驟

(1) mockMvc.perform執(zhí)行一個請求。

(2) MockMvcRequestBuilders.get("XXX")構(gòu)造一個請求。

(3) ResultActions.param添加請求傳值

(4) ResultActions.accept()設(shè)置返回類型

(5) ResultActions.andExpect添加執(zhí)行完成后的斷言。

(6) ResultActions.andDo添加一個結(jié)果處理器，表示要對結(jié)果做點什么事情，比如處使用print()輸出整個響應(yīng)結(jié)果信息。

(7) ResultActions.andReturn表示執(zhí)行完成后返回相應(yīng)的結(jié)果。