Register test manager when using the new language server (#2187)

Author: DonJayamanne

news/2 Fixes/2186.md

@@ -0,0 +1 @@
+Register test manager when using the new language server.

src/client/activation/languageServer.ts

@@ -11,14 +11,15 @@ import { isTestExecution, STANDARD_OUTPUT_CHANNEL } from '../common/constants';
 import { createDeferred, Deferred } from '../common/helpers';
 import { IFileSystem, IPlatformService } from '../common/platform/types';
 import { StopWatch } from '../common/stopWatch';
-import { IConfigurationService, IExtensionContext, IOutputChannel, IPythonSettings } from '../common/types';
+import { IConfigurationService, IExtensionContext, ILogger, IOutputChannel, IPythonSettings } from '../common/types';
 import { IServiceContainer } from '../ioc/types';
 import {
     PYTHON_LANGUAGE_SERVER_DOWNLOADED,
     PYTHON_LANGUAGE_SERVER_ENABLED,
     PYTHON_LANGUAGE_SERVER_ERROR
 } from '../telemetry/constants';
 import { getTelemetryReporter } from '../telemetry/telemetry';
+import { IUnitTestManagementService } from '../unittests/types';
 import { LanguageServerDownloader } from './downloader';
 import { InterpreterData, InterpreterDataService } from './interpreterDataService';
 import { PlatformData } from './platformData';
@@ -89,6 +90,11 @@ export class LanguageServerExtensionActivator implements IExtensionActivator {
         if (!clientOptions) {
             return false;
         }
+
+        const testManagementService = this.services.get<IUnitTestManagementService>(IUnitTestManagementService);
+        testManagementService.activate()
+            .catch(ex => this.services.get<ILogger>(ILogger).logError('Failed to activate Unit Tests', ex));
+
         return this.startLanguageServer(clientOptions);
     }