@@ -10,7 +10,8 @@ import { Mock, vi } from "vitest";
1010import { VisualBuilderPostMessageEvents } from "../../../utils/types/postMessage.types" ;
1111import { VisualBuilder } from "../../../index" ;
1212import { triggerAndWaitForClickAction } from "../../../../__test__/utils" ;
13- import { act } from "preact/test-utils" ;
13+
14+ const EXAMPLE_STAGE_NAME = "Example Stage" ;
1415
1516const VALUES = {
1617 singleLine : "Single line" ,
@@ -99,23 +100,28 @@ describe("When an element is clicked in visual builder mode", () => {
99100 beforeAll ( async ( ) => {
100101 ( visualBuilderPostMessage ?. send as Mock ) . mockImplementation (
101102 ( eventName : string ) => {
102- if (
103- eventName ===
104- VisualBuilderPostMessageEvents . GET_FIELD_DATA
105- ) {
106- return Promise . resolve ( {
107- fieldData : VALUES . singleLine ,
108- } ) ;
109- } else if (
110- eventName ===
111- VisualBuilderPostMessageEvents . GET_FIELD_DISPLAY_NAMES
112- ) {
113- return Promise . resolve ( {
114- "all_fields.bltapikey.en-us.single_line" :
115- "Single Line" ,
116- } ) ;
103+ switch ( eventName ) {
104+ case VisualBuilderPostMessageEvents . GET_FIELD_DATA :
105+ return Promise . resolve ( {
106+ fieldData : VALUES . singleLine ,
107+ } ) ;
108+ case VisualBuilderPostMessageEvents . GET_FIELD_DISPLAY_NAMES :
109+ return Promise . resolve ( {
110+ "all_fields.bltapikey.en-us.single_line" :
111+ "Single Line" ,
112+ } ) ;
113+ case VisualBuilderPostMessageEvents . GET_WORKFLOW_STAGE_DETAILS :
114+ return Promise . resolve ( {
115+ stage : { name : EXAMPLE_STAGE_NAME } ,
116+ permissions : {
117+ entry : {
118+ update : true ,
119+ } ,
120+ } ,
121+ } ) ;
122+ default :
123+ return Promise . resolve ( { } ) ;
117124 }
118- return Promise . resolve ( { } ) ;
119125 }
120126 ) ;
121127
@@ -205,20 +211,32 @@ describe("When an element is clicked in visual builder mode", () => {
205211 beforeAll ( async ( ) => {
206212 ( visualBuilderPostMessage ?. send as Mock ) . mockImplementation (
207213 ( eventName : string , args ) => {
208- if (
209- eventName ===
210- VisualBuilderPostMessageEvents . GET_FIELD_DATA
211- ) {
212- const values : Record < string , any > = {
213- single_line_textbox_multiple_ : [ "Hello" , "world" ] ,
214- "single_line_textbox_multiple_.0" : "Hello" ,
215- "single_line_textbox_multiple_.1" : "world" ,
216- } ;
217- return Promise . resolve ( {
218- fieldData : values [ args . entryPath ] ,
219- } ) ;
214+ switch ( eventName ) {
215+ case VisualBuilderPostMessageEvents . GET_FIELD_DATA : {
216+ const values : Record < string , any > = {
217+ single_line_textbox_multiple_ : [
218+ "Hello" ,
219+ "world" ,
220+ ] ,
221+ "single_line_textbox_multiple_.0" : "Hello" ,
222+ "single_line_textbox_multiple_.1" : "world" ,
223+ } ;
224+ return Promise . resolve ( {
225+ fieldData : values [ args . entryPath ] ,
226+ } ) ;
227+ }
228+ case VisualBuilderPostMessageEvents . GET_WORKFLOW_STAGE_DETAILS :
229+ return Promise . resolve ( {
230+ stage : { name : EXAMPLE_STAGE_NAME } ,
231+ permissions : {
232+ entry : {
233+ update : true ,
234+ } ,
235+ } ,
236+ } ) ;
237+ default :
238+ return Promise . resolve ( { } ) ;
220239 }
221- return Promise . resolve ( { } ) ;
222240 }
223241 ) ;
224242
0 commit comments