From 24ec642d495198749f9c3da37ecf42b42d869ff6 Mon Sep 17 00:00:00 2001
From: Madeline <46743919+MaddyUnderStars@users.noreply.github.com>
Date: Mon, 7 Aug 2023 11:59:48 +1000
Subject: * add TOKEN and ITER env vars to identify test script * kill child
 process server when tests are done, not just parent

---
 scripts/stress/identify.js | 5 ++---
 scripts/test.js            | 1 +
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/scripts/stress/identify.js b/scripts/stress/identify.js
index 7dea5e4d..9c74c634 100644
--- a/scripts/stress/identify.js
+++ b/scripts/stress/identify.js
@@ -4,9 +4,8 @@ require("dotenv").config();
 const { OPCODES } = require("../../dist/gateway/util/Constants.js");
 const WebSocket = require("ws");
 const ENDPOINT = `ws://localhost:3002?v=9&encoding=json`;
-const TOKEN =
-	"eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpZCI6IjEwOTMxMTgwMjgzNjA1MzYxMDYiLCJpYXQiOjE2ODA2OTE5MDB9.9ByCqDvC4mIutW8nM7WhVCtGuKW08UimPnmBeNw-K0E";
-const TOTAL_ITERATIONS = 500;
+const TOKEN = process.env.TOKEN;
+const TOTAL_ITERATIONS = process.env.ITER ? parseInt(process.env.ITER) : 500;
 
 const doTimedIdentify = () =>
 	new Promise((resolve) => {
diff --git a/scripts/test.js b/scripts/test.js
index 28ac3778..69e9fdd6 100644
--- a/scripts/test.js
+++ b/scripts/test.js
@@ -34,6 +34,7 @@ server.stdout.on("data", (data) => {
 	if (data.toString().toLowerCase().includes("listening")) {
 		// we good :)
 		console.log("we good");
+		server.kill();
 		process.exit();
 	}
 });
-- 
cgit 1.4.1