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

您无法模拟局部变量。但是，你可以做的是将其创建提取为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
}

Answer 2

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

请注意，新的Java 8时间API定义了这样的Clock class，专门用于此目的。

Answer 3

对不起，我的回复很晚。

这可能太麻烦了，但是如果您模拟可以提供局部变量的对象，则可以返回它的模拟。我宁愿不重组代码以简化测试，但要考虑一下。

/etc/hosts

在测试中，您可以执行以下操作：public class A { DateTimeFactory factory; private void method() { DateTime dateTime = factory.getDateTime(); //use dateTime to perform some operations } }

工厂模拟必须以某种方式注入到类中。

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

3 个答案: