import asyncio
from app.services.conversation_service import conversation_service
import time

def test_redis_persistence():
    print("\n--- Testing Redis Persistence ---")
    
    # 1. Check Connection Mode
    if conversation_service.use_redis:
        print("✅ Using Redis Backend")
    else:
        print("⚠️ Using In-Memory Fallback (Redis not reachable?)")
        
    # 2. Set Identity
    print("Setting Identity to 'RedisUser'...")
    conversation_service.set_user_identity("RedisUser")
    
    # 3. Retrieve Identity
    identity = conversation_service.get_user_identity()
    print(f"Retrieved Identity: {identity}")
    assert identity["name"] == "RedisUser"
    
    # 4. Set Context
    print("Setting Context to 'RedisTopic'...")
    conversation_service.update_context({"name": "RedisTopic", "notes": "Testing Persistence"})
    
    # 5. Retrieve Context
    context = conversation_service.get_context()
    print(f"Retrieved Context: {context}")
    assert context["name"] == "RedisTopic"
    
    print("\n✅ Verification Successful!")

if __name__ == "__main__":
    test_redis_persistence()