Junit 5中@ParameterizedTest与@EnumSource结合使用

概述

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

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，需要添加如下依赖：

  org.junit.jupiter
  junit-jupiter-params
  5.2.0
  test

总结

以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，谢谢大家对脚本之家的支持。如果你想了解更多相关内容请查看下面相关链接