refactor:add integration test for the imageGenerations() entry point

Author: renatoarg

ychat/src/commonTest/kotlin/co/yml/ychat/domain/mapper/ImageGenerationsMapperTest.kt

@@ -15,7 +15,7 @@ class ImageGenerationsMapperTest {
             created = 12345,
             data = listOfImageGeneratedDto
         )
-        assertEquals(listOfImageGeneratedDto, imageGenerationsDto.toImageGenerated())
+        assertEquals(listOfImageGeneratedDto.map { it.url }, imageGenerationsDto.toImageGenerated())
     }
 
     @Test

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

@@ -37,7 +37,7 @@ class ImageGenerationsUseCaseTest {
         val result = runBlocking { imageGenerationsUseCase.requestImageGenerations(params) }
 
         // assert
-        assertEquals("https://image-generated.test", result.last().url)
+        assertEquals("https://image-generated.test", result.last())
     }
 
     @Test

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

@@ -73,6 +73,26 @@ class YChatTest {
         assertEquals("This in indeed a test", result)
     }
 
+    @Test
+    fun `on imageGenerations execute method should return result successfully`() {
+        // arrange
+        val textResult = "This in indeed a test"
+        val imageGenerationsSuccessResult = MockStorage.imageGenerationsSuccessResult(textResult)
+        mockHttpEngine(imageGenerationsSuccessResult)
+
+        // act
+        val result = runBlocking {
+            yChat.imageGenerations()
+                .setResults(1)
+                .setSize("256x256")
+                .setResponseFormat("url")
+                .execute("/image ocean")
+        }
+
+        // assert
+        assertEquals("https://testlink.com/image-test.jps", result.first())
+    }
+
     private fun mockHttpEngine(result: String) {
         val httpEngine = MockEngine {
             respond(

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

@@ -13,4 +13,6 @@ object MockStorage {
         "\"usage\":{\"prompt_tokens\":13,\"completion_tokens\":12,\"total_tokens\":25}," +
         "\"choices\":[{\"message\":{\"role\":\"assistant\",\"content\":\"$text\"}," +
         "\"finish_reason\":\"stop\",\"index\":0}]}"
+
+    fun imageGenerationsSuccessResult(text: String) ="{\"created\":1678805561,\"data\":[{\"url\":\"https://testlink.com/image-test.jps\"}]}"
 }