前言

我們在寫單元測試時，有一個比較重要的要求是可以重復(fù)運行，即只要外部參數(shù)不變，那么一定可以獲取確定的結(jié)果。 那么這樣就會有一個比較麻煩的問題：數(shù)據(jù)污染。 以數(shù)據(jù)庫操作為例，對于一些查詢類的測試用例倒還好，因為只要保證數(shù)據(jù)存在，那么查詢操作天生就是冪等的，不管你查多少次，數(shù)據(jù)都不會變。 但是對于寫入操作就不友好了，每一次運行單元測試，都會在數(shù)據(jù)庫中產(chǎn)生新的數(shù)據(jù)，可能在第一次運行時正常，第二次運行時就由于數(shù)據(jù)沖突導(dǎo)致失敗。那對于這種情況，我們應(yīng)該如何去解決？

下面分享兩種我使用過的數(shù)據(jù)隔離的方式，供大家參考。

數(shù)據(jù)隔離

測試事務(wù)

這種方式比較常規(guī)，適用場景也比較廣泛，即我們在測試用例中，通過@Transactional注解開始事務(wù)，此時整個單元測試方法會被包裹在事務(wù)中，并且會在運行結(jié)束后，自動回滾。

我們可以在類或方法上添加@Transactional注解

/**
 * 在類上添加 @Transactional 注解，可以讓測試方法在執(zhí)行完畢后自動回滾，不會對數(shù)據(jù)庫造成影響。
 * 如果在方法上添加，則只影響對應(yīng)的方法
 */
@SpringBootTest
@Transactional
public class TransactionalTests extends AbstractTransactionalJUnit4SpringContextTests {
    @Resource
    private UserService userService;
    @Test
    void createRecord() {
        User user = userService.createUser();
        //to other things
    }
}

或者繼承AbstractTransactionalJUnit4SpringContextTests類

/**
 * 在類上添加 @Transactional 注解，可以讓測試方法在執(zhí)行完畢后自動回滾，不會對數(shù)據(jù)庫造成影響。
 * 如果在方法上添加，則只影響對應(yīng)的方法
 */
@SpringBootTest
public class TransactionalTests extends AbstractTransactionalJUnit4SpringContextTests {
    @Resource
    private UserService userService;
    @Test
    void createRecord() {
        User user = userService.createUser();
        //to other things
    }
}

AbstractTransactionalJUnit4SpringContextTests實際上也是添加了@Transactional注解，只不過在此之外，它還提供了一些JDBC接口，可以讓我們更方便的操作數(shù)據(jù)庫。

使用這種方式來實現(xiàn)單元測試的話，由于測試用例中的所有數(shù)據(jù)庫操作都在事務(wù)中進行，然后運行結(jié)束后事務(wù)回滾，就可以保證不會染污數(shù)據(jù)庫數(shù)據(jù)，做到數(shù)據(jù)隔離。但是它也存在一些問題：

• 影響單元測試結(jié)果，在方法上層添加事務(wù)，本質(zhì)上還是改變了方法的邏輯。比如說我被測試的方法中用到了不同的數(shù)據(jù)源（主從數(shù)據(jù)庫），一但被包裹了事務(wù)，那么意味著我在這個事務(wù)內(nèi)，都是使用同一條連接，這會對我們實際上期望運行的邏輯生產(chǎn)影響。
• 排查問題困難，因為在單元測試運行結(jié)束之后，事務(wù)最終會回滾，也就意味著我們DB最終是沒有數(shù)據(jù)的，很難排查問題。

數(shù)據(jù)預(yù)處理與清理

上面說到事務(wù)會存在一些問題，那么我們不使用事務(wù)，怎么保證數(shù)據(jù)隔離？我們可以通過幾種方式來在單元測試運行前后手動的增加或清理數(shù)據(jù)：

通過@Sql注解來定義單元測試前后需要執(zhí)行的SQL腳本

    @Sql(
            scripts = "create-data.sql",
            config = @SqlConfig(transactionMode = ISOLATED)
    )
    @Sql(
            scripts = "delete-data.sql",
            config = @SqlConfig(transactionMode = ISOLATED),
            executionPhase = AFTER_TEST_METHOD
    )
    @Test
    void createRecord() {
        User user = userService.createUser();
        //to other things
    }

我們通過@Sql注解定義指定了兩個SQL腳本，分別用于在運行單元測試前創(chuàng)建測試數(shù)據(jù)，以及在運行結(jié)束后，刪除數(shù)據(jù)，這樣就可以保證我們的每次運行單元測試時的數(shù)據(jù)都是隔離的。

@Sql只是一種方式，我們還可以通過其它各種方式來執(zhí)行，比如@BeforeEach、@AfterEach來定義單元測試執(zhí)行前后的攔截器，或者Junit的@ExtendWith來定義外部的監(jiān)聽器等各種方式來處理數(shù)據(jù)的預(yù)處理與清理。

總結(jié)

上面分享了兩種數(shù)據(jù)隔離的方式，一般情況下都可以滿足我們的單元測試需求，但是實際上兩種方式都還存在一些問題，比如不管是哪種方式，都依賴于外部的數(shù)據(jù)庫，如果我們依賴的數(shù)據(jù)庫出現(xiàn)了異常，也會影響到我們的單元測試運行。而且使用外部數(shù)據(jù)庫也是無法完全做到數(shù)據(jù)隔離的，還是會有數(shù)據(jù)沖突的風險。那還有沒有更好的方法？ 如果大家的單元測試環(huán)境有docker環(huán)境的話，那么可以考慮引入Testcontainers，可以很好的解決這個問題。下篇文章我會詳細講一下Testcontainers在單測數(shù)據(jù)隔離中的實踐。

以上就是SpringBoot單元測試之數(shù)據(jù)隔離詳解的詳細內(nèi)容，更多關(guān)于SpringBoot單元測試的資料請關(guān)注腳本之家其它相關(guān)文章！

最新評論