103 lines
3.5 KiB
TypeScript
Executable File
103 lines
3.5 KiB
TypeScript
Executable File
#!/usr/bin/env bun
|
|
/**
|
|
* Gateway Load Test Runner
|
|
*
|
|
* Entry point for running gateway load tests.
|
|
*
|
|
* Usage:
|
|
* bun load-tests/run.ts --scenario connections --concurrency 100 --duration 30
|
|
* bun load-tests/run.ts --scenario chat --concurrency 20 --rps 5 --duration 60
|
|
* bun load-tests/run.ts --scenario auth-stress --concurrency 10 --duration 30
|
|
*/
|
|
|
|
import { DEFAULT_CONFIG, formatMetricsConsole, parseArgs } from "./config.js";
|
|
import type { LoadTestConfig, LoadTestMetrics } from "./config.js";
|
|
import { runConnectionsTest } from "./gateway-ws.js";
|
|
import { runChatTest } from "./gateway-chat.js";
|
|
import { runAuthStressTest } from "./gateway-auth.js";
|
|
|
|
async function main(): Promise<void> {
|
|
const args = process.argv.slice(2);
|
|
const parsedArgs = parseArgs(args);
|
|
|
|
const config: LoadTestConfig = {
|
|
...DEFAULT_CONFIG,
|
|
...parsedArgs,
|
|
};
|
|
|
|
// Validate config
|
|
if (!config.gatewayUrl) {
|
|
console.error("Error: Gateway URL is required (--url or -u)");
|
|
process.exit(1);
|
|
}
|
|
|
|
if (!["connections", "chat", "auth-stress"].includes(config.scenario)) {
|
|
console.error(`Error: Invalid scenario "${config.scenario}". Use: connections, chat, auth-stress`);
|
|
process.exit(1);
|
|
}
|
|
|
|
console.log("");
|
|
console.log("╔══════════════════════════════════════════════════════════════════╗");
|
|
console.log("║ Moltbot Gateway Load Test ║");
|
|
console.log("╚══════════════════════════════════════════════════════════════════╝");
|
|
console.log("");
|
|
console.log(` Scenario: ${config.scenario}`);
|
|
console.log(` Gateway: ${config.gatewayUrl}`);
|
|
console.log(` Concurrency: ${config.concurrency}`);
|
|
console.log(` Duration: ${config.durationSeconds}s`);
|
|
console.log(` Ramp-up: ${config.rampUpSeconds}s`);
|
|
if (config.scenario === "chat") {
|
|
console.log(` RPS/user: ${config.rpsPerUser}`);
|
|
}
|
|
console.log(` Auth: ${config.token ? "token provided" : "none"}`);
|
|
console.log("");
|
|
|
|
let metrics: LoadTestMetrics;
|
|
|
|
try {
|
|
switch (config.scenario) {
|
|
case "connections":
|
|
metrics = await runConnectionsTest(config);
|
|
break;
|
|
case "chat":
|
|
metrics = await runChatTest(config);
|
|
break;
|
|
case "auth-stress":
|
|
metrics = await runAuthStressTest(config);
|
|
break;
|
|
}
|
|
} catch (err) {
|
|
console.error(`\nLoad test failed: ${err instanceof Error ? err.message : String(err)}`);
|
|
process.exit(1);
|
|
}
|
|
|
|
// Output results
|
|
if (config.outputFormat === "json") {
|
|
console.log(JSON.stringify(metrics, null, 2));
|
|
} else {
|
|
console.log(formatMetricsConsole(metrics));
|
|
}
|
|
|
|
// Exit with error code if significant failures
|
|
const failureRate = metrics.requestsTotal > 0
|
|
? metrics.requestsFailed / metrics.requestsTotal
|
|
: 0;
|
|
|
|
if (failureRate > 0.1) {
|
|
console.log("⚠️ Warning: Failure rate > 10%");
|
|
process.exit(1);
|
|
}
|
|
|
|
if (metrics.connectionsFailed > 0 && metrics.connectionsSucceeded === 0) {
|
|
console.log("⚠️ Warning: All connections failed");
|
|
process.exit(1);
|
|
}
|
|
|
|
console.log("✓ Load test completed successfully");
|
|
}
|
|
|
|
main().catch((err) => {
|
|
console.error(`Fatal error: ${err instanceof Error ? err.message : String(err)}`);
|
|
process.exit(1);
|
|
});
|