我們平時寫單元測試時經(jīng)常會遇到調(diào)用抽象類或父類的方法，這些抽象方法可能是調(diào)用底層接口或數(shù)據(jù)庫，需要mock掉，讓抽象方法返回一個我們指定的值，以便測試當(dāng)前代碼邏輯的場景。

下面講下Spock如何結(jié)合power mock實(shí)現(xiàn)動態(tài)mock抽象方法

一. 抽象方法或父類方法動態(tài)mock

AbstractService?是個抽象類，我們需要把它的方法?parentMethod?模擬掉，返回我們預(yù)先設(shè)置的"期望值"?

代碼示例：

public?abstract?class?AbstractService?{String parentMethod(){// 發(fā)起接口調(diào)用或數(shù)據(jù)庫操作return?"parentMethod value";
????}
}

SubService是繼承AbstractService?的子類，在doSomething方法體里會調(diào)用抽象類的?parentMethod?方法邏輯，然后根據(jù)抽象方法?parentMethod?的返回值走不同的邏輯：

public?class?SubService?extends?AbstractService?{
????@Autowired
????MoneyDAO moneyDAO;public?String doSomething() {
????????String parent?= super.parentMethod(); // 調(diào)用抽象類或父類方法if?("parent1".equals(parent)) {// 執(zhí)行parent1分支邏輯return?"sub1";
????????}if?("parent2".equals(parent)) {// 執(zhí)行parent2分支邏輯return?"sub2";
????????}if?("parent3".equals(parent)) {// 執(zhí)行parent3分支邏輯return?"sub3";
????????}return?"other";
????}
}

如果要mock掉抽象類AbstractService?中的?parentMethod?方法, 并且每次mock的值不一樣, 可以使用spock + powermock來實(shí)現(xiàn):

單元測試代碼如下：

/**
?* 測試抽象類方法或父類方法
?* @Author: www.javakk.com
?* @Description: 公眾號:Java老K
?* @Date: Created in 14:53 2020/10/05
?* @Modified?By:
?*/@RunWith(PowerMockRunner.class)@PowerMockRunnerDelegate(Sputnik.class)@PrepareForTest([SubService.class])class?AbstractServiceTest?extends?Specification?{@Unroll
????def "測試抽象方法"() {
????????given: "mock抽象類方法"
????????def sub = PowerMockito.mock(SubService)
????????PowerMockito.when(sub.parentMethod()).thenReturn(parentValue) // mock掉抽象類的parentMethod, 返回動態(tài)mock值:mockParentReturn
????????PowerMockito.when(sub.doSomething()).thenCallRealMethod()
????????expect: "調(diào)用doSomething方法"
????????sub.doSomething() == result
????????where: "驗(yàn)證分支場景"
????????parentValue | result"parent1"???| "sub1""parent2"???| "sub2""parent3"???| "sub3""parent4"???| "other"
????}
}

使用power mock模擬掉抽象類的方法，返回一個變量parentValue，然后再放在Spock的where標(biāo)簽里，即可實(shí)現(xiàn)動態(tài)mock的效果，即每次調(diào)用返回的mock值都不一樣：parent1、parent2、parent3 ...

二. 抽象方法+實(shí)例方法的動態(tài)mock

如果在SubService中還有引用其他實(shí)例對象的方法，比如下面的業(yè)務(wù)代碼：

public?class?SubService?extends?AbstractService?{
????@Autowired
????MoneyDAO moneyDAO; // 金額換算對象public?String doSomethingAndDao() {
????????String parent?= super.parentMethod(); // 調(diào)用抽象類或父類方法
????????BigDecimal money = moneyDAO.getExchangeByCountry(parent); // 獲取對應(yīng)國家的金額if?("parent1".equals(parent)) {return?money + " CNY";
????????}if?("parent2".equals(parent)) {return?money + " USD";
????????}if?("parent3".equals(parent)) {return?money + " EUR";
????????}return?money.toString();
????}
}

如果即要mock掉抽象類AbstractService?中的?parentMethod?方法，又要mockmoneyDAO對象，可以使用 Whitebox.setInternalState 方式

單元測試代碼如下：

/**
?* 測試抽象類方法或父類方法
?* @Author: www.javakk.com
?* @Description: 公眾號:Java老K
?* @Date: Created in 14:53 2020/10/05
?* @Modified?By:
?*/@RunWith(PowerMockRunner.class)@PowerMockRunnerDelegate(Sputnik.class)@PrepareForTest([SubService.class])class?AbstractServiceTest?extends?Specification?{@Unroll
????def "測試抽象方法和實(shí)例方法"() {
????????given: "mock抽象類方法"
????????def sub = PowerMockito.mock(SubService)// mock掉抽象類的parentMethod, 返回動態(tài)mock值:mockParentReturn
????????PowerMockito.when(sub.parentMethod()).thenReturn(parentValue)
????????PowerMockito.when(sub.doSomethingAndDao()).thenCallRealMethod()
????????def moneyDAO = Mock(MoneyDAO)//將Spockmock的對象moneyDAO使用powermock賦值給SubService的引用moneyDAO
????????Whitebox.setInternalState(sub, "moneyDAO", moneyDAO)
????????moneyDAO.getExchangeByCountry(_) >> money // 這樣就可以使用spock的動態(tài)mock
????????expect: "調(diào)用doSomething方法"
????????sub.doSomethingAndDao() == result
????????where: "驗(yàn)證分支場景"
????????parentValue | money || result"parent1"???| 100???|| "100 CNY""parent2"???| 200???|| "200 USD""parent3"???| 300???|| "300 EUR""parent4"???| 400???|| "400"
????}
}

關(guān)于動態(tài)mock的更多用法可以參考這篇文章：Spock高級用法 - 動態(tài)mock

(完整的代碼已上傳到github，在公眾號里回復(fù)spock即可獲取github項(xiàng)目地址)

-END-推薦閱讀：Spock高級用法 - 動態(tài)mockSpock代碼講解 - 靜態(tài)方法測試Spock代碼講解 - void方法測試Spock代碼講解 - 異常測試Spock代碼講解 - if esle 分支場景測試

---

互聯(lián)網(wǎng)一線java開發(fā)老兵，工作10年有余，夢想敲一輩子代碼，以夢為碼，不負(fù)韶華。

掃碼關(guān)注Java老K，獲取更多Java干貨。

總結(jié)

以上是生活随笔為你收集整理的mockito mock void方法_Spock如何模拟抽象类方法的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。