Ensure conda version picker handles array response.

Author: karthiknadig

src/client/pythonEnvironments/creation/provider/condaUtils.ts

@@ -1,6 +1,7 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License.
 
+import { isArray } from 'lodash';
 import { CancellationToken, QuickPickItem, Uri } from 'vscode';
 import { Common } from '../../../browser/localize';
 import { Octicons } from '../../../common/constants';
@@ -51,6 +52,9 @@ export async function pickPythonVersion(token?: CancellationToken): Promise<stri
         token,
     );
 
+    if (isArray(selection) && selection.length > 0) {
+        return (selection[0] as QuickPickItem).description;
+    }
     if (selection) {
         return (selection as QuickPickItem).description;
     }

src/test/pythonEnvironments/creation/provider/condaUtils.unit.test.ts

@@ -0,0 +1,52 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+import { assert } from 'chai';
+import * as sinon from 'sinon';
+import { CancellationTokenSource } from 'vscode';
+import * as windowApis from '../../../../client/common/vscodeApis/windowApis';
+import { pickPythonVersion } from '../../../../client/pythonEnvironments/creation/provider/condaUtils';
+
+suite('Venv Utils test', () => {
+    let showQuickPickWithBackStub: sinon.SinonStub;
+
+    setup(() => {
+        showQuickPickWithBackStub = sinon.stub(windowApis, 'showQuickPickWithBack');
+    });
+
+    teardown(() => {
+        sinon.restore();
+    });
+
+    test('No version selected or user pressed escape', async () => {
+        showQuickPickWithBackStub.resolves(undefined);
+
+        const actual = await pickPythonVersion();
+        assert.isUndefined(actual);
+    });
+
+    test('User selected a version', async () => {
+        showQuickPickWithBackStub.resolves({ label: 'Python', description: '3.10' });
+
+        const actual = await pickPythonVersion();
+        assert.equal(actual, '3.10');
+    });
+
+    test('User selected a version (array)', async () => {
+        showQuickPickWithBackStub.resolves([{ label: 'Python', description: '3.10' }]);
+
+        const actual = await pickPythonVersion();
+        assert.equal(actual, '3.10');
+    });
+
+    test('With cancellation', async () => {
+        const source = new CancellationTokenSource();
+
+        showQuickPickWithBackStub.callsFake(() => {
+            source.cancel();
+        });
+
+        const actual = await pickPythonVersion(source.token);
+        assert.isUndefined(actual);
+    });
+});