@@ -18,6 +18,7 @@ import {
1818 InstallPackageTool ,
1919 IResourceReference ,
2020} from '../features/copilotTools' ;
21+ import { EnvironmentManagers , InternalEnvironmentManager } from '../internal.api' ;
2122
2223suite ( 'InstallPackageTool Tests' , ( ) => {
2324 let installPackageTool : InstallPackageTool ;
@@ -249,6 +250,8 @@ suite('GetEnvironmentInfoTool Tests', () => {
249250 let getEnvironmentInfoTool : GetEnvironmentInfoTool ;
250251 let mockApi : typeMoq . IMock < PythonProjectEnvironmentApi & PythonPackageGetterApi & PythonPackageManagementApi > ;
251252 let mockEnvironment : typeMoq . IMock < PythonEnvironment > ;
253+ let em : typeMoq . IMock < EnvironmentManagers > ;
254+ let manager : typeMoq . IMock < InternalEnvironmentManager > ;
252255
253256 setup ( ( ) => {
254257 // Create mock functions
@@ -261,7 +264,16 @@ suite('GetEnvironmentInfoTool Tests', () => {
261264 mockEnvironment . setup ( ( x : any ) => x . then ) . returns ( ( ) => undefined ) ;
262265
263266 // Create an instance of GetEnvironmentInfoTool with the mock functions
264- getEnvironmentInfoTool = new GetEnvironmentInfoTool ( mockApi . object ) ;
267+ manager = typeMoq . Mock . ofType < InternalEnvironmentManager > ( ) ;
268+ manager . setup ( ( m ) => m . id ) . returns ( ( ) => 'ms-python.python:venv' ) ;
269+ manager . setup ( ( m ) => m . name ) . returns ( ( ) => 'venv' ) ;
270+ manager . setup ( ( m ) => m . displayName ) . returns ( ( ) => 'Test Manager' ) ;
271+
272+ em = typeMoq . Mock . ofType < EnvironmentManagers > ( ) ;
273+ em . setup ( ( e ) => e . managers ) . returns ( ( ) => [ manager . object ] ) ;
274+ em . setup ( ( e ) => e . getEnvironmentManager ( typeMoq . It . isAnyString ( ) ) ) . returns ( ( ) => manager . object ) ;
275+
276+ getEnvironmentInfoTool = new GetEnvironmentInfoTool ( mockApi . object , em . object ) ;
265277
266278 // runConfig valid / not valid
267279 // const runConfigValid: PythonCommandRunConfiguration = {
@@ -390,7 +402,8 @@ suite('GetEnvironmentInfoTool Tests', () => {
390402 mockEnvironmentSuccess . setup ( ( x : any ) => x . then ) . returns ( ( ) => undefined ) ;
391403 mockEnvironmentSuccess . setup ( ( x ) => x . version ) . returns ( ( ) => '3.12.1' ) ;
392404 const mockEnvId = typeMoq . Mock . ofType < PythonEnvironmentId > ( ) ;
393- mockEnvId . setup ( ( x ) => x . managerId ) . returns ( ( ) => 'ms-python.python:sys' ) ;
405+ mockEnvId . setup ( ( x ) => x . managerId ) . returns ( ( ) => 'ms-python.python:system' ) ;
406+ manager . setup ( ( m ) => m . name ) . returns ( ( ) => 'system' ) ;
394407 mockEnvironmentSuccess . setup ( ( x ) => x . envId ) . returns ( ( ) => mockEnvId . object ) ;
395408 mockEnvironmentSuccess
396409 . setup ( ( x ) => x . execInfo )
0 commit comments