@@ -7,9 +7,15 @@ import * as TypeMoq from 'typemoq';
7
7
import { Disposable , Terminal , Uri } from 'vscode' ;
8
8
import { IActiveResourceService , ICommandManager , IWorkspaceService } from '../../client/common/application/types' ;
9
9
import { Commands } from '../../client/common/constants' ;
10
+ import { TerminalEnvVarActivation } from '../../client/common/experiments/groups' ;
10
11
import { TerminalService } from '../../client/common/terminal/service' ;
11
12
import { ITerminalActivator , ITerminalServiceFactory } from '../../client/common/terminal/types' ;
12
- import { IConfigurationService , IPythonSettings , ITerminalSettings } from '../../client/common/types' ;
13
+ import {
14
+ IConfigurationService ,
15
+ IExperimentService ,
16
+ IPythonSettings ,
17
+ ITerminalSettings ,
18
+ } from '../../client/common/types' ;
13
19
import { IServiceContainer } from '../../client/ioc/types' ;
14
20
import { TerminalProvider } from '../../client/providers/terminalProvider' ;
15
21
@@ -18,13 +24,17 @@ suite('Terminal Provider', () => {
18
24
let commandManager : TypeMoq . IMock < ICommandManager > ;
19
25
let workspace : TypeMoq . IMock < IWorkspaceService > ;
20
26
let activeResourceService : TypeMoq . IMock < IActiveResourceService > ;
27
+ let experimentService : TypeMoq . IMock < IExperimentService > ;
21
28
let terminalProvider : TerminalProvider ;
22
29
const resource = Uri . parse ( 'a' ) ;
23
30
setup ( ( ) => {
24
31
serviceContainer = TypeMoq . Mock . ofType < IServiceContainer > ( ) ;
25
32
commandManager = TypeMoq . Mock . ofType < ICommandManager > ( ) ;
33
+ experimentService = TypeMoq . Mock . ofType < IExperimentService > ( ) ;
34
+ experimentService . setup ( ( e ) => e . inExperimentSync ( TerminalEnvVarActivation . experiment ) ) . returns ( ( ) => false ) ;
26
35
activeResourceService = TypeMoq . Mock . ofType < IActiveResourceService > ( ) ;
27
36
workspace = TypeMoq . Mock . ofType < IWorkspaceService > ( ) ;
37
+ serviceContainer . setup ( ( c ) => c . get ( IExperimentService ) ) . returns ( ( ) => experimentService . object ) ;
28
38
serviceContainer . setup ( ( c ) => c . get ( ICommandManager ) ) . returns ( ( ) => commandManager . object ) ;
29
39
serviceContainer . setup ( ( c ) => c . get ( IWorkspaceService ) ) . returns ( ( ) => workspace . object ) ;
30
40
serviceContainer . setup ( ( c ) => c . get ( IActiveResourceService ) ) . returns ( ( ) => activeResourceService . object ) ;
0 commit comments