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

时间:2016-04-01 12:06:51

标签: java testing collections mockito

对不起,我尽量清楚这个标题。不要犹豫,编辑以推动它。

我的问题是我想测试这个结构的内容,更具体地测试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的列表。通过测试这个结构,我想确定有两个B用于KEY2和第一个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"
                    )
            ); 

我认为测试有点长,我想找到一种改进方法。你有任何解决方案吗?

感谢您的帮助

1 个答案:

答案 0 :(得分: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;
}

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

很抱歉代码如此之久。希望你明白这个主意。这是否更优雅取决于你自己决定。