net.serenitybdd.junit5.extensions.TemporaryFolder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of serenity-junit5 Show documentation

There is a newer version: 4.2.9

package net.serenitybdd.junit5.extensions;

import java.io.File;
import java.io.IOException;
import java.nio.file.FileVisitResult;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.SimpleFileVisitor;
import java.nio.file.attribute.BasicFileAttributes;

import static java.nio.file.FileVisitResult.CONTINUE;

public class TemporaryFolder {
  private File rootFolder;

  public File newFile( String name ) throws IOException {
    File result = new File( rootFolder, name );
    result.createNewFile();
    return result;
  }

  public File newFolder( String name ) throws IOException {
    File result = new File( rootFolder, name );
    result.mkdirs();
    return result;
  }
  
  void prepare() {
    try {
      rootFolder = File.createTempFile( "junit5-", ".tmp" );
    } catch( IOException ioe ) {
      throw new RuntimeException( ioe );
    }
    rootFolder.delete();
    rootFolder.mkdir();
  }
  
  void cleanUp() {
    try {
      Files.walkFileTree( rootFolder.toPath(), new DeleteAllVisitor() );
    } catch( IOException ioe ) {
      throw new RuntimeException( ioe );
    }
  }

  private static class DeleteAllVisitor extends SimpleFileVisitor {
    @Override
    public FileVisitResult visitFile( Path file, BasicFileAttributes attributes ) throws IOException {
      Files.delete( file );
      return CONTINUE;
    }
  
    @Override
    public FileVisitResult postVisitDirectory( Path directory, IOException exception ) throws IOException {
      Files.delete( directory );
      return CONTINUE;
    }
  }
  
}