Fix diagnostics tests

jupyter-lsp · Feb 24, 2024 · 7713b97 · 7713b97
1 parent 3cc80d5
commit 7713b97
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/packages/jupyterlab-lsp/src/features/diagnostics/diagnostics.spec.ts b/packages/jupyterlab-lsp/src/features/diagnostics/diagnostics.spec.ts
@@ -70,6 +70,7 @@ describe('Diagnostics', () => {
         editorExtensionRegistry: env.editorExtensionRegistry,
         themeManager: null
       });
+      env.featureManager.register(feature);
       await env.init();
     });
     afterEach(() => {
@@ -222,6 +223,7 @@ describe('Diagnostics', () => {
         editorExtensionRegistry: env.editorExtensionRegistry,
         themeManager: null
       });
+      env.featureManager.register(feature);
       await env.init();
     });
     afterEach(() => {

diff --git a/packages/jupyterlab-lsp/src/testutils.ts b/packages/jupyterlab-lsp/src/testutils.ts
@@ -24,7 +24,8 @@ import {
   DocumentConnectionManager,
   FeatureManager,
   ISocketConnectionOptions,
-  ILSPOptions
+  ILSPOptions,
+  ILSPFeatureManager
 } from '@jupyterlab/lsp';
 import { LSPConnection } from '@jupyterlab/lsp/lib/connection';
 import * as nbformat from '@jupyterlab/nbformat';
@@ -200,6 +201,7 @@ export abstract class TestEnvironment implements ITestEnvironment {
   documentOptions: VirtualDocument.IOptions;
   editorExtensionRegistry: EditorExtensionRegistry;
   editorServices: IEditorServices;
+  featureManager: ILSPFeatureManager;
 
   constructor(protected options?: TestEnvironment.IOptions) {
     this.editorExtensionRegistry = new EditorExtensionRegistry();
@@ -230,6 +232,7 @@ export abstract class TestEnvironment implements ITestEnvironment {
       ...this.getDefaults(),
       ...(options?.document || {})
     };
+    this.featureManager = new FeatureManager();
   }
 
   protected abstract createWidget(): IDocumentWidget;
@@ -280,7 +283,7 @@ export abstract class TestEnvironment implements ITestEnvironment {
       docRegistry,
       connectionManager: this.connectionManager,
       codeOverridesManager: overridesManager,
-      featureManager: new FeatureManager(),
+      featureManager: this.featureManager,
       foreignCodeExtractorsManager: foreignCodeExtractors,
       translator: nullTranslator
     });