add install for envs ext

Author: eleanorjboyd

src/test/constants.ts

@@ -33,6 +33,8 @@ function isMultiRootTest() {
 export const EXTENSION_ROOT_DIR_FOR_TESTS = path.join(__dirname, '..', '..');
 export const PVSC_EXTENSION_ID_FOR_TESTS = 'ms-python.python';
 
+export const PVSC_ENVS_EXTENSION_ID_FOR_TESTS = 'ms-python.vscode-python-envs';
+
 export const SMOKE_TEST_EXTENSIONS_DIR = path.join(
     EXTENSION_ROOT_DIR_FOR_TESTS,
     'tmp',

src/test/runTest.ts

@@ -2,7 +2,7 @@ import * as cp from 'child_process';
 import * as path from 'path';
 
 import { downloadAndUnzipVSCode, resolveCliArgsFromVSCodeExecutablePath, runTests } from '@vscode/test-electron';
-import { PVSC_EXTENSION_ID_FOR_TESTS } from './constants';
+import { PVSC_ENVS_EXTENSION_ID_FOR_TESTS, PVSC_EXTENSION_ID_FOR_TESTS } from './constants';
 import { OSType, getOSType } from '../extension/common/platform';
 
 async function main() {
@@ -20,16 +20,24 @@ async function main() {
         // Use cp.spawn / cp.exec for custom setup
         if (getOSType() === OSType.Windows) {
             const exec = path.basename(cliPath);
-            cp.spawnSync(exec, [...args, '--install-extension', PVSC_EXTENSION_ID_FOR_TESTS], {
-                cwd: path.dirname(cliPath),
-                encoding: 'utf-8',
-                stdio: 'inherit',
-            });
+            cp.spawnSync(
+                exec,
+                [...args, '--install-extension', PVSC_EXTENSION_ID_FOR_TESTS, PVSC_ENVS_EXTENSION_ID_FOR_TESTS],
+                {
+                    cwd: path.dirname(cliPath),
+                    encoding: 'utf-8',
+                    stdio: 'inherit',
+                },
+            );
         } else {
-            cp.spawnSync(cliPath, [...args, '--install-extension', PVSC_EXTENSION_ID_FOR_TESTS], {
-                encoding: 'utf-8',
-                stdio: 'inherit',
-            });
+            cp.spawnSync(
+                cliPath,
+                [...args, '--install-extension', PVSC_EXTENSION_ID_FOR_TESTS, PVSC_ENVS_EXTENSION_ID_FOR_TESTS],
+                {
+                    encoding: 'utf-8',
+                    stdio: 'inherit',
+                },
+            );
         }
 
         // Run the extension test