Fix python.interpreterPath resolving wrong interpreter in tasks.json when other variables are present

Signed-off-by: Mohit Yadav <ymohit799057@gmail.com>

Author: mohityadav8

src/client/interpreter/interpreterPathCommand.ts

@@ -43,6 +43,17 @@ export class InterpreterPathCommand implements IExtensionSingleActivationService
             if (Array.isArray(workspace.workspaceFolders) && workspace.workspaceFolders.length > 0) {
                 workspaceFolder = workspace.workspaceFolders[0].uri.fsPath;
             }
+        } else if (
+            Array.isArray(args) &&
+            Array.isArray(workspace.workspaceFolders) &&
+            workspace.workspaceFolders.length === 1
+        ) {
+            // tasks.json case: args[1] isn't always populated by VS Code's task variable resolver
+            // when other variable kinds (${input:...}, ${env:...}) are resolved alongside this
+            // command in the same task. Single-root workspace is unambiguous, so fall back to it
+            // instead of resolving the global interpreter. Multi-root stays undefined below --
+            // we can't guess which folder without args.
+            workspaceFolder = workspace.workspaceFolders[0].uri.fsPath;
         } else {
             workspaceFolder = undefined;
         }

src/test/interpreters/interpreterPathCommand.unit.test.ts

@@ -5,7 +5,7 @@
 
 import { assert, expect } from 'chai';
 import * as sinon from 'sinon';
-import { anything, instance, mock, when } from 'ts-mockito';
+import { anything, instance, mock, reset, when } from 'ts-mockito';
 import * as TypeMoq from 'typemoq';
 import { Uri } from 'vscode';
 import { IDisposable } from '../../client/common/types';
@@ -14,6 +14,7 @@ import { InterpreterPathCommand } from '../../client/interpreter/interpreterPath
 import { IInterpreterService } from '../../client/interpreter/contracts';
 import { PythonEnvironment } from '../../client/pythonEnvironments/info';
 import * as workspaceApis from '../../client/common/vscodeApis/workspaceApis';
+import { mockedVSCodeNamespaces } from '../vscode-mock';
 
 suite('Interpreter Path Command', () => {
     let interpreterService: IInterpreterService;
@@ -30,6 +31,7 @@ suite('Interpreter Path Command', () => {
 
     teardown(() => {
         sinon.restore();
+        reset(mockedVSCodeNamespaces.workspace);
     });
 
     test('Ensure command is registered with the correct callback handler', async () => {
@@ -68,6 +70,35 @@ suite('Interpreter Path Command', () => {
         expect(setting).to.equal('settingValue');
     });
 
+    test('If `args[1]` is missing but there is exactly one workspace folder, fall back to it (tasks.json case)', async () => {
+        when(mockedVSCodeNamespaces.workspace!.workspaceFolders).thenReturn([
+            { uri: Uri.file('onlyFolderPath'), name: 'onlyFolder', index: 0 },
+        ]);
+
+        const args = ['command'];
+        when(interpreterService.getActiveInterpreter(anything())).thenCall((arg) => {
+            assert.deepEqual(arg, Uri.file('onlyFolderPath'));
+
+            return Promise.resolve({ path: 'settingValue' }) as unknown;
+        });
+        const setting = await interpreterPathCommand._getSelectedInterpreterPath(args);
+        expect(setting).to.equal('settingValue');
+    });
+
+    test('If `args[1]` is missing and there are multiple (or zero) workspace folders, value of workspace folder is `undefined`', async () => {
+        when(mockedVSCodeNamespaces.workspace!.workspaceFolders).thenReturn([
+            { uri: Uri.file('folderOne'), name: 'folderOne', index: 0 },
+            { uri: Uri.file('folderTwo'), name: 'folderTwo', index: 1 },
+        ]);
+
+        const args = ['command'];
+        when(interpreterService.getActiveInterpreter(undefined)).thenReturn(
+            Promise.resolve({ path: 'settingValue' }) as Promise<PythonEnvironment | undefined>,
+        );
+        const setting = await interpreterPathCommand._getSelectedInterpreterPath(args);
+        expect(setting).to.equal('settingValue');
+    });
+
     test('If interpreter path contains spaces, double quote it before returning', async () => {
         const args = ['command', 'folderPath'];
         when(interpreterService.getActiveInterpreter(anything())).thenCall((arg) => {