#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Тестирование нового формата сообщений для AI-КАЛОРИЯ бота
"""

import os
import sys
from pathlib import Path

# Добавляем текущую директорию в путь
sys.path.insert(0, str(Path(__file__).parent))

# Загружаем переменные окружения
from dotenv import load_dotenv
load_dotenv()

def test_new_format():
    """Тестирование нового формата сообщений"""
    print("🧪 Тестирование нового формата сообщений...")
    
    try:
        from telegram_bot import format_food_record
        
        # Тестовые данные
        test_cases = [
            {
                "food_name": "мороженное",
                "amount": 1,
                "unit": "стаканчик",
                "calories": 200
            },
            {
                "food_name": "роллы филадельфия",
                "amount": 4,
                "unit": "шт",
                "calories": 280
            },
            {
                "food_name": "гречка",
                "amount": 350,
                "unit": "мл",
                "calories": 385
            }
        ]
        
        for i, food_info in enumerate(test_cases, 1):
            print(f"\n📝 Тест {i}: {food_info['food_name']}")
            formatted = format_food_record(food_info, i)
            print("Результат:")
            print(formatted)
            print("-" * 50)
        
        return True
        
    except Exception as e:
        print(f"❌ Ошибка при тестировании: {e}")
        return False

def main():
    """Основная функция тестирования"""
    print("🧪 Запуск тестов нового формата\n")
    
    success = test_new_format()
    
    if success:
        print("\n🎉 Тестирование завершено!")
    else:
        print("\n⚠️ Тестирование завершено с ошибками")
    
    return 0 if success else 1

if __name__ == '__main__':
    exit_code = main()
    sys.exit(exit_code)