hermes-agent/tests/gateway/test_sticker_cache.py

"""Tests for gateway/sticker_cache.py — sticker description cache."""

import json
import time
from unittest.mock import patch

from gateway.sticker_cache import (
    _load_cache,
    _save_cache,
    get_cached_description,
    cache_sticker_description,
    build_sticker_injection,
    build_animated_sticker_injection,
    STICKER_VISION_PROMPT,
)


class TestLoadSaveCache:
    def test_load_missing_file(self, tmp_path):
        with patch("gateway.sticker_cache.CACHE_PATH", tmp_path / "nope.json"):
            assert _load_cache() == {}

    def test_load_corrupt_file(self, tmp_path):
        bad_file = tmp_path / "bad.json"
        bad_file.write_text("not json{{{")
        with patch("gateway.sticker_cache.CACHE_PATH", bad_file):
            assert _load_cache() == {}

    def test_save_and_load_roundtrip(self, tmp_path):
        cache_file = tmp_path / "cache.json"
        data = {"abc123": {"description": "A cat", "emoji": "", "set_name": "", "cached_at": 1.0}}
        with patch("gateway.sticker_cache.CACHE_PATH", cache_file):
            _save_cache(data)
            loaded = _load_cache()
        assert loaded == data

    def test_save_creates_parent_dirs(self, tmp_path):
        cache_file = tmp_path / "sub" / "dir" / "cache.json"
        with patch("gateway.sticker_cache.CACHE_PATH", cache_file):
            _save_cache({"key": "value"})
        assert cache_file.exists()


class TestCacheSticker:
    def test_cache_and_retrieve(self, tmp_path):
        cache_file = tmp_path / "cache.json"
        with patch("gateway.sticker_cache.CACHE_PATH", cache_file):
            cache_sticker_description("uid_1", "A happy dog", emoji="🐕", set_name="Dogs")
            result = get_cached_description("uid_1")

        assert result is not None
        assert result["description"] == "A happy dog"
        assert result["emoji"] == "🐕"
        assert result["set_name"] == "Dogs"
        assert "cached_at" in result

    def test_missing_sticker_returns_none(self, tmp_path):
        cache_file = tmp_path / "cache.json"
        with patch("gateway.sticker_cache.CACHE_PATH", cache_file):
            result = get_cached_description("nonexistent")
        assert result is None

    def test_overwrite_existing(self, tmp_path):
        cache_file = tmp_path / "cache.json"
        with patch("gateway.sticker_cache.CACHE_PATH", cache_file):
            cache_sticker_description("uid_1", "Old description")
            cache_sticker_description("uid_1", "New description")
            result = get_cached_description("uid_1")

        assert result["description"] == "New description"

    def test_multiple_stickers(self, tmp_path):
        cache_file = tmp_path / "cache.json"
        with patch("gateway.sticker_cache.CACHE_PATH", cache_file):
            cache_sticker_description("uid_1", "Cat")
            cache_sticker_description("uid_2", "Dog")
            r1 = get_cached_description("uid_1")
            r2 = get_cached_description("uid_2")

        assert r1["description"] == "Cat"
        assert r2["description"] == "Dog"


class TestBuildStickerInjection:
    def test_exact_format_no_context(self):
        result = build_sticker_injection("A cat waving")
        assert result == '[The user sent a sticker~ It shows: "A cat waving" (=^.w.^=)]'

    def test_exact_format_emoji_only(self):
        result = build_sticker_injection("A cat", emoji="😀")
        assert result == '[The user sent a sticker 😀~ It shows: "A cat" (=^.w.^=)]'

    def test_exact_format_emoji_and_set_name(self):
        result = build_sticker_injection("A cat", emoji="😀", set_name="MyPack")
        assert result == '[The user sent a sticker 😀 from "MyPack"~ It shows: "A cat" (=^.w.^=)]'

    def test_set_name_without_emoji_ignored(self):
        """set_name alone (no emoji) produces no context — only emoji+set_name triggers 'from' clause."""
        result = build_sticker_injection("A cat", set_name="MyPack")
        assert result == '[The user sent a sticker~ It shows: "A cat" (=^.w.^=)]'
        assert "MyPack" not in result

    def test_description_with_quotes(self):
        result = build_sticker_injection('A "happy" dog')
        assert '"A \\"happy\\" dog"' not in result  # no escaping happens
        assert 'A "happy" dog' in result

    def test_empty_description(self):
        result = build_sticker_injection("")
        assert result == '[The user sent a sticker~ It shows: "" (=^.w.^=)]'


class TestBuildAnimatedStickerInjection:
    def test_exact_format_with_emoji(self):
        result = build_animated_sticker_injection(emoji="🎉")
        assert result == (
            "[The user sent an animated sticker 🎉~ "
            "I can't see animated ones yet, but the emoji suggests: 🎉]"
        )

    def test_exact_format_without_emoji(self):
        result = build_animated_sticker_injection()
        assert result == "[The user sent an animated sticker~ I can't see animated ones yet]"

    def test_empty_emoji_same_as_no_emoji(self):
        result = build_animated_sticker_injection(emoji="")
        assert result == build_animated_sticker_injection()