引言

在編寫測試用例的時(shí)候遇到有異步或者隊(duì)列處理的時(shí)候經(jīng)常會(huì)用到 Thread.sleep() 等待來進(jìn)行測試。例如：DLedger 測試選舉的過程。當(dāng)DLedger Leader下線。此時(shí)DLedger會(huì)重新發(fā)起選舉，這個(gè)選舉的過程是需要一定時(shí)間。很多時(shí)候在測試代碼中就會(huì)使用 Thread.sleep 。

由于選舉需要的時(shí)間多少不確定所以sleep時(shí)間就會(huì)設(shè)置為開發(fā)者經(jīng)驗(yàn)的最大值。這樣會(huì)造成測試代碼會(huì)變得很慢。 當(dāng)然開發(fā)者可以通過自己輪詢來實(shí)現(xiàn)減少時(shí)間的消耗。

下面介紹一個(gè)處理這個(gè)一類問題的工具：awaitility

1. awaitility入門

Maven：

<dependency>
      <groupId>org.awaitility</groupId>
      <artifactId>awaitility</artifactId>
      <version>xxxx</version>
      <scope>test</scope>
</dependency>

文章編寫的時(shí)候版本為：4.2.0

1.1 靜態(tài)導(dǎo)入

為了有效地使用Awaitility，建議從Awaitility框架中靜態(tài)地導(dǎo)入以下方法：

org.awaitility.Awaitility.*

在使用的時(shí)候需要搭配Java的時(shí)間相關(guān)的類以及Junit相關(guān)類：

java.time.Duration.*

java.util.concurrent.TimeUnit.*

org.junit.Assert.*

1.2 簡單例子

例子1：

await().until(newUserIsAdded());

等待直到執(zhí)行newUserIsAdded()返回true. 這個(gè)是沒有返回值的。

例子2：

await().atMost(5, SECONDS).until(newUserWasAdded());

最多等待5秒，等待直到執(zhí)行newUserIsAdded()返回true. 這個(gè)是沒有返回值的。

例子3：

await().until( userRepositorySize(), equalTo(1) );

等待直到執(zhí)行userRepositorySize()返回方法對(duì)應(yīng)的值. 這個(gè)是有返回值

更多的例子可以參照官網(wǎng) 使用例子

2. awaitility在RocketMQ中的實(shí)戰(zhàn)

在RocketMQ的test cases 中有一些使用了 Thread.sleep，接下來我們看看如何使用awaitility進(jìn)行優(yōu)化，減少測試用例的執(zhí)行時(shí)間。以ControllerManagerTest測試用例為例子來解決，在代碼中可以看到有這樣的代碼：

上圖框出來的代碼主要的作用是什么呢？等待Broker的Master過期，但是過期的時(shí)間我們根據(jù)設(shè)置的心跳的過期時(shí)間來預(yù)估時(shí)間。所以這里填寫的是6秒當(dāng)然你也可以填寫10秒或者更長。

解決之前的執(zhí)行時(shí)間：

使用awaitility對(duì)代碼進(jìn)行改造重構(gòu)：

重構(gòu)后的代碼，如上圖的紅線框出來部分。當(dāng)然我這里還對(duì)其他的進(jìn)行處理。

有興趣的可以關(guān)注一下RocketMQ的這個(gè)ISSUE:github.com/apache/rock…

使用awaitility重構(gòu)后的執(zhí)行時(shí)間：

時(shí)間有明顯的下降。相比之前的下降了5秒左右。

3. 總結(jié)

• 在測試過程中引入awaitility能夠很大程度上方便測試，無需要每次都憑經(jīng)驗(yàn)去預(yù)估時(shí)間。并且很多時(shí)候這個(gè)Thread.sleep的時(shí)間不是很好估算。減少了單元測試執(zhí)行的時(shí)間。特別是像RocketMQ這樣大型的項(xiàng)目單元測試很多。并且很多都是去測試分布式的，如果使用Thread.sleep會(huì)導(dǎo)致整個(gè)單元測試的時(shí)間很長。
• 無需自己去實(shí)現(xiàn)輪詢來減少單元測試的時(shí)間。

以上就是Awaitility同步異步工具實(shí)戰(zhàn)示例詳解的詳細(xì)內(nèi)容，更多關(guān)于Awaitility同步異步工具實(shí)的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

最新評(píng)論