测试对象集合中集合属性的内容

Question

很抱歉我试图尽可能清楚地说出这个标题。不要犹豫，编辑以提高它。

我的问题是，我想测试这个结构的内容，更特别的是B对象的内容，并确保它对应于特定的A值：

public class A {
    String key;
    List<B> bs;
}

我的数据是这样的

List<A> as = [
    {
        key : "KEY1", 
        bs: [
            {val1:"val1", val2:"val2}
        ]
    },
    {
        key : "KEY2", 
        bs: [
            {val1:"val3", val2:"val4"}, 
            {val1:"val5", val2:"val6"}
        ]
    },
];

在使用Mockito进行单元测试的情况下，我希望能够在不进行预处理的情况下测试该结构以获得B的列表。通过测试该结构，我希望确保KEY2有两个B，第一个B有val3和val4，第二个有val5和val6。

目前，我必须按键创建一个映射，并测试每个条目。如果有的话，我想用一种更直接的方式。

这是我的实际测试。

List<A> as = captor.getAllValues();
assertThat(as)
            .isNotNull()
            .hasSize(2)
            .extracting("key")
            .containsOnlyOnce(
                    tuple("KEY1"),
                    tuple("KEY2")
            );
    Map<String, A> estimationParPlateforme = indexBy(as, new Indexer<String, A>() {
        @Override
        public String apply(A a) {
            return a.getKey();
        }
    });
assertThat(as.get("KEY1").getBs())
            .isNotEmpty()
            .extracting(
                    "val1",
                    "val2"
            )
            .containsExactly(
                    tuple(
                            "val1",
                            "val2"
                    )
            );
assertThat(as.get("KEY2").getBs())
            .isNotEmpty()
            .extracting(
                    "val1",
                    "val2"
            )
            .containsExactly(
                    tuple(
                            "val3",
                            "val4"
                    ),
                    tuple(
                            "val5",
                            "val6"
                    )
            ); 

我认为测试时间有点长，我想找到一种改进的方法。你有什么解决方案吗？

感谢您的帮助

Answer 1

您必须定义数据结构的外观。目前，所有这些信息都分散在测试语句中。这让它变得非常丑陋，正如你所想的那样。我喜欢做的是有一些数据结构，我可以用我期望的数据填充。然后我可以使用它来测试我的数据是否有效。一个例子：

static Map<String, B[]> expectedBs = new HashMap<String, B[]>();
public static void main(String[] args) {
    //put the expected values into a map
    expectedBs.put("KEY1", new B[] { new B("val1", "val2") });
    expectedBs.put("KEY2", new B[] { new B("val3", "val4"), new B("val5", "val6") });
    //build a valid datastructure
    List<A> as = new ArrayList<A>();
    List<B> bs1 = new ArrayList<B>();
    bs1.add(new B("val1", "val2"));
    List<B> bs2 = new ArrayList<B>();
    bs2.add(new B("val3", "val4"));
    bs2.add(new B("val5", "val6"));
    as.add(new A("KEY1", bs1));
    as.add(new A("KEY2", bs2));

    //test validity of as
    System.out.println(isValid(as)); //prints true
    //modify the datastructure such that it is not valid anymore
    bs1.get(0).val1 = "val2";
    //test validity once more
    System.out.println(isValid(as)); //prints false
}
static boolean isValid(List<A> as) {
    for (A a : as) {
        B[] expected = expectedBs.get(a.key);
        if (!equals(a.bs, expected))
            return false;
    }
    return true;
}
static boolean equals(List<B> x, B[] y) {
    if (x == null)
        return false;
    if (y == null)
        return false;
    if (x.size() != y.length)
        return false;
    for (int i = 0; i < x.size(); i++)
        if (!x.get(i).equals(y[i]))
            return false;
    return true;
}

A:

public class A {
    String key;
    List<B> bs;
    public A(String key, List<B> bs) {
        this.key = key;
        this.bs = bs;
    }
}

B:

public class B {
    String val1, val2;
    public B(String val1, String val2) {
        this.val1 = val1;
        this.val2 = val2;
    }
    @Override
    public boolean equals(Object obj) {
        if (obj == null)
            return false;
        if (!(obj instanceof B))
            return false;
        B objB = (B) obj;
        if (!objB.val1.equals(this.val1))
            return false;
        if (!objB.val2.equals(this.val2))
            return false;
        return true;
    }
}

抱歉代码太长了。希望你明白。这是否更优雅取决于你自己。