使用Mockito模拟方法的局部变量

Question

我有一个类a需要测试。以下是A的定义：

public class A {
    public void methodOne(int argument) {
        //some operations
        methodTwo(int argument);
        //some operations
    }

    private void methodTwo(int argument) {
        DateTime dateTime = new DateTime();
        //use dateTime to perform some operations
    }
}

并根据datetime值从数据库中检索一些要操作的数据。对于该数据库，值通过JSON文件持久化。

这使事情复杂化了。我需要的是在测试datetime时将其设置为某个特定的日期。有没有办法可以用mockito模拟局部变量的值？

Answer 1

这可能太麻烦了，但是如果您模拟可以给出局部变量的对象，那么您可以返回它的一个模拟。我宁愿不重新构造代码以使测试更容易，但这是需要考虑的。

public class A {
    
    DateTimeFactory factory;

    private void method() {
        DateTime dateTime = factory.getDateTime();
        //use dateTime to perform some operations
    }
}

在测试中，可以执行以下操作:when(factorymock.getDateTime()).doReturn(dateTimeMock)

将需要以某种方式将工厂模拟注入到类中。

Answer 2

处理此类问题的最佳方法是使用注入的clock服务，用于获取DateTime的新实例。这样，您的测试就可以注入一个模拟时钟，它返回一个特定的DateTime而不是当前时间。

注意，新的Java 8time API定义了这样一个时钟类，专门用于此目的。

Answer 3

不能模拟局部变量。但是，您可以做的是将其创建提取到protected方法中，并将其提取到spy中：

public class A {
  public void methodOne(int argument) {
    //some operations
    methodTwo(int argument);
    //some operations
  }

  private void methodTwo(int argument) {
    DateTime dateTime = createDateTime();
    //use dateTime to perform some operations
  }

  protected DateTime createDateTime() {
    return new DateTime();
  }
}

public class ATest {
  @Test
  public void testMethodOne() {
    DateTime dt = new DateTime (/* some known parameters... */);
    A a = Mockito.spy(new A());
    doReturn(dt).when(a).createDateTime();
    int arg = 0; // Or some meaningful value...
    a.methodOne(arg);
    // assert the result
}