test_inventory_sync.py

"""
🧪 Test Inventory Sync System
Тестовый скрипт для проверки динамической синхронизации
"""
import asyncio
import httpx
import json
from typing import Optional

BASE_URL = "http://localhost:8000"

class Colors:
    GREEN = '\033[92m'
    RED = '\033[91m'
    YELLOW = '\033[93m'
    BLUE = '\033[94m'
    END = '\033[0m'

def print_success(msg: str):
    print(f"{Colors.GREEN}✅ {msg}{Colors.END}")

def print_error(msg: str):
    print(f"{Colors.RED}❌ {msg}{Colors.END}")

def print_info(msg: str):
    print(f"{Colors.BLUE}ℹ️  {msg}{Colors.END}")

def print_warning(msg: str):
    print(f"{Colors.YELLOW}⚠️  {msg}{Colors.END}")

async def test_inventory_sync():
    """Main test function"""
    async with httpx.AsyncClient() as client:
        print("\n" + "="*60)
        print("🧪 Testing Inventory Sync System")
        print("="*60 + "\n")
        
        # Test 1: Check API is running
        print("Test 1: Check API Status")
        try:
            response = await client.get(f"{BASE_URL}/api/inventory/test")
            if response.status_code == 200:
                print_success("API is running")
            else:
                print_error(f"API returned status {response.status_code}")
                return
        except Exception as e:
            print_error(f"Cannot connect to API: {e}")
            print_warning("Make sure to run: python main.py")
            return
        
        # Test 2: Get initial inventory
        print("\nTest 2: Get Initial Inventory")
        try:
            response = await client.get(f"{BASE_URL}/api/inventory/items?user_id=1")
            data = response.json()
            print_info(f"Current items: {data.get('count', 0)}")
            if data.get('items'):
                for item in data['items']:
                    print(f"  • {item['name']} - {item['quantity']}")
        except Exception as e:
            print_error(f"Error: {e}")
        
        # Test 3: Add item to inventory
        print("\nTest 3: Add Item to Inventory")
        print_info("Adding 'яйцо' (10 шт) to inventory...")
        try:
            response = await client.post(
                f"{BASE_URL}/api/inventory/items/add",
                json={
                    "name": "яйцо",
                    "quantity": "10 шт",
                    "category": "молочные продукты",
                    "user_id": 1
                }
            )
            data = response.json()
            
            if data.get('success'):
                print_success(f"Item added: {data['message']}")
                
                sync_result = data.get('data', {}).get('sync_result', {})
                if sync_result.get('synced'):
                    print_info(f"Sync action: {sync_result.get('action')}")
                    
                    removed_items = sync_result.get('items_removed', [])
                    if removed_items:
                        print_success(f"Removed from shopping list: {', '.join(removed_items)}")
                    
                    if sync_result.get('cart_updated'):
                        print_success("Cart updated successfully")
                        print_info(f"Remaining items: {sync_result.get('remaining_items')}")
                        print_info(f"New total: {sync_result.get('new_total')}₽")
            else:
                print_error(f"Failed: {data}")
                
        except Exception as e:
            print_error(f"Error: {e}")
        
        # Test 4: Check sync status
        print("\nTest 4: Check Sync Status")
        try:
            response = await client.get(f"{BASE_URL}/api/inventory/sync/status?user_id=1")
            data = response.json()
            
            if data.get('success'):
                status = data['data']
                
                print_info("Current Status:")
                print(f"\n📦 Inventory:")
                print(f"   Items: {status['inventory']['count']}")
                if status['inventory']['items']:
                    for item in status['inventory']['items'][:5]:
                        print(f"   • {item}")
                
                print(f"\n🛒 Shopping List:")
                print(f"   Items: {status['shopping_list']['count']}")
                print(f"   Total: {status['shopping_list']['total_cost']}₽")
                if status['shopping_list']['items']:
                    for item in status['shopping_list']['items'][:5]:
                        print(f"   • {item}")
                
                print(f"\n🛍️ Cart:")
                print(f"   Items: {status['cart']['count']}")
                if status['cart']['link']:
                    print(f"   Link: {status['cart']['link']}")
                else:
                    print_warning("   No cart link available")
                    print_info("   Generate shopping list first: POST /api/shopping/generate")
                
        except Exception as e:
            print_error(f"Error: {e}")
        
        # Test 5: Remove item
        print("\nTest 5: Remove Item from Inventory")
        print_info("Removing 'яйцо' from inventory...")
        try:
            response = await client.post(
                f"{BASE_URL}/api/inventory/items/remove",
                json={
                    "name": "яйцо",
                    "user_id": 1
                }
            )
            data = response.json()
            
            if data.get('success'):
                print_success(f"Item removed: {data['message']}")
                
                sync_result = data.get('data', {}).get('sync_result', {})
                if sync_result.get('synced'):
                    print_info(f"Sync action: {sync_result.get('action')}")
                    if sync_result.get('message'):
                        print_warning(sync_result['message'])
            else:
                print_warning(f"Item not found or already removed")
                
        except Exception as e:
            print_error(f"Error: {e}")
        
        print("\n" + "="*60)
        print("✨ Test Complete!")
        print("="*60 + "\n")
        
        print("📚 Next Steps:")
        print("1. Generate shopping list: POST /api/shopping/generate")
        print("2. Add items to inventory: POST /api/inventory/items/add")
        print("3. Check sync status: GET /api/inventory/sync/status")
        print("\n📖 Full docs: INVENTORY_SYNC_GUIDE.md\n")


if __name__ == "__main__":
    print("""
╔══════════════════════════════════════════════════════════╗
║     🔄 Inventory Sync System - Test Suite              ║
║                                                          ║
║  Тестирование динамической синхронизации:                ║
║  • Холодильник ↔ Список покупок ↔ Корзина ВкусВилл     ║
╚══════════════════════════════════════════════════════════╝
    """)
    
    asyncio.run(test_inventory_sync())