Java 測試自定義斷言實戰：用 AssertJ 打造領域語言，提升可讀性與維護性

痛點：基礎斷言的侷限

JUnit `assertEquals/assertTrue` 導致測試代碼充斥底層細節，重複代碼多、錯誤訊息不明確（如 `Expected: 1343044800000 Actual: 1343041200000`），難以表達業務意圖。

漸進式改進

1. 私有輔助方法：抽取 `assertThatRangeExists` 減少重複，但參數多、命名難、邏輯易膨脹。
2. 匹配器庫：AssertJ（FEST 分支）或 Hamcrest。AssertJ 基於 Fluent API，IDE 自動提示極佳。

AssertJ 核心語法

assertThat(myDouble).isLessThanOrEqualTo(2.0);
assertThat(myList).contains("a");
assertThat("text").isNotEmpty().startsWith("some").hasLength(9); // 鏈式

自定義斷言：RangeAssert 實戰

public class RangeAssert extends ListAssert {
    protected RangeAssert(List actual) { super(actual); }
    public static RangeAssert assertThat(List actual) { return new RangeAssert(actual); }

    public RangeAssert hasSize(int expected) {
        isNotNull();
        Assertions.assertThat(actual.size()).as("range count").isEqualTo(expected);
        return this;
    }
    public RangeAssert isSortedAscending() { ... return this; }
    public RangeAssert hasRange(String from, String to) {
        isNotNull();
        boolean found = actual.stream().anyMatch(r -> r.getStart()==fromTime && r.getEnd()==toTime);
        Assertions.assertThat(found)
            .overridingErrorMessage("ranges %s do not contain %s-%s", actual, from, to)
            .isTrue();
        return this;
    }
}

測試代碼對比

// 原始：大量 assertEquals
// 自定義斷言：
RangeAssert.assertThat(ranges)
    .hasSize(3)
    .isSortedAscending()
    .hasRange("2012-07-23 12:00", "2012-07-23 13:00")
    .hasRange("2012-07-23 13:00", "2012-07-23 14:00")
    .hasRange("2012-07-23 14:00", "2012-07-23 15:00");

錯誤訊息變為：`ranges [Range{12:00-13:00}, ...] do not contain 2012-07-23 16:00-2012-07-23 14:00`，業務語義清晰。

最佳實踐建議

• 停用 JUnit/TestNG 內建斷言，全面遷移 AssertJ/Hamcrest
• 單元測試通常無需自定義斷言
• 整合測試、端到端測試**強烈建議**編寫自定義斷言：封裝技術細節、使用領域語言、易維護

結語

自定義斷言成本低、收益高，是測試代碼從「能跑」進化為「可讀、可維護」的關鍵一步。

出處：bianceng.cn