import pytest
import sys
from pathlib import Path
from unittest.mock import Mock, AsyncMock, patch

sys.path.insert(0, str(Path(__file__).parent.parent / "src"))

from promptforge.providers.factory import ProviderFactory
from promptforge.providers.openai import OpenAIProvider
from promptforge.providers.anthropic import AnthropicProvider
from promptforge.providers.ollama import OllamaProvider


class TestProviderFactory:
    def test_create_openai(self):
        provider = ProviderFactory.create("openai", model="gpt-4")
        assert isinstance(provider, OpenAIProvider)
        assert provider.model == "gpt-4"

    def test_create_anthropic(self):
        provider = ProviderFactory.create("anthropic", model="claude-3")
        assert isinstance(provider, AnthropicProvider)

    def test_create_ollama(self):
        provider = ProviderFactory.create("ollama", model="llama2")
        assert isinstance(provider, OllamaProvider)

    def test_create_unknown(self):
        with pytest.raises(Exception):
            ProviderFactory.create("unknown")


class TestOpenAIProvider:
    def test_provider_name(self):
        provider = OpenAIProvider(api_key="test-key")
        assert provider.name == "openai"

    def test_provider_defaults(self):
        provider = OpenAIProvider()
        assert provider.model == "gpt-4"
        assert provider.temperature == 0.7


class TestAnthropicProvider:
    def test_provider_name(self):
        provider = AnthropicProvider(api_key="test-key")
        assert provider.name == "anthropic"


class TestOllamaProvider:
    def test_provider_name(self):
        provider = OllamaProvider()
        assert provider.name == "ollama"

    def test_provider_defaults(self):
        provider = OllamaProvider(model="llama2")
        assert provider.model == "llama2"