Skip to content

Commit 739f6d1

Browse files
test: fix flaky test
1 parent 91f2d78 commit 739f6d1

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

test/util.ts

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,9 @@ function createServerAndClient(pool: Pool) {
3333
}),
3434
});
3535

36+
// calling init() sends an INITIAL_HEARTBEAT message to the other nodes, so each node knows about the others
37+
io.of("/").adapter.init();
38+
3639
return new Promise<{
3740
io: Server;
3841
socket: ServerSocket;
@@ -55,6 +58,7 @@ function createServerAndClient(pool: Pool) {
5558

5659
function isInitComplete(servers: Server[]) {
5760
return servers.every((server) => {
61+
// @ts-expect-error nodesMap is private
5862
return server.of("/").adapter.nodesMap.size === servers.length - 1;
5963
});
6064
}
@@ -86,8 +90,6 @@ export async function setup() {
8690
const serverSockets = results.map(({ socket }) => socket);
8791
const clientSockets = results.map(({ clientSocket }) => clientSocket);
8892

89-
servers.forEach((server) => server.of("/").adapter.init());
90-
9193
while (!isInitComplete(servers)) {
9294
await sleep(20);
9395
}

0 commit comments

Comments
 (0)