feat: refactor tests

Author: vladislavkeblysh

src/courseware/course/sequence/sequence-navigation/UnitButton.test.jsx

@@ -5,6 +5,14 @@ import {
 } from '../../../../setupTest';
 import UnitButton from './UnitButton';
 
+jest.mock('react-router-dom', () => {
+  const actual = jest.requireActual('react-router-dom');
+  return {
+    ...actual,
+    useLocation: () => ({ pathname: '/preview/anything' }),
+  };
+});
+
 describe('Unit Button', () => {
   let mockData;
   const courseMetadata = Factory.build('courseMetadata');
@@ -31,6 +39,10 @@ describe('Unit Button', () => {
     };
   });
 
+  afterEach(() => {
+    jest.resetModules();
+  });
+
   it('hides title by default', () => {
     render(<UnitButton {...mockData} />, { wrapWithRouter: true });
     expect(screen.getByRole('tab')).not.toHaveTextContent(unit.display_name);
@@ -99,4 +111,19 @@ describe('Unit Button', () => {
 
     expect(screen.getByRole('tab')).toBeInTheDocument();
   });
+
+  it('prepends /preview to the unit path if in preview mode', () => {
+    const onClick = jest.fn();
+
+    render(
+      <UnitButton {...mockData} onClick={onClick} />,
+      {
+        wrapWithRouter: true,
+        initialEntries: ['/preview/some/path'],
+      },
+    );
+
+    const button = screen.getByRole('tab');
+    expect(button.closest('a')).toHaveAttribute('href', expect.stringContaining('/preview/course/'));
+  });
 });