JUnit4 @Rule 用于创建临时文件或文件夹

文
章
目
录

很多时候，我们需要创建JUnit单元测试，在此过程中我们需要创建临时文件夹或临时文件以执行测试用例。很多时候，我们依赖于在特定位置有一个临时文件夹，并在那里生成所有临时文件。嗯，它有自己的缺点。主要的缺点是你需要手动清理这些文件。

考虑升级到JUnit 5。它也有类似的功能，使用@TempDir注解。

1.JUnit TemporaryFolder

JUnit有TemporaryFolder类，你可以用它来生成临时文件夹。

TemporaryFolder规则允许创建在测试方法结束时（无论通过还是失败）应被删除的文件和文件夹。这个规则不会检查删除操作是否成功，也不会在删除失败时抛出异常。

使用TemporaryFolder规则的语法

TemporaryFolder规则的使用示例：

public static class AppTest {
        @Rule
        public TemporaryFolder folder= new TemporaryFolder();
        @Test
        public void testUsingTempFolder() throws IOException {
                File createdFile= folder.newFile("myfile.txt");
                File createdFolder= folder.newFolder("subfolder");
                // ...
        }
 }

2.JUnit 测试创建临时文件

让我们创建一个快速的测试用例来看看它是如何工作的。

public class TemporaryFolderTest {

   @Rule
   public TemporaryFolder tempFolder = new TemporaryFolder();

   @Test
   public void testWrite() throws IOException {

     // 创建一个临时文件
     final File tempFile = tempFolder.newFile("tempFile.txt");

     // 写一些内容进去
     FileUtils.writeStringToFile(tempFile, "hello world");

     //从临时文件读取内容
     final String s = FileUtils.readFileToString(tempFile);

     // 验证内容
     Assert.assertEquals("hello world", s);

     //注意：测试结束后，文件保证会被删除。
   }
 }

这是一个非常简单但有用的JUnit功能。下次使用它，你会发现它非常有帮助。