"""
边缘端运行测试脚本 - 推理测试
运行 main.py 并测试 30 秒
"""

import subprocess
import sys
import os
import time

def run_test():
    print("=" * 60)
    print("边缘端运行测试 - 推理测试")
    print("=" * 60)
    print(f"测试时长: 30 秒")
    print(f"测试时间: {time.strftime('%Y-%m-%d %H:%M:%S')}")
    print("=" * 60)
    
    env = os.environ.copy()
    env['PATH'] = r"C:\Users\16337\miniconda3\envs\yolo;" + env.get('PATH', '')
    
    cmd = [
        sys.executable, "main.py"
    ]
    
    process = subprocess.Popen(
        cmd,
        stdout=subprocess.PIPE,
        stderr=subprocess.STDOUT,
        text=True,
        bufsize=1,
        cwd=os.path.dirname(os.path.abspath(__file__)),
        env=env
    )
    
    output_lines = []
    start_time = time.time()
    
    try:
        while True:
            line = process.stdout.readline()
            if not line and process.poll() is not None:
                break
            
            if line:
                output_lines.append(line.strip())
                print(line.strip())
            
            elapsed = time.time() - start_time
            if elapsed >= 30:
                print(f"\n[INFO] 测试达到 30 秒，停止进程...")
                process.terminate()
                try:
                    process.wait(timeout=5)
                except subprocess.TimeoutExpired:
                    process.kill()
                break
    
    except KeyboardInterrupt:
        print("\n[INFO] 用户中断测试")
        process.terminate()
    
    return output_lines

if __name__ == "__main__":
    run_test()
    print("\n" + "=" * 60)
    print("测试完成")
    print("=" * 60)