import pytest
from app.services.vision import has_images
from app.services.cache_layer import make_cache_key, compute_embedding, cosine_similarity


class TestProxyRouter:
    def test_has_images_text_only(self):
        messages = [{"role": "user", "content": "Hello"}]
        assert has_images(messages) is False

    def test_has_images_with_image_url(self):
        messages = [{"role": "user", "content": [{"type": "image_url", "image_url": {"url": "https://example.com/img.jpg"}}]}]
        assert has_images(messages) is True

    def test_has_images_with_content_list_text(self):
        messages = [{"role": "user", "content": [{"type": "text", "text": "describe this"}]}]
        assert has_images(messages) is False

    def test_make_cache_key_deterministic(self):
        messages = [{"role": "user", "content": "hello"}]
        key1 = make_cache_key("gpt-4", messages)
        key2 = make_cache_key("gpt-4", messages)
        assert key1 == key2

    def test_make_cache_key_different_model(self):
        messages = [{"role": "user", "content": "hello"}]
        key1 = make_cache_key("gpt-4", messages)
        key2 = make_cache_key("gpt-3.5", messages)
        assert key1 != key2

    def test_compute_embedding(self):
        emb = compute_embedding("hello world")
        assert len(emb) == 256
        assert all(isinstance(v, float) for v in emb)

    def test_cosine_similarity_identical(self):
        emb = compute_embedding("test")
        sim = cosine_similarity(emb, emb)
        assert abs(sim - 1.0) < 0.001

    def test_cosine_similarity_orthogonal(self):
        a = [1.0] + [0.0] * 255
        b = [0.0] * 256
        b[0] = 1.0
        sim = cosine_similarity(a, b)
        assert abs(sim - 1.0) < 0.001

    def test_cosine_similarity_different(self):
        a = compute_embedding("hello world")
        b = compute_embedding("goodbye universe")
        assert cosine_similarity(a, b) < 0.99