actionhero/__tests__/integration/sendBuffer.ts

The reusable, scalable, and quick node.js API server for stateless and stateful applications

import * as request from "request-promise-native";
import * as stream from "stream";
import { api, Process, config } from "./../../src/index";

const actionhero = new Process();
let url;

describe("Server: sendBuffer", () => {
  beforeAll(async () => {
    process.env.AUTOMATIC_ROUTES = "get";
    await actionhero.start();
    url = "http://localhost:" + config.servers.web.port;
  });

  afterAll(async () => {
    await actionhero.stop();
  });

  beforeAll(() => {
    api.actions.versions.sendBufferTest = [1];
    api.actions.actions.sendBufferTest = {
      // @ts-ignore
      1: {
        name: "sendBufferTest",
        description: "sendBufferTest",
        version: 1,
        run: async (data) => {
          const buffer = "Example of data buffer";
          const bufferStream = new stream.PassThrough();
          data.connection.rawConnection.responseHeaders.push([
            "Content-Disposition",
            "attachment; filename=test.csv",
          ]);
          api.servers.servers.web.sendFile(
            data.connection,
            null,
            bufferStream,
            "text/csv",
            buffer.length,
            new Date()
          );
          data.toRender = false;
          bufferStream.end(buffer);
        },
      },
    };

    api.actions.versions.sendUnknownLengthBufferTest = [1];
    api.actions.actions.sendUnknownLengthBufferTest = {
      // @ts-ignore
      1: {
        name: "sendUnknownLengthBufferTest",
        description: "sendUnknownLengthBufferTest",
        version: 1,
        run: async (data) => {
          const bufferStream = new stream.PassThrough();
          api.servers.servers.web.sendFile(
            data.connection,
            null,
            bufferStream,
            "text/plain",
            null,
            new Date()
          );
          const buffer = "Example of unknown length data buffer";
          data.toRender = false;
          bufferStream.end(buffer);
        },
      },
    };

    api.routes.loadRoutes();
  });

  afterAll(() => {
    delete api.actions.actions.sendBufferTest;
    delete api.actions.versions.sendBufferTest;
    delete api.actions.versions.sendUnknownLengthBufferTest;
    delete api.actions.versions.sendUnknownLengthBufferTest;
    api.routes.loadRoutes();
  });

  test("Server should sendBuffer", async () => {
    const body = await request.get(url + "/api/sendBufferTest");
    expect(body).toEqual("Example of data buffer");
  });

  test("Server should send a stream with no specified length", async () => {
    const { body, headers } = await request.get({
      uri: url + "/api/sendUnknownLengthBufferTest",
      resolveWithFullResponse: true,
    });

    expect(headers).not.toHaveProperty("content-length");
    expect(body).toEqual("Example of unknown length data buffer");
  });
});