ant-design/tests/shared/imageTest.tsx

import React from 'react';
// Reference: https://github.com/ant-design/ant-design/pull/24003#discussion_r427267386
// eslint-disable-next-line import/no-unresolved
import { createCache, extractStyle, StyleProvider } from '@ant-design/cssinjs';
import dayjs from 'dayjs';
import { globSync } from 'glob';
import { configureToMatchImageSnapshot } from 'jest-image-snapshot';
import MockDate from 'mockdate';
import ReactDOMServer from 'react-dom/server';
import { App, ConfigProvider, theme } from '../../components';

const toMatchImageSnapshot = configureToMatchImageSnapshot({
  customSnapshotsDir: `${process.cwd()}/imageSnapshots`,
  customDiffDir: `${process.cwd()}/imageDiffSnapshots`,
});

expect.extend({ toMatchImageSnapshot });

const themes = {
  default: theme.defaultAlgorithm,
  dark: theme.darkAlgorithm,
  compact: theme.compactAlgorithm,
};

export default function imageTest(component: React.ReactElement) {
  it(`component image screenshot should correct`, async () => {
    await resetPage();
    await page.setRequestInterception(true);
    const onRequestHandle = (request: any) => {
      if (['image'].includes(request.resourceType())) {
        request.abort();
      } else {
        request.continue();
      }
    };

    MockDate.set(dayjs('2016-11-22').valueOf());
    page.on('request', onRequestHandle);
    await page.goto(`file://${process.cwd()}/tests/index.html`);
    await page.addStyleTag({ path: `${process.cwd()}/dist/reset.css` });

    const cache = createCache();

    const element = (
      <StyleProvider cache={cache}>
        <App>
          {Object.entries(themes).map(([key, algorithm]) => (
            <div
              style={{ background: key === 'dark' ? '#000' : '', padding: `24px 12px` }}
              key={key}
            >
              <ConfigProvider theme={{ algorithm }}>{component}</ConfigProvider>
            </div>
          ))}
        </App>
      </StyleProvider>
    );

    const html = ReactDOMServer.renderToString(element);
    const styleStr = extractStyle(cache);

    await page.evaluate(
      (innerHTML, ssrStyle) => {
        document.querySelector('#root')!.innerHTML = innerHTML;

        const head = document.querySelector('head')!;
        head.innerHTML += ssrStyle;
      },
      html,
      styleStr,
    );

    const image = await page.screenshot({
      fullPage: true,
    });

    expect(image).toMatchImageSnapshot();

    MockDate.reset();
    page.removeListener('request', onRequestHandle);
  });
}

type Options = {
  skip?: boolean | string[];
};

export function imageDemoTest(component: string, options: Options = {}) {
  let describeMethod = options.skip === true ? describe.skip : describe;
  const files = globSync(`./components/${component}/demo/*.tsx`);

  files.forEach((file) => {
    if (Array.isArray(options.skip) && options.skip.some((c) => file.includes(c))) {
      describeMethod = describe.skip;
    } else {
      describeMethod = describe;
    }
    describeMethod(`Test ${file} image`, async () => {
      // eslint-disable-next-line global-require,import/no-dynamic-require
      let { default: Demo } = await import(`../../${file}`);
      if (typeof Demo === 'function') {
        Demo = <Demo />;
      }
      imageTest(Demo);
    });
  });
}