使用 AssertJ 进行测试时从实际和预期对象中提取字段

Question

我想检查一个列表是否包含另一个列表的子列表。 这是我目前如何做的一个虚拟示例：

@Test
public void testExtracting() throws Exception {
    final List<User> users = new ArrayList<>();
    users.add(new User(5234, "Adam"));
    users.add(new User(4635, "David"));

    final List<User> newUsers = new ArrayList<>();
    newUsers.add(new User(6143, "Bob"));
    newUsers.add(new User(3465, "Cindy"));

    users.addAll(newUsers);

    assertThat(users).extracting("id").contains(newUsers.get(0).id, newUsers.get(1).id);
}

public static class User {
    public long id;
    public String name;

    public User(int id, String name) {
        this.id = id;
        this.name = name;
    }
}

是否有可能以更紧凑的方式实现相同的结果，而不覆盖User类的equals方法？ 我正在寻找以下方面的东西：

assertThat(users).contains(newUsers.get(0), newUsers.get(1)).extracting("id");

或者，更好的是：

assertThat(users).contains(newUsers.subList(0, 2)).extracting("id");

Answer 1

如果您可以将getId()方法添加到User ，则可以使用提取器解决此问题：

assertThat(users)
    .extracting("id")
    .containsAll(extractProperty("id").from(newUsers));

如果无法更改User （因此不能覆盖hashcode()或equals()或添加getter），则需要从两个集合中提取id并进行比较。 这对我有用，但是我承认这有点丑陋。

assertThat(users)
    .extracting("id")
    .containsAll(newUsers.stream().map(it -> it.id).collect(Collectors.toSet()));

我正在做的是使用assertJ从用户集中提取ID，并使用Java 8流/映射在一个集合中获取所有newUser ID。 这样，您的newUsers集合可以任意大，而您不必自己对映射进行硬编码。

请注意，第二个示例中的流/映射工作与第一个示例中的extractProperty完成的工作相同，我们只是手动进行。

Answer 2

您可以为id属性创建一个 getter 方法并执行以下操作：

assertThat(users).usingElementComparator(Comparator.comparing(User::getId)).containsAll(newUsers);