OpenFlare/frontend/app/(main)/waf/rules/editor/components/unsaved-changes.test.tsx

import {render} from '@testing-library/react';
import {afterEach, expect, it, vi} from 'vitest';

import {UnsavedChanges} from './unsaved-changes';

afterEach(() => vi.restoreAllMocks());

it('blocks same-origin application links when dirty and confirmation is declined', () => {
  vi.spyOn(window, 'confirm').mockReturnValue(false);
  const {container} = render(<><UnsavedChanges dirty/><a href="/waf">WAF</a></>);
  const event = new MouseEvent('click', {bubbles: true, cancelable: true, button: 0});
  container.querySelector('a')!.dispatchEvent(event);
  expect(window.confirm).toHaveBeenCalledOnce();
  expect(event.defaultPrevented).toBe(true);
});

it('does not block application links without changes', () => {
  const confirm = vi.spyOn(window, 'confirm');
  const {getByRole} = render(<><UnsavedChanges dirty={false}/><a href="/waf">WAF</a></>);
  const event = new MouseEvent('click', {bubbles: true, cancelable: true, button: 0});
  event.preventDefault();
  getByRole('link').dispatchEvent(event);
  expect(confirm).not.toHaveBeenCalled();
});

it('restores declined Back and Forward transitions by indexed delta', () => {
  vi.spyOn(window, 'confirm').mockReturnValue(false);
  const go = vi.spyOn(history, 'go').mockImplementation(() => undefined);
  history.replaceState({__wafEditorIndex: 4}, '');
  render(<UnsavedChanges dirty/>);
  window.dispatchEvent(new PopStateEvent('popstate', {state: {__wafEditorIndex: 3}}));
  expect(go).toHaveBeenLastCalledWith(1);
  window.dispatchEvent(new PopStateEvent('popstate', {state: {__wafEditorIndex: 4}}));
  window.dispatchEvent(new PopStateEvent('popstate', {state: {__wafEditorIndex: 6}}));
  expect(go).toHaveBeenLastCalledWith(-2);
});

it('prompts and restores the current URL for an unknown unindexed history entry', () => {
  vi.spyOn(window, 'confirm').mockReturnValue(false);
  history.replaceState({__wafEditorIndex: 4}, '', '/waf/rules/editor?id=9');
  const push = vi.spyOn(history, 'pushState');
  render(<UnsavedChanges dirty/>);
  window.dispatchEvent(new PopStateEvent('popstate', {state: {legacy: true}}));
  expect(window.confirm).toHaveBeenCalledOnce();
  expect(push).toHaveBeenCalledWith(expect.objectContaining({__wafEditorIndex: 4}), '', '/waf/rules/editor?id=9');
});