test_api.py

#!/usr/bin/env python3
"""
Script de test complet pour l'API Gamma
Ce script teste tous les endpoints disponibles de l'API
"""

import sys
import os
from rich.console import Console
from rich.table import Table
from rich.panel import Panel
from rich import box
from gamma_cli.client import GammaClient, GammaAPIError
from gamma_cli.config import config

console = Console()


def print_header(title: str):
    """Affiche un en-tête formaté"""
    console.print(f"\n[bold cyan]{'=' * 60}[/bold cyan]")
    console.print(f"[bold yellow]{title}[/bold yellow]")
    console.print(f"[bold cyan]{'=' * 60}[/bold cyan]\n")


def test_api_connection(client: GammaClient):
    """Test 1: Vérifier la connexion à l'API"""
    print_header("TEST 1: Vérification de la connexion à l'API")

    try:
        result = client.verify_connection()
        if result:
            console.print("✅ [green]Connexion à l'API réussie![/green]")
            return True
        else:
            console.print("❌ [red]Échec de la connexion à l'API[/red]")
            return False
    except GammaAPIError as e:
        console.print(f"❌ [red]Erreur de connexion: {e.message}[/red]")
        if e.status_code:
            console.print(f"   Code HTTP: {e.status_code}")
        return False


def test_list_themes(client: GammaClient):
    """Test 2: Lister les thèmes disponibles"""
    print_header("TEST 2: Liste des thèmes disponibles")

    try:
        themes = client.list_themes()
        console.print(f"✅ [green]Récupération de {len(themes)} thèmes réussie![/green]\n")

        if themes:
            table = Table(title="Thèmes disponibles", box=box.ROUNDED)
            table.add_column("ID", style="cyan", no_wrap=True)
            table.add_column("Nom", style="yellow")
            table.add_column("Description", style="white")

            for theme in themes[:10]:  # Afficher les 10 premiers
                table.add_row(
                    theme.id,
                    theme.name or "N/A",
                    (theme.description or "N/A")[:50] + "..." if theme.description and len(theme.description) > 50 else (theme.description or "N/A")
                )

            console.print(table)
            if len(themes) > 10:
                console.print(f"\n[dim]... et {len(themes) - 10} autres thèmes[/dim]")
        else:
            console.print("[yellow]Aucun thème trouvé[/yellow]")

        return True, themes
    except GammaAPIError as e:
        console.print(f"❌ [red]Erreur lors de la récupération des thèmes: {e.message}[/red]")
        if e.status_code:
            console.print(f"   Code HTTP: {e.status_code}")
        return False, []


def test_list_folders(client: GammaClient):
    """Test 3: Lister les dossiers disponibles"""
    print_header("TEST 3: Liste des dossiers disponibles")

    try:
        folders = client.list_folders()
        console.print(f"✅ [green]Récupération de {len(folders)} dossiers réussie![/green]\n")

        if folders:
            table = Table(title="Dossiers disponibles", box=box.ROUNDED)
            table.add_column("ID", style="cyan", no_wrap=True)
            table.add_column("Nom", style="yellow")

            for folder in folders:
                table.add_row(folder.id, folder.name or "N/A")

            console.print(table)
        else:
            console.print("[yellow]Aucun dossier trouvé (normal pour un nouveau compte)[/yellow]")

        return True, folders
    except GammaAPIError as e:
        console.print(f"❌ [red]Erreur lors de la récupération des dossiers: {e.message}[/red]")
        if e.status_code:
            console.print(f"   Code HTTP: {e.status_code}")
        return False, []


def test_generate_gamma(client: GammaClient, theme_id=None):
    """Test 4: Générer un gamma de test"""
    print_header("TEST 4: Génération d'un gamma de test")

    test_prompt = """Create a brief presentation about AI and Machine Learning with the following structure:

1. Introduction to AI
   - What is Artificial Intelligence?
   - Brief history

2. Machine Learning Basics
   - What is Machine Learning?
   - Types of ML (supervised, unsupervised, reinforcement)

3. Applications
   - Real-world examples
   - Future perspectives
"""

    console.print(f"[cyan]Génération d'un gamma avec le texte:[/cyan]")
    console.print(Panel(test_prompt, title="Input Text", border_style="blue"))

    try:
        console.print("\n[yellow]⏳ Génération en cours (cela peut prendre quelques secondes)...[/yellow]")

        result = client.generate(
            input_text=test_prompt,
            text_mode="generate",
            theme_id=theme_id
        )

        console.print(f"\n✅ [green]Gamma généré avec succès![/green]")
        console.print(f"   [cyan]ID:[/cyan] {result.id}")
        console.print(f"   [cyan]Titre:[/cyan] {result.title or 'N/A'}")
        console.print(f"   [cyan]Statut:[/cyan] {result.status or 'N/A'}")

        if result.url:
            console.print(f"   [cyan]URL:[/cyan] {result.url}")
            console.print(f"\n[green]🔗 Ouvrez ce lien dans votre navigateur pour voir le gamma:[/green]")
            console.print(f"   [bold blue]{result.url}[/bold blue]")

        return True, result
    except GammaAPIError as e:
        console.print(f"\n❌ [red]Erreur lors de la génération: {e.message}[/red]")
        if e.status_code:
            console.print(f"   Code HTTP: {e.status_code}")
        if e.details:
            console.print(f"   Détails: {e.details}")
        return False, None


def test_get_gamma(client: GammaClient, gamma_id: str):
    """Test 5: Récupérer un gamma par son ID"""
    print_header("TEST 5: Récupération d'un gamma par ID")

    try:
        console.print(f"[cyan]Récupération du gamma ID: {gamma_id}[/cyan]\n")
        result = client.get_gamma(gamma_id)

        console.print(f"✅ [green]Gamma récupéré avec succès![/green]")
        console.print(f"   [cyan]ID:[/cyan] {result.id}")
        console.print(f"   [cyan]Titre:[/cyan] {result.title or 'N/A'}")
        console.print(f"   [cyan]Statut:[/cyan] {result.status or 'N/A'}")
        if result.url:
            console.print(f"   [cyan]URL:[/cyan] {result.url}")

        return True
    except GammaAPIError as e:
        console.print(f"❌ [red]Erreur lors de la récupération: {e.message}[/red]")
        if e.status_code:
            console.print(f"   Code HTTP: {e.status_code}")
        return False


def main():
    """Fonction principale pour exécuter tous les tests"""
    console.print(Panel.fit(
        "[bold yellow]Test de l'API Gamma[/bold yellow]\n"
        "Ce script va tester tous les endpoints disponibles",
        border_style="green"
    ))

    # Vérifier si une clé API est configurée
    if not config.is_configured():
        console.print("\n❌ [red]Aucune clé API configurée![/red]")
        console.print("\n[yellow]Pour configurer votre clé API:[/yellow]")
        console.print("  1. Via la CLI: [cyan]python gamma config set-key YOUR_API_KEY[/cyan]")
        console.print("  2. Via variable d'environnement: [cyan]export GAMMA_API_KEY='your-key'[/cyan]")
        console.print("  3. Via fichier .env: créez un fichier .env avec [cyan]GAMMA_API_KEY=your-key[/cyan]")
        console.print("\n[dim]Obtenez votre clé API sur: https://gamma.app/settings/billing[/dim]")
        return 1

    # Créer le client avec mode verbose pour plus de détails
    try:
        console.print("\n[cyan]Initialisation du client API...[/cyan]")
        client = GammaClient(verbose=False)  # Mettre à True pour voir les détails des requêtes
        console.print("✅ [green]Client initialisé avec succès![/green]")
    except GammaAPIError as e:
        console.print(f"❌ [red]Erreur d'initialisation: {e.message}[/red]")
        return 1

    # Compteur de tests réussis
    tests_passed = 0
    tests_total = 0

    # Test 1: Connexion
    tests_total += 1
    if test_api_connection(client):
        tests_passed += 1

    # Test 2: Liste des thèmes
    tests_total += 1
    success, themes = test_list_themes(client)
    if success:
        tests_passed += 1

    # Test 3: Liste des dossiers
    tests_total += 1
    success, folders = test_list_folders(client)
    if success:
        tests_passed += 1

    # Test 4: Génération d'un gamma (avec le premier thème si disponible)
    tests_total += 1
    theme_id = themes[0].id if themes else None
    success, generated_gamma = test_generate_gamma(client, theme_id)
    if success:
        tests_passed += 1

    # Test 5: Récupération du gamma généré
    if generated_gamma:
        tests_total += 1
        if test_get_gamma(client, generated_gamma.id):
            tests_passed += 1

    # Résumé final
    print_header("RÉSUMÉ DES TESTS")

    percentage = (tests_passed / tests_total * 100) if tests_total > 0 else 0

    if tests_passed == tests_total:
        style = "green"
        emoji = "🎉"
        message = "Tous les tests ont réussi!"
    elif tests_passed > 0:
        style = "yellow"
        emoji = "⚠️"
        message = "Certains tests ont échoué"
    else:
        style = "red"
        emoji = "❌"
        message = "Tous les tests ont échoué"

    console.print(f"\n{emoji} [{style}]{message}[/{style}]")
    console.print(f"[bold]Tests réussis: {tests_passed}/{tests_total} ({percentage:.1f}%)[/bold]")

    if generated_gamma and generated_gamma.url:
        console.print(f"\n[green]🔗 Gamma généré disponible ici:[/green]")
        console.print(f"   [bold blue]{generated_gamma.url}[/bold blue]")

    return 0 if tests_passed == tests_total else 1


if __name__ == "__main__":
    sys.exit(main())