✅ Add a test to guarantee buffers order (#157)

frinyvonnick · Feb 8, 2022 · 1a9bf4f · 1a9bf4f
1 parent ab7cce1
commit 1a9bf4f
Show file tree

Hide file tree

Showing 2 changed files with 55 additions and 0 deletions.
diff --git a/src/main.spec.ts → src/main.integration.spec.ts b/src/main.spec.ts → src/main.integration.spec.ts
diff --git a/src/main.unit.spec.ts b/src/main.unit.spec.ts
@@ -0,0 +1,55 @@
+import { nodeHtmlToImage } from "./main";
+import { Cluster } from "puppeteer-cluster";
+
+import { Screenshot } from "./models/Screenshot";
+
+const sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
+
+describe("node-html-to-image | Unit", () => {
+  let mockExit;
+  const buffer1 = Buffer.alloc(1);
+  const buffer2 = Buffer.alloc(1);
+  const html = "<html><body>{{message}}</body></html>";
+
+  beforeEach(() => {
+    jest.spyOn(Cluster, "launch").mockImplementation(
+      // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+      // @ts-ignore
+      jest.fn(() => ({
+        execute: jest
+          .fn()
+          .mockImplementationOnce(async () => {
+            const screenshot = new Screenshot({ html });
+            screenshot.setBuffer(buffer1);
+            await sleep(10);
+            return screenshot;
+          })
+          .mockImplementationOnce(() => {
+            const screenshot = new Screenshot({ html });
+            screenshot.setBuffer(buffer2);
+            return screenshot;
+          }),
+        idle: jest.fn(),
+        close: jest.fn(),
+      }))
+    );
+    mockExit = jest.spyOn(process, "exit").mockImplementation((number) => {
+      throw new Error("process.exit: " + number);
+    });
+  });
+
+  afterEach(() => {
+    mockExit.mockRestore();
+  });
+
+  it("should sort buffer in the right order", async () => {
+    const result = await nodeHtmlToImage({
+      html,
+      content: [{ message: "Hello world!" }, { message: "Bonjour monde!" }],
+    });
+
+    expect(result).toEqual([buffer1, buffer2]);
+  });
+});
+
+jest.mock("puppeteer-cluster");