diff --git a/core/src/test/scala/spark/Utils.scala b/core/src/test/scala/spark/UtilsSuite.scala similarity index 56% rename from core/src/test/scala/spark/Utils.scala rename to core/src/test/scala/spark/UtilsSuite.scala index 4e852903be663423e2686806e0a82dec2bf55ba9..f31251e509a9c14460a573f7584f42d206362e4e 100644 --- a/core/src/test/scala/spark/Utils.scala +++ b/core/src/test/scala/spark/UtilsSuite.scala @@ -1,7 +1,8 @@ package spark import org.scalatest.FunSuite - +import java.io.{ByteArrayOutputStream, ByteArrayInputStream} +import util.Random class UtilsSuite extends FunSuite { @@ -14,5 +15,15 @@ class UtilsSuite extends FunSuite { assert(Utils.memoryBytesToString(5368709120L) === "5.0GB") } + test("copyStream") { + //input array initialization + val bytes = Array.ofDim[Byte](9000) + Random.nextBytes(bytes) + + val os = new ByteArrayOutputStream() + Utils.copyStream(new ByteArrayInputStream(bytes), os) + + assert(os.toByteArray.toList.equals(bytes.toList)) + } }