@@ -10,7 +10,7 @@ import { getDefaultConfig } from "../../configManager/config.default";
1010import Config from "../../configManager/configManager" ;
1111import { PublicLogger } from "../../logger/logger" ;
1212import { ILivePreviewWindowType } from "../../types/types" ;
13- import * as utils from " ../../utils" ;
13+ import { addLivePreviewQueryTags } from ' ../../utils/addLivePreviewQueryTags' ;
1414import livePreviewPostMessage from "../eventManager/livePreviewEventManager" ;
1515import { LIVE_PREVIEW_POST_MESSAGE_EVENTS } from "../eventManager/livePreviewEventManager.constant" ;
1616import {
@@ -20,6 +20,9 @@ import {
2020import LivePreview from "../live-preview" ;
2121import { mockLivePreviewInitEventListener } from "./mock" ;
2222
23+ vi . mock ( "../../utils/addLivePreviewQueryTags" , ( ) => ( {
24+ addLivePreviewQueryTags : vi . fn ( ) ,
25+ } ) ) ;
2326vi . mock ( "../../visualBuilder/utils/visualBuilderPostMessage" , async ( ) => {
2427 const { getAllContentTypes } = await vi . importActual <
2528 typeof import ( "../../__test__/data/contentType" )
@@ -429,15 +432,15 @@ describe("incoming postMessage", () => {
429432
430433describe ( "testing window event listeners" , ( ) => {
431434 let addEventListenerMock : any ;
432- let sendInitEvent : any ;
435+ let sendInitEvent = vi . fn ( ) . mockImplementation ( mockLivePreviewInitEventListener ) ;
433436 let livePreviewInstance : LivePreview ;
434437
435438 beforeEach ( ( ) => {
436439 Config . reset ( ) ;
437440 livePreviewPostMessage ?. destroy ( { soft : true } ) ;
438441 livePreviewPostMessage ?. on (
439442 LIVE_PREVIEW_POST_MESSAGE_EVENTS . INIT ,
440- mockLivePreviewInitEventListener
443+ sendInitEvent
441444 ) ;
442445
443446 const titlePara = document . createElement ( "h3" ) ;
@@ -457,13 +460,10 @@ describe("testing window event listeners", () => {
457460 document . body . appendChild ( descPara ) ;
458461 document . body . appendChild ( linkPara ) ;
459462
460- Config . set ( "windowType" , ILivePreviewWindowType . PREVIEW ) ;
461-
462463 addEventListenerMock = vi . spyOn ( window , "addEventListener" ) ;
463464 } ) ;
464465
465466 afterEach ( ( ) => {
466- vi . restoreAllMocks ( ) ;
467467 document . getElementsByTagName ( "html" ) [ 0 ] . innerHTML = "" ;
468468 } ) ;
469469
@@ -473,11 +473,7 @@ describe("testing window event listeners", () => {
473473 } ) ;
474474
475475 test ( "should attach a load event to call requestDataSync if document is not yet loaded" , ( ) => {
476- Object . defineProperty ( document , "readyState" , {
477- get ( ) {
478- return "loading" ;
479- } ,
480- } ) ;
476+ const readyState = vi . spyOn ( document , 'readyState' , 'get' ) . mockReturnValue ( 'loading' ) ;
481477
482478 Config . replace ( {
483479 enable : true ,
@@ -489,28 +485,32 @@ describe("testing window event listeners", () => {
489485 "load" ,
490486 expect . any ( Function )
491487 ) ;
488+ readyState . mockRestore ( ) ;
492489 } ) ;
493- //TODO: fix this test
494- test . skip ( "should handle link click event if ssr is set to true" , async ( ) => {
495- vi . spyOn ( utils , "addLivePreviewQueryTags" ) ;
490+ test ( "should handle link click event if ssr is set to true" , async ( ) => {
491+
496492 Config . replace ( {
497493 enable : true ,
498494 ssr : true ,
495+ debug : true ,
499496 } ) ;
500497
501498 const targetElement = document . createElement ( "a" ) ;
502499 targetElement . href = "http://localhost:3000/" ;
503500
504501 document . body . appendChild ( targetElement ) ;
505502 await act ( async ( ) => {
506- livePreviewInstance = new LivePreview ( ) ;
503+ livePreviewInstance = new LivePreview ( ) ;
507504 } ) ;
505+ await waitFor ( ( ) => {
506+ expect ( sendInitEvent ) . toBeCalled ( ) ;
507+ } )
508508 await waitFor ( ( ) => {
509509 expect ( Config . get ( ) . stackDetails . contentTypeUid ) . toBe ( 'contentTypeUid' ) ;
510510 } )
511511 await act ( async ( ) => {
512512 fireEvent . click ( targetElement ) ;
513513 } ) ;
514- expect ( utils . addLivePreviewQueryTags ) . toBeCalled ( ) ;
514+ expect ( addLivePreviewQueryTags ) . toBeCalled ( ) ;
515515 } ) ;
516516} ) ;
0 commit comments