#!/bin/bash

# WhaleTown API 测试工具
echo "🐋 WhaleTown API 测试工具"
echo "========================"

# 检查Python是否安装
if ! command -v python3 &> /dev/null; then
    echo "❌ 错误: 未找到 python3，请先安装Python"
    exit 1
fi

# 检查是否在正确的目录
if [ ! -f "quick_test.py" ]; then
    echo "❌ 错误: 请在 tests/api 目录下运行此脚本"
    exit 1
fi

echo ""
echo "请选择要运行的测试："
echo "1. 快速测试 (推荐)"
echo "2. 完整测试套件"
echo "3. 简单连接测试"
echo "4. 安装依赖"
echo "5. 退出"
echo ""

read -p "请输入选择 (1-5): " choice

case $choice in
    1)
        echo ""
        echo "🚀 运行快速测试..."
        python3 quick_test.py
        ;;
    2)
        echo ""
        echo "🚀 运行完整测试套件..."
        python3 api_client_test.py
        ;;
    3)
        echo ""
        echo "🚀 运行简单连接测试..."
        python3 simple_api_test.py
        ;;
    4)
        echo ""
        echo "📦 安装Python依赖..."
        if command -v pip3 &> /dev/null; then
            pip3 install -r requirements.txt
        elif command -v pip &> /dev/null; then
            pip install -r requirements.txt
        else
            echo "❌ 错误: 未找到 pip，请手动安装依赖"
            exit 1
        fi
        echo "✅ 依赖安装完成！"
        ;;
    5)
        echo "👋 再见！"
        exit 0
        ;;
    *)
        echo "❌ 无效选择，请重新运行脚本"
        exit 1
        ;;
esac

echo ""
echo "测试完成！按任意键继续..."
read -n 1