"""Test ChatWatsonx API wrapper."""

import os

from langchain_ibm import ChatWatsonx

os.environ.pop("WATSONX_APIKEY", None)
os.environ.pop("WATSONX_PROJECT_ID", None)

MODEL_ID = "mistralai/mixtral-8x7b-instruct-v01"


def test_initialize_chat_watsonx_bad_path_without_url() -> None:
    try:
        ChatWatsonx(
            model_id=MODEL_ID,
        )
    except ValueError as e:
        assert "WATSONX_URL" in e.__str__()


def test_initialize_chat_watsonx_cloud_bad_path() -> None:
    try:
        ChatWatsonx(model_id=MODEL_ID, url="https://us-south.ml.cloud.ibm.com")  # type: ignore[arg-type]
    except ValueError as e:
        assert "WATSONX_APIKEY" in e.__str__()


def test_initialize_chat_watsonx_cpd_bad_path_without_all() -> None:
    try:
        ChatWatsonx(
            model_id=MODEL_ID,
            url="https://cpd-zen.apps.cpd48.cp.fyre.ibm.com",  # type: ignore[arg-type]
        )
    except ValueError as e:
        assert (
            "WATSONX_APIKEY" in e.__str__()
            and "WATSONX_PASSWORD" in e.__str__()
            and "WATSONX_TOKEN" in e.__str__()
        )


def test_initialize_chat_watsonx_cpd_bad_path_password_without_username() -> None:
    try:
        ChatWatsonx(
            model_id=MODEL_ID,
            url="https://cpd-zen.apps.cpd48.cp.fyre.ibm.com",  # type: ignore[arg-type]
            password="test_password",  # type: ignore[arg-type]
        )
    except ValueError as e:
        assert "WATSONX_USERNAME" in e.__str__()


def test_initialize_chat_watsonx_cpd_bad_path_apikey_without_username() -> None:
    try:
        ChatWatsonx(
            model_id=MODEL_ID,
            url="https://cpd-zen.apps.cpd48.cp.fyre.ibm.com",  # type: ignore[arg-type]
            apikey="test_apikey",  # type: ignore[arg-type]
        )
    except ValueError as e:
        assert "WATSONX_USERNAME" in e.__str__()