44import { expect , use } from 'chai' ;
55import * as chaiAsPromised from 'chai-as-promised' ;
66import * as path from 'path' ;
7- import { anything , instance , mock , when } from 'ts-mockito' ;
7+ import { anything } from 'ts-mockito' ;
88import { ConfigurationTarget , Disposable , Uri , workspace } from 'vscode' ;
99import { WorkspaceService } from '../../../client/common/application/workspace' ;
1010import { PlatformService } from '../../../client/common/platform/platformService' ;
@@ -14,14 +14,14 @@ import { getSearchPathEnvVarNames } from '../../../client/common/utils/exec';
1414import { EnvironmentVariablesService } from '../../../client/common/variables/environment' ;
1515import { EnvironmentVariablesProvider } from '../../../client/common/variables/environmentVariablesProvider' ;
1616import { EnvironmentVariables } from '../../../client/common/variables/types' ;
17- import { EnvironmentActivationService } from '../../../client/interpreter/activation/service' ;
1817import { IEnvironmentActivationService } from '../../../client/interpreter/activation/types' ;
1918import { IInterpreterAutoSelectionService } from '../../../client/interpreter/autoSelection/types' ;
2019import { clearPythonPathInWorkspaceFolder , isOs , OSType , updateSetting } from '../../common' ;
2120import { closeActiveWindows , initialize , initializeTest , IS_MULTI_ROOT_TEST } from '../../initialize' ;
2221import { MockAutoSelectionService } from '../../mocks/autoSelector' ;
2322import { MockProcess } from '../../mocks/process' ;
2423import { UnitTestIocContainer } from '../../testing/serviceRegistry' ;
24+ import { createTypeMoq } from '../../mocks/helper' ;
2525
2626use ( chaiAsPromised . default ) ;
2727
@@ -47,12 +47,21 @@ suite('Multiroot Environment Variables Provider', () => {
4747 ioc . registerProcessTypes ( ) ;
4848 ioc . registerInterpreterStorageTypes ( ) ;
4949 await ioc . registerMockInterpreterTypes ( ) ;
50- const mockEnvironmentActivationService = mock ( EnvironmentActivationService ) ;
51- when ( mockEnvironmentActivationService . getActivatedEnvironmentVariables ( anything ( ) ) ) . thenResolve ( ) ;
52- ioc . serviceManager . rebindInstance < IEnvironmentActivationService > (
53- IEnvironmentActivationService ,
54- instance ( mockEnvironmentActivationService ) ,
55- ) ;
50+ const mockEnvironmentActivationService = createTypeMoq < IEnvironmentActivationService > ( ) ;
51+ mockEnvironmentActivationService
52+ . setup ( ( m ) => m . getActivatedEnvironmentVariables ( anything ( ) ) )
53+ . returns ( ( ) => Promise . resolve ( { } ) ) ;
54+ if ( ioc . serviceManager . tryGet < IEnvironmentActivationService > ( IEnvironmentActivationService ) ) {
55+ ioc . serviceManager . rebindInstance < IEnvironmentActivationService > (
56+ IEnvironmentActivationService ,
57+ mockEnvironmentActivationService . object ,
58+ ) ;
59+ } else {
60+ ioc . serviceManager . addSingletonInstance (
61+ IEnvironmentActivationService ,
62+ mockEnvironmentActivationService . object ,
63+ ) ;
64+ }
5665 return initializeTest ( ) ;
5766 } ) ;
5867 suiteTeardown ( closeActiveWindows ) ;
0 commit comments