單元測試之Mockito與PowerMock

什么是Mock?

在單元測試中,我們往往想去獨立地去測一個類中的某個方法,但是這個類可不是獨立的,它會去調(diào)用一些其它類的方法和service,這也就導致了以下兩個問題:

  • 外部服務可能無法在單元測試的環(huán)境中正常工作,因為它們可能需要訪問數(shù)據(jù)庫或者使用一些其它的外部系統(tǒng)。
  • 我們的測試關注點在于這個類的實現(xiàn)上,外部類的一些行為可能會影響到我們對本類的測試,那也就失去了我們進行單測的意義。

為了解決這種問題,Mockito和PowerMock誕生了。這兩種工具都可以用一個虛擬的對象來替換那些外部的、不容易構造的對象,便于我們進行單元測試。兩者的不同點在于Mockito對于大多數(shù)的標準單測case都很適用,而PowerMock可以去解決一些更難的問題(比如靜態(tài)方法、私有方法等)。

Mockito

話不多說,就上干貨。在下面這個例子中,我們要測試的class是DummyService,DummyRepository用來從數(shù)據(jù)庫中獲取數(shù)據(jù)。getDummyNameUpperCase()方法會根據(jù)id獲取數(shù)據(jù),且該方法有一些數(shù)據(jù)處理的邏輯,如果我們只想測試這個邏輯:

public class DummyServiceTest {
 
    @Mock
    private DummyRepository dummyRepository;
 
    @InjectMocks
    private DummyService dummyService;
 
    @Before
    public void setUp() throws Exception {
        MockitoAnnotations.initMocks(this);
    }
 
    @Test
    public void getDummyNameUpperCase() {
 
        Dummy mockedDummy = new Dummy("dm1", "Dummy 1", "This is dummy 1", 2000, true);
        when(dummyRepository.findById("dm1")).thenReturn(mockedDummy);
 
        String result = dummyService.getDummyNameUpperCase("dm1");

        verify(dummyRepository, times(1)).findById("dm1");
        assertThat(result, is("DUMMY 1"));
    }
}

根據(jù)經(jīng)驗,在大多數(shù)項目中,都會使用一些依賴注入框架(PicoContainer、Avalon、Excalibur等),因此這里的例子是基于Spring且我們使用了對應的Mockito注釋@Mock來創(chuàng)建虛擬對象。對于被測class,使用@injectMocks進行注釋即可。

實際上,mock的發(fā)生時機是當程序執(zhí)行至”when-method-call-then-return”之類的語句時,上例中,也就是執(zhí)行到when(dummyRepository.findById("dm1")).thenReturn(mockedDummy)時,mock就已經(jīng)發(fā)生了;當程序運行至verify(dummyRepository, times(1)).findById("dm1")類的語句時,我們可以確認的是方法已經(jīng)被調(diào)用了,這在測試一個類的邏輯流程時格外地有效。

上例展示了Mockito的基本用法,然后實際測試可能不會這么簡單。比如我們需要將對象傳遞給mock的對象作為參數(shù)時,需要使用doAnswer

@Test
public void testAddressCombination() {
 

    Customer customer = new Customer("100089", "Zhang San", "1 Zhongguancun East Rd");
 
    doAnswer(new Answer<Customer>() {
        @Override
        public Customer answer(InvocationOnMock invocation) throws Throwable {
            Object originalArgument = (invocation.getArguments())[0];
            Customer param = (Customer) originalArgument;
            param.setHost("TestHost");
            return null;
        }
    }).when(hostService).expand(any(Customer.class));
 
    when(addressService.getPLZForCustomer(customer)).thenReturn(47891);
    doNothing().when(addressService).updateExternalSystems(customer);
 

    String address = customerService.getAddressCombination(customer, true);
 

    Mockito.verify(addressService, times(1)).updateExternalSystems(any(Customer.class));
    assertThat(address, is("47891_1 Zhongguancun East Rd_TestHost"));
}

以上的兩個例子足以延伸并cover掉大部分普通的單測case,但是如果方法是靜態(tài)的該怎么辦呢?那就是——PowerMock。

PowerMock

PowerMock基本上cover了所有Mockito不能支持的case(大多數(shù)情況也就是靜態(tài)方法,但其實也可以支持私有方法和構造函數(shù)的調(diào)用)。PowerMock使用了字節(jié)碼操作,因此它是自帶Junit runner的。在使用PowerMock時,必須使用@PrepareForTest注釋被測類,mock才會被執(zhí)行:

@RunWith(PowerMockRunner.class)
@PrepareForTest({StaticService.class})
public class DummyServiceTest {
 
    @Mock
    private DummyRepository dummyRepository;
 
    @InjectMocks
    private DummyService dummyService;
 
    @Before
    public void setUp() throws Exception {
        MockitoAnnotations.initMocks(this);
    }
 
    @Test
    public void readDummyDescriptionWithoutIOException() throws IOException {
 
        String fileName = "DummyName";
 
        mockStatic(StaticService.class);
        when(StaticService.readFile(fileName)).thenReturn("Dummy");
 
        String value = dummyService.readDummyDescription(fileName);
 
        verifyStatic(times(1));
        StaticService.readFile(fileName);
        assertThat(value, equalTo("Dummy"));
    }
}

從上例可以看出,PowerMock的寫法與Mockito基本相同,主要原因是PowerMock其實是從Mockito的一個特殊的API衍化而來,我們從Maven的摘錄中可以看到,我們不僅導入了PowerMock Junit-module,還導入了Mockito-API:

<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-module-junit4</artifactId>
    <version>1.6.4</version>
    <scope>test</scope>
</dependency>
 
<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-api-mockito</artifactId>
    <version>1.6.4</version>
    <scope>test</scope>
    </dependency>

注意,被測類如果包含了靜態(tài)方法,必須使用mockStatic()方法;與Mockito相似,當程序執(zhí)行至when-then-mocking類語句時,我們可以認為方法被成功調(diào)用了。

PowerMock另外一個很好用的特性是當使用@PowerMockRunnerDelegate作為代理。比如在Spring中一般使用的是SpringJUnit4ClassRunner,但是這樣@PrepairForTest就不能用了,因此:

@RunWith(PowerMockRunner.class)
@PowerMockRunnerDelegate(Parameterized.class)
@PrepareForTest({StaticService.class})
public class DummyServiceTest {
...
}

總結

Mockito提供了可讀性較強、易于使用的mocking測試方法,而PowerMock提供了Mockito-like API是它使用方法基本與Mockito有很大類似,且它能解決靜態(tài)方法、私有方法等“問題方法”

最后編輯于
?著作權歸作者所有,轉載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 227,663評論 6 531
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 98,125評論 3 414
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事?!?“怎么了?”我有些...
    開封第一講書人閱讀 175,506評論 0 373
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經(jīng)常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 62,614評論 1 307
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,402評論 6 404
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 54,934評論 1 321
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,021評論 3 440
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,168評論 0 287
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 48,690評論 1 333
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 40,596評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,784評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,288評論 5 357
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 44,027評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,404評論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,662評論 1 280
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,398評論 3 390
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,743評論 2 370

推薦閱讀更多精彩內(nèi)容