#!/usr/bin/env python3
"""
Test script for memory tools.
"""

import sys
from pathlib import Path

# Add parent directories to path
sys.path.insert(0, str(Path(__file__).parent.parent))

from tools.memory_tools import StoreMemoryTool, GetMemoryTool, SearchMemoryTool, ListMemoryTool

def test_memory_tools():
    """Test memory tools."""
    print("=" * 60)
    print("Memory Tools Test")
    print("=" * 60)
    
    # Test StoreMemoryTool
    print("\n1. Testing StoreMemoryTool...")
    store_tool = StoreMemoryTool()
    result = store_tool.execute({
        "category": "preferences",
        "key": "favorite_color",
        "value": "blue",
        "confidence": 1.0
    })
    print(f"   ✅ Store memory: {result['message']}")
    print(f"      Entry ID: {result['entry_id']}")
    
    # Test GetMemoryTool
    print("\n2. Testing GetMemoryTool...")
    get_tool = GetMemoryTool()
    result = get_tool.execute({
        "category": "preferences",
        "key": "favorite_color"
    })
    if result["found"]:
        print(f"   ✅ Get memory: {result['key']} = {result['value']}")
        print(f"      Confidence: {result['confidence']}")
    else:
        print(f"   ❌ Memory not found")
    
    # Test SearchMemoryTool
    print("\n3. Testing SearchMemoryTool...")
    search_tool = SearchMemoryTool()
    result = search_tool.execute({
        "query": "blue",
        "limit": 5
    })
    print(f"   ✅ Search memory: Found {result['count']} results")
    for entry in result['results'][:3]:
        print(f"      - {entry['key']}: {entry['value']}")
    
    # Test ListMemoryTool
    print("\n4. Testing ListMemoryTool...")
    list_tool = ListMemoryTool()
    result = list_tool.execute({
        "category": "preferences",
        "limit": 10
    })
    print(f"   ✅ List memory: {result['count']} entries in preferences")
    for entry in result['entries'][:3]:
        print(f"      - {entry['key']}: {entry['value']}")
    
    print("\n" + "=" * 60)
    print("✅ Memory tools tests complete!")
    print("=" * 60)

if __name__ == "__main__":
    test_memory_tools()