// File generated from our OpenAPI spec by Stainless.

package com.openai.services.blocking

import com.openai.TestServerExtension
import com.openai.client.okhttp.OpenAIOkHttpClient
import com.openai.models.images.ImageCreateVariationParams
import com.openai.models.images.ImageEditParams
import com.openai.models.images.ImageGenerateParams
import com.openai.models.images.ImageModel
import org.junit.jupiter.api.Disabled
import org.junit.jupiter.api.Test
import org.junit.jupiter.api.extension.ExtendWith

@ExtendWith(TestServerExtension::class)
internal class ImageServiceTest {

    @Test
    fun createVariation() {
        val client =
            OpenAIOkHttpClient.builder()
                .baseUrl(TestServerExtension.BASE_URL)
                .apiKey("My API Key")
                .adminApiKey("My Admin API Key")
                .build()
        val imageService = client.images()

        val imagesResponse =
            imageService.createVariation(
                ImageCreateVariationParams.builder()
                    .image("Example data".byteInputStream())
                    .model(ImageModel.GPT_IMAGE_1_5)
                    .n(1L)
                    .responseFormat(ImageCreateVariationParams.ResponseFormat.URL)
                    .size(ImageCreateVariationParams.Size._1024X1024)
                    .user("user-1234")
                    .build()
            )

        imagesResponse.validate()
    }

    @Test
    fun edit() {
        val client =
            OpenAIOkHttpClient.builder()
                .baseUrl(TestServerExtension.BASE_URL)
                .apiKey("My API Key")
                .adminApiKey("My Admin API Key")
                .build()
        val imageService = client.images()

        val imagesResponse =
            imageService.edit(
                ImageEditParams.builder()
                    .image("Example data".byteInputStream())
                    .prompt("A cute baby sea otter wearing a beret")
                    .background(ImageEditParams.Background.TRANSPARENT)
                    .inputFidelity(ImageEditParams.InputFidelity.HIGH)
                    .mask("Example data".byteInputStream())
                    .model(ImageModel.GPT_IMAGE_1_5)
                    .n(1L)
                    .outputCompression(100L)
                    .outputFormat(ImageEditParams.OutputFormat.PNG)
                    .partialImages(1L)
                    .quality(ImageEditParams.Quality.HIGH)
                    .responseFormat(ImageEditParams.ResponseFormat.URL)
                    .size(ImageEditParams.Size._1024X1024)
                    .user("user-1234")
                    .build()
            )

        imagesResponse.validate()
    }

    @Disabled("Prism does not support SSE streaming mock responses")
    @Test
    fun editStreaming() {
        val client =
            OpenAIOkHttpClient.builder()
                .baseUrl(TestServerExtension.BASE_URL)
                .apiKey("My API Key")
                .adminApiKey("My Admin API Key")
                .build()
        val imageService = client.images()

        val imagesResponseStreamResponse =
            imageService.editStreaming(
                ImageEditParams.builder()
                    .image("Example data".byteInputStream())
                    .prompt("A cute baby sea otter wearing a beret")
                    .background(ImageEditParams.Background.TRANSPARENT)
                    .inputFidelity(ImageEditParams.InputFidelity.HIGH)
                    .mask("Example data".byteInputStream())
                    .model(ImageModel.GPT_IMAGE_1_5)
                    .n(1L)
                    .outputCompression(100L)
                    .outputFormat(ImageEditParams.OutputFormat.PNG)
                    .partialImages(1L)
                    .quality(ImageEditParams.Quality.HIGH)
                    .responseFormat(ImageEditParams.ResponseFormat.URL)
                    .size(ImageEditParams.Size._1024X1024)
                    .user("user-1234")
                    .build()
            )

        imagesResponseStreamResponse.use {
            imagesResponseStreamResponse.stream().forEach { imagesResponse ->
                imagesResponse.validate()
            }
        }
    }

    @Test
    fun generate() {
        val client =
            OpenAIOkHttpClient.builder()
                .baseUrl(TestServerExtension.BASE_URL)
                .apiKey("My API Key")
                .adminApiKey("My Admin API Key")
                .build()
        val imageService = client.images()

        val imagesResponse =
            imageService.generate(
                ImageGenerateParams.builder()
                    .prompt("A cute baby sea otter")
                    .background(ImageGenerateParams.Background.TRANSPARENT)
                    .model(ImageModel.GPT_IMAGE_1_5)
                    .moderation(ImageGenerateParams.Moderation.LOW)
                    .n(1L)
                    .outputCompression(100L)
                    .outputFormat(ImageGenerateParams.OutputFormat.PNG)
                    .partialImages(1L)
                    .quality(ImageGenerateParams.Quality.MEDIUM)
                    .responseFormat(ImageGenerateParams.ResponseFormat.URL)
                    .size(ImageGenerateParams.Size._1024X1024)
                    .style(ImageGenerateParams.Style.VIVID)
                    .user("user-1234")
                    .build()
            )

        imagesResponse.validate()
    }

    @Disabled("Prism does not support SSE streaming mock responses")
    @Test
    fun generateStreaming() {
        val client =
            OpenAIOkHttpClient.builder()
                .baseUrl(TestServerExtension.BASE_URL)
                .apiKey("My API Key")
                .adminApiKey("My Admin API Key")
                .build()
        val imageService = client.images()

        val imagesResponseStreamResponse =
            imageService.generateStreaming(
                ImageGenerateParams.builder()
                    .prompt("A cute baby sea otter")
                    .background(ImageGenerateParams.Background.TRANSPARENT)
                    .model(ImageModel.GPT_IMAGE_1_5)
                    .moderation(ImageGenerateParams.Moderation.LOW)
                    .n(1L)
                    .outputCompression(100L)
                    .outputFormat(ImageGenerateParams.OutputFormat.PNG)
                    .partialImages(1L)
                    .quality(ImageGenerateParams.Quality.MEDIUM)
                    .responseFormat(ImageGenerateParams.ResponseFormat.URL)
                    .size(ImageGenerateParams.Size._1024X1024)
                    .style(ImageGenerateParams.Style.VIVID)
                    .user("user-1234")
                    .build()
            )

        imagesResponseStreamResponse.use {
            imagesResponseStreamResponse.stream().forEach { imagesResponse ->
                imagesResponse.validate()
            }
        }
    }
}
