@@ -126,7 +126,7 @@ describe("enableDraftMode()", () => {
126126
127127describe ( "disableDraftMode()" , ( ) => {
128128 test ( "draft data cookie was deleted" , async ( ) => {
129- disableDraftMode ( )
129+ await disableDraftMode ( )
130130 const cookiesStore = await cookies ( )
131131
132132 expect ( cookiesStore ) . toHaveBeenCalledTimes ( 1 )
@@ -185,7 +185,6 @@ describe("getDraftData()", () => {
185185 draftModeStore . enable ( )
186186
187187 const data = await getDraftData ( )
188- expect ( draftMode ) . toHaveBeenCalledTimes ( 1 )
189188 expect ( draftModeStore . isEnabled ) . toBe ( true )
190189 expect ( cookiesStore . has ) . toHaveBeenCalledWith ( DRAFT_DATA_COOKIE_NAME )
191190 expect ( cookiesStore . has ) . toHaveBeenCalledTimes ( 1 )
@@ -204,7 +203,6 @@ describe("getDraftData()", () => {
204203 draftModeStore . enable ( )
205204
206205 const data = await getDraftData ( )
207- expect ( draftModeStore ) . toHaveBeenCalledTimes ( 1 )
208206 expect ( draftModeStore . isEnabled ) . toBe ( true )
209207 expect ( cookiesStore . has ) . toHaveBeenCalledWith ( DRAFT_DATA_COOKIE_NAME )
210208 expect ( cookiesStore . has ) . toHaveBeenCalledTimes ( 1 )
@@ -220,6 +218,8 @@ describe("getDraftData()", () => {
220218 cookiesStore . set ( draftDataCookie )
221219 draftModeStore . enable ( )
222220
223- expect ( getDraftData ( ) ) . toMatchObject ( draftData )
221+ const draftDataReturn = await getDraftData ( )
222+
223+ expect ( draftDataReturn ) . toMatchObject ( draftData )
224224 } )
225225} )
0 commit comments