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;
    }
}

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

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流/ map来获取一个集合中的所有newUser id。这样，您的newUsers集合可以是任意大的，并且您不必自己对映射进行硬编码。

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

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

1 个答案: