如何在java中使用mockito测试文件大小?

时间:2012-11-13 06:40:03

标签: java file testing junit mockito

我需要创建JUnit测试的方法:

public class MyClass {
  private String file1;
  private String file2;

  public void myMethodSpaceCheck(){
    if (new File(file1).size() > new File(file2).size() {
       throw new Exception .....  
    } 
  }
}

是否可以使用Mockito创建JUnit测试?

1 个答案:

答案 0 :(得分:2)

在处理Java中的文件时,我首选的选择是使用Apache VFS,因为我可以将它们视为任何其他POJO。显然,当你已经坚持使用File API时,这是很多工作。

另一个选择是完全忘记Mockito并在系统上写下这些文件。我通常会避免这种情况,因为它有时会使某些系统上的测试并行运行变得更加困难。

对于这种特定情况,我的解决方案通常是提供一个特殊的类,比如FileBuilder,它可以实例化新的文件:

public class FileBuilder {
  public java.io.File newFile(String pathname) {
    return new java.io.File(pathname);
  }
}

然后在将它传递给MyClass之前模拟这个类,并在适当的时候进行检测:

@Test(expected = Exception.class)
public void should_fail_when_file1_is_bigger_than_file2() {
  FileBuilder mockFile1 = file(2L);
  FileBuilder mockFile2 = file(1L);
  FileBuilder mockFileBuilder = mock(FileBuilder.class);
  when(mockFileBuilder.newFile("file1").thenReturn(mockFile1);
  when(mockFileBuilder.newFile("file2").thenReturn(mockFile2);

  new MyClass(mockFileBuilder).myMethodSpaceCheck();
}

private static File file(long length) {
    File mockFile = mock(File.class);
    when(mockFile.length()).thenReturn(length);
    return mockFile;
}

(你的例子提到File.size();我假设你的意思是File.length())

MyClass的实际实现如下:

public class MyClass {
  private String file1;
  private String file2;
  private final FileBuilder fileBuilder;

  public MyClass() {
    this(new FileBuilder());
  }

  @VisibleForTesting
  MyClass(FileBuilder fileBuilder) {
    this.fileBuilder = fileBuilder;
  }

  public void myMethodSpaceCheck() //...
}