UNPKG

1.58 kBJavaScriptView Raw
1import { readdir, stat } from 'fs';
2import { createRequire } from 'module';
3import { resolve, join } from 'path';
4import { promisify } from 'util';
5
6const ls = promisify(readdir);
7const toStat = promisify(stat);
8const toRegex = x => new RegExp(x, 'i');
9
10export async function parse(dir, pattern, opts = {}) {
11 if (pattern) pattern = toRegex(pattern);
12 else if (dir) pattern = /(((?:[^\/]*(?:\/|$))*)[\\\/])?\w+\.([mc]js|[jt]sx?)$/;
13 else pattern = /((\/|^)(tests?|__tests?__)\/.*|\.(tests?|spec)|^\/?tests?)\.([mc]js|[jt]sx?)$/i;
14 dir = resolve(opts.cwd || '.', dir || '.');
15
16 let suites = [];
17 let requires = [].concat(opts.require || []).filter(Boolean);
18 let ignores = ['^.git', 'node_modules'].concat(opts.ignore || []).map(toRegex);
19
20 if (requires.length) {
21 let $require = createRequire(import.meta.url);
22 requires.forEach(name => {
23 try { return $require(name) }
24 catch (e) { throw new Error(`Cannot find module "${name}"`) }
25 });
26 }
27
28 // NOTE: Node 8.x support
29 // @modified lukeed/totalist
30 await (async function collect(d, p) {
31 await ls(d).then(files => {
32 return Promise.all(
33 files.map(async str => {
34 let name = join(p, str);
35 for (let i = ignores.length; i--;) {
36 if (ignores[i].test(name)) return;
37 }
38
39 let file = join(d, str);
40 let stats = await toStat(file);
41 if (stats.isDirectory()) return collect(file, name);
42 else if (pattern.test(name)) suites.push({ name, file });
43 })
44 );
45 });
46 })(dir, '');
47
48 suites.sort((a, b) => a.name.localeCompare(b.name));
49
50 return { dir, suites, requires: requires.length > 0 };
51}