摘要：在對邏輯進行重構的時候可以直接通過單元測試能把控整個重構的邏輯不發生重大偏差，給后來者造福了。服務單元測試下面是一段獲取緩存邏輯的單元測試。

單元測試在開發的過程中可能會被很多的人所忽略，其實也不是真的被忽略可能是由于巨大的業務壓力而導致沒有時間去寫那些測試，而是直接通過集成測試完沒問題就提交給測試進行測試后上線了。這樣其實很多時候反而會浪費大量的時間在測試上面，而適量的寫好單元測試有以下一些優勢：

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

在對邏輯進行重構的時候可以直接通過單元測試能把控整個重構的邏輯不發生重大偏差，給后來者造福了。

下面是一段獲取緩存邏輯的單元測試。CacheManager中封裝了邏輯從緩沖中取數據，如果數據沒有取到則從repositoryCache.loadStoreByKey(key)中load數據。

public void getValueIfNullLoadTest() {
    /* 從repository里面取數據 */
    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,當調用repositoryCache的loadStoreByKey或者getCacheName方法后返回測試數據。

后面執行CacheManager.getValueIfNullLoad中正常的業務邏輯。

最后判斷結果是否符合預期。

上面一小節只是針對普通的服務單元進行測試，但是遇到http的接口測試就無能為力了，下面來介紹下如果寫http接口的單元測試。

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();
                    //針對結果進行驗證。
                });
        Mockito.verify(userService, times(1)).getAll();
        Mockito.verifyNoMoreInteractions(userService);
    }

}

mock一個服務userService

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

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

后面就是模擬發送http請求，然后驗證response的操作和上一節的類似。

最后是驗證mock服務的接口調用次數。

可能你也會在項目中看到很多的單元測試，但是隨著項目的迭代那些單元測試已經失效，又或者是你會看到在進行單元測試的時候會去啟動整個項目容器去做運行，不過只要有就是好的，后面在進行項目迭代的過程中別忘了把unit test寫上吧。

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