ant-design/tests/shared/demoTest.tsx

134 lines
4.1 KiB
TypeScript

/* eslint-disable react/jsx-no-constructed-context-values */
import * as React from 'react';
import glob from 'glob';
import { render } from 'enzyme';
import MockDate from 'mockdate';
import dayjs from 'dayjs';
import { StyleProvider, createCache } from '@ant-design/cssinjs';
import type { TriggerProps } from 'rc-trigger';
import { excludeWarning } from './excludeWarning';
export const TriggerMockContext = React.createContext<Partial<TriggerProps> | undefined>(undefined);
type CheerIO = ReturnType<typeof render>;
type CheerIOElement = CheerIO[0];
// We should avoid use it in 4.0. Reopen if can not handle this.
const USE_REPLACEMENT = false;
const testDist = process.env.LIB_DIR === 'dist';
/**
* Rc component will generate id for aria usage. It's created as `test-uuid` when env === 'test'. Or
* `f7fa7a3c-a675-47bc-912e-0c45fb6a74d9`(randomly) when not test env. So we need hack of this to
* modify the `aria-controls`.
*/
function ariaConvert(wrapper: CheerIO) {
if (!testDist || !USE_REPLACEMENT) return wrapper;
const matches = new Map();
function process(entry: CheerIOElement) {
if (entry.type === 'text' || entry.type === 'comment') {
return;
}
const { attribs, children } = entry;
if (matches.has(entry)) return;
matches.set(entry, true);
// Change aria
if (attribs && attribs['aria-controls']) {
attribs['aria-controls'] = ''; // Remove all the aria to keep render sync in jest & jest node
}
// Loop children
if (!children) {
return;
}
(Array.isArray(children) ? children : [children]).forEach(process);
}
wrapper.each((_, entry) => process(entry));
return wrapper;
}
type Options = {
skip?: boolean | string[];
};
function baseText(doInject: boolean, component: string, options: Options = {}) {
const files = glob.sync(`./components/${component}/demo/*.md`);
files.forEach(file => {
let testMethod = options.skip === true ? test.skip : test;
if (Array.isArray(options.skip) && options.skip.some(c => file.includes(c))) {
testMethod = test.skip;
}
if (!doInject) {
testMethod(`cssinjs should not warn in ${file}`, () => {
const errSpy = jest.spyOn(console, 'error');
MockDate.set(dayjs('2016-11-22').valueOf());
let Demo = require(`../.${file}`).default; // eslint-disable-line global-require, import/no-dynamic-require
// Inject Trigger status unless skipped
Demo = typeof Demo === 'function' ? <Demo /> : Demo;
// Inject cssinjs cache to avoid create <style /> element
Demo = <StyleProvider cache={createCache()}>{Demo}</StyleProvider>;
render(Demo);
expect(errSpy).not.toHaveBeenCalledWith(expect.stringContaining('[Ant Design CSS-in-JS]'));
MockDate.reset();
errSpy.mockRestore();
});
}
// function doTest(name: string, openTrigger = false) {
testMethod(
doInject ? `renders ${file} extend context correctly` : `renders ${file} correctly`,
() => {
const errSpy = excludeWarning();
MockDate.set(dayjs('2016-11-22').valueOf());
let Demo = require(`../.${file}`).default; // eslint-disable-line global-require, import/no-dynamic-require
// Inject Trigger status unless skipped
Demo = typeof Demo === 'function' ? <Demo /> : Demo;
if (doInject) {
Demo = (
<TriggerMockContext.Provider
value={{
popupVisible: true,
}}
>
{Demo}
</TriggerMockContext.Provider>
);
}
// Inject cssinjs cache to avoid create <style /> element
Demo = <StyleProvider cache={createCache()}>{Demo}</StyleProvider>;
const wrapper = render(Demo);
// Convert aria related content
ariaConvert(wrapper);
expect(wrapper).toMatchSnapshot();
MockDate.reset();
errSpy();
},
);
});
}
export function extendTest(component: string, options: Options = {}) {
baseText(true, component, options);
}
export default function demoTest(component: string, options: Options = {}) {
baseText(false, component, options);
}