fix(language-service): Increase project/script version in MockHost.reset() (#33200)

PR Close #33200
This commit is contained in:
Keen Yee Liau
2019-10-16 11:01:31 -07:00
committed by Matias Niemelä
parent becd62d4a1
commit 43241a560a
2 changed files with 34 additions and 17 deletions

View File

@ -92,7 +92,7 @@ const COMPILER_OPTIONS: Readonly<ts.CompilerOptions> = {
};
export class MockTypescriptHost implements ts.LanguageServiceHost {
private angularPath?: string;
private readonly angularPath: string;
private readonly nodeModulesPath: string;
private readonly scriptVersion = new Map<string, number>();
private readonly overrides = new Map<string, string>();
@ -127,14 +127,16 @@ export class MockTypescriptHost implements ts.LanguageServiceHost {
}
addScript(fileName: string, content: string) {
if (this.scriptVersion.has(fileName)) {
throw new Error(`${fileName} is already in the root files.`);
}
this.scriptVersion.set(fileName, 0);
this.projectVersion++;
this.overrides.set(fileName, content);
this.overrideDirectory.add(path.dirname(fileName));
this.scriptNames.push(fileName);
}
forgetAngular() { this.angularPath = undefined; }
overrideOptions(options: Partial<ts.CompilerOptions>) {
this.options = {...this.options, ...options};
this.projectVersion++;
@ -185,6 +187,16 @@ export class MockTypescriptHost implements ts.LanguageServiceHost {
* Reset the project to its original state, effectively removing all overrides.
*/
reset() {
// project version and script version must be monotonically increasing,
// they must not be reset to zero.
this.projectVersion++;
for (const fileName of this.overrides.keys()) {
const version = this.scriptVersion.get(fileName);
if (version === undefined) {
throw new Error(`No prior version found for ${fileName}`);
}
this.scriptVersion.set(fileName, version + 1);
}
// Remove overrides from scriptNames
let length = 0;
for (let i = 0; i < this.scriptNames.length; ++i) {
@ -251,7 +263,7 @@ export class MockTypescriptHost implements ts.LanguageServiceHost {
return result;
}
}
if (this.angularPath && name.startsWith('/' + node_modules + at_angular)) {
if (name.startsWith('/' + node_modules + at_angular)) {
return this.myPath.posix.join(
this.angularPath, name.substr(node_modules.length + at_angular.length + 1));
}