Junit 5中@ParameterizedTest结合@EnumSource

最新推荐文章于 2025-10-27 05:03:14 发布

原创最新推荐文章于 2025-10-27 05:03:14 发布 · 2.2k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Junit

测试专栏收录该内容

13 篇文章

订阅专栏

概述

有时候业务代码里会根据一个枚举值来区分业务场景，比如说：

public enum ActivityLimitEnum {
    LIMIT(1,"封顶"),
    UNLIMIT(0,"上不封顶");
}

如果编写单元测试来验证业务代码，至少需要写两个测试方法,一个验收封顶逻辑，一个验收不封顶逻辑。

@Test
    @DisplayName("封顶")
    void testLimit() {
        
    }

    @Test
    @DisplayName("不封顶")
    void testUnLimit() {

    }

由于这两个场景只是区分封顶和不封顶，其他的都是一样的，如果写两个测试方法，testLimit和testUnLimit方法里会有很多重复代码。这个时候Junit 5 中@ParameterizedTest和@EnumSource就可以派上用场了。

    @ParameterizedTest
    @EnumSource(ActivityLimitEnum.class)
    @DisplayName("封顶和不封顶")
    void test(ActivityLimitEnum activityLimitEnum) {
        if (ActivityLimitEnum.LIMIT.equals(activityLimitEnum)) {
            assertFalse(false);
        } 
        else if (ActivityLimitEnum.UNLIMIT.equals(activityLimitEnum)) {
            assertTrue(true);
        }
    }

Junit 5在执行上面的test方法的时候，由于ActivityLimitEnum中有LIMIT和UNLIMIT两个枚举，先按照LIMIT类型执行一次test方法，再按照UNLIMIT类型再次执行test方法，共执行两次test方法，因此我们需要在test方法里，区分一下类型，按照实际情况编写assert代码进行验证即可。

我们用Intellij IDEA执行上面的单元测试，效果如下：
在这里插入图片描述

可以看到test方法确实执行了两次,LIMIT和UNLIMIT各执行一次。

如果枚举中的枚举值非常多，而单元测试里只是想验证其中部分枚举值，那可以用下面的方式：

    @ParameterizedTest
    @EnumSource(value = ActivityLimitEnum.class,names = {"LIMIT"})
    @DisplayName("封顶和不封顶")
    void test(ActivityLimitEnum activityLimitEnum) {
        if (ActivityLimitEnum.LIMIT.equals(activityLimitEnum)) {
            assertFalse(false);
        }
        else if (ActivityLimitEnum.UNLIMIT.equals(activityLimitEnum)) {
            assertTrue(true);
        }
    }

这样就只会执行LIMIT枚举值对应的场景。

如果要在Junit 5里使用@ParameterizedTest和@EnumSource，需要添加如下依赖：

<dependency>
    <groupId>org.junit.jupiter</groupId>
    <artifactId>junit-jupiter-params</artifactId>
    <version>5.2.0</version>
    <scope>test</scope>
</dependency>