chore: Multiple machine instances per-container

inferablehq · Dec 2, 2024 · ecd8bd2 · ecd8bd2
1 parent c82649c
commit ecd8bd2
Show file tree

Hide file tree

Showing 5 changed files with 26 additions and 6 deletions.
diff --git a/load-tests/Dockerfile.machine b/load-tests/Dockerfile.machine
@@ -8,7 +8,8 @@ COPY --link package-lock.json ./
 RUN npm ci
 
 COPY machine.ts ./
+COPY entrypoint.sh ./
 
 RUN npm install -g tsx
 
-CMD ["tsx", "machine.ts"]
+CMD ["entrypoint.sh"]
diff --git a/load-tests/entrypoint.sh b/load-tests/entrypoint.sh
@@ -0,0 +1,18 @@
+#!/bin/sh
+
+set -e
+
+PIDS=()
+
+# Start 20 instances of the machine
+for i in $(seq 1 20); do
+  tsx machine.ts &
+  echo "Started instance $i"
+  PIDS+=($!)
+done
+
+# Wait for all instances to finish
+for pid in "${PIDS[@]}"; do
+  wait $pid
+done
+
diff --git a/load-tests/machine.ts b/load-tests/machine.ts
@@ -4,6 +4,7 @@ const API_SECRET = process.env.INFERABLE_TEST_API_SECRET
 
 const client = new Inferable({
   apiSecret: API_SECRET,
+  machineId: `load-test-${Math.floor(Math.random() * 1000000)}`,
 })
 
 client.default.register({

diff --git a/load-tests/package-lock.json b/load-tests/package-lock.json
diff --git a/load-tests/package.json b/load-tests/package.json
@@ -10,7 +10,7 @@
   },
   "homepage": "https://github.com/inferablehq/inferable#readme",
   "dependencies": {
-    "inferable": "^0.30.55"
+    "inferable": "^0.30.57"
   },
   "devDependencies": {
     "@types/node": "^22.10.1",