摘要：在對(duì)邏輯進(jìn)行重構(gòu)的時(shí)候可以直接通過單元測(cè)試能把控整個(gè)重構(gòu)的邏輯不發(fā)生重大偏差，給后來者造福了。服務(wù)單元測(cè)試下面是一段獲取緩存邏輯的單元測(cè)試。

單元測(cè)試在開發(fā)的過程中可能會(huì)被很多的人所忽略，其實(shí)也不是真的被忽略可能是由于巨大的業(yè)務(wù)壓力而導(dǎo)致沒有時(shí)間去寫那些測(cè)試，而是直接通過集成測(cè)試完沒問題就提交給測(cè)試進(jìn)行測(cè)試后上線了。這樣其實(shí)很多時(shí)候反而會(huì)浪費(fèi)大量的時(shí)間在測(cè)試上面，而適量的寫好單元測(cè)試有以下一些優(yōu)勢(shì)：

有單元測(cè)試可能會(huì)提高整個(gè)集成測(cè)試的進(jìn)度而且最重要的是做到對(duì)自己寫的代碼知根知底，更有底氣的推到線上去。

在對(duì)邏輯進(jìn)行重構(gòu)的時(shí)候可以直接通過單元測(cè)試能把控整個(gè)重構(gòu)的邏輯不發(fā)生重大偏差，給后來者造福了。

下面是一段獲取緩存邏輯的單元測(cè)試。CacheManager中封裝了邏輯從緩沖中取數(shù)據(jù)，如果數(shù)據(jù)沒有取到則從repositoryCache.loadStoreByKey(key)中l(wèi)oad數(shù)據(jù)。

public void getValueIfNullLoadTest() {
    /* 從repository里面取數(shù)據(jù) */
    RepositoryCache repositoryCache = mock(RepositoryCache.class);
    String key = anyString();
    when(repositoryCache.loadStoreByKey(key)).thenReturn("test");
    when(repositoryCache.getCacheName()).thenReturn(Constants.CACHE);

    CacheManager.addRepository(repositoryCache);
    String value = CacheManager.getValueIfNullLoad(Constants.CACHE, key, String.class);

    Assert.assertTrue(StringUtils.equals(value, "test"));

    /* 從緩存中取 */
    value = CacheManager.getValueByCache(Constants.CACHE, key, String.class);
    Assert.assertTrue(StringUtils.equals(value, "test"));

}

mockrepositoryCache,當(dāng)調(diào)用repositoryCache的loadStoreByKey或者getCacheName方法后返回測(cè)試數(shù)據(jù)。

后面執(zhí)行CacheManager.getValueIfNullLoad中正常的業(yè)務(wù)邏輯。

最后判斷結(jié)果是否符合預(yù)期。

上面一小節(jié)只是針對(duì)普通的服務(wù)單元進(jìn)行測(cè)試，但是遇到http的接口測(cè)試就無能為力了，下面來介紹下如果寫http接口的單元測(cè)試。

public class UserControllerUnitTest {

    private MockMvc mockMvc;

    @Mock
    private UserService userService;

    @InjectMocks
    private UserController userController;

    @Before
    public void init(){
        MockitoAnnotations.initMocks(this);
        mockMvc = MockMvcBuilders
                .standaloneSetup(userController)
                .addFilters(new CORSFilter())
                .build();
    }

    @Test
    public void test_get_all_success() throws Exception {
        List users = Arrays.asList(
                new User(1, "Daenerys Targaryen"),
                new User(2, "John Snow"));
        when(userService.getAll()).thenReturn(users);
        mockMvc.perform(MockMvcRequestBuilders.get("/users"))
                .andExpect(MockMvcResultMatchers.status().isOk())
                .andExpect(content().contentType(MediaType.APPLICATION_JSON_UTF8_VALUE))
                .andExpect(mvcResult -> {
                    String responseStr = mvcResult.getResponse().getContentAsString();
                    //針對(duì)結(jié)果進(jìn)行驗(yàn)證。
                });
        Mockito.verify(userService, times(1)).getAll();
        Mockito.verifyNoMoreInteractions(userService);
    }

}

mock一個(gè)服務(wù)userService

InjectMocks userController后通過MockitoAnnotations.initMocks(this) 將userService這個(gè)mock的服務(wù)注入到userController中。

MockMvcBuilders.standaloneSetup(userController).build(); 創(chuàng)建MockMvc。

后面就是模擬發(fā)送http請(qǐng)求，然后驗(yàn)證response的操作和上一節(jié)的類似。

最后是驗(yàn)證mock服務(wù)的接口調(diào)用次數(shù)。

可能你也會(huì)在項(xiàng)目中看到很多的單元測(cè)試，但是隨著項(xiàng)目的迭代那些單元測(cè)試已經(jīng)失效，又或者是你會(huì)看到在進(jìn)行單元測(cè)試的時(shí)候會(huì)去啟動(dòng)整個(gè)項(xiàng)目容器去做運(yùn)行，不過只要有就是好的，后面在進(jìn)行項(xiàng)目迭代的過程中別忘了把unit test寫上吧。

Unit Test Spring MVC Rest Service: MockMVC, JUnit, Mockito