import { check } from "k6";
import ws from "k6/ws";

export const options = {
  // Key configurations for avg load test in this section
  stages: [
    { duration: "30s", target: 200 }, // traffic ramp-up from 1 to 100 users over 30 seconds.
    { duration: "1m", target: 200 }, // keep traffic at 100 users over 1 minute.
    { duration: "30s", target: 0 }, // traffic ramp-up from 1 to 100 users over 30 seconds.
  ],
};

export default function () {
  const url = "ws://localhost:8080/subscriptions";
  const token = null; // replace with your auth token
  const operation = `
  subscription {
    EVM(network: eth, mempool: true) {
      Transactions {
        count
      }
    }
}`;
  const headers = {
    "Sec-WebSocket-Protocol": "graphql-ws",
  };

  if (token != null)
    Object.assign(headers, { Authorization: `Bearer ${token}` });

  const res = ws.connect(
    url,
    {
      headers,
    },
    (socket) => {
      socket.on("message", (msg) => {
        const message = JSON.parse(msg);
        if (message.type == "connection_ack")
          console.log("Connection Established with WebSocket");
        if (message.type == "data") console.log(`Message Received: ${message}`);
      });

      socket.on("error", (e) => {
        console.error(e);
      });

      socket.on("open", () => {
        socket.send(
          JSON.stringify({
            type: "connection_init",
            payload: {},
          })
        );
        socket.send(
          JSON.stringify({
            type: "start",
            payload: {
              query: operation,
              variables: {},
              extensions: {},
              operationName: null,
            },
          })
        );
      });
    }
  );

  check(res, { "Connected successfully": (r) => r && r.status === 101 });
}