package com.openai.core.http

import com.openai.azure.AzureUrlPathMode
import com.openai.azure.credential.AzureApiKeyCredential
import com.openai.client.okhttp.OkHttpClient
import com.openai.core.ClientOptions
import com.openai.credential.BearerTokenCredential
import java.util.stream.Stream
import kotlin.test.Test
import org.assertj.core.api.Assertions.assertThat
import org.assertj.core.api.Assertions.assertThatThrownBy
import org.junit.jupiter.params.ParameterizedTest
import org.junit.jupiter.params.provider.MethodSource

internal class ClientOptionsTest {

    companion object {
        private const val FAKE_API_KEY = "test-api-key"

        @JvmStatic
        private fun createOkHttpClient(): OkHttpClient {
            return OkHttpClient.builder().build()
        }

        @JvmStatic
        private fun provideBaseUrls(): Stream<String> {
            return Stream.of(
                "https://api.openai.com/v1",
                "https://example.openai.azure.com",
                "https://example.azure-api.net",
            )
        }
    }

    fun clientOptionsWithoutBaseUrl() {
        // Arrange
        val apiKey = FAKE_API_KEY

        // Act
        val clientOptions =
            ClientOptions.builder()
                .httpClient(createOkHttpClient())
                .credential(BearerTokenCredential.create(apiKey))
                .build()

        // Assert
        assertThat(clientOptions.baseUrl()).isEqualTo(ClientOptions.PRODUCTION_URL)
    }

    @ParameterizedTest
    @MethodSource("provideBaseUrls")
    fun throwExceptionWhenNullCredential(baseUrl: String) {
        // Act
        val clientOptionsBuilder =
            ClientOptions.builder().httpClient(createOkHttpClient()).baseUrl(baseUrl)

        // Assert
        assertThatThrownBy { clientOptionsBuilder.build() }
            .isInstanceOf(IllegalStateException::class.java)
            .hasMessage("Either credential (apiKey) or workloadIdentity must be specified")
    }

    @Test
    fun throwExceptionWhenEmptyCredential() {
        assertThatThrownBy { AzureApiKeyCredential.create("") }
            .isInstanceOf(IllegalArgumentException::class.java)
            .hasMessage("Azure API key cannot be empty.")
    }

    @Test
    fun azureUrlPathMode_setToLegacy() {
        val clientOptions =
            ClientOptions.builder()
                .httpClient(createOkHttpClient())
                .credential(BearerTokenCredential.create(FAKE_API_KEY))
                .azureUrlPathMode(AzureUrlPathMode.LEGACY)
                .build()

        assertThat(clientOptions.azureUrlPathMode).isEqualTo(AzureUrlPathMode.LEGACY)
    }

    @Test
    fun azureUrlPathMode_setToUnified() {
        val clientOptions =
            ClientOptions.builder()
                .httpClient(createOkHttpClient())
                .credential(BearerTokenCredential.create(FAKE_API_KEY))
                .azureUrlPathMode(AzureUrlPathMode.UNIFIED)
                .build()

        assertThat(clientOptions.azureUrlPathMode).isEqualTo(AzureUrlPathMode.UNIFIED)
    }

    @Test
    fun azureUrlPathMode_defaultsToAuto() {
        val clientOptions =
            ClientOptions.builder()
                .httpClient(createOkHttpClient())
                .credential(BearerTokenCredential.create(FAKE_API_KEY))
                .build()

        assertThat(clientOptions.azureUrlPathMode).isEqualTo(AzureUrlPathMode.AUTO)
    }
}
