Sort_YM/tests/test_analyze.py

import json
import re
from pathlib import Path

from sort_ym import analyze


def test_theme_pattern_accepts_english_snake_case():
    assert re.match(analyze.THEME_PATTERN, "digital_aging")
    assert re.match(analyze.THEME_PATTERN, "modern_loneliness")


def test_theme_pattern_rejects_pilot_failure_modes():
    # Пилот без паттерна давал кириллицу и составные значения через слэш вперемешку
    # с английским в одном прогоне - агрегировать такое невозможно.
    assert not re.match(analyze.THEME_PATTERN, "одиночество")
    assert not re.match(analyze.THEME_PATTERN, "эмоциональное выгорание")
    assert not re.match(analyze.THEME_PATTERN, "Digital_Aging")
    assert not re.match(analyze.THEME_PATTERN, "ethereal/hopeful")
    assert not re.match(analyze.THEME_PATTERN, "ab")


def test_mood_enum_excludes_stance_axis():
    # Смешение осей в пилоте давало "aggressive/triumphant"; ирония/сарказм живут в stance.
    assert "ironic" not in analyze.MOODS
    assert "sarcastic" not in analyze.MOODS
    assert "sincere" not in analyze.MOODS
    assert len(analyze.MOODS) == len(set(analyze.MOODS))


def test_response_schema_constrains_mood_and_themes():
    props = analyze.RESPONSE_SCHEMA["properties"]
    assert props["mood"]["items"]["enum"] == analyze.MOODS
    assert props["mood"]["maxItems"] == 3
    assert props["themes"]["items"]["pattern"] == analyze.THEME_PATTERN
    assert set(analyze.RESPONSE_SCHEMA["required"]) == set(props)


def test_lyrics_hash_changes_with_text():
    assert analyze.lyrics_hash("текст") == analyze.lyrics_hash("текст")
    assert analyze.lyrics_hash("текст") != analyze.lyrics_hash("другой текст")


def test_verify_key_line_accepts_verbatim_quote_ignoring_whitespace_and_case():
    text = "Первая строка\nА я иду, шагаю по Москве\nТретья строка"

    assert analyze.verify_key_line("А я иду, шагаю по Москве", text)
    assert analyze.verify_key_line("а я  иду,  шагаю по москве", text)


def test_verify_key_line_rejects_paraphrase_and_empty():
    text = "А я иду, шагаю по Москве"

    assert not analyze.verify_key_line("Герой идёт по городу", text)
    assert not analyze.verify_key_line("", text)


def test_is_fresh_requires_model_prompt_version_and_hash_match():
    entry = {"model": "m1", "prompt_version": 1, "lyrics_hash": "h1"}

    assert analyze.is_fresh(entry, "m1", 1, "h1")
    assert not analyze.is_fresh(entry, "m2", 1, "h1")
    assert not analyze.is_fresh(entry, "m1", 2, "h1")
    assert not analyze.is_fresh(entry, "m1", 1, "h2")
    assert not analyze.is_fresh(None, "m1", 1, "h1")


def test_is_fresh_false_for_error_marker():
    # Ошибочные записи перезапрашиваются при следующем прогоне - это и есть механизм
    # "повторить упавшие треки" вместо отдельного флага.
    entry = {"model": "m1", "prompt_version": 1, "lyrics_hash": "h1", "error": "TimeoutError: ..."}

    assert not analyze.is_fresh(entry, "m1", 1, "h1")


def test_top_themes_orders_by_frequency_and_limits():
    cache = {
        "1": {"themes": ["loneliness", "city"]},
        "2": {"themes": ["loneliness", "memory"]},
        "3": {"themes": ["loneliness", "city"]},
        "4": {"error": "TimeoutError: ..."},
    }

    assert analyze.top_themes(cache, limit=2) == ["loneliness", "city"]


def test_load_analysis_returns_empty_dict_when_missing(tmp_path: Path):
    assert analyze.load_analysis(tmp_path) == {}


class FakeResponse:
    def __init__(self, payload):
        self._payload = payload

    def read(self):
        return json.dumps(self._payload).encode("utf-8")

    def __enter__(self):
        return self

    def __exit__(self, *exc):
        return False


def _settings(**overrides):
    base = dict(
        host="http://localhost:11434",
        model="qwen3.6-35b-a3b:latest",
        prompt_version=1,
        timeout=600,
        keep_alive="30m",
    )
    base.update(overrides)
    return analyze.OllamaSettings(**base)


def test_call_ollama_sends_schema_constrained_thinking_request(monkeypatch):
    captured = {}

    def fake_urlopen(req, timeout=None):
        captured["url"] = req.full_url
        captured["timeout"] = timeout
        captured["body"] = json.loads(req.data.decode("utf-8"))
        return FakeResponse(
            {"message": {"thinking": "рассуждение", "content": json.dumps({"mood": ["grief"]})}}
        )

    monkeypatch.setattr(analyze.urllib.request, "urlopen", fake_urlopen)

    result = analyze.call_ollama(_settings(), "промпт")

    assert captured["url"] == "http://localhost:11434/api/chat"
    assert captured["timeout"] == 600
    body = captured["body"]
    assert body["model"] == "qwen3.6-35b-a3b:latest"
    assert body["stream"] is False
    assert body["think"] is True
    assert body["format"] == analyze.RESPONSE_SCHEMA
    assert body["keep_alive"] == "30m"
    assert body["options"] == {"num_predict": -1, "temperature": 0.65}
    assert body["messages"] == [{"role": "user", "content": "промпт"}]
    assert result == {"mood": ["grief"]}, "в кэш идёт только content, не thinking"


def test_call_ollama_strips_trailing_slash_from_host(monkeypatch):
    captured = {}

    def fake_urlopen(req, timeout=None):
        captured["url"] = req.full_url
        return FakeResponse({"message": {"content": "{}"}})

    monkeypatch.setattr(analyze.urllib.request, "urlopen", fake_urlopen)

    analyze.call_ollama(_settings(host="http://localhost:11434/"), "промпт")

    assert captured["url"] == "http://localhost:11434/api/chat"


def test_build_prompt_includes_lyrics_title_and_known_themes():
    prompt = analyze.build_prompt("Тоска", ["Артист"], "Текст песни", ["loneliness", "city"])

    assert "Тоска" in prompt
    assert "Артист" in prompt
    assert "Текст песни" in prompt
    assert "loneliness, city" in prompt


def test_build_prompt_omits_known_themes_section_when_empty():
    prompt = analyze.build_prompt("Тоска", ["Артист"], "Текст песни", [])

    assert "Уже использованные темы" not in prompt