Function generateTestFile

src/generators/test/test.file.ts:125–159

(exports: Export[], importFrom: string, to: vscode.Uri, isRouteFile: boolean)

};

export const generateTestFile = async (exports: Export[], importFrom: string, to: vscode.Uri, isRouteFile: boolean) => {
  if (!exports.length) {
    return;
  }

  let existingContent = await tryReadFile(to.path);

  const testFileContent = generateTestFileContent(exports, isRouteFile);
  // Creates the dependencies to import, eg. default as TestRoute, utilFunction, etc...
  const exportsToImport = exports
    .map((toImport) => (toImport.isDefault ? `default as ${toImport.name}` : toImport.name))
    .join(", ");

  // File already exists, append stuff to it
  if (existingContent) {
    const fileName = to.path.split("/").pop();
    // We want to match an existing import partially so we just need the name of the file, eg ../module.name.test.tsx => module.name
    const importStatement = `${fileName!.replace(".test.tsx", "").replace(".test.ts", "")}`;
    // Adds the imports to the existing import statement
    existingContent = addImportsToImportStatement(importStatement, exportsToImport, existingContent);
    // Adds import { render } from "@testing-library/react" if it doesn't exist and is needed
    const additionalImports = isRouteFile && !existingContent.includes(renderImport) ? `${renderImport};\n` : "";
    // Appends the new test suites to the end of the file and writes back the existing content with modified imports
    await vscode.workspace.fs.writeFile(to, Buffer.from(additionalImports + existingContent + "\n" + testFileContent));
    // Formats file
    return await formatFile(to.path);
  }
  // Creating a new file, add the imports and the test suites
  const importStatement = `${importFrom.replace(".tsx", "").replace(".ts", "")}`;
  const dependencies = `import { ${exportsToImport} } from "${importStatement}";${
    isRouteFile && exports.some((exp) => exp.isDefault || exp.name === "ErrorBoundary") ? `\n${renderImport};\n` : ""
  }\n\n`;
  await vscode.workspace.fs.writeFile(to, Buffer.from(dependencies + testFileContent));
  await formatFile(to.path);
};