如何使用Junit超越或跳过方法

Question

我有一个我需要测试的代码块。

Class MainClass{    
   public void startProcess() {     
      ----Some Logic to generate fileName; 
          uploadFile(fileName);     
       }
  private static void uploadFile(String key) {    
      fileUpload();    
      deleteFile();    
  }    
}

我想编写一个junit测试，该测试将调用startprocess，但要么跳过上传file行，要么只是忽略uploadfile方法中存在的任何行。

我尝试使用PowerMock，但它不起作用。以下是我的代码

@RunWith(PowerMockRunner.class)
@PrepareForTest(MainClass.class)
public class MainClassTest {         
   @Test
   public void teststartProcess() throws Exception {
     processor=PowerMock.createPartialMock(MainClass.class,"uploadFile");
     PowerMock.expectPrivate(processor , "uploadFile", "xyz").andAnswer(
       new IAnswer<Void>() {
       @Override
       public Void answer() throws Throwable {
         System.out.println("Invoked!");
         return null;
       }
    }).atLeastOnce();
  }
}

，但它不会覆盖方法上传file即可打印。它调用fileupload并删除filetefile，而不仅仅是跳过线条并打印调用。

我的基本目标是模拟方法上载文件以打印

private void uploadFile(String key) {
  System.out.println("Invoked");
}

我知道可以使用Mockito，但是我们可以使用PowerMock或EasyMock。

Answer 1

，而不是尝试通过过度复杂的测试来重构代码，以便可以测试。根据您提供的内容，我会将文件功能移至另一个类。这样的东西：

public class MainClass {
    private final FileUploader fileUploader;
    public MainClass(FileUploader fileUploader) {
        this.fileUploader= fileUploader;
    }
    public void startProcess() {
        fileUploader.uploadFile(fileName);
    }
}

通过此重构，您有可能将普通模拟用于测试：

String fileName = "foo";
FileUploader fileUploader = mock(FileUploader.class);
MainClass classUnderTest = new MainClass(fileUploader);
classUnderTest.startProcess();
verify(fileUploader, times(1)).uploadFile(fileName);

另一个好处是测试FileUploader也变得容易。如您所见，我还摆脱了所有复杂的部分模拟和私人测试。

Answer 2

您必须在调用测试的方法之前将模拟对象更改为重播模式，并且必须告诉PowerMock以静态方法工作：

mockStaticStrict(MainClass.class,method(MainClass.class, "uploadFile"));
PowerMock.replay(processor);
processor.startProcess();

有关此信息的更多信息，请参见模拟私人方法。

i 高度建议重构此代码并可以测试。