Unit tests for core code and bug fixes

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()
+  }
+}