test: edit unit tests

Author: diana-villalvazo-wgu

src/dateExtensions/DateExtensionsPage.test.tsx

@@ -1,11 +1,13 @@
 import { render, screen } from '@testing-library/react';
+import userEvent from '@testing-library/user-event';
 import { IntlProvider } from '@openedx/frontend-base';
 import { MemoryRouter, Route, Routes } from 'react-router-dom';
 import DateExtensionsPage from './DateExtensionsPage';
-import { useDateExtensions } from '../data/apiHook';
+import { useDateExtensions, useResetDateExtensionMutation } from '../data/apiHook';
 
 jest.mock('../data/apiHook', () => ({
   useDateExtensions: jest.fn(),
+  useResetDateExtensionMutation: jest.fn(),
 }));
 
 const mockDateExtensions = [
@@ -25,6 +27,9 @@ describe('DateExtensionsPage', () => {
       data: mockDateExtensions,
       isLoading: false,
     });
+    (useResetDateExtensionMutation as jest.Mock).mockReturnValue({
+      mutate: jest.fn(),
+    });
   });
 
   const RenderWithRouter = () => (
@@ -67,4 +72,37 @@ describe('DateExtensionsPage', () => {
     const resetLinks = screen.getAllByRole('button', { name: 'Reset Extensions' });
     expect(resetLinks).toHaveLength(mockDateExtensions.length);
   });
+
+  it('opens reset modal when reset button is clicked', async () => {
+    render(<RenderWithRouter />);
+    const user = userEvent.setup();
+    const resetButton = screen.getByRole('button', { name: 'Reset Extensions' });
+    await user.click(resetButton);
+    expect(screen.getByRole('dialog')).toBeInTheDocument();
+    expect(screen.getByText(/reset extensions for/i)).toBeInTheDocument();
+    const confirmButton = screen.getByRole('button', { name: /reset due date/i });
+    expect(confirmButton).toBeInTheDocument();
+  });
+
+  it('calls reset mutation when confirm reset is clicked', async () => {
+    const mutateMock = jest.fn();
+    (useResetDateExtensionMutation as jest.Mock).mockReturnValue({ mutate: mutateMock });
+    render(<RenderWithRouter />);
+    const user = userEvent.setup();
+    const resetButton = screen.getByRole('button', { name: 'Reset Extensions' });
+    await user.click(resetButton);
+    const confirmButton = screen.getByRole('button', { name: /reset due date/i });
+    await user.click(confirmButton);
+    expect(mutateMock).toHaveBeenCalled();
+  });
+
+  it('closes reset modal when cancel is clicked', async () => {
+    render(<RenderWithRouter />);
+    const user = userEvent.setup();
+    const resetButton = screen.getByRole('button', { name: 'Reset Extensions' });
+    await user.click(resetButton);
+    const cancelButton = screen.getByRole('button', { name: /cancel/i });
+    await user.click(cancelButton);
+    expect(screen.queryByRole('dialog')).not.toBeInTheDocument();
+  });
 });