diff options
Diffstat (limited to 'crashboxd/src/test/scala/io/crashbox/ci/TestUtil.scala')
-rw-r--r-- | crashboxd/src/test/scala/io/crashbox/ci/TestUtil.scala | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/crashboxd/src/test/scala/io/crashbox/ci/TestUtil.scala b/crashboxd/src/test/scala/io/crashbox/ci/TestUtil.scala new file mode 100644 index 0000000..b5bacf2 --- /dev/null +++ b/crashboxd/src/test/scala/io/crashbox/ci/TestUtil.scala @@ -0,0 +1,26 @@ +package io.crashbox.ci + +import java.io.{ ByteArrayOutputStream, File } +import java.nio.file.Files + + +object TestUtil { + + def withTempFile[A](action: File => A): A = { + def rm(parent: File): Unit = if (parent.isDirectory) { + parent.listFiles.foreach{ child => + rm(child) + } + } + val dir = Files.createTempDirectory("crashbox-test").toFile + try action(dir) + finally rm(dir) + } + + def withTempStream[A](action: ByteArrayOutputStream => A, size: Int = 1024): A = { + val out = new ByteArrayOutputStream(size) + try action(out) + finally out.close() + } + +} |