Test restarting

Author: jakebailey

src/test/activation/languageServer/manager.unit.test.ts

@@ -42,6 +42,7 @@ suite('Language Server - Manager', () => {
     let configService: IConfigurationService;
     let commandManager: ICommandManager;
     const languageClientOptions = ({ x: 1 } as any) as LanguageClientOptions;
+    let restartCommand: () => void;
     setup(() => {
         serviceContainer = mock(ServiceContainer);
         analysisOptions = mock(DotNetLanguageServerAnalysisOptions);
@@ -53,7 +54,12 @@ suite('Language Server - Manager', () => {
 
         commandManager = mock(CommandManager);
         const disposable = mock(Disposable);
-        when(commandManager.registerCommand(Commands.RestartLS, anything())).thenReturn(instance(disposable));
+        when(commandManager.registerCommand(Commands.RestartLS, anything())).thenCall(
+            (_command: string, func: () => void) => {
+                restartCommand = func;
+                return instance(disposable);
+            }
+        );
 
         manager = new DotNetLanguageServerManager(
             instance(serviceContainer),
@@ -189,5 +195,17 @@ suite('Language Server - Manager', () => {
 
             verify(languageServer.loadExtension(args)).once();
         });
+        test('Restart LS command restarts LS', async () => {
+            await startLanguageServer();
+
+            restartCommand();
+            await sleep(50);
+
+            verify(languageServer.dispose()).once();
+
+            verify(analysisOptions.getAnalysisOptions()).twice();
+            verify(serviceContainer.get<ILanguageServerProxy>(ILanguageServerProxy)).twice();
+            verify(languageServer.start(resource, undefined, languageClientOptions)).twice();
+        });
     });
 });