test: Add unit test

Author: osugikoji

ychat/src/commonTest/kotlin/co/yml/ychat/di/LibraryModuleTest.kt

@@ -4,11 +4,13 @@ import co.yml.ychat.data.api.ChatGptApi
 import co.yml.ychat.data.infrastructure.ApiExecutor
 import co.yml.ychat.data.storage.ChatLogStorage
 import co.yml.ychat.di.module.LibraryModule
+import co.yml.ychat.domain.usecases.AudioUseCase
 import co.yml.ychat.domain.usecases.ChatCompletionsUseCase
 import co.yml.ychat.domain.usecases.CompletionUseCase
 import co.yml.ychat.domain.usecases.EditsUseCase
 import co.yml.ychat.domain.usecases.ImageGenerationsUseCase
 import co.yml.ychat.domain.usecases.ListModelsUseCase
+import co.yml.ychat.entrypoint.features.AudioTranscriptions
 import co.yml.ychat.entrypoint.features.ChatCompletions
 import co.yml.ychat.entrypoint.features.Completion
 import co.yml.ychat.entrypoint.features.Edits
@@ -42,6 +44,7 @@ class LibraryModuleTest : KoinTest {
         get<ImageGenerations>()
         get<Edits>()
         get<ListModels>()
+        get<AudioTranscriptions>()
     }
 
     @Test
@@ -51,6 +54,7 @@ class LibraryModuleTest : KoinTest {
         get<ChatCompletionsUseCase>()
         get<ImageGenerationsUseCase>()
         get<EditsUseCase>()
+        get<AudioUseCase>()
     }
 
     @Test

ychat/src/commonTest/kotlin/co/yml/ychat/domain/model/AudioParamsTest.kt

@@ -0,0 +1,20 @@
+package co.yml.ychat.domain.model
+
+import kotlin.test.Test
+import kotlin.test.assertEquals
+
+class AudioParamsTest {
+
+    @Test
+    fun `on AudioParams verify default values`() {
+        // arrange
+        val params = AudioParams()
+
+        // assert
+        assertEquals("whisper-1", params.model)
+        assertEquals("", params.prompt)
+        assertEquals("json", params.responseFormat)
+        assertEquals(0.0, params.temperature)
+        assertEquals("en", params.language)
+    }
+}

ychat/src/commonTest/kotlin/co/yml/ychat/domain/usecases/AudioUseCaseTest.kt

@@ -0,0 +1,61 @@
+package co.yml.ychat.domain.usecases
+
+import co.yml.ychat.data.api.ChatGptApi
+import co.yml.ychat.data.dto.AudioResultDto
+import co.yml.ychat.data.exception.ChatGptException
+import co.yml.ychat.data.infrastructure.ApiResult
+import co.yml.ychat.domain.model.AudioParams
+import co.yml.ychat.domain.model.FileBytes
+import io.mockk.coEvery
+import io.mockk.mockk
+import kotlin.test.BeforeTest
+import kotlin.test.Test
+import kotlin.test.assertEquals
+import kotlinx.coroutines.runBlocking
+
+class AudioUseCaseTest {
+
+    private lateinit var useCase: AudioUseCase
+
+    private val chatGptApiMock = mockk<ChatGptApi>()
+
+    @BeforeTest
+    fun setup() {
+        useCase = AudioUseCase(chatGptApiMock)
+    }
+
+    @Test
+    fun `on requestAudioTranscription when request succeed then should return formatted result`() {
+        // arrange
+        val fileName = "audio-test.m4a"
+        val audioFile = ByteArray(1024) as FileBytes
+        val audioParams = AudioParams()
+        val audioResultDto = AudioResultDto("this is a test.")
+        val apiResult = ApiResult(body = audioResultDto)
+        coEvery { chatGptApiMock.audioTranscriptions(any()) } returns apiResult
+
+        // act
+        val result = runBlocking { useCase.requestAudioTranscription(fileName, audioFile, audioParams) }
+
+        // assert
+        assertEquals("this is a test.", result)
+    }
+
+    @Test
+    fun `on requestAudioTranscription when not request succeed then should throw an exception`() {
+        // arrange
+        val fileName = "audio-test.m4a"
+        val audioFile = ByteArray(1024) as FileBytes
+        val audioParams = AudioParams()
+        val apiResult = ApiResult<AudioResultDto>(exception = ChatGptException())
+        coEvery { chatGptApiMock.audioTranscriptions(any()) } returns apiResult
+
+        // act
+        val result = runCatching {
+            runBlocking { useCase.requestAudioTranscription(fileName, audioFile, audioParams) }
+        }
+
+        // assert
+        assertEquals(true, result.exceptionOrNull() is ChatGptException)
+    }
+}

ychat/src/commonTest/kotlin/co/yml/ychat/entrypoint/YChatTest.kt

@@ -1,6 +1,7 @@
 package co.yml.ychat.entrypoint
 
 import co.yml.ychat.YChat
+import co.yml.ychat.domain.model.FileBytes
 import co.yml.ychat.entrypoint.impl.YChatImpl
 import infrastructure.MockStorage
 import io.ktor.client.engine.HttpClientEngine
@@ -143,6 +144,28 @@ class YChatTest {
         assertEquals(expectedResult, result.id)
     }
 
+    @Test
+    fun `on audioTranscriptions execute method should return result successfully`() {
+        // arrange
+        val expectedResult = "This is a test."
+        val imageGenerationsSuccessResult =
+            MockStorage.audioTranscriptionsSuccessResult(expectedResult)
+        val audioFile = ByteArray(1024) as FileBytes
+        mockHttpEngine(imageGenerationsSuccessResult)
+
+        // act
+        val result = runBlocking {
+            yChat.audioTranscriptions()
+                .setTemperature(0.0)
+                .setModel("model-1")
+                .setPrompt("Test")
+                .execute("file.mp4", audioFile)
+        }
+
+        // assert
+        assertEquals(expectedResult, result)
+    }
+
     private fun mockHttpEngine(result: String) {
         val httpEngine = MockEngine {
             respond(

ychat/src/commonTest/kotlin/infrastructure/MockStorage.kt

@@ -135,4 +135,10 @@ object MockStorage {
         "parent": null
     }
     """
+
+    fun audioTranscriptionsSuccessResult(text: String) = """
+    {
+        "text": "$text"
+    }
+    """
 }