跳到主要內容

Mockito - Get Mock Class Type

Problem

在mock factory所產生出來的東西時，會希望寫成generic形式以reuse:

private <T extends Command> CommandCreator mockCommandCreator(T mockCommand){
	CommandCreator cmdCreator = mock(CommandCreator.class);
 
	oReturn(mockCommand).when(cmdCreator);
	cmdCreator.createCommand(mockCommand.getClass());
 
	return cmdCreator;
}

然而，針對mockCommand去getClass通常會拿到以下結果:

class org.tonylin.mockito.BatchCommand$$EnhancerByMockitoWithCGLIB$$6f5074eb

也因此無法達到mock效果。本篇文章分享解決這個問題的過程。

Lib Version

我們所使用的powermockito與mockito版本大致如下: (powermock項目比較多，省略部分)

POWERMOCK_MOCKITO_VERSION = "1.6.6"
POWERMOCK_API_MOCKITO_DEP = "org.powermock:powermock-api-mockito:$POWERMOCK_MOCKITO_VERSION"
POWERMOCK_API_MOCKITO_COM_DEP = "org.powermock:powermock-api-mockito-common:$POWERMOCK_MOCKITO_VERSION"

MOCKITO_VERSION = "1.10.19"
MOCKITO_CORE_DEP = "org.mockito:mockito-core:$MOCKITO_VERSION"
	
HAMCREST_VERSION = "1.3"
HAMCREST_DEP = "org.hamcrest:hamcrest-library:$HAMCREST_VERSION"

How to?

GetSuperclass

首先我嘗試使用getSuperclass作法，假如目標類別是concrete class就可以達到我要的效果:

// concrete class
BatchCommand mockCommand2 = mock(BatchCommandImpl.class); 
System.out.println(mockCommand2.getClass());
System.out.println(mockCommand2.getClass().getSuperclass());
// outut
class org.tonylin.mockito.BatchCommandImpl$$EnhancerByMockitoWithCGLIB$$b692c9d5
class org.tonylin.mockito.BatchCommandImpl

但如果目標是interface取到的SuperClass會是class java.lang.Object。如果你要mock的目標只有concrete class，可以透過mockingDetails去區別input的是否為mock object:

Class<? extends BatchCommandImpl> mockType = null;
if(mockingDetails(mockCommand).isMock()){
	mockType = (Class<? extends BatchCommandImpl>)mockCommand.getClass().getSuperclass();
} else {
	mockType = mockCommand.getClass();
}
 
doReturn(mockCommand).when(cmdCreator);
cmdCreator.createCommand(mockType);

Update Libraries

因為前一個方法我並不滿意，因此我找尋網路資料看看是否有解。後來在這篇文章中，看到mockingDetails中應有提供getMockedType的功能；然而實際試驗後，發現這個API只有在2.x beta版本中出現(參考javadoc)。

最後在mockito issue中，發現可以透過mockingDetails.getMockCreationSettings去拿到我要的資料。因此我將mockito升級到以下版本:

MOCKITO_VERSION = "2.18.3"
MOCKITO_CORE_DEP = "org.mockito:mockito-core:$MOCKITO_VERSION"

並撰寫以下程式碼實驗:

BatchCommand mockCommand = mock(BatchCommand.class); 
System.out.println(mockCommand.getClass());
System.out.println(mockCommand.getClass().getSuperclass());
System.out.println(mockingDetails(mockCommand).getMockCreationSettings().getTypeToMock());

結果出現以下例外: (只擷取有用部分)

Caused by: java.lang.NoClassDefFoundError: org/mockito/cglib/proxy/MethodInterceptor
	at org.powermock.api.mockito.internal.mockmaker.PowerMockMaker.<init>(PowerMockMaker.java:43)

我推測是library版本問題並找一下資料，只要替換powermock-api-mockito為powermock-api-mockito2就可以解決:

POWERMOCK_MOCKITO2_VERSION = "1.7.4"
POWERMOCK_API_MOCKITO2_DEP = "org.powermock:powermock-api-mockito2:$POWERMOCK_MOCKITO2_VERSION"

第二個方法目前可以正常使用。

Reference

標籤：

留言

張貼留言

這個網誌中的熱門文章

解決RobotFramework從3.1.2升級到3.2.2之後，Choose File突然會整個Hand住的問題

考慮到自動測試環境的維護，我們很久以前就使用java去執行robot framework。前陣子開始處理從3.1.2升級到3.2.2的事情，主要先把明確的runtime語法錯誤與deprecate item處理好，這部分內容可以參考: link 。 直到最近才發現，透過SeleniumLibrary執行Choose File去上傳檔案的動作，會導致測試案例timeout。本篇文章主要分享心路歷程與解決方法，我也送了一條issue給robot framework: link 。 我的環境如下: RobotFramework: 3.2.2 Selenium: 3.141.0 SeleniumLibrary: 3.3.1 Remote Selenium Version: selenium-server-standalone-3.141.59 首先並非所有Choose File的動作都會hang住，有些測試案例是可以執行的，但是上傳一個作業系統ISO檔案一定會發生問題。後來我透過wireshark去比對新舊版本的上傳動作，因為我使用 Remote Selenium ，所以Selenium會先把檔案透過REST API發送到Remote Selenium Server上。從下圖我們可以發現，在3.2.2的最後一個TCP封包，比3.1.2大概少了500個bytes。 於是就開始了我trace code之路。包含SeleniumLibrary產生要送給Remote Selenium Server的request內容，還有HTTP Content-Length的計算，我都確認過沒有問題。 最後發現問題是出在socket API的使用上，就是下圖的這支code: 最後發現可能因為開始使用nio的方式送資料，但沒處理到尚未送完的資料內容，而導致發生問題。加一個loop去做計算就可以解決了。 最後我有把解法提供給robot framework官方，在他們出新的版本之前，我是將改完的_socket.py放在我們自己的Lib底下，好讓我們測試可以正常進行。(shutil.py應該也是為了解某個bug而產生的樣子..)
張貼留言
閱讀完整內容

Show NIC selection when setting the network command with the device option

 Problem  在answer file中設定網卡名稱後，安裝時會停在以下畫面: 所使用的command參數如下: network --onboot = yes --bootproto =dhcp --ipv6 =auto --device =eth1 Diagnostic Result 這樣的參數，以前試驗過是可以安裝完成的。因此在發生這個問題後，我檢查了它的debug console: 從console得知，eth1可能是沒有連接網路線或者是網路太慢而導致的問題。後來和Ivy再三確認，有問題的是有接網路線的網卡，且問題是發生在activate階段: Solution 我想既然有retry應該就有次數或者timeout限制，因此發現在Anaconda的說明文件中( link )，有提到dhcptimeout這個boot參數。看了一些人的使用範例，應該是可以直接串在isolinux.cfg中，如下: default linux ksdevice = link ip =dhcp ks =cdrom: / ks.cfg dhcptimeout = 90 然而我在RHEL/CentOS 6.7與6.8試驗後都無效。 因此我就拿了顯示的錯誤字串，問問Google大師，想找一下Anaconda source code來看一下。最後找到別人根據Anaconda code修改的版本: link ，關鍵在於setupIfaceStruct函式中的setupIfaceStruct與readNetConfig: setupIfaceStruct: 會在dhcp時設定dhcptimeout。 readNetConfig: 在writeEnabledNetInfo將timeout寫入dhclient config中；在wait_for_iface_activation內會根據timeout做retry。 再來從log與code可以得知，它讀取的檔案是answer file而不是boot command line。因此我接下來的測試，就是在answer file的network command上加入dhcptimeout: network --onboot = yes --bootproto =dhcp --ipv6 =auto --device =eth1 --...
張貼留言
閱讀完整內容

Robot Framework - Evaluate該怎麼用?

Evaluate該怎麼用? 前言 Builtin的RobotFramework Library提供了Evaluate Keyword。它所提供的功能是「執行Python描述句」。但實際上到底有什麼用途呢?原本我僅僅拿來將string轉為int的功用，經過一些查詢與試驗，我將心得整理給大家。 Builtin Builtin的function可以參考Library Doc for Evaluate。我以有使用過的function做說明。 數字轉換 Python提供了int、long、float與complex等function讓你可以將字串轉為數字，也可以透過它們做四則運算。首先以字串轉數字為例，我將8設於${num_str}中，再透過Evaluate+int轉為數字。這裡必須注意的是: 「int()中放變數必須以單引號'括起」。否則，假如你設定的數字為08，在轉換int時會出現Syntax Error。 ${num_str} | Set Variable | 8 ${num} | Evaluate | int('${num_str}') 其中int與long的第二個參數為base，這是根據你的input所決定: Comment | num = 9 ${num} | Evaluate | int('11', 8) Comment | num = 11 ${num} | Evaluate | int('11', 10) Comment | num = 17 ${num} | Evaluate | int('11', 16) 其它還有像bin、oct、hex，可以將整數轉為2、8、16進位。 運算 四則運算: 直接將運算子加上即可: ${num} | Evaluate | int('${hour}')*60 + int('${min}') 指數: 可以用pow。以下面兩個例子來說，第一個是2的3次方為8，第二個是2的3次方再mod 7為1。需注意的是: 「傳入值必須是數字不可為字串」。 ${num} | Evaluate | pow(2,3) ${num} | Evaluate | pow(2,3,7) 取最大最小值: 使用max/min，可以選擇丟一個array的方式...
張貼留言
閱讀完整內容