如何在 Java 中为 Java 8 Predicate 和 distinctByKeys 编写 Mockito 测试



我有通过使用多个键(如员工 ID、姓名和年龄(来获取唯一元素的方法。写了一个测试用例,但过滤器部分没有覆盖,那么如何覆盖它?

List<Employee> employees = getEmployees().stream()
.filter(distinctByKey(Employee::getEmployeeId, Employee::getEmployeeName, Employee::getEmployeeAge))
.collect(Collectors.toList());

private <T> Predicate<T> distinctByKeys(Function<? super T, ?>... keyExtractors) {
final Map<List<?>, Boolean> seen = new ConcurrentHashMap<>();
return t -> {
final List<?> keys = Arrays.stream(keyExtractors)
.map(ke -> ke.apply(t))
.collect(Collectors.toList());
return seen.putIfAbsent(keys, Boolean.TRUE) == null;
};
}

如果确实需要这部分代码,可以使用带有注释@VisibleForTesting的lib Guava,并且可以在测试中调用此方法。

https://guava.dev/releases/19.0/api/docs/com/google/common/annotations/VisibleForTesting.html

最新更新