test: fix flaky test

Author: rpenido

src/course-unit/CourseUnit.test.jsx

@@ -581,17 +581,16 @@ describe('<CourseUnit />', () => {
       published_preview_link: publishedPreviewLink,
     } = courseSectionVerticalMock;
 
-    await waitFor(async () => {
-      const viewLiveButton = screen.getByRole('button', { name: headerNavigationsMessages.viewLiveButton.defaultMessage });
-      await user.click(viewLiveButton);
-      expect(window.open).toHaveBeenCalled();
-      expect(window.open).toHaveBeenCalledWith(publishedPreviewLink, '_blank');
+    const viewLiveButton = await screen.findByRole('button', { name: headerNavigationsMessages.viewLiveButton.defaultMessage });
 
-      const previewButton = screen.getByRole('button', { name: headerNavigationsMessages.previewButton.defaultMessage });
-      await user.click(previewButton);
-      expect(window.open).toHaveBeenCalled();
-      expect(window.open).toHaveBeenCalledWith(draftPreviewLink, '_blank');
-    });
+    await user.click(viewLiveButton);
+    expect(window.open).toHaveBeenCalled();
+    expect(window.open).toHaveBeenCalledWith(publishedPreviewLink, '_blank');
+
+    const previewButton = screen.getByRole('button', { name: headerNavigationsMessages.previewButton.defaultMessage });
+    await user.click(previewButton);
+    expect(window.open).toHaveBeenCalled();
+    expect(window.open).toHaveBeenCalledWith(draftPreviewLink, '_blank');
 
     window.open = open;
   });