#!/usr/bin/env python3
"""
Script para testar a API de mensagem de boas-vindas
"""
import requests

def test_welcome_api():
    base_url = "http://localhost:8888"
    
    agent_ids = [1, 2, 3, 25]
    
    for agent_id in agent_ids:
        url = f"{base_url}/agente/api/agent/{agent_id}/welcome"
        print(f"Testando: {url}")
        
        try:
            response = requests.get(url)
            print(f"Status: {response.status_code}")
            
            if response.status_code == 200:
                data = response.json()
                print(f"Agente: {data.get('agent_name')}")
                print(f"Mensagem: {data.get('welcome_message')}")
            elif response.status_code == 401:
                print("Erro: Não autorizado (precisa estar logado)")
            elif response.status_code == 404:
                print("Erro: Agente não encontrado")
            else:
                print(f"Erro: {response.text}")
                
        except Exception as e:
            print(f"Erro de conexão: {e}")
        
        print("-" * 50)

if __name__ == "__main__":
    test_welcome_api()