"""
Unit tests for ModelFactory with new default model (gemini-3.6-flash-high)
"""
import pytest
from unittest.mock import patch, Mock
from src.models.factory import ModelFactory

def test_model_factory_default():
    factory = ModelFactory()
    model = factory.get_model()
    # Check that default model loaded is gemini-3.6-flash-high
    assert model.id == "gemini-3.6-flash-high"
    assert model.base_url == "http://ai.newhorizonco.uk/v1"

def test_model_factory_get_specific_model():
    factory = ModelFactory()
    model = factory.get_model("gemini_flash")
    assert model.id == "gemini-3.6-flash-high"