babel/packages/babel-helper-annotate-as-pure/test/index.js

import annotateAsPure from "../lib/index.js";

describe("@babel/helper-annotate-as-pure", () => {
  it("will add leading comment", () => {
    const node = {};
    annotateAsPure(node);

    expect(node.leadingComments).toEqual([
      {
        type: "CommentBlock",
        value: "#__PURE__",
      },
    ]);
  });

  it("will not add an extra leading comment", () => {
    const node = {
      leadingComments: [
        {
          type: "CommentBlock",
          value: "#__PURE__",
        },
      ],
    };

    annotateAsPure(node);

    expect(node.leadingComments).toEqual([
      {
        type: "CommentBlock",
        value: "#__PURE__",
      },
    ]);
  });
});