junit 5中的tmpdir注解

Posted 2022-12-02

在junit 5中，如何设置针对文件的测试呢？可以使用@TempDir这个注解，比如要测试一个文件的操作，如下：
 

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;

public class FileWriter 

    public void writeTo(String path, String content) throws IOException 
        Path target = Paths.get(path);
        if (Files.exists(target)) 
            throw new IOException("file already exists");
        
        Files.copy(new ByteArrayInputStream(content.getBytes(StandardCharsets.UTF_8)), target);
    

测试用例：
 

import org.junit.jupiter.api.io.TempDir;
 
@Test
void writesContentToFile(@TempDir Path tempDir) throws IOException 
   
    Path output = tempDir
            .resolve("output.txt");
 
    //执行
    fileWriter.writeTo(output.toString(), "test");
 

    assertAll(
            () -> assertTrue(Files.exists(output)),
            () -> assertLinesMatch(List.of("test"), Files.readAllLines(output))
    );

注意这里都是用到不少NIO的知识了。
   也可以把tmpdir注解注解字段类型，比如：
   

import org.junit.jupiter.api.io.TempDir;
 
class FileWriterTest 
 
    private FileWriter fileWriter = new FileWriter();
 
    @TempDir
    Path tempDir;
 
    @BeforeEach
    void beforeEach() 
        assertTrue(Files.isDirectory(this.tempDir));
    
 
    @RepeatedTest(3)
    void throwsErrorWhenTargetFileExists() throws IOException 
        // arrange
        Path output = Files.createFile(
                tempDir.resolve("output.txt")
        );
 
        // act & assert
        IOException expectedException = assertThrows(IOException.class, () -> fileWriter.writeTo(output.toString(), "test"));
        assertEquals("file already exists", expectedException.getMessage());
    

 这里是在tempdir变量中加了注解，然后重复三次建立文件，并进行断言判断。