shlink-web-client/test/mercure/helpers/index.test.tsx

import { EventSourcePolyfill as EventSource } from 'event-source-polyfill';
import { Mock } from 'ts-mockery';
import { identity } from 'ramda';
import { bindToMercureTopic } from '../../../src/mercure/helpers';
import { MercureInfo } from '../../../src/mercure/reducers/mercureInfo';

jest.mock('event-source-polyfill');

describe('helpers', () => {
  afterEach(jest.resetAllMocks);

  describe('bindToMercureTopic', () => {
    const onMessage = jest.fn();
    const onTokenExpired = jest.fn();

    it.each([
      [ Mock.of<MercureInfo>({ loading: true, error: false, mercureHubUrl: 'foo' }) ],
      [ Mock.of<MercureInfo>({ loading: false, error: true, mercureHubUrl: 'foo' }) ],
      [ Mock.of<MercureInfo>({ loading: true, error: true, mercureHubUrl: 'foo' }) ],
      [ Mock.of<MercureInfo>({ loading: false, error: false, mercureHubUrl: undefined }) ],
      [ Mock.of<MercureInfo>({ loading: true, error: true, mercureHubUrl: undefined }) ],
    ])('does not bind an EventSource when loading, error or no hub URL', (mercureInfo) => {
      bindToMercureTopic(mercureInfo, '', identity, identity);

      expect(EventSource).not.toHaveBeenCalled();
      expect(onMessage).not.toHaveBeenCalled();
      expect(onTokenExpired).not.toHaveBeenCalled();
    });

    it('binds an EventSource when mercure info is properly loaded', () => {
      const token = 'abc.123.efg';
      const mercureHubUrl = 'https://example.com/.well-known/mercure';
      const topic = 'foo';
      const hubUrl = new URL(mercureHubUrl);

      hubUrl.searchParams.append('topic', topic);

      const callback = bindToMercureTopic({
        loading: false,
        error: false,
        mercureHubUrl,
        token,
      }, topic, onMessage, onTokenExpired);

      expect(EventSource).toHaveBeenCalledWith(hubUrl, {
        headers: {
          Authorization: `Bearer ${token}`,
        },
      });

      const [ es ] = EventSource.mock.instances;

      es.onmessage({ data: '{"foo": "bar"}' });
      es.onerror({ status: 401 });
      expect(onMessage).toHaveBeenCalledWith({ foo: 'bar' });
      expect(onTokenExpired).toHaveBeenCalled();

      callback?.();
      expect(es.close).toHaveBeenCalled();
    });
  });
});