123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117
// Copyright 2018-2022 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,  });}