from traderai.memory import MemoryStore


def test_memory_store_recalls_saved_fact(tmp_path):
    store = MemoryStore(str(tmp_path / "memory.sqlite3"))
    store.remember("preference", "The user prefers Polaris Bits searches to include barter listings.", importance=5)

    results = store.recall("Polaris barter")

    assert results
    assert "Polaris Bits" in results[0]["content"]


def test_memory_store_clear_selected_sections(tmp_path):
    store = MemoryStore(str(tmp_path / "memory.sqlite3"))
    store.remember("note", "Forgettable note")
    store.add_conversation("user", "hello")
    store.set_profile("configured_name", "Hudson")

    deleted = store.clear(include_profile=False)
    snapshot = store.inspect()

    assert deleted["memories"] == 1
    assert deleted["conversations"] == 1
    assert snapshot["memories"] == []
    assert snapshot["conversations"] == []
    assert snapshot["profile"][0]["key"] == "configured_name"