Fix unit tests

microsoft · Apr 6, 2023 · d357974 · d357974
1 parent d4c8b9a
commit d357974
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/vs/editor/contrib/semanticTokens/test/browser/documentSemanticTokens.test.ts b/src/vs/editor/contrib/semanticTokens/test/browser/documentSemanticTokens.test.ts
@@ -96,6 +96,8 @@ suite('ModelSemanticColoring', () => {
 			}));
 
 			const textModel = disposables.add(modelService.createModel('Hello world', languageService.createById('testMode')));
+			// pretend the text model is attached to an editor (so that semantic tokens are computed)
+			textModel.onBeforeAttached();
 
 			// wait for the provider to be called
 			await inFirstCall.wait();
@@ -151,6 +153,8 @@ suite('ModelSemanticColoring', () => {
 			}));
 
 			const textModel = disposables.add(modelService.createModel('', languageService.createById('testMode')));
+			// pretend the text model is attached to an editor (so that semantic tokens are computed)
+			textModel.onBeforeAttached();
 
 			// wait for the semantic tokens to be fetched
 			await Event.toPromise(textModel.onDidChangeTokens);
@@ -192,7 +196,9 @@ suite('ModelSemanticColoring', () => {
 				}
 			}));
 
-			disposables.add(modelService.createModel('', languageService.createById('testMode')));
+			const textModel = disposables.add(modelService.createModel('', languageService.createById('testMode')));
+			// pretend the text model is attached to an editor (so that semantic tokens are computed)
+			textModel.onBeforeAttached();
 
 			await timeout(5000);
 			assert.deepStrictEqual(requestCount, 2);