EvalueTonSavoir/server/__tests__/socket.test.js

const http = require("http");
const { Server } = require("socket.io");
const Client = require("socket.io-client");
const { setupWebsocket } = require("../socket/socket");

process.env.NODE_ENV = "test";

// pick a random port number for testing
const BACKEND_PORT = Math.ceil(Math.random() * 1000 + 3000);
const BACKEND_URL = "http://localhost";

const BACKEND_API = `${BACKEND_URL}:${BACKEND_PORT}`;

describe("websocket server", () => {
  let ioServer, server, teacherSocket, studentSocket;

  beforeAll((done) => {
    const httpServer = http.createServer();
    ioServer = new Server(httpServer, {
      path: "/socket.io",
      cors: {
        origin: "*",
        methods: ["GET", "POST"],
        credentials: true,
      },
    });
    setupWebsocket(ioServer);
    server = httpServer.listen(BACKEND_PORT, () => done());
  });

  afterAll(() => {
    ioServer.close();
    server.close();
    if (teacherSocket) {
      console.log("teacherSocket disconnect");
      teacherSocket.disconnect();
      if (studentSocket) {
        console.log("studentSocket disconnect");
        studentSocket.disconnect();
      }
    }
  });

  test("should connect to the server", (done) => {
    teacherSocket = new Client(BACKEND_API, {
      path: "/socket.io",
      transports: ["websocket"],
    });
    studentSocket = new Client(BACKEND_API, {
      path: "/socket.io",
      transports: ["websocket"],
    });
    studentSocket.on("connect", () => {
      expect(studentSocket.connected).toBe(true);
    });
    teacherSocket.on("connect", () => {
      expect(teacherSocket.connected).toBe(true);
      done();
    });
  });

  test("should create a room", (done) => {
    teacherSocket.on("create-success", (roomName) => {
      expect(roomName).toBe("ROOM1");
      done();
    });
    teacherSocket.emit("create-room", "room1");
  });

  test("should not create a room if it already exists", (done) => {
    teacherSocket.on("create-failure", () => {
      done();
    });
    teacherSocket.emit("create-room", "room1");
  });

  test("should join a room", (done) => {
    studentSocket.on("join-success", (roomName) => {
      expect(roomName).toBe("ROOM1");
      done();
    });
    studentSocket.emit("join-room", {
      enteredRoomName: "room1",
      username: "student1",
    });
  });

  test("should not join a room if it does not exist", (done) => {
    studentSocket.on("join-failure", () => {
      done();
    });
    studentSocket.emit("join-room", {
      enteredRoomName: "ROOM2",
      username: "student1",
    });
  });

  test("should launch student mode", (done) => {
    studentSocket.on("launch-student-mode", (questions) => {
      expect(questions).toEqual([
        { question: "question1" },
        { question: "question2" },
      ]);
      done();
    });
    teacherSocket.emit("launch-student-mode", {
      roomName: "ROOM1",
      questions: [{ question: "question1" }, { question: "question2" }],
    });
  });

  test("should launch teacher mode", (done) => {
    studentSocket.on("launch-teacher-mode", (questions) => {
      expect(questions).toEqual([
        { question: "question1" },
        { question: "question2" },
      ]);
      done();
    });
    teacherSocket.emit("launch-teacher-mode", {
      roomName: "ROOM1",
      questions: [{ question: "question1" }, { question: "question2" }],
    });
  });

  test("should send next question", (done) => {
    studentSocket.on("next-question", ( question ) => {
      expect(question).toBe("question2");
      done();
    });
    teacherSocket.emit("next-question", { roomName: "ROOM1", question: 'question2'},
    );
  });

  test("should send answer", (done) => {
    teacherSocket.on("submit-answer-room", (answer) => {
      expect(answer).toEqual({
        idUser: studentSocket.id,
        username: "student1",
        answer: "answer1",
        idQuestion: 1,
      });
      done();
    });
    studentSocket.emit("submit-answer", {
      roomName: "ROOM1",
      username: "student1",
      answer: "answer1",
      idQuestion: 1,
    });
  });

  test("should not join a room if no room name is provided", (done) => {
    studentSocket.on("join-failure", () => {
      done();
    });
    studentSocket.emit("join-room", {
      enteredRoomName: "",
      username: "student1",
    });
  });

  test("should not join a room if the username is not provided", (done) => {
    studentSocket.on("join-failure", () => {
      done();
    });
    studentSocket.emit("join-room", { enteredRoomName: "ROOM2", username: "" });
  });

  test("should end quiz", (done) => {
    studentSocket.on("end-quiz", () => {
      done();
    });
    teacherSocket.emit("end-quiz", {
      roomName: "ROOM1",
    });
  });

  test("should disconnect", (done) => {
    teacherSocket.disconnect();
    studentSocket.disconnect();
    done();
  });
});