You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

58 lines
1.6 KiB
TypeScript

import { PropsWithChildren } from 'react';
import { SWRConfig } from 'swr';
// 全局的 SWR 配置
const swrConfig = {
provider: () => new Map(),
};
export const withSWR = ({ children }: PropsWithChildren) => (
<SWRConfig value={swrConfig}>{children}</SWRConfig>
);
interface TestServiceOptions {
/** 是否检查 async */
checkAsync?: boolean;
/** 自定义的额外检查 */
extraChecks?: (method: string, func: () => any) => void;
/** 是否跳过某些方法 */
skipMethods?: string[];
}
const builtinSkipProps = new Set(['userId']);
export const testService = (ServiceClass: new () => any, options: TestServiceOptions = {}) => {
const { checkAsync = true, skipMethods = ['userId'], extraChecks } = options;
describe(ServiceClass.name, () => {
it('should implement all methods as arrow functions', () => {
const service = new ServiceClass();
const methods = Object.getOwnPropertyNames(service).filter(
(method) => !builtinSkipProps.has(method) || !skipMethods.includes(method),
);
methods.forEach((method) => {
const func = service[method];
// 检查是否为函数
expect(typeof func).toBe('function');
const funcString = func.toString();
// 验证是否是箭头函数
expect(funcString).toContain('=>');
// 可选的 async 检查
if (checkAsync) {
expect(funcString).toMatch(/^async.*=>/);
}
// 运行额外的自定义检查
if (extraChecks) {
extraChecks(method, func);
}
});
});
});
};