@@ -85,7 +85,10 @@ describe('<IframePreviewLibraryXBlockChanges />', () => {
8585 const acceptBtn = await screen . findByRole ( 'button' , { name : 'Accept changes' } ) ;
8686 userEvent . click ( acceptBtn ) ;
8787 await waitFor ( ( ) => {
88- expect ( mockSendMessageToIframe ) . toHaveBeenCalledWith ( messageTypes . refreshXBlock , null ) ;
88+ expect ( mockSendMessageToIframe ) . toHaveBeenCalledWith (
89+ messageTypes . completeXBlockEditing ,
90+ { locator : usageKey } ,
91+ ) ;
8992 expect ( axiosMock . history . post . length ) . toEqual ( 1 ) ;
9093 expect ( axiosMock . history . post [ 0 ] . url ) . toEqual ( libraryBlockChangesUrl ( usageKey ) ) ;
9194 } ) ;
@@ -100,7 +103,6 @@ describe('<IframePreviewLibraryXBlockChanges />', () => {
100103 const acceptBtn = await screen . findByRole ( 'button' , { name : 'Accept changes' } ) ;
101104 userEvent . click ( acceptBtn ) ;
102105 await waitFor ( ( ) => {
103- expect ( mockSendMessageToIframe ) . not . toHaveBeenCalledWith ( messageTypes . refreshXBlock , null ) ;
104106 expect ( axiosMock . history . post . length ) . toEqual ( 1 ) ;
105107 expect ( axiosMock . history . post [ 0 ] . url ) . toEqual ( libraryBlockChangesUrl ( usageKey ) ) ;
106108 } ) ;
@@ -118,7 +120,10 @@ describe('<IframePreviewLibraryXBlockChanges />', () => {
118120 const ignoreConfirmBtn = await screen . findByRole ( 'button' , { name : 'Ignore' } ) ;
119121 userEvent . click ( ignoreConfirmBtn ) ;
120122 await waitFor ( ( ) => {
121- expect ( mockSendMessageToIframe ) . toHaveBeenCalledWith ( messageTypes . refreshXBlock , null ) ;
123+ expect ( mockSendMessageToIframe ) . toHaveBeenCalledWith (
124+ messageTypes . completeXBlockEditing ,
125+ { locator : usageKey } ,
126+ ) ;
122127 expect ( axiosMock . history . delete . length ) . toEqual ( 1 ) ;
123128 expect ( axiosMock . history . delete [ 0 ] . url ) . toEqual ( libraryBlockChangesUrl ( usageKey ) ) ;
124129 } ) ;
0 commit comments