Java單元測試Powermockito和Mockito使用總結(jié)

更新時間：2021年09月08日 11:54:48 作者：LJWLgl

公司單元測試框架選用了Junit 4.12，Mock框架選用了Mockito和PowerMock，本文主要介紹了Java單元測試Powermockito和Mockito使用總結(jié)，感興趣的可以了解一下

最近公司在推進(jìn)Java應(yīng)用的單元測試，要求將單元測試的覆蓋率提高到50%以上，保證上線代碼充分自測。公司單元測試框架選用了Junit 4.12，Mock框架選用了Mockito和PowerMock，同時選用JaCoCo來做覆蓋率檢測，下面詳細(xì)介紹一下我在使用這幾個框架的一些經(jīng)驗。

依賴引入

<dependency>
    <groupId>org.mockito</groupId>
    <artifactId>mockito-core</artifactId>
    <version>2.8.9</version>
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-module-junit4</artifactId>
    <version>1.7.4</version>
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-api-mockito2</artifactId>
    <version>1.7.4</version>
    <scope>test</scope>
</dependency>

PowerMockito的使用

Mockito、EasyMock、JMock等比較流行Mock框架有個共同的缺點(diǎn)，都不能mock靜態(tài)、final、私有方法等，而PowerMock可以完美解決以上框架的不足，接下來讓我們看看無所不能的PowerMock是如何解決上述問題，我們先從一個實例來看，下面的代碼是需要單測的類，被測類中既有對Spring Bean的調(diào)用，同時又包含對Redis的靜態(tài)讀取，另外還有g(shù)etInstance()單例類的使用，現(xiàn)在，我們一一來mock上述的外部類。

被單測類（主要測試queryStudentScoreByKeyword方法）

@Component
public class StudentService {

    @Resource
    private StudentDao studentDao;

    public List<StudentBo> queryStudentScoreByKeyword(String name) {
        System.out.println("invoke StudentService.queryStudentScoreByKeyword ...");

        List<StudentBo> cacheList = RedisUtils.getArray(name, StudentBo.class);
        if (CollectionUtils.isNotEmpty(cacheList)) {
            return cacheList;
        }
        String keyword = processKeyword(name);
        List<Student> students = studentDao.queryStudentByKeyWord(keyword);
        List<Integer> ids = students.stream().map(Student::getId).collect(Collectors.toList());
        List<Person> personList = SchoolManageProxy.getInstance().queryPerson(ids);

        List<StudentBo> studentBos = CommonUtils.toBo(personList, students);
        // 高亮結(jié)果
        highlightResult(studentBos, name);
        // 緩存到Redis
        RedisUtils.setArray(name, studentBos, 10 * 60);
        return studentBos;
    }

    private String processKeyword(String name) {
        System.out.println("invoke StudentService.processKeyword ...");
        String newName = name;
        // do somethings
        return newName;
    }

    private void highlightResult(List<StudentBo> result, String name) {
        System.out.println("invoke StudentService.highlightResult ...");
        // do keyword highlight
    }

}

單測類

@RunWith(PowerMockRunner.class)
@PrepareForTest({SchoolManageProxy.class, RedisUtils.class, StudentService.class})
// @PowerMockIgnore({"javax.management.*", "javax.net.ssl.*"})
@SuppressStaticInitializationFor({"cn.ganzhiqiang.ares.unittest.SchoolManageProxy"})
public class StudentServiceTest {

    @Mock
    private StudentDao mockStudentDao;

    @InjectMocks
    private StudentService studentServiceUnderTest;

    @Before
    public void setUp() {
        initMocks(this);
    }

    @Test
    public void testQueryStudentScoreByKeyword() throws Exception {
        studentServiceUnderTest = PowerMockito.spy(studentServiceUnderTest);
        PowerMockito.mockStatic(RedisUtils.class);
        PowerMockito.mockStatic(SchoolManageProxy.class);

        // mock單例調(diào)用
        SchoolManageProxy mockSchoolManageProxy = PowerMockito.mock(SchoolManageProxy.class);
        PowerMockito.when(SchoolManageProxy.getInstance()).thenReturn(mockSchoolManageProxy);
        when(mockSchoolManageProxy.queryPerson(anyList())).thenReturn(Collections.emptyList());

        // mock掉對Redis的靜態(tài)調(diào)用
        PowerMockito.when(RedisUtils.getArray(eq("tom"), eq(StudentBo.class))).thenReturn(Collections.emptyList());
        // 顯示的mock掉靜態(tài)的void的方法（可以不mock）
        PowerMockito.doNothing().when(RedisUtils.class, "setArray", anyString(), anyList(), anyInt());

        // mock私有方法processKeyword
        PowerMockito.doReturn("tom").when(studentServiceUnderTest, "processKeyword", anyString());

        // 跳過私有方法highlightResult的執(zhí)行
        PowerMockito.suppress(PowerMockito.method(StudentService.class, "highlightResult"));

        // 使用Mockito來mock服務(wù)的調(diào)用
        when(mockStudentDao.queryStudentByKeyWord(anyString())).thenReturn(Collections.emptyList());

        // Run the test
        final List<StudentBo> result = studentServiceUnderTest.queryStudentScoreByKeyword("tom");

    }
}

使用mockito來mock實例

首選我們先用Mockito來mock對Spring Bean的調(diào)用，Mockito.mock可以mock一個實例，我們這里選用@Mock注解，效果是一樣的。

// 使用Mockito來mock服務(wù)的調(diào)用 
when(mockStudentDao.queryStudentByKeyWord(anyString())).thenReturn(Collections.emptyList());

mock對Redis的靜態(tài)調(diào)用

接下來我們使用PowerMock來mock對靜態(tài)方法的調(diào)用，注意需要將RedisUtils類，加入@PrepareForTest注解中，我們既mock了getArray方法，也mock了setArray方法，其實setArray不需要mock這里顯式的mock了

PowerMockito.mockStatic(RedisUtils.class);
// mock掉對Redis的靜態(tài)調(diào)用
PowerMockito.when(RedisUtils.getArray(eq("tom"), eq(StudentBo.class))).thenReturn(Collections.emptyList());
// 顯式的mock掉靜態(tài)的void的方法（可以不mock）
PowerMockito.doNothing().when(RedisUtils.class, "setArray", anyString(), anyList(), anyInt());

mock單例類

mock單例類相對來說復(fù)雜一點(diǎn)，邏輯上先用Powermock mock出單例類，然后再給單例類的getInstance方法打樁，返回之前mock，再對mock類實際調(diào)用的方法打樁即可，代碼如下

PowerMockito.mockStatic(SchoolManageProxy.class);
// Powermock mock出單例類
SchoolManageProxy mockSchoolManageProxy = PowerMockito.mock(SchoolManageProxy.class);
// 給單例類的getInstance方法打樁

PowerMockito.when(SchoolManageProxy.getInstance()).thenReturn(mockSchoolManageProxy);
// 對mock類queryPerson的方法打樁
when(mockSchoolManageProxy.queryPerson(anyList())).thenReturn(Collections.emptyList());

mock私有方法

可以看到queryStudentScoreByKeyword方法調(diào)用了該類的私有方法processKeyword，如果該方法耗時過長，使用powermock也可以mock該私有方法，需要注意的studentServiceUnderTest需要用spy()來mock

// mock 實例
// spy的標(biāo)準(zhǔn)是：如果不打樁，默認(rèn)執(zhí)行真實的方法，如果打樁則返回樁實現(xiàn)。
studentServiceUnderTest = PowerMockito.spy(studentServiceUnderTest);
// mock私有方法processKeyword
// doReturn(...) when(...)不做真實調(diào)用，但是when(...) thenReturn(...)還是會真實調(diào)用原方法，只是返回了指定的結(jié)果
PowerMockito.doReturn("tom").when(studentServiceUnderTest, "processKeyword", anyString());

PowerMock跳過方法執(zhí)行

使用PowerMock也可以跳過私有方法的執(zhí)行

// 跳過私有方法highlightResult的執(zhí)行
PowerMockito.suppress(PowerMockito.method(StudentService.class, "highlightResult"));

總結(jié)

筆者之前寫代碼很少會寫單測，自從公司強(qiáng)制要求提高單測覆蓋率之后，雖然開發(fā)效率變慢了，但是確實引起我對單測的重視，進(jìn)而研究了一下PowerMockito、Mokcito等Mock框架。Powermock之所以無所不能，是因為它使用了自定義的加載器和字節(jié)碼操作技術(shù)，與此同時，它還十分簡單易用，確實是個很優(yōu)秀的框架。

Demo地址：https://github.com/LJWLgl/mock-data

參考文檔

PowerMock
powermockito單元測試之深入實踐
 淺談測試之PowerMock
無所不能的PowerMock，mock私有方法，靜態(tài)方法，測試私有方法，final類
 Mock和Spy的區(qū)別

到此這篇關(guān)于Java單元測試Powermockito和Mockito使用總結(jié)的文章就介紹到這了,更多相關(guān)Java Powermockito和Mockito 內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: