时间戳

首页 > 解决方案 > 如何对要使用 EasyMock 模拟的类的多个内部调用的方法进行单元测试

java - 如何对要使用 EasyMock 模拟的类的多个内部调用的方法进行单元测试

问题描述

我想对一个方法进行单元测试，该方法对我想使用的类进行多次内部调用EasyMock。

测试方法实际上运行了 5 次并调用了 mocked 方法。

在每个循环中，我将创建一些对象，它们都属于同一个类（比如说 A 类）。私有方法将调用获取类 A 实例的模拟对象方法，对其进行评估并返回结果。

最后，公共方法将返回一个结果列表。

我尝试了标准EasyMock.expect(MockClass.method(A)).andReturn()，但它不起作用，因为没有实现equals()for class A：

// this is the method example I am trying to test
public methodToTest(){
     // some logic
      privateMethodToTest(x);
     // some logic 
}
private List<B> privateMethodToTest(int x){
  List<B> list = new ArrayList<>();
  List<A> all = getObjects(x); //getObjects private method
  for (A a:all){
    list.add(objectToMock.methodToMock(a));
    return list;
  }

这就是我希望它的工作方式：

EasyMock.createMock(ObjectToMock.class);
EasyMock.expect(ObjectToMock.methodToMock(A)/* when A.getValue() == 1 */.andReturn("B object number 1")
EasyMock.expect(ObjectToMock.methodToMock(A)/* when A.getValue() == 2 */.andReturn("B object number 2")
//... and so on 
//object of class A does not implement equals()

我不知道该怎么做，也找不到任何类似的例子或回答我的问题。

标签： javaunit-testingeasymock

解决方案

你需要另一个匹配器。默认情况下，EasyMock 确实会使用equals. 但你不能那样做。您的基本选择是：

  1. 你不关心精确匹配

如果对你来说似乎是最简单的。这意味着做：

expect(objectToMock.methodToMock(anyObject()).andReturn("B object number 1");
expect(objectToMock.methodToMock(anyObject()).andReturn("B object number 2");
  1. 使用比较器

根据您的评论，您实际上可能更喜欢这个

expect(mock.methodToTest(EasyMock.cmp(new A(1), Comparator.comparingInt(A::getValue), LogicalOperator.EQUAL))).andReturn(1);

唯一的问题是您需要A与正确的值进行比较。

  1. 要获得简化版本，您可以使用自己的匹配器

使用下面的自定义匹配器的期望。

expect(mock.methodToTest(cmp(0))).andReturn(3);

public static <T> T cmp(int value) {
    reportMatcher(new IArgumentMatcher() {
        @Override
        public boolean matches(Object argument) {
            return value == ((A) argument).getValue();
        }

        @Override
        public void appendTo(StringBuffer buffer) {
            buffer.append("A.value=").append(value);
        }
    });
    return null;
}

推荐阅读