The socket-server stops after 1 message is emitted about mock-socket HOT 4 CLOSED

import { readFileSync } from "fs";
import { WebSocket, Server } from "mock-socket";
import { AIS_API_URL } from "../lib/test-mock.js";

const server = new Server(AIS_API_URL);

let count = 0;
function callback(message) {
  count++;
  if (count === 5) {
    socket.close();
  }
  console.log(message, count);
}

const messageExamples = JSON.parse(readFileSync("message_examples.json"));

server.on("connection", (socket) => {
  const messageType = Object.keys(messageExamples);
  const message =
    messageExamples[messageType[(messageType.length * Math.random()) << 0]];
  socket.send(JSON.stringify(message));
});

const socket = new WebSocket(AIS_API_URL);

socket.addEventListener("error", (event) => {
  console.error(event);
});
socket.addEventListener("message", (event) => {
  const aisMessage = JSON.parse(event.data);
  callback(aisMessage);
});
socket.addEventListener("close", () => {
 console.log("Closing with ", count, " calls" );
});

server.on("connection", (socket) => {
  console.log("connected " + socket.url);
  for (const message of Object.keys(messageExamples)) {
    socket.send(JSON.stringify(messageExamples[message]));
  }
});