"""Tests for memory_store — port of long_term_memory tests in rust."""

import json
import os
import pytest


def test_remember_recall_round_trip(isolated_state):
    m = isolated_state('memory_store', ['_PATH'])
    m.remember("любимый чай", "улун")
    assert m.recall("любимый чай") == "улун"
    assert m.recall("Любимый Чай") == "улун"  # case-insensitive


def test_recall_returns_none_for_missing(isolated_state):
    m = isolated_state('memory_store', ['_PATH'])
    assert m.recall("nonexistent") is None


def test_forget_removes(isolated_state):
    m = isolated_state('memory_store', ['_PATH'])
    m.remember("foo", "bar")
    assert m.forget("foo") is True
    assert m.recall("foo") is None
    assert m.forget("foo") is False  # idempotent


def test_search_by_key_and_value(isolated_state):
    m = isolated_state('memory_store', ['_PATH'])
    m.remember("любимый чай", "улун")
    m.remember("любимый кофе", "арабика")
    m.remember("питомец", "собака бася")

    hits = m.search("чай", 5)
    assert len(hits) == 1
    assert hits[0]['key'] == "любимый чай"

    hits = m.search("бася", 5)
    assert len(hits) == 1
    assert hits[0]['key'] == "питомец"


def test_search_respects_limit(isolated_state):
    m = isolated_state('memory_store', ['_PATH'])
    for i in range(10):
        m.remember(f"key{i}", "value")
    hits = m.search("value", 3)
    assert len(hits) == 3


def test_search_empty_query_returns_nothing(isolated_state):
    m = isolated_state('memory_store', ['_PATH'])
    m.remember("foo", "bar")
    assert m.search("", 5) == []
    assert m.search("   ", 5) == []


def test_persists_across_module_reload(isolated_state, tmp_path):
    m = isolated_state('memory_store', ['_PATH'])
    m.remember("persistent", "yes")
    json_file = m._PATH
    assert os.path.isfile(json_file)
    with open(json_file, encoding='utf-8') as f:
        data = json.load(f)
    assert data['entries']['persistent']['value'] == 'yes'


def test_build_llm_context_format(isolated_state):
    m = isolated_state('memory_store', ['_PATH'])
    m.remember("любимый чай", "улун")
    # Query "чай" is substring of key "любимый чай" → matches.
    ctx = m.build_llm_context("чай", 5)
    assert "любимый чай = улун" in ctx
    assert ctx.startswith("Известные факты")


def test_build_llm_context_empty(isolated_state):
    m = isolated_state('memory_store', ['_PATH'])
    assert m.build_llm_context("anything", 5) == ""