import asyncio
import sys
import logging
from unittest.mock import AsyncMock

from mnemocore.core.async_storage import AsyncRedisStorage

logging.basicConfig(level=logging.DEBUG)

async def main():
    print("Starting debug_async.py...")
    try:
        mock_client = AsyncMock()
        print("Mock client created.")

        storage = AsyncRedisStorage(client=mock_client)
        print("AsyncRedisStorage initialized.")

        node_id = "mem_debug"
        data = {"content": "debug"}

        print("Calling store_memory...")
        await storage.store_memory(node_id, data)
        print("store_memory returned.")

        mock_client.set.assert_called_once()
        print("Assertion passed.")

    except Exception as e:
        print(f"Error: {e}")
        sys.exit(1)

if __name__ == "__main__":
    asyncio.run(main())