IIIFIFrameCommunication.test.jsx

import { render, screen } from '@tests/utils/test-utils';
import { IIIFIFrameCommunication } from '../../../src/components/IIIFIFrameCommunication';

/** */
function createWrapper(props) {
  render(
    <IIIFIFrameCommunication
      src="https://iiifauth.digtest.co.uk/auth/token/login/01_Icarus_Breughel.jpg?origin=http://localhost:4444&messageId=https://iiifauth.digtest.co.uk/auth/token/login/01_Icarus_Breughel.jpg"
      title="AccessTokenSender"
      handleReceiveMessage={() => {}}
      {...props}
    />,
  );
}

describe('IIIFIFrameCommunication', () => {
  it('should render an iframe', () => {
    createWrapper();
    expect(screen.getByTitle('AccessTokenSender', { hidden: true })).toBeInTheDocument();
  });
});

describe('Register event listener', () => {
  afterEach(() => {
    vi.restoreAllMocks();
  });
  it('should call handleReceiveMessage on message event', () => {
    const events = {};
    vi.spyOn(window, 'addEventListener').mockImplementation((event, onReceiveMessage) => {
      events[event] = onReceiveMessage;
    });
    vi.spyOn(window, 'removeEventListener').mockImplementation((event, onReceiveMessage) => {
      events[event] = undefined;
    });
    const props = { handleReceiveMessage: vi.fn() };
    const view = render(<IIIFIFrameCommunication {...props} />);
    events.message();

    expect(props.handleReceiveMessage).toBeCalledTimes(1);
    expect(window.addEventListener).toBeCalledWith('message', expect.any(Function));

    view.unmount();
    expect(window.removeEventListener).toBeCalledWith('message', expect.any(Function), false);
  });
});