From 5d00ae0c3cf3ae7a30d37345808fffd894e6f1d5 Mon Sep 17 00:00:00 2001
From: slam <slamhan1987@gmail.com>
Date: Wed, 21 Aug 2024 13:55:58 +0800
Subject: [PATCH] feat(gateway): fix ut.

---
 .../com/tddworks/openai/api/OpenAIITest.kt    |  4 +--
 .../com/tddworks/openai/api/OpenAITest.kt     |  2 +-
 .../api/AnthropicOpenAIProviderTest.kt        | 27 ++++++++++---------
 3 files changed, 18 insertions(+), 15 deletions(-)

diff --git a/openai-client/openai-client-core/src/jvmTest/kotlin/com/tddworks/openai/api/OpenAIITest.kt b/openai-client/openai-client-core/src/jvmTest/kotlin/com/tddworks/openai/api/OpenAIITest.kt
index a05e2c3..088aa94 100644
--- a/openai-client/openai-client-core/src/jvmTest/kotlin/com/tddworks/openai/api/OpenAIITest.kt
+++ b/openai-client/openai-client-core/src/jvmTest/kotlin/com/tddworks/openai/api/OpenAIITest.kt
@@ -34,7 +34,7 @@ class OpenAIITest : AutoCloseKoinTest() {
             ChatCompletionRequest(
                 messages = listOf(ChatMessage.UserMessage("hello")),
                 maxTokens = 1024,
-                openAIModel = OpenAIModel.GPT_3_5_TURBO
+                model = OpenAIModel.GPT_3_5_TURBO
             )
         ).test(timeout = 10.seconds) {
             assertNotNull(awaitItem())
@@ -49,7 +49,7 @@ class OpenAIITest : AutoCloseKoinTest() {
             ChatCompletionRequest(
                 messages = listOf(ChatMessage.UserMessage("hello")),
                 maxTokens = 1024,
-                openAIModel = OpenAIModel.GPT_3_5_TURBO
+                model = OpenAIModel.GPT_3_5_TURBO
             )
         )
         assertNotNull(response)
diff --git a/openai-client/openai-client-core/src/jvmTest/kotlin/com/tddworks/openai/api/OpenAITest.kt b/openai-client/openai-client-core/src/jvmTest/kotlin/com/tddworks/openai/api/OpenAITest.kt
index 08649e3..8f86ac2 100644
--- a/openai-client/openai-client-core/src/jvmTest/kotlin/com/tddworks/openai/api/OpenAITest.kt
+++ b/openai-client/openai-client-core/src/jvmTest/kotlin/com/tddworks/openai/api/OpenAITest.kt
@@ -121,7 +121,7 @@ class OpenAITest {
                         )
                     )
                 ),
-                openAIModel = OpenAIModel.GPT4_VISION_PREVIEW
+                model = OpenAIModel.GPT4_VISION_PREVIEW
             )
         ).toList().first().content()
         assertEquals("", response)
diff --git a/openai-gateway/openai-gateway-core/src/jvmTest/kotlin/com/tddworks/openai/gateway/api/AnthropicOpenAIProviderTest.kt b/openai-gateway/openai-gateway-core/src/jvmTest/kotlin/com/tddworks/openai/gateway/api/AnthropicOpenAIProviderTest.kt
index 528305c..ba41987 100644
--- a/openai-gateway/openai-gateway-core/src/jvmTest/kotlin/com/tddworks/openai/gateway/api/AnthropicOpenAIProviderTest.kt
+++ b/openai-gateway/openai-gateway-core/src/jvmTest/kotlin/com/tddworks/openai/gateway/api/AnthropicOpenAIProviderTest.kt
@@ -5,27 +5,28 @@ import com.tddworks.anthropic.api.Anthropic
 import com.tddworks.anthropic.api.AnthropicModel
 import com.tddworks.anthropic.api.messages.api.*
 import com.tddworks.openai.api.chat.api.ChatCompletionRequest
+import com.tddworks.openai.api.chat.api.OpenAIModel
 import com.tddworks.openai.api.legacy.completions.api.CompletionRequest
 import kotlinx.coroutines.flow.flow
 import kotlinx.coroutines.test.runTest
 import org.junit.jupiter.api.Assertions.assertEquals
+import org.junit.jupiter.api.BeforeEach
 import org.junit.jupiter.api.Test
-import org.junit.jupiter.api.extension.ExtendWith
-import org.mockito.InjectMocks
-import org.mockito.Mock
-import org.mockito.junit.jupiter.MockitoExtension
+import org.mockito.Mockito.mock
 import org.mockito.kotlin.whenever
 import kotlin.test.assertFalse
 import kotlin.test.assertTrue
-import com.tddworks.openai.api.chat.api.OpenAIModel as OpenAIModel
 
-@ExtendWith(MockitoExtension::class)
 class AnthropicOpenAIProviderTest {
-    @Mock
-    lateinit var client: Anthropic
+    private lateinit var client: Anthropic
 
-    @InjectMocks
-    lateinit var provider: AnthropicOpenAIProvider
+    private lateinit var provider: AnthropicOpenAIProvider
+
+    @BeforeEach
+    fun setUp() {
+        client = mock()
+        provider = AnthropicOpenAIProvider(client)
+    }
 
     @Test
     fun `should throw not supported when invoke completions`() = runTest {
@@ -73,7 +74,8 @@ class AnthropicOpenAIProviderTest {
     @Test
     fun `should fetch completions from OpenAI API`() = runTest {
         // given
-        val request = ChatCompletionRequest.dummy(OpenAIModel(AnthropicModel.CLAUDE_3_HAIKU.value))
+        val request =
+            ChatCompletionRequest.dummy(OpenAIModel(AnthropicModel.CLAUDE_3_HAIKU.value))
         val response = CreateMessageResponse(
             id = "msg_1nZdL29xx5MUA1yADyHTEsnR8uuvGzszyY",
             type = "message",
@@ -99,7 +101,8 @@ class AnthropicOpenAIProviderTest {
     @Test
     fun `should stream completions for chat`() = runTest {
         // given
-        val request = ChatCompletionRequest.dummy(OpenAIModel(AnthropicModel.CLAUDE_3_HAIKU.value))
+        val request =
+            ChatCompletionRequest.dummy(OpenAIModel(AnthropicModel.CLAUDE_3_HAIKU.value))
 
         val contentBlockStart = ContentBlockStart(
             type = "content_block_start",