quick_test.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
WAF API 快速测试脚本
"""

import subprocess
import time
import requests
import sys
from config import API_CONFIG

def start_server():
    """启动服务器"""
    print("启动WAF API服务器...")
    try:
        # 启动服务器进程
        process = subprocess.Popen([sys.executable, "start_server.py"], 
                                 stdout=subprocess.PIPE, 
                                 stderr=subprocess.PIPE)
        
        # 等待服务器启动
        print("等待服务器启动...")
        time.sleep(3)
        
        return process
    except Exception as e:
        print(f"启动服务器失败: {e}")
        return None

def test_server():
    """测试服务器"""
    base_url = f"http://localhost:{API_CONFIG['port']}"
    
    try:
        # 测试基本连接
        response = requests.get(f"{base_url}/", timeout=5)
        if response.status_code == 200:
            print("✓ 服务器启动成功")
            return True
        else:
            print(f"✗ 服务器响应异常: {response.status_code}")
            return False
    except Exception as e:
        print(f"✗ 无法连接到服务器: {e}")
        return False

def main():
    """主函数"""
    print("=" * 50)
    print("WAF API 快速测试")
    print("=" * 50)
    
    # 启动服务器
    process = start_server()
    if not process:
        return False
    
    try:
        # 测试服务器
        if test_server():
            print("✓ 系统测试通过")
            print("\n系统已准备就绪!")
            print(f"管理界面: http://localhost:{API_CONFIG['port']}/admin")
            print("默认管理员: admin / waf123456")
            print("\n按 Ctrl+C 停止服务器")
            
            # 保持服务器运行
            try:
                process.wait()
            except KeyboardInterrupt:
                print("\n正在停止服务器...")
                process.terminate()
                process.wait()
                print("服务器已停止")
        else:
            print("✗ 系统测试失败")
            process.terminate()
            return False
            
    except KeyboardInterrupt:
        print("\n正在停止服务器...")
        process.terminate()
        process.wait()
        print("服务器已停止")
    except Exception as e:
        print(f"测试过程中出错: {e}")
        process.terminate()
        return False
    
    return True

if __name__ == "__main__":
    main()