Refactor test helpers (#53918)

Author: sheetalkamat

src/testRunner/unittests/helpers/baseline.ts

@@ -0,0 +1,21 @@
+import * as ts from "../../_namespaces/ts";
+import { libFile } from "./virtualFileSystemWithWatch";
+
+export function compilerOptionsToConfigJson(options: ts.CompilerOptions) {
+    return ts.optionMapToObject(ts.serializeCompilerOptions(options));
+}
+
+export const libContent = `${libFile.content}
+interface ReadonlyArray<T> {}
+declare const console: { log(msg: any): void; };`;
+
+export const symbolLibContent = `
+interface SymbolConstructor {
+    readonly species: symbol;
+    readonly toStringTag: symbol;
+}
+declare var Symbol: SymbolConstructor;
+interface Symbol {
+    readonly [Symbol.toStringTag]: string;
+}
+`;

src/testRunner/unittests/helpers/contents.ts

@@ -0,0 +1,63 @@
+import * as fakes from "../../_namespaces/fakes";
+import * as ts from "../../_namespaces/ts";
+import { commandLineCallbacks } from "./baseline";
+import {
+    makeSystemReadyForBaseline,
+    TscCompileSystem,
+} from "./tsc";
+import {
+    changeToHostTrackingWrittenFiles,
+    createWatchedSystem,
+    FileOrFolderOrSymLink,
+    FileOrFolderOrSymLinkMap,
+    TestServerHost,
+    TestServerHostCreationParameters,
+} from "./virtualFileSystemWithWatch";
+
+export function createSolutionBuilderHostForBaseline(
+    sys: TscCompileSystem | TestServerHost,
+    versionToWrite?: string,
+    originalRead?: (TscCompileSystem | TestServerHost)["readFile"]
+) {
+    if (sys instanceof fakes.System) makeSystemReadyForBaseline(sys, versionToWrite);
+    const { cb } = commandLineCallbacks(sys, originalRead);
+    const host = ts.createSolutionBuilderHost(sys,
+        /*createProgram*/ undefined,
+        ts.createDiagnosticReporter(sys, /*pretty*/ true),
+        ts.createBuilderStatusReporter(sys, /*pretty*/ true)
+    );
+    host.afterProgramEmitAndDiagnostics = cb;
+    host.afterEmitBundle = cb;
+    return host;
+}
+
+export function createSolutionBuilder(system: TestServerHost, rootNames: readonly string[], originalRead?: TestServerHost["readFile"]) {
+    const host = createSolutionBuilderHostForBaseline(system, /*versionToWrite*/ undefined, originalRead);
+    return ts.createSolutionBuilder(host, rootNames, {});
+}
+
+export function ensureErrorFreeBuild(host: TestServerHost, rootNames: readonly string[]) {
+    // ts build should succeed
+    solutionBuildWithBaseline(host, rootNames);
+    assert.equal(host.getOutput().length, 0, JSON.stringify(host.getOutput(), /*replacer*/ undefined, " "));
+}
+
+export function solutionBuildWithBaseline(sys: TestServerHost, solutionRoots: readonly string[], originalRead?: TestServerHost["readFile"]) {
+    const originalReadFile = sys.readFile;
+    const originalWrite = sys.write;
+    const originalWriteFile = sys.writeFile;
+    ts.Debug.assert(sys.writtenFiles === undefined);
+    const solutionBuilder = createSolutionBuilder(changeToHostTrackingWrittenFiles(
+        fakes.patchHostForBuildInfoReadWrite(sys)
+    ), solutionRoots, originalRead);
+    solutionBuilder.build();
+    sys.readFile = originalReadFile;
+    sys.write = originalWrite;
+    sys.writeFile = originalWriteFile;
+    sys.writtenFiles = undefined;
+    return sys;
+}
+
+export function createSystemWithSolutionBuild(solutionRoots: readonly string[], files: FileOrFolderOrSymLinkMap | readonly FileOrFolderOrSymLink[], params?: TestServerHostCreationParameters) {
+    return solutionBuildWithBaseline(createWatchedSystem(files, params), solutionRoots);
+}