fix: Register code action provider after server mode change (#1224)

microsoft · Jun 28, 2021 · 50119b3 · 50119b3
1 parent f13ac5c
commit 50119b3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/extension.ts b/src/extension.ts
@@ -85,6 +85,7 @@ async function doActivate(_operationId: string, context: ExtensionContext): Prom
                     await testSourceProvider.initialize();
                     await testFileWatcher.registerListeners();
                     await testCodeLensController.registerCodeLensProvider();
+                    await registerTestCodeActionProvider();
                 }
             }));
         }

diff --git a/src/provider/codeActionProvider.ts b/src/provider/codeActionProvider.ts
@@ -29,9 +29,8 @@ export class TestCodeActionProvider implements CodeActionProvider {
 }
 
 export async function registerTestCodeActionProvider(): Promise<Disposable> {
-    if (provider) {
-        provider.dispose();
-    }
+    disposeCodeActionProvider();
+
     provider = languages.registerCodeActionsProvider(
         {language: 'java', scheme: 'file', pattern: '**/*.java'},
         new TestCodeActionProvider(),