regeneration-handler.test.ts

import { RegenerationEvent } from "../../src";
import { createCloudFrontEvent } from "../test-utils";

// eslint-disable-next-line @typescript-eslint/no-var-requires
jest.mock("node-fetch", () => require("fetch-mock-jest").sandbox());

jest.mock(
  "../../src/prerender-manifest.json",
  () => require("./prerender-manifest.json"),
  {
    virtual: true
  }
);

jest.mock(
  "../../src/images-manifest.json",
  () => require("./images-manifest.json"),
  {
    virtual: true
  }
);

const sqsHandlerEvent = (body: RegenerationEvent) => {
  return {
    Records: [
      {
        body: JSON.stringify(body)
      }
    ]
  };
};

const mockPageRequire = (mockPagePath: string): void => {
  jest.mock(
    `../../src/${mockPagePath}`,
    () => require(`../shared-fixtures/built-artifact/${mockPagePath}`),
    {
      virtual: true
    }
  );
};

describe("Regeneration Handler", () => {
  let consoleWarnSpy: jest.SpyInstance;

  beforeEach(() => {
    consoleWarnSpy = jest.spyOn(console, "error").mockReturnValue();

    jest.mock(
      "../../src/manifest.json",
      () => require("./default-build-manifest.json"),
      {
        virtual: true
      }
    );

    jest.mock(`../../src/s3/s3StorePage`);
  });

  afterEach(() => {
    consoleWarnSpy.mockRestore();
  });

  it.each`
    basePath
    ${"/custom"}
    ${""}
  `(
    "should generate correct page when basePath = $basePath",
    async ({ basePath }) => {
      mockPageRequire("pages/preview.js");

      const regenerationHandler =
        require("../../src/regeneration-handler").handler; // eslint-disable-line @typescript-eslint/no-var-requires

      // eslint-disable-next-line @typescript-eslint/no-var-requires
      const s3StorePage = require("../../src/s3/s3StorePage").s3StorePage;

      const event = createCloudFrontEvent({
        uri: `/preview`,
        host: "mydistribution.cloudfront.net",
        config: {
          eventType: "origin-request"
        } as AWSLambda.CloudFrontEvent["config"],
        querystring: undefined,
        requestHeaders: {}
      });

      await regenerationHandler(
        sqsHandlerEvent({
          basePath,
          bucketName: "my-bucket",
          cloudFrontEventRequest: event.Records[0].cf.request,
          region: "us-west-2",
          pageS3Path: `static-pages/build-id/preview.js`,
          pagePath: "pages/preview.js"
        })
      );

      expect(s3StorePage).toBeCalledTimes(1);
      expect(s3StorePage).toBeCalledWith(
        expect.objectContaining({
          basePath,
          uri: "/preview",
          pageData: { page: "pages/preview.js" }
        })
      );
    }
  );
});