gitlab-org--gitlab-foss/spec/frontend/helpers/startup_css_helper_spec.js

import { waitForCSSLoaded } from '~/helpers/startup_css_helper';

describe('waitForCSSLoaded', () => {
  let mockedCallback;

  beforeEach(() => {
    mockedCallback = jest.fn();
  });

  describe('Promise-like api', () => {
    it('can be used with a callback', async () => {
      await waitForCSSLoaded(mockedCallback);
      expect(mockedCallback).toHaveBeenCalledTimes(1);
    });

    it('can be used as a promise', async () => {
      await waitForCSSLoaded().then(mockedCallback);
      expect(mockedCallback).toHaveBeenCalledTimes(1);
    });
  });

  describe('with startup css disabled', () => {
    gon.features = {
      startupCss: false,
    };

    it('should invoke the action right away', async () => {
      const events = waitForCSSLoaded(mockedCallback);
      await events;

      expect(mockedCallback).toHaveBeenCalledTimes(1);
    });
  });

  describe('with startup css enabled', () => {
    gon.features = {
      startupCss: true,
    };

    it('should dispatch CSSLoaded when the assets are cached or already loaded', async () => {
      setFixtures(`
        <link href="one.css" data-startupcss="loaded">
        <link href="two.css" data-startupcss="loaded">
      `);
      await waitForCSSLoaded(mockedCallback);

      expect(mockedCallback).toHaveBeenCalledTimes(1);
    });

    it('should wait to call CssLoaded until the assets are loaded', async () => {
      setFixtures(`
        <link href="one.css" data-startupcss="loading">
        <link href="two.css" data-startupcss="loading">
      `);
      const events = waitForCSSLoaded(mockedCallback);
      document
        .querySelectorAll('[data-startupcss="loading"]')
        .forEach(elem => elem.setAttribute('data-startupcss', 'loaded'));
      document.dispatchEvent(new CustomEvent('CSSStartupLinkLoaded'));
      await events;

      expect(mockedCallback).toHaveBeenCalledTimes(1);
    });
  });
});