get_test_runner_code.ts

// Copyright 2018-2023 the Deno authors. All rights reserved. MIT license.

import { CodeBlockWriter } from "../mod.deps.ts";
import { runTestDefinitions } from "./test_runner.ts";

export function getTestRunnerCode(options: {
  testEntryPoints: string[];
  denoTestShimPackageName: string | undefined;
  includeEsModule: boolean | undefined;
  includeScriptModule: boolean | undefined;
}) {
  const usesDenoTest = options.denoTestShimPackageName != null;
  const writer = createWriter();
  writer.writeLine(`const chalk = require("chalk");`)
    .writeLine(`const process = require("process");`);
  if (usesDenoTest) {
    writer.writeLine(`const { pathToFileURL } = require("url");`);
    writer.writeLine(
      `const { testDefinitions } = require("${options.denoTestShimPackageName}");`,
    );
  }
  writer.blankLine();

  writer.writeLine("const filePaths = [");
  writer.indent(() => {
    for (const entryPoint of options.testEntryPoints) {
      writer.quote(entryPoint.replace(/\.ts$/, ".js")).write(",").newLine();
    }
  });
  writer.writeLine("];").newLine();

  writer.write("async function main()").block(() => {
    if (usesDenoTest) {
      writer.write("const testContext = ").inlineBlock(() => {
        writer.writeLine("process,");
        writer.writeLine("chalk,");
      }).write(";").newLine();
    }
    writer.write("for (const [i, filePath] of filePaths.entries())")
      .block(() => {
        writer.write("if (i > 0)").block(() => {
          writer.writeLine(`console.log("");`);
        }).blankLine();

        if (options.includeScriptModule) {
          writer.writeLine(`const scriptPath = "./script/" + filePath;`);
          writer.writeLine(
            `console.log("Running tests in " + chalk.underline(scriptPath) + "...\\n");`,
          );
          writer.writeLine(`process.chdir(__dirname + "/script");`);
          if (usesDenoTest) {
            writer.write(`const scriptTestContext = `).inlineBlock(() => {
              writer.writeLine("origin: pathToFileURL(filePath).toString(),");
              writer.writeLine("...testContext,");
            }).write(";").newLine();
          }
          writer.write("try ").inlineBlock(() => {
            writer.writeLine(`require(scriptPath);`);
          }).write(" catch(err)").block(() => {
            writer.writeLine("console.error(err);");
            writer.writeLine("process.exit(1);");
          });
          if (usesDenoTest) {
            writer.writeLine(
              "await runTestDefinitions(testDefinitions.splice(0, testDefinitions.length), scriptTestContext);",
            );
          }
        }

        if (options.includeEsModule) {
          if (options.includeScriptModule) {
            writer.blankLine();
          }
          writer.writeLine(`const esmPath = "./esm/" + filePath;`);
          writer.writeLine(
            `console.log("\\nRunning tests in " + chalk.underline(esmPath) + "...\\n");`,
          );
          writer.writeLine(`process.chdir(__dirname + "/esm");`);
          if (usesDenoTest) {
            writer.write(`const esmTestContext = `).inlineBlock(() => {
              writer.writeLine("origin: pathToFileURL(filePath).toString(),");
              writer.writeLine("...testContext,");
            }).write(";").newLine();
          }
          writer.writeLine(`await import(esmPath);`);
          if (usesDenoTest) {
            writer.writeLine(
              "await runTestDefinitions(testDefinitions.splice(0, testDefinitions.length), esmTestContext);",
            );
          }
        }
      });
  });
  writer.blankLine();

  if (options.denoTestShimPackageName != null) {
    writer.writeLine(`${getRunTestDefinitionsCode()}`);
    writer.blankLine();
  }

  writer.writeLine("main();");
  return writer.toString();
}

function getRunTestDefinitionsCode() {
  return runTestDefinitions.toString().replace(
    "export async function",
    "async function",
  );
}

function createWriter() {
  return new CodeBlockWriter({
    indentNumberOfSpaces: 2,
  });
}