@halsp/native/test/startup.test.ts

Halsp 的 NodeJS 原生环境

import "../src";
import request from "supertest";
import { Startup } from "@halsp/core";
import { Server } from "http";

describe("startup", () => {
  it("should listen with empty options", async () => {
    const startup = new Startup().useNative({
      port: 23331,
    });
    const server = await startup
      .use(async (ctx) => {
        ctx.res.ok({
          content: "BODY",
        });
      })
      .listen();
    const res = await request(server).get("");

    expect(res.status).toBe(200);
    expect(res.headers["content-type"]).toBe("application/json; charset=utf-8");
    expect(res.body).toEqual({
      content: "BODY",
    });
    await startup.close();
  });

  it("should listen with https", async () => {
    const server = await new Startup()
      .useNative({
        port: 0,
        https: {},
      })
      .listen();
    expect(server).not.toBeUndefined();
    expect(server.listening).toBeTruthy();
    server.close();
  });

  it("should throw error when the port is already in use", async () => {
    const startup = new Startup().useNative();
    const server = startup["nativeServer"] as Server;

    let options: any;
    server.listen = (opts, cb) => {
      options = opts;
      cb();
      return server;
    };

    await startup.listen();

    expect(options.port).toBe(9504);
  }, 10000);
});

describe("write end", () => {
  it("should not send body after stream ended", async () => {
    const server = await new Startup()
      .useNative({
        port: 0,
      })
      .useNative()
      .use(async (ctx, next) => {
        ctx.resStream.end();
        expect(!!ctx.reqStream).toBeTruthy();
        await next();
      })
      .use(async (ctx) => {
        ctx.res.ok("BODY");
      })
      .listen();
    const res = await request(server).get("");
    server.close();

    expect(res.status).toBe(404);
    expect(res.body).not.toBe("BODY");
  });

  it("should not set header after writeHead called", async () => {
    const startup = new Startup()
      .useNative({
        port: 0,
      })
      .use(async (ctx, next) => {
        ctx.resStream.writeHead(200);
        await next();
      })
      .use(async (ctx) => {
        ctx.res.setHeader("h1", "1");
      });
    await startup.listen();
    const server = startup["nativeServer"];
    const res = await request(server).get("");
    server.close();

    expect(res.status).toBe(200);
    expect(res.headers.h1).toBeUndefined();
  });
});