test: update XBlock publish state reset test

Author: bra-i-am

src/course-unit/CourseUnit.test.jsx

@@ -2361,9 +2361,6 @@ describe('<CourseUnit />', () => {
   });
 
   it('resets XBlock publish state after saving XBlock changes', async () => {
-    store = initializeStore();
-    mockedUsedNavigate.mockClear();
-
     axiosMock
       .onGet(getCourseSectionVerticalApiUrl(blockId))
       .reply(200, {
@@ -2377,13 +2374,15 @@ describe('<CourseUnit />', () => {
 
     render(<RootWrapper />);
 
-    const xblocksIframe = await screen.findByTitle(xblockContainerIframeMessages.xblockIframeTitle.defaultMessage);
-    expect(xblocksIframe).toBeInTheDocument();
+    waitFor(() => {
+      const xblocksIframe = screen.findByTitle(xblockContainerIframeMessages.xblockIframeTitle.defaultMessage);
+      expect(xblocksIframe).toBeInTheDocument();
+    });
 
     const courseUnitSidebar = await screen.findByTestId('course-unit-sidebar');
 
     await waitFor(() => {
-    // Verify the publish button is present (indicates unpublished changes)
+      // Verify the publish button is present (indicates unpublished changes)
       expect(within(courseUnitSidebar)
         .getByRole('button', { name: sidebarMessages.actionButtonPublishTitle.defaultMessage })).toBeInTheDocument();