initial working test for OpenAIServiceWrapper (cequence-io#24)

Author: phelps-sg

build.sbt

@@ -10,17 +10,27 @@ ThisBuild / scalaVersion := scala212
 ThisBuild / version := "0.3.3"
 ThisBuild / isSnapshot := false
 
+lazy val commonSettings = Seq(
+  libraryDependencies += "org.scalactic" %% "scalactic" % "3.2.16",
+  libraryDependencies += "org.scalatest" %% "scalatest" % "3.2.16" % Test,
+  libraryDependencies += "org.mockito" %% "mockito-scala-scalatest" % "1.17.14" % Test
+)
+
 lazy val core = (project in file("openai-core"))
+  .settings(commonSettings: _*)
 
 lazy val client = (project in file("openai-client"))
+  .settings(commonSettings: _*)
   .dependsOn(core)
   .aggregate(core)
 
 lazy val client_stream = (project in file("openai-client-stream"))
+  .settings(commonSettings: _*)
   .dependsOn(client)
   .aggregate(client)
 
 lazy val guice = (project in file("openai-guice"))
+  .settings(commonSettings: _*)
   .dependsOn(client)
   .aggregate(client_stream)
 

openai-core/src/test/scala/io/cequence/openaiscala/service/OpenAIServiceWrapperSpec.scala

@@ -1,5 +1,48 @@
 package io.cequence.openaiscala.service
 
-class OpenAIServiceWrapperSpec extends AnySpec with should.Matchers {
+import io.cequence.openaiscala.domain.response.{ModelInfo, Permission}
+import org.mockito.scalatest.MockitoSugar
+import org.scalatest.concurrent.ScalaFutures.convertScalaFuture
+import org.scalatest.matchers.should
+import org.scalatest.wordspec.AnyWordSpecLike
+
+import scala.concurrent.Future
+import scala.concurrent.duration.DurationInt
+
+class OpenAIServiceWrapperSpec
+    extends AnyWordSpecLike
+    with should.Matchers
+    with MockitoSugar {
+
+  "OpenAIServiceWrapper" should {
+
+    class MockWrapper(val underlying: OpenAIService)
+        extends OpenAIServiceWrapper {
+      override protected def wrap[T](fun: OpenAIService => Future[T]): Future[T] = {
+        fun(underlying)
+      }
+
+      override def close: Unit = {}
+    }
+
+    "call wrap for listModels" in {
+      val mockService = mock[OpenAIService]
+      val wrapper = new MockWrapper(mockService)
+      val testModels = Array(
+        ModelInfo(
+          "test",
+          new java.util.Date(0L),
+          owned_by = "test_owner",
+          root = "test_root",
+          parent = None,
+          permission = Array[Permission]()
+        )
+      )
+      when(mockService.listModels).thenReturn(Future.successful {
+        testModels
+      })
+      wrapper.listModels.futureValue should be(testModels)
+    }
+  }
 
 }