#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
测试修复后的系统功能
"""

import requests
import time
import json

def test_system():
    """测试系统功能"""
    base_url = "http://localhost:5000"
    
    print("🧪 开始测试修复后的系统...")
    print("=" * 50)
    
    # 测试1: 检查系统状态
    print("1. 测试系统状态...")
    try:
        response = requests.get(f"{base_url}/api/status", timeout=10)
        if response.status_code == 200:
            status = response.json()
            print(f"   ✅ 系统状态: {status}")
        else:
            print(f"   ❌ 状态检查失败: {response.status_code}")
    except Exception as e:
        print(f"   ❌ 状态检查异常: {e}")
    
    # 测试2: 检查数据统计
    print("\n2. 测试数据统计...")
    try:
        response = requests.get(f"{base_url}/api/data/stats", timeout=10)
        if response.status_code == 200:
            stats = response.json()
            print(f"   ✅ 数据统计: {stats}")
        else:
            print(f"   ❌ 统计检查失败: {response.status_code}")
    except Exception as e:
        print(f"   ❌ 统计检查异常: {e}")
    
    # 测试3: 检查数据列表
    print("\n3. 测试数据列表...")
    try:
        response = requests.get(f"{base_url}/api/data/list", timeout=10)
        if response.status_code == 200:
            data_list = response.json()
            print(f"   ✅ 数据列表: {data_list}")
        else:
            print(f"   ❌ 列表检查失败: {response.status_code}")
    except Exception as e:
        print(f"   ❌ 列表检查异常: {e}")
    
    # 测试4: 测试文本搜索（如果系统已初始化）
    print("\n4. 测试文本搜索...")
    try:
        search_data = {
            "query": "测试查询",
            "top_k": 3
        }
        response = requests.post(f"{base_url}/api/search/text_to_text", 
                               json=search_data, timeout=10)
        if response.status_code == 200:
            result = response.json()
            print(f"   ✅ 文本搜索: {result}")
        else:
            print(f"   ❌ 文本搜索失败: {response.status_code}")
    except Exception as e:
        print(f"   ❌ 文本搜索异常: {e}")
    
    print("\n" + "=" * 50)
    print("🎉 测试完成！")

if __name__ == "__main__":
    # 等待系统启动
    print("⏳ 等待系统启动...")
    time.sleep(5)
    
    test_system()