如何处理 JUnit 测试用例中的 wait() 方法调用？

Question

我有一个方法有另一个方法调用，它的定义有一个 SomeIntegerObject.wait() 调用。现在，当我运行我的测试用例时，由于这种等待方法，测试用例一直在等待并且不执行。

我尝试在 @Test 注释处添加超时，但这会导致“测试超时”之类的异常。

测试方法：

private PastingResult result;
public PastingResult pasteImages(IIOImage[] images, byte pasteProcessType){
     result = new PastingResult();
     _reprocessingProvider.setListLogicHandler(this);
     //--some more method calls--
    waitForResult(); //causes my test case to wait forever and pauses execution
    return result;
}

测试用例：

@Test
    public void testPasteImagesIIOImageArrayByte() throws Exception 
    {
        ReprocessManager _reprocessingProvider1=Mockito.mock(ReprocessManager.class);
        Whitebox.setInternalState(imagePasterIfImpl, "_reprocessingProvider", _reprocessingProvider1);

        IIOImage[] images=new IIOImage[]{Mockito.mock(IIOImage.class),Mockito.mock(IIOImage.class)};
        byte pasteProcessType = 02;

        PastingResult result= imagePasterIfImpl.pasteImages(images, pasteProcessType);  
        Mockito.verify(_reprocessingProvider1).setListLogicHandler(imagePasterIfImpl);
        System.out.println(result+"is the result");
    }

waitForResult();的定义

private Integer processingWaitMonitor = new Integer(0);
private void waitForResult() {
       // synchronize to wait on the Monitor..
       synchronized(processingWaitMonitor) {
           try {
               processingWaitMonitor.wait();
           }
           catch(Exception e) {
               System.out.println("Exception during Wait !"+e);
           }
       }
   }

Answer 1

这里有两个答案：

• 在第一层：如果您需要控制被测对象“内部”的某些东西；您可以求助于 Mockito Spys 或 PowerMock 或 JMockit；因为这些框架允许在这些方面“模拟”（也就是获得控制权）。但在大多数情况下，这并不是真正的好习惯。如果有的话，您应该将所需的功能“外包”到另一个类；然后您使用依赖注入来为您的测试设置提供该类的 mocked 对象。而不是真正的等待，您只需在一个什么都不做的模拟对象上调用一些等待方法。
• 除此之外：您可以退后一步，考虑重新设计您的整个方法。你看，使用诸如等待/通知之类的“低级”原语不再是好的做法了。有ExecutorService、Futures、Promises 等抽象。关键是：使用 ExecutorService，您可能能够重写代码以使用 Same-Thread-Executor-Service 进行测试。含义：可以重写多线程代码，以便仅通过为测试提供不同的执行器，所有事情都发生在单个线程上。无需再等待，获得有保证的可靠结果。

关于您的评论；你的代码说：

private Integer processingWaitMonitor = new Integer(0);
private void waitForResult() {

含义：你在你的测试类中有一个字段+私有方法。假设你创建了

public interface<E> WaitService {
   public E waitForResult();
}

或类似的东西。进而;而不是在你的生产类中拥有一个锁对象和那个等待方法，你只在那里保留那个服务的“一些”实例。但是，当然，这又需要您更改生产代码。

如果您真的无法更改该代码（这有点可惜 - 当您无法更改代码时，测试代码有什么意义？）...那么 Mockito/PowerMock/JMockit 是您唯一的选择在那里。

Answer 2

正在等待的模块，将其放入junit测试用例的新线程中。间隔后，检查线程是否完成。设置间隔的最大超时。超时发生后，检查线程完成情况并决定是通过还是失败。

例如：

private static final int MAX_WAIT_TIME = 20000; //total time out in milliseconds
private static final int WAIT_INTERVAL = 1000; //wait interval till MAX_WAIT_TIME

@Test
public void testPasteImagesIIOImageArrayByte() throws Exception 
{
  Thread t = new MyThread();
  t.start();    
  int totalWaitTime = 0;
  while ((!t.isCompleted()) && totalWaitTime <= MAX_WAIT_TIME) {
    try {
        totalWaitTime += WAIT_INTERVAL;
        Thread.sleep(WAIT_INTERVAL);
    } catch (java.lang.InterruptedException e) {
        //LOG OR THROW
    }
    if(!t.isCompleted()){
      if(t.hasExceptionOccurred()){
        //fail the case
      }else{
        //take your decision here, if you want to pass it or fail it
      }
    }
  }
}

class MyThead extends Thread{
  private volatile boolean isCompleted = false;
  private volatile Throwable exception = null;
  private volatile boolean hasExceptionOccured = false;

  public void run(){
    try{
        ReprocessManager _reprocessingProvider1=Mockito.mock(ReprocessManager.class);
        Whitebox.setInternalState(imagePasterIfImpl, "_reprocessingProvider", _reprocessingProvider1);

        IIOImage[] images=new IIOImage[]{Mockito.mock(IIOImage.class),Mockito.mock(IIOImage.class)};
        byte pasteProcessType = 02;

        PastingResult result= imagePasterIfImpl.pasteImages(images, pasteProcessType);  
        Mockito.verify(_reprocessingProvider1).setListLogicHandler(imagePasterIfImpl);
        System.out.println(result+"is the result");

        synchronized(this){
          isCompleted = true;
        }
    }catch(Throwable ex){
        synchronized(this){
            exception = ex;
            hasExceptionOccured = true;
        }
    }
  }

  public synchronized boolean isCompleted(){
    return isCompleted;
  }

  public synchronized boolean hasExceptionOccurred(){
    return hasExceptionOccured;
  }

  public synchronized Throwable getExceptionOccurred(){
    return exception;
  }
}