Function main

examples/Asio/Server/test_stress.py:272–326

()

def main() -> int:
    parser = argparse.ArgumentParser(
        description="Stress test FastLED Network HTTP server"
    )
    parser.add_argument("--host", default="localhost", help="Server host")
    parser.add_argument("--port", type=int, default=8080, help="Server port")
    parser.add_argument(
        "--connections", type=int, default=20, help="Number of concurrent connections"
    )
    parser.add_argument(
        "--requests", type=int, default=50, help="Number of sequential requests"
    )
    args = parser.parse_args()

    config = StressTestConfig(
        host=args.host,
        port=args.port,
        num_connections=args.connections,
        num_requests=args.requests,
    )

    console.print("[bold]FastLED Network HTTP Server - Phase 3 Stress Test[/bold]")
    console.print(f"Server: http://{config.host}:{config.port}\n")

    console.print("[cyan]Starting server...[/cyan]")
    server_proc = start_server(config)
    console.print("[green]✓ Server started and responding[/green]")

    results = StressTestResults()

    try:
        # Test 1: Concurrent connections
        stress_test_concurrent(config, results)

        # Test 2: Rapid sequential requests
        stress_test_rapid(config, results)

        # Display results
        exit_code = display_results(results)

        return exit_code

    except KeyboardInterrupt:
        console.print("\n[yellow]Test interrupted by user[/yellow]")
        _thread.interrupt_main()
        return 130

    finally:
        console.print("\n[cyan]Stopping server...[/cyan]")
        server_proc.terminate()
        try:
            server_proc.wait(timeout=5)
        except subprocess.TimeoutExpired:
            server_proc.kill()
        console.print("[green]✓ Server stopped[/green]")


if __name__ == "__main__":