如何模拟和验证在子类中调用的ScheduledExcecutorService方法

时间:2017-05-28 18:26:29

标签: java unit-testing inheritance mockito scheduledexecutorservice

我有一个看起来像这样的基类:

public abstract class BaseClass implements Runnable {
 final ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1);

 @Override
 public void run() {
     someFunction();
 }
 protected abstract void someFunction();
}

然后我有一个像这样的子课:

public class ChildClass extends BaseClass {
 functionNeedsToBeTested() {
  scheduler.scheduleAtFixedRate(this, 0L, 5L, TimeUnit.HOURS)
 }
 someFunction() {
  //Does Something
 }
}

当我尝试编写测试时,问题就来了,我无法验证scheduleAtFixedRate方法的调用。我的测试看起来像这样:

@RunWith(MockitoJUnitRunner.class)
public class TestClass {
 @Mock
 private ScheduledExecutorService scheduler;

 @InjectMocks
 private ChildClass obj;

 @Test
 public void testFunc() {
   obj.functionNeedsToBeTested();
   Mockito.verify(scheduler).scheduleAtFixedRate(Mockito.any(ChildClass.class, Mockito.anyLong(), Mockito.anyLong(), Mockito.any(TimeUnit.class)));
 }
}

测试给了我这个错误:

junit.framework.AssertionFailedError:
    Wanted but not invoked:
    scheduler.scheduleAtFixedRate(
        <any>,
        <any>,
        <any>,
        <any>
     );

1 个答案:

答案 0 :(得分:3)

在测试中,会创建一个模拟调度程序,但测试对象不会使用它。

如果注入调度程序,则可以使方法可测试,而不是在基类中实例化一个。您可以要求调度程序作为BaseClass构造函数的参数,例如