test(lib-storage): speed up assertion of 5MB buffer

Author: AllanZhengYP

lib/lib-storage/src/Upload.spec.ts

@@ -51,6 +51,7 @@ jest.mock("@aws-sdk/client-s3", () => ({
 }));
 
 import { CompleteMultipartUploadCommandOutput, S3 } from "@aws-sdk/client-s3";
+import { createHash } from "crypto";
 import { Readable } from "stream";
 
 import { Progress, Upload } from "./index";
@@ -76,6 +77,24 @@ describe(Upload.name, () => {
     Body: "this-is-a-sample-payload",
   };
 
+  expect.extend({
+    toHaveSameHashAsBuffer: (received: Uint8Array, expected: Uint8Array) => {
+      const receivedHash = createHash("sha256").update(received).digest("hex");
+      const expectHash = createHash("sha256").update(expected).digest("hex");
+      if (expectHash === receivedHash) {
+        return {
+          message: () => "received buffer has the correct hash",
+          pass: true,
+        };
+      } else {
+        return {
+          message: () => `received buffer hash is incorrect, expect ${expectHash}, got ${receivedHash}.`,
+          pass: false,
+        };
+      }
+    },
+  });
+
   it("correctly exposes the event emitter API", () => {
     const upload = new Upload({
       params,
@@ -237,33 +256,30 @@ describe(Upload.name, () => {
       params: actionParams,
       client: new S3({}),
     });
-
     await upload.done();
-
     expect(sendMock).toHaveBeenCalledTimes(4);
     // create multipartMock is called correctly.
     expect(createMultipartMock).toHaveBeenCalledTimes(1);
     expect(createMultipartMock).toHaveBeenCalledWith({
       ...actionParams,
       Body: undefined,
     });
-
     // upload parts is called correctly.
     expect(uploadPartMock).toHaveBeenCalledTimes(2);
     expect(uploadPartMock).toHaveBeenNthCalledWith(1, {
       ...actionParams,
-      Body: firstBuffer,
+      // @ts-ignore extended custom matcher
+      Body: expect.toHaveSameHashAsBuffer(firstBuffer),
       PartNumber: 1,
       UploadId: "mockuploadId",
     });
-
     expect(uploadPartMock).toHaveBeenNthCalledWith(2, {
       ...actionParams,
-      Body: secondBuffer,
+      // @ts-ignore extended custom matcher
+      Body: expect.toHaveSameHashAsBuffer(secondBuffer),
       PartNumber: 2,
       UploadId: "mockuploadId",
     });
-
     // complete multipart upload is called correctly.
     expect(completeMultipartMock).toHaveBeenCalledTimes(1);
     expect(completeMultipartMock).toHaveBeenLastCalledWith({
@@ -320,14 +336,16 @@ describe(Upload.name, () => {
     expect(uploadPartMock).toHaveBeenCalledTimes(2);
     expect(uploadPartMock).toHaveBeenNthCalledWith(1, {
       ...actionParams,
-      Body: firstBuffer,
+      // @ts-ignore extended custom matcher
+      Body: expect.toHaveSameHashAsBuffer(firstBuffer),
       PartNumber: 1,
       UploadId: "mockuploadId",
     });
 
     expect(uploadPartMock).toHaveBeenNthCalledWith(2, {
       ...actionParams,
-      Body: secondBuffer,
+      // @ts-ignore extended custom matcher
+      Body: expect.toHaveSameHashAsBuffer(secondBuffer),
       PartNumber: 2,
       UploadId: "mockuploadId",
     });
@@ -418,6 +436,8 @@ describe(Upload.name, () => {
     expect(putObjectTaggingMock).toHaveBeenCalledTimes(1);
     expect(putObjectTaggingMock).toHaveBeenCalledWith({
       ...actionParams,
+      // @ts-ignore extended custom matcher
+      Body: expect.toHaveSameHashAsBuffer(largeBuffer),
       Tagging: {
         TagSet: tags,
       },