使用 AssertJ 断言 Java 枚举值和字段

Question

给定一个像这样的简单枚举：

@AllArgsConstructor
@Getter
public enum PaymentMethod {

    CREDITCARD(1),
    PAYPAL(2),
    APPLE_PAY(3),
    GOOGLE_PAY(4);

    private final long id;
}

如何使用 AssertJ 断言编写 JUnit 测试，检查枚举值及其字段是否与枚举中定义的完全一致？ 这样最终测试会检查每种付款类型：

• 枚举中是否存在付款类型？
• 付款类型 X 是否有 id Y？

我目前的方法有效，但感觉有点麻烦。 我很确定 AssertJ 提供了更简单的语句，但我还没有找到解决方案。

class PaymentTypeTest {

    private static final Map<Long, PaymentType> expectedPaymentTypes = new HashMap<>();

    @BeforeAll
    static void beforeAll() {
        expectedPaymentTypes.put(1L, CREDITCARD);
        expectedPaymentTypes.put(2L, PAYPAL);
        expectedPaymentTypes.put(3L, APPLE_PAY);
        expectedPaymentTypes.put(4L, GOOGLE_PAY);    
    }

    @Test
    void should_contain_exact_values() {
        for (Map.Entry<Long, PaymentType> entry : expectedPaymentTypes.entrySet()) {
            final PaymentType paymentType = entry.getValue();
            assertThat(paymentType.getId()).isEqualTo(entry.getKey());
        }
    }
}```

Answer 1

带有@CsvSource的 JUnit 5 参数化测试可能会有所帮助：

@ParameterizedTest
@CsvSource({
  "CREDITCARD, 1",
  "PAYPAL, 2",
  "APPLE_PAY, 3",
  "GOOGLE_PAY, 4"
})
void getId_should_return_expected_id(PaymentType paymentType, long id) {
  assertThat(paymentType.getId()).isEqualTo(id);
}

为了确保所有 Enum 值都经过测试，可以将映射定义为@MethodSource并且再进行一项测试可以强制执行完全覆盖：

@ParameterizedTest
@MethodSource("mappings")
void getId_should_return_expected_id(PaymentType paymentType, long id) {
  assertThat(paymentType.getId()).isEqualTo(id);
}

@Test
void all_values_should_be_tested() {
  List<PaymentType> values = mappings() // same mappings used above
    .map(arguments -> arguments.get()[0])
    .map(PaymentType.class::cast)
    .collect(Collectors.toList());

  assertThat(values).contains(PaymentType.values());
}

static Stream<Arguments> mappings() {
  return Stream.of(
    arguments(CREDITCARD, 1L),
    arguments(PAYPAL, 2L),
    arguments(APPLE_PAY, 3L),
    arguments(GOOGLE_PAY, 4L)
  );
}