support listTestName transform

Author: XMadrid

bin/cli.js

@@ -16,7 +16,8 @@ program
   .option("--temp <path>", "test template file folder")
   .option("--output <path>", "coverage report output folder")
   .option("--mode <output mode>", "test result output format")
-  .option("--coverageLimit [error warning...]", "set warn(yellow) and error(red) upper limit in coverage report");
+  .option("--coverageLimit [error warning...]", "set warn(yellow) and error(red) upper limit in coverage report")
+  .option("--testNamePattern <test name pattern>", "run only tests with a name that matches the regex pattern");
 
 program.parse(process.argv);
 const options = program.opts();
@@ -53,9 +54,11 @@ let outputFolder = options.output || config.output || "coverage";
 let errorLimit = options.coverageLimit?.at(0);
 let warnLimit = options.coverageLimit?.at(1);
 
+let testNamePattern = options.testNamePattern;
+
 validatArgument(includes, excludes);
 start_unit_test(
-  { includes, excludes, testcases },
+  { includes, excludes, testcases, testNamePattern },
   { flags, imports },
   { tempFolder, outputFolder, mode, warnLimit, errorLimit }
 )

src/core/precompile.ts

@@ -9,34 +9,73 @@ import { getIncludeFiles } from "../utils/pathResolver.js";
 import { SourceFunctionInfo, UnittestPackage } from "../interface.js";
 import { projectRoot } from "../utils/projectRoot.js";
 
-const sourceFunctions = new Map<string, SourceFunctionInfo[]>();
 export async function precompile(
   includes: string[],
   excludes: string[],
   testcases: string[] | undefined,
-  flags: string,
-  transformFunction = join(projectRoot, "transform", "listFunctions.mjs")
+  testNamePattern: string | undefined,
+  flags: string
 ): Promise<UnittestPackage> {
   // if specify testcases, use testcases for unittest
   // otherwise, get testcases(*.test.ts) in includes directory
   const testCodePaths = testcases ?? getRelatedFiles(includes, excludes, (path: string) => path.endsWith(".test.ts"));
 
-  const sourceCodePaths = getRelatedFiles(includes, excludes, (path: string) => !path.endsWith(".test.ts"));
+  const matchedTestNames: string[] = [];
+  if (testNamePattern) {
+    const testNameInfos = new Map<string, string[]>();
+    const testNameTransformFunction = join(projectRoot, "transform", "listTestNames.mjs");
+    for (const testCodePath of testCodePaths) {
+      await transform(testNameTransformFunction, testCodePath, flags, () => {
+        testNameInfos.set(testCodePath, testNames);
+      });
+    }
+    const regexPattern = new RegExp(testNamePattern);
+    for (const [fileName, testNames] of testNameInfos) {
+      for (const testName of testNames) {
+        if (regexPattern.test(testName)) {
+          matchedTestNames.push(testName);
+        }
+      }
+    }
+  }
 
+  const sourceCodePaths = getRelatedFiles(includes, excludes, (path: string) => !path.endsWith(".test.ts"));
+  const sourceFunctions = new Map<string, SourceFunctionInfo[]>();
+  const sourceTransformFunction = join(projectRoot, "transform", "listFunctions.mjs");
   // The batchSize = 2 is empirical data after benchmarking
   const batchSize = 2;
   for (let i = 0; i < sourceCodePaths.length; i += batchSize) {
     await Promise.all(
-      sourceCodePaths.slice(i, i + batchSize).map((sourcePath) => transform(sourcePath, transformFunction, flags))
+      sourceCodePaths.slice(i, i + batchSize).map((sourcePath) =>
+        transform(sourceTransformFunction, sourcePath, flags, () => {
+          sourceFunctions.set(sourcePath, functionInfos);
+        })
+      )
     );
   }
 
   return {
     testCodePaths,
+    matchedTestNames,
     sourceFunctions,
   };
 }
 
+async function transform(transformFunction: string, codePath: string, flags: string, collectCallback: () => void) {
+  let ascArgv = [codePath, "--noEmit", "--disableWarning", "--transform", transformFunction, "-O0"];
+  if (flags) {
+    const argv = flags.split(" ");
+    ascArgv = ascArgv.concat(argv);
+  }
+  const { error, stderr } = await main(ascArgv);
+  if (error) {
+    // eslint-disable-next-line @typescript-eslint/no-base-to-string
+    console.error(stderr.toString());
+    throw error;
+  }
+  collectCallback();
+}
+
 // a. include in config
 // b. exclude in config
 export function getRelatedFiles(includes: string[], excludes: string[], filter: (path: string) => boolean) {
@@ -58,18 +97,3 @@ export function getRelatedFiles(includes: string[], excludes: string[], filter:
   }
   return result;
 }
-
-async function transform(sourceCodePath: string, transformFunction: string, flags: string) {
-  let ascArgv = [sourceCodePath, "--noEmit", "--disableWarning", "--transform", transformFunction, "-O0"];
-  if (flags) {
-    const argv = flags.split(" ");
-    ascArgv = ascArgv.concat(argv);
-  }
-  const { error, stderr } = await main(ascArgv);
-  if (error) {
-    // eslint-disable-next-line @typescript-eslint/no-base-to-string
-    console.error(stderr.toString());
-    throw error;
-  }
-  sourceFunctions.set(sourceCodePath, functionInfos);
-}

src/index.ts

@@ -55,6 +55,7 @@ export interface FileOption {
   includes: string[];
   excludes: string[];
   testcases: string[] | undefined;
+  testNamePattern: string | undefined;
 }
 export interface TestOption {
   flags: string;
@@ -75,7 +76,7 @@ export type OutputMode = "html" | "json" | "table";
 export async function start_unit_test(fo: FileOption, to: TestOption, oo: OutputOption): Promise<boolean> {
   emptydirSync(oo.outputFolder);
   emptydirSync(oo.tempFolder);
-  const unittestPackage = await precompile(fo.includes, fo.excludes, fo.testcases, to.flags);
+  const unittestPackage = await precompile(fo.includes, fo.excludes, fo.testcases, fo.testNamePattern, to.flags);
   console.log(chalk.blueBright("code analysis: ") + chalk.bold.greenBright("OK"));
   const wasmPaths = await compile(unittestPackage.testCodePaths, oo.tempFolder, to.flags);
   console.log(chalk.blueBright("compile testcases: ") + chalk.bold.greenBright("OK"));

src/interface.ts

@@ -150,6 +150,7 @@ export class CodeCoverage {
 
 export interface UnittestPackage {
   readonly testCodePaths: string[];
+  readonly matchedTestNames: string[];
   readonly sourceFunctions: Map<string, SourceFunctionInfo[]>;
 }
 
@@ -158,5 +159,10 @@ export interface SourceFunctionInfo {
   range: [number, number];
 }
 
+export interface TestNameInfo {
+  testName: string;
+  testFilePath: string;
+}
+
 export const OrganizationName = "wasm-ecosystem";
 export const Repository = "https://github.com/wasm-ecosystem/assemblyscript-unittest-framework";

src/type/global.d.ts

@@ -3,6 +3,8 @@ import { SourceFunctionInfo } from "../interface.ts";
 declare global {
   // store listFunctions transform results in global
   let functionInfos: SourceFunctionInfo[];
+  // store listTestNames transform results in global
+  let testNames: string[];
 }
 
 export {};