com.swoval.test.platform.package.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of testing_2.10 Show documentation
testing
The newest version!
package com.swoval.test

import java.io.IOException
import java.nio.file.attribute.BasicFileAttributes
import java.nio.file.{
  DirectoryNotEmptyException,
  FileVisitResult,
  FileVisitor,
  Path,
  Paths,
  Files
}

import scala.concurrent.ExecutionContext
import scala.concurrent.duration._

package object platform {
  def sleep(duration: FiniteDuration) = Thread.sleep(duration.toMillis)
  def executionContext: ExecutionContext = ExecutionContext.global
  def createTempFile(dir: String, prefix: String): (String, Thread) =
    deleteOnExit(Files.createTempFile(Paths.get(dir), prefix, "").toRealPath())

  def createTempDirectory(): (String, Thread) = {
    val base = Paths.get(System.getProperty("java.io.tmpdir")).resolve("swoval")
    Files.createDirectories(base)
    deleteOnExit(Files.createTempDirectory(base, "dir").toRealPath())
  }

  def createTempSubdirectory(dir: String): (String, Thread) =
    deleteOnExit(Files.createTempDirectory(Paths.get(dir), "subdir").toRealPath())

  val shutdownHooks = new ShutdownHooks {
    override def add(thread: Thread): Unit = Runtime.getRuntime.addShutdownHook(thread)

    override def remove(thread: Thread): Unit = Runtime.getRuntime.removeShutdownHook(thread)
  }

  def delete(dir: String): Unit = {
    Files.walkFileTree(
      Paths.get(dir),
      new FileVisitor[Path] {
        override def preVisitDirectory(dir: Path, attrs: BasicFileAttributes): FileVisitResult =
          FileVisitResult.CONTINUE
        override def visitFile(file: Path, attrs: BasicFileAttributes): FileVisitResult = {
          Files.deleteIfExists(file)
          FileVisitResult.CONTINUE
        }
        override def visitFileFailed(file: Path, exc: IOException): FileVisitResult = {
          FileVisitResult.CONTINUE
        }
        override def postVisitDirectory(dir: Path, exc: IOException): FileVisitResult = {
          try {
            Files.deleteIfExists(dir)
          } catch {
            case _: DirectoryNotEmptyException => delete(dir.toString)
          }
          FileVisitResult.CONTINUE
        }
      }
    )
  }
  def mkdir(path: String): String = Files.createDirectories(Paths.get(path)).toRealPath().toString
  private def deleteOnExit(path: Path): (String, Thread) = {
    val thread = new Thread() {
      override def run(): Unit = {
        delete(path.toString)
      }
    }
    Runtime.getRuntime().addShutdownHook(thread)
    (path.toString, thread)
  }

}