mirror of
https://gitee.com/antv/g6.git
synced 2024-11-30 02:38:20 +08:00
111 lines
3.7 KiB
JavaScript
111 lines
3.7 KiB
JavaScript
/**
|
||
* 为指定路径下的示例添加测试用例
|
||
* @param {string} path demo 路径
|
||
* @param {string} prifix 前缀
|
||
* @param {string} data 数据文件的
|
||
* @example
|
||
* - node scripts/demo-to-test/index net/radialLayout layout radial.json
|
||
* - explain: 为 site/exmples/net/radialLayout 路径下的 demo 示例添加测试文件；layout 为文件名、Fn前缀；radial.json 为数据文件
|
||
*/
|
||
const fs = require('fs');
|
||
const path = require('path');
|
||
const prettier = require('prettier');
|
||
const { uniq, camelCase, kebabCase } = require('lodash');
|
||
const { parser, transformSign } = require('./core');
|
||
const specTemplate = require('./template/spec');
|
||
const testTemplate = require('./template/test');
|
||
const itTemplate = require('./template/it');
|
||
|
||
const args = process.argv.slice(2);
|
||
const demoPath = args[0];
|
||
const name = kebabCase(demoPath.split('/').pop());
|
||
const prefix = args[1];
|
||
const dataFile = args[2];
|
||
|
||
const prePath = `../../packages/`;
|
||
const demoDir = path.resolve(__dirname, prePath, `site/examples/${demoPath}/demo`);
|
||
const testDir = path.resolve(__dirname, prePath, `g6/__tests__/demo/case`);
|
||
const unitDir = path.resolve(__dirname, prePath, `g6/__tests__/unit/${prefix}s`);
|
||
const exportDir = path.resolve(__dirname, prePath, `g6/__tests__/demo/case/index.ts`);
|
||
|
||
const unitMeta = [];
|
||
|
||
/**
|
||
* 移除 object string 左右 {}，直接插入到已有 object 中
|
||
* @param text
|
||
*/
|
||
const removeFirstAndLastTwoChars = (text) => {
|
||
if (text.length < 4) {
|
||
return '';
|
||
}
|
||
return text.substring(2, text.length - 2);
|
||
};
|
||
|
||
const formatCode = async (code) => {
|
||
const formattedContent = await prettier.format(code, {
|
||
semi: true,
|
||
singleQuote: true,
|
||
printWidth: 120,
|
||
parser: 'typescript',
|
||
});
|
||
return formattedContent;
|
||
};
|
||
|
||
const initUnit = async () => {
|
||
const unitFiles = fs.readdirSync(unitDir);
|
||
let content = '';
|
||
let imported = '';
|
||
unitMeta.forEach(async (kebabName) => {
|
||
const camelName = camelCase(kebabName);
|
||
const sampleKebabName = kebabName.replace(`${prefix}-${name}-`, '');
|
||
const config = { camelName, sampleKebabName, sampleKebabNameWithSpace: sampleKebabName.replace(/-/g, ' ') };
|
||
let exist = !!imported;
|
||
content += exist ? '\n\n' + itTemplate(config) : itTemplate(config);
|
||
imported += exist ? ',\n' + camelName : camelName;
|
||
fs.appendFileSync(exportDir, `export * from './${kebabName}';\n`, 'utf-8');
|
||
});
|
||
const filename = `./${name}.spec.ts`;
|
||
if (!unitFiles.includes(filename)) {
|
||
const formattedContent = await formatCode(
|
||
specTemplate({
|
||
content,
|
||
imported,
|
||
describe: `${uniq(name.split('-')).join(' ')}`,
|
||
}),
|
||
);
|
||
fs.writeFileSync(path.resolve(unitDir, filename), formattedContent, 'utf-8');
|
||
}
|
||
};
|
||
|
||
const initTest = async (dir) => {
|
||
const files = await fs.readdirSync(dir);
|
||
const testFiles = fs.readdirSync(testDir);
|
||
files.forEach(async (file, index) => {
|
||
const filePath = path.resolve(dir, file);
|
||
const stats = fs.statSync(filePath);
|
||
const kebabName = uniq(`${prefix}-${name}-${kebabCase(file.replace('.ts', ''))}`.split('-')).join('-');
|
||
const camelName = camelCase(kebabName);
|
||
if (stats.isFile() && file.endsWith('.ts')) {
|
||
unitMeta.push(kebabName);
|
||
const { config, success } = parser(filePath);
|
||
if (success) {
|
||
const formattedContent = await formatCode(
|
||
testTemplate({
|
||
config: removeFirstAndLastTwoChars(transformSign(JSON.stringify(config, null, 2))),
|
||
filename: camelName,
|
||
dataFile,
|
||
}),
|
||
);
|
||
if (!testFiles.includes(`${kebabName}.ts`)) {
|
||
fs.writeFileSync(path.resolve(testDir, `./${kebabName}.ts`), formattedContent, 'utf-8');
|
||
}
|
||
}
|
||
}
|
||
if (index === files.length - 1) {
|
||
initUnit();
|
||
}
|
||
});
|
||
};
|
||
|
||
initTest(demoDir);
|