Fix tests.

Author: karthiknadig

src/test/linters/lint.args.test.ts

@@ -11,7 +11,13 @@ import { CancellationTokenSource, TextDocument, Uri, WorkspaceFolder } from 'vsc
 import { IDocumentManager, IWorkspaceService } from '../../client/common/application/types';
 import '../../client/common/extensions';
 import { IFileSystem, IPlatformService } from '../../client/common/platform/types';
-import { IConfigurationService, IInstaller, ILintingSettings, IPythonSettings } from '../../client/common/types';
+import {
+    IConfigurationService,
+    IExtensions,
+    IInstaller,
+    ILintingSettings,
+    IPythonSettings,
+} from '../../client/common/types';
 import {
     IInterpreterAutoSelectionService,
     IInterpreterAutoSelectionProxyService,
@@ -53,6 +59,7 @@ suite('Linting - Arguments', () => {
                     let serviceContainer: ServiceContainer;
                     let document: TypeMoq.IMock<TextDocument>;
                     let workspaceService: TypeMoq.IMock<IWorkspaceService>;
+                    let extensionsService: TypeMoq.IMock<IExtensions>;
                     const cancellationToken = new CancellationTokenSource().token;
                     suiteSetup(initialize);
                     setup(async () => {
@@ -122,6 +129,10 @@ suite('Linting - Arguments', () => {
                         const platformService = TypeMoq.Mock.ofType<IPlatformService>();
                         serviceManager.addSingletonInstance<IPlatformService>(IPlatformService, platformService.object);
 
+                        extensionsService = TypeMoq.Mock.ofType<IExtensions>();
+                        extensionsService.setup((e) => e.getExtension(TypeMoq.It.isAny())).returns(() => undefined);
+                        serviceManager.addSingletonInstance<IExtensions>(IExtensions, extensionsService.object);
+
                         lm = new LinterManager(configService.object);
                         serviceManager.addSingletonInstance<ILinterManager>(ILinterManager, lm);
                         document = TypeMoq.Mock.ofType<TextDocument>();

src/test/linters/pylint.test.ts

@@ -18,7 +18,7 @@ import { LanguageServerType } from '../../client/activation/types';
 import { IWorkspaceService } from '../../client/common/application/types';
 import { IFileSystem, IPlatformService } from '../../client/common/platform/types';
 import { IPythonToolExecutionService } from '../../client/common/process/types';
-import { IConfigurationService, IInstaller, IPythonSettings } from '../../client/common/types';
+import { IConfigurationService, IExtensions, IInstaller, IPythonSettings } from '../../client/common/types';
 import {
     IInterpreterAutoSelectionService,
     IInterpreterAutoSelectionProxyService,
@@ -40,6 +40,7 @@ suite('Linting - Pylint', () => {
     let workspaceConfig: TypeMoq.IMock<WorkspaceConfiguration>;
     let pythonSettings: TypeMoq.IMock<IPythonSettings>;
     let serviceContainer: ServiceContainer;
+    let extensionsService: TypeMoq.IMock<IExtensions>;
 
     setup(() => {
         fileSystem = TypeMoq.Mock.ofType<IFileSystem>();
@@ -50,6 +51,9 @@ suite('Linting - Pylint', () => {
         platformService = TypeMoq.Mock.ofType<IPlatformService>();
         platformService.setup((x) => x.isWindows).returns(() => false);
 
+        extensionsService = TypeMoq.Mock.ofType<IExtensions>();
+        extensionsService.setup((e) => e.getExtension(TypeMoq.It.isAny())).returns(() => undefined);
+
         workspace = TypeMoq.Mock.ofType<IWorkspaceService>();
         execService = TypeMoq.Mock.ofType<IPythonToolExecutionService>();
 
@@ -72,6 +76,7 @@ suite('Linting - Pylint', () => {
             IInterpreterAutoSelectionProxyService,
             MockAutoSelectionService,
         );
+        serviceManager.addSingletonInstance<IExtensions>(IExtensions, extensionsService.object);
 
         pythonSettings = TypeMoq.Mock.ofType<IPythonSettings>();
         pythonSettings.setup((p) => p.languageServer).returns(() => LanguageServerType.Jedi);

src/test/linters/pylint.unit.test.ts

@@ -8,7 +8,7 @@ import * as TypeMoq from 'typemoq';
 import * as vscode from 'vscode';
 import { IWorkspaceService } from '../../client/common/application/types';
 import { IFileSystem, IPlatformService } from '../../client/common/platform/types';
-import { IConfigurationService, IPythonSettings } from '../../client/common/types';
+import { IConfigurationService, IExtensions, IPythonSettings } from '../../client/common/types';
 import { IServiceContainer } from '../../client/ioc/types';
 import { IToolsExtensionPrompt } from '../../client/linters/prompts/types';
 import { Pylint } from '../../client/linters/pylint';
@@ -22,6 +22,7 @@ suite('Pylint - Function runLinter()', () => {
     let manager: TypeMoq.IMock<ILinterManager>;
     let _info: TypeMoq.IMock<ILinterInfo>;
     let platformService: TypeMoq.IMock<IPlatformService>;
+    let extensionsService: TypeMoq.IMock<IExtensions>;
     let run: sinon.SinonStub;
     let parseMessagesSeverity: sinon.SinonStub;
     let extensionPrompt: TypeMoq.IMock<IToolsExtensionPrompt>;
@@ -77,6 +78,7 @@ suite('Pylint - Function runLinter()', () => {
         serviceContainer = TypeMoq.Mock.ofType<IServiceContainer>();
         workspaceService = TypeMoq.Mock.ofType<IWorkspaceService>();
         configService = TypeMoq.Mock.ofType<IConfigurationService>();
+        extensionsService = TypeMoq.Mock.ofType<IExtensions>();
         manager = TypeMoq.Mock.ofType<ILinterManager>();
         serviceContainer.setup((c) => c.get(TypeMoq.It.isValue(ILinterManager))).returns(() => manager.object);
         serviceContainer
@@ -89,6 +91,7 @@ suite('Pylint - Function runLinter()', () => {
         serviceContainer
             .setup((c) => c.get(TypeMoq.It.isValue(IPlatformService)))
             .returns(() => platformService.object);
+        serviceContainer.setup((c) => c.get(TypeMoq.It.isValue(IExtensions))).returns(() => extensionsService.object);
         fileSystem = TypeMoq.Mock.ofType<IFileSystem>();
         fileSystem
             .setup((x) => x.arePathsSame(TypeMoq.It.isAnyString(), TypeMoq.It.isAnyString()))