Java单元测试冗余代码简化方法咨询（新手提问）

嘿，完全理解你的困扰——重复代码不仅写着烦，维护起来也头疼！作为Java新手能注意到这个问题已经很棒啦～针对你遇到的同一对象不同属性的null校验重复代码问题，这里有几个实用的简化方案：

方案一：提取公共校验方法 #

最直接的方式是把重复的校验逻辑抽成一个私有方法，然后在各个测试用例中调用它。这样只需要维护一份校验逻辑，修改起来更方便。

示例代码：

// 假设你的对象是这样的
class MyObj {
    private String a;
    private Integer b;

    public void setA(String a) {
        if (a == null) throw new IllegalArgumentException("A cannot be null");
        this.a = a;
    }

    public void setB(Integer b) {
        if (b == null) throw new IllegalArgumentException("B cannot be null");
        this.b = b;
    }
}

class MyObjTest {
    private MyObj obj = new MyObj();

    // 提取公共校验方法
    private void testNullProperty(Consumer<MyObj> nullSetter, String expectedMessage) {
        IllegalArgumentException exception = assertThrows(IllegalArgumentException.class, () -> {
            nullSetter.accept(obj);
        });
        assertEquals(expectedMessage, exception.getMessage());
    }

    @Test
    void testInvalidA() {
        testNullProperty(obj -> obj.setA(null), "A cannot be null");
    }

    @Test
    void testInvalidB() {
        testNullProperty(obj -> obj.setB(null), "B cannot be null");
    }
}

方案二：使用JUnit 5参数化测试（更优雅的方式） #

如果你的项目用的是JUnit 5，参数化测试可以让你用单份测试代码覆盖所有属性的校验场景，彻底消除重复。核心思路是把不同的属性设置逻辑和预期信息作为参数传递给测试方法。

示例代码：

import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.MethodSource;
import java.util.stream.Stream;

class MyObjParamTest {
    private MyObj obj = new MyObj();

    // 定义测试参数流：每个元素包含属性设置器和预期异常信息
    static Stream<Object[]> nullPropertyCases() {
        return Stream.of(
            new Object[]{(Consumer<MyObj>) obj -> obj.setA(null), "A cannot be null"},
            new Object[]{(Consumer<MyObj>) obj -> obj.setB(null), "B cannot be null"}
        );
    }

    @ParameterizedTest
    @MethodSource("nullPropertyCases")
    void testNullProperties(Consumer<MyObj> nullSetter, String expectedMessage) {
        IllegalArgumentException exception = assertThrows(IllegalArgumentException.class, () -> {
            nullSetter.accept(obj);
        });
        assertEquals(expectedMessage, exception.getMessage());
    }
}

这样运行测试时，JUnit会自动为每个参数生成独立的测试用例，你在测试报告里能看到每个属性校验的结果，既简洁又清晰。

额外小提示 #

如果你的属性非常多，还可以考虑用反射来动态设置属性并校验，但反射相对复杂一些，对于新手来说先掌握上面两种方法就足够啦。

刚开始写测试的时候能想到优化重复代码，真的很赞，慢慢积累这些小技巧，后续写测试会越来越顺手的～

内容的提问来源于stack exchange，提问作者Sydney