feat: refactor tests

Author: vladislavkeblysh

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

@@ -83,6 +83,15 @@ describe('Unit Button', () => {
     expect(onClick).toHaveBeenCalledTimes(1);
   });
 
+  it('calls onClick with correct unitId when clicked', () => {
+    const onClick = jest.fn();
+    render(<UnitButton {...mockData} onClick={onClick} />, { wrapWithRouter: true });
+
+    fireEvent.click(screen.getByRole('tab'));
+
+    expect(onClick).toHaveBeenCalledWith(mockData.unitId);
+  });
+
   it('renders with no unitId and does not crash', async () => {
     render(<UnitButton unitId={undefined} onClick={jest.fn()} contentType="video" title="Unit" />, {
       wrapWithRouter: true,