bun.sh/test/bun.js/socket/socket.test.ts

import { expect, it } from "bun:test";
import { bunExe } from "../bunExe";
import { connect, spawn } from "bun";

it("should keep process alive only when active", async () => {
  const { exited, stdout, stderr } = spawn({
    cmd: [ bunExe(), "echo.js" ],
    cwd: import.meta.dir,
    stdout: "pipe",
    stdin: null,
    stderr: "pipe",
    env: {
      BUN_DEBUG_QUIET_LOGS: 1,
    },
  });
  expect(await exited).toBe(0);
  expect(await new Response(stderr).text()).toBe("");
  var lines = (await new Response(stdout).text()).split(/\r?\n/);
  expect(lines.filter(function(line) {
    return line.startsWith("[Server]");
  })).toEqual([
    "[Server] OPENED",
    "[Server] GOT request",
    "[Server] CLOSED",
  ]);
  expect(lines.filter(function(line) {
    return line.startsWith("[Client]");
  })).toEqual([
    "[Client] OPENED",
    "[Client] GOT response",
    "[Client] ENDED",
    "[Client] CLOSED",
  ]);
});

it("should handle connection error", done => {
  var data = {};
  connect({
    data,
    hostname: "localhost",
    port: 55555,
    socket: {
      connectError(socket, error) {
        expect(socket).toBeDefined();
        expect(socket.data).toBe(data);
        expect(error).toBeDefined();
        expect(error.name).toBe("SystemError");
        expect(error.message).toBe("Failed to connect");
        done();
      },
      data() {
        done(new Error("Unexpected data()"));
      },
      drain() {
        done(new Error("Unexpected drain()"));
      },
      close() {
        done(new Error("Unexpected close()"));
      },
      end() {
        done(new Error("Unexpected end()"));
      },
      error() {
        done(new Error("Unexpected error()"));
      },
      open() {
        done(new Error("Unexpected open()"));
      },
    },
  });
});