github/socket.io
1
1
Fork 0
mirror of https://github.com/socketio/socket.io.git synced 2026-01-08 22:48:20 -05:00
Code Issues Packages Projects Releases Wiki Activity

test(cluster-engine): fix flaky test cleanup

Browse Source
This commit is contained in:
Damien Arrachequesne
2024-07-26 09:26:47 +02:00
parent b79d80aa59
commit 582655f679
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
packages/socket.io-cluster-engine/test/cluster.ts
View File

@@ -3,6 +3,8 @@ import expect = require("expect.js");
import { handshake, url } from "./util";
import { setupPrimary } from "../lib";
const WORKER_COUNT = 3;
cluster.setupPrimary({
exec: "./test/worker.js",
// @ts-expect-error
@@ -13,7 +15,7 @@ setupPrimary();
describe("cluster", () => {
beforeEach((done) => {
for (let i = 0; i < 3; i++) {
for (let i = 0; i < WORKER_COUNT; i++) {
const worker = cluster.fork();
if (i === 2) {
@@ -23,16 +25,18 @@ describe("cluster", () => {
});
afterEach((done) => {
for (const worker of Object.values(cluster.workers)) {
worker.kill();
}
let i = 0;
function onExit() {
if (Object.keys(cluster.workers).length === 0) {
if (++i === WORKER_COUNT) {
cluster.off("exit", onExit);
done();
}
}
cluster.on("exit", onExit);
for (const worker of Object.values(cluster.workers)) {
worker.kill();
}
});
it("should ping/pong", (done) => {