test: update mock type casting in visual builder tests

Author: hitesh-shetty-cstk

src/visualBuilder/utils/__test__/getFieldData.test.ts

@@ -3,6 +3,7 @@ import { getFieldData } from "../getFieldData";
 import visualBuilderPostMessage from "../visualBuilderPostMessage";
 import { VisualBuilderPostMessageEvents } from "../types/postMessage.types";
 import { hasPostMessageError } from "../errorHandling";
+import { Mock } from "vitest";
 
 vi.mock("../../utils/visualBuilderPostMessage", async () => {
     return {
@@ -85,7 +86,7 @@ describe("getFieldData", () => {
     });
 
     test("should return an empty string if there is a post message error", async () => {
-        (hasPostMessageError as vi.Mock).mockReturnValueOnce(true);
+        (hasPostMessageError as Mock).mockReturnValueOnce(true);
 
         const fieldData = await getFieldData(mockFieldMetadata);
         expect(fieldData).toBe("");

src/visualBuilder/utils/__test__/getPsuedoEditableStylesElement.test.ts

@@ -1,4 +1,4 @@
-import { describe, it, expect, vi } from "vitest";
+import { describe, it, expect, vi, Mock } from "vitest";
 import { getPsuedoEditableElementStyles } from "../getPsuedoEditableStylesElement";
 import getCamelCaseStyles from "../getCamelCaseStyles";
 import getStyleOfAnElement from "../getStyleOfAnElement";
@@ -23,7 +23,7 @@ describe("getPsuedoEditableElementStyles", () => {
             fontSize: "16px",
         };
 
-        (getStyleOfAnElement as vi.Mock).mockReturnValue(mockStyles);
+        (getStyleOfAnElement as Mock).mockReturnValue(mockStyles);
 
         const result = getPsuedoEditableElementStyles(mockElement);
 
@@ -60,8 +60,8 @@ describe("getPsuedoEditableElementStyles", () => {
             fontSize: "16px",
         };
 
-        (getStyleOfAnElement as vi.Mock).mockReturnValue(mockStyles);
-        (getCamelCaseStyles as vi.Mock).mockReturnValue(mockCamelCaseStyles);
+        (getStyleOfAnElement as Mock).mockReturnValue(mockStyles);
+        (getCamelCaseStyles as Mock).mockReturnValue(mockCamelCaseStyles);
 
         const result = getPsuedoEditableElementStyles(mockElement, true);
 

src/visualBuilder/utils/__test__/handleIndividualFields.test.ts

@@ -1,4 +1,4 @@
-import { describe, it, expect, vi, beforeEach } from "vitest";
+import { describe, it, expect, vi, beforeEach, Mock } from "vitest";
 import { handleIndividualFields, cleanIndividualFieldResidual } from "../handleIndividualFields";
 import { VisualBuilderCslpEventDetails } from "../../types/visualBuilder.types";
 import { FieldSchemaMap } from "../fieldSchemaMap";
@@ -59,10 +59,10 @@ describe("handleIndividualFields", () => {
         const fieldType = "text";
         const isDisabled = { isDisabled: false };
 
-        (FieldSchemaMap.getFieldSchema as vi.Mock).mockResolvedValue(fieldSchema);
-        (getFieldData as vi.Mock).mockResolvedValue(expectedFieldData);
-        (getFieldType as vi.Mock).mockReturnValue(fieldType);
-        (isFieldDisabled as vi.Mock).mockReturnValue(isDisabled);
+        (FieldSchemaMap.getFieldSchema as Mock).mockResolvedValue(fieldSchema);
+        (getFieldData as Mock).mockResolvedValue(expectedFieldData);
+        (getFieldType as Mock).mockReturnValue(fieldType);
+        (isFieldDisabled as Mock).mockReturnValue(isDisabled);
 
         await act(async () => {
             await handleIndividualFields(eventDetails, elements);
@@ -81,10 +81,10 @@ describe("handleIndividualFields", () => {
         const fieldType = "blocks";
         const isDisabled = { isDisabled: false };
 
-        (FieldSchemaMap.getFieldSchema as vi.Mock).mockResolvedValue(fieldSchema);
-        (getFieldData as vi.Mock).mockResolvedValue(expectedFieldData);
-        (getFieldType as vi.Mock).mockReturnValue(fieldType);
-        (isFieldDisabled as vi.Mock).mockReturnValue(isDisabled);
+        (FieldSchemaMap.getFieldSchema as Mock).mockResolvedValue(fieldSchema);
+        (getFieldData as Mock).mockResolvedValue(expectedFieldData);
+        (getFieldType as Mock).mockReturnValue(fieldType);
+        (isFieldDisabled as Mock).mockReturnValue(isDisabled);
 
         await handleIndividualFields(eventDetails, elements);
 
@@ -98,16 +98,15 @@ describe("handleIndividualFields", () => {
         const fieldType = FieldDataType.SINGLELINE;
         const isDisabled = { isDisabled: false };
 
-        (FieldSchemaMap.getFieldSchema as vi.Mock).mockResolvedValue(fieldSchema);
-        (getFieldData as vi.Mock).mockResolvedValue(expectedFieldData);
-        (getFieldType as vi.Mock).mockReturnValue(fieldType);
-        (isFieldDisabled as vi.Mock).mockReturnValue(isDisabled);
+        (FieldSchemaMap.getFieldSchema as Mock).mockResolvedValue(fieldSchema);
+        (getFieldData as Mock).mockResolvedValue(expectedFieldData);
+        (getFieldType as Mock).mockReturnValue(fieldType);
+        (isFieldDisabled as Mock).mockReturnValue(isDisabled);
 
         await act(async () => {
             await handleIndividualFields(eventDetails, elements);
         })
 
-        screen.debug(eventDetails.editableElement);
         expect(eventDetails.editableElement.getAttribute(VISUAL_BUILDER_FIELD_TYPE_ATTRIBUTE_KEY)).toBe(fieldType);
         expect(eventDetails.editableElement.getAttribute("contenteditable")).toBe("true");
     });
@@ -159,7 +158,7 @@ describe("cleanIndividualFieldResidual", () => {
         cleanIndividualFieldResidual(elements);
 
         expect(elements.focusedToolbar?.innerHTML).toBe("");
-        expect(visualBuilderPostMessage.unregisterEvent).toHaveBeenCalledWith(VisualBuilderPostMessageEvents.DELETE_INSTANCE);
-        expect(visualBuilderPostMessage.unregisterEvent).toHaveBeenCalledWith(VisualBuilderPostMessageEvents.UPDATE_DISCUSSION_ID);
+        expect(visualBuilderPostMessage?.unregisterEvent).toHaveBeenCalledWith(VisualBuilderPostMessageEvents.DELETE_INSTANCE);
+        expect(visualBuilderPostMessage?.unregisterEvent).toHaveBeenCalledWith(VisualBuilderPostMessageEvents.UPDATE_DISCUSSION_ID);
     });
 });