diff options
Diffstat (limited to 'src/test/scala/com/drivergrp/core/CoreTest.scala')
-rw-r--r-- | src/test/scala/com/drivergrp/core/CoreTest.scala | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/src/test/scala/com/drivergrp/core/CoreTest.scala b/src/test/scala/com/drivergrp/core/CoreTest.scala new file mode 100644 index 0000000..005cda5 --- /dev/null +++ b/src/test/scala/com/drivergrp/core/CoreTest.scala @@ -0,0 +1,30 @@ +package com.drivergrp.core + +import java.io.ByteArrayOutputStream + +import org.scalatest.mock.MockitoSugar +import org.scalatest.{FlatSpec, Matchers} +import org.mockito.Mockito._ + +class CoreTest extends FlatSpec with Matchers with MockitoSugar { + + "'make' function" should "allow initialization for objects" in { + + val createdAndInitializedValue = make(new ByteArrayOutputStream(128)) { baos => + baos.write(Array(1.toByte, 1.toByte, 0.toByte)) + } + + createdAndInitializedValue.toByteArray should be(Array(1.toByte, 1.toByte, 0.toByte)) + } + + "'using' function" should "call close after performing action on resource" in { + + val baos = mock[ByteArrayOutputStream] + + using(baos /* usually new ByteArrayOutputStream(128) */ ) { baos => + baos.write(Array(1.toByte, 1.toByte, 0.toByte)) + } + + verify(baos).close() + } +} |