@@ -79,9 +79,9 @@ suite('TerminalEnvVarInjector - Core Functionality', () => {
7979 envVarCollection . setup ( ( x ) => x . getScoped ( typeMoq . It . isAny ( ) ) ) . returns ( ( ) => mockScopedCollection ) ;
8080 envVarCollection . setup ( ( x ) => x . clear ( ) ) . returns ( ( ) => { } ) ;
8181
82- // Setup minimal mocks for event subscriptions - return a function matching Event signature
82+ // Setup minimal mocks for event subscriptions - return disposable when handler is registered
8383 const mockDisposable : Disposable = { dispose : ( ) => { } } ;
84- envVarManager . setup ( ( m ) => m . onDidChangeEnvironmentVariables ) . returns ( ( ) => ( ) => mockDisposable ) ;
84+ envVarManager . setup ( ( m ) => m . onDidChangeEnvironmentVariables ( typeMoq . It . isAny ( ) ) ) . returns ( ( ) => mockDisposable ) ;
8585 } ) ;
8686
8787 teardown ( ( ) => {
@@ -316,18 +316,12 @@ suite('TerminalEnvVarInjector - Core Functionality', () => {
316316 } ) ;
317317
318318 test ( 'should register environment variable change event handler' , ( ) => {
319- // Arrange
320- let eventHandlerRegistered = false ;
321- envVarManager . reset ( ) ;
322- // Setup minimal mocks for event subscriptions - return a function matching Event signature
323- const mockDisposable : Disposable = { dispose : ( ) => { } } ;
324- envVarManager . setup ( ( m ) => m . onDidChangeEnvironmentVariables ) . returns ( ( ) => ( ) => mockDisposable ) ;
325-
319+ // Arrange - Use the global setup's mock configuration
326320 // Act
327321 injector = new TerminalEnvVarInjector ( envVarCollection . object , envVarManager . object ) ;
328322
329- // Assert
330- sinon . assert . match ( eventHandlerRegistered , true ) ;
323+ // Assert - Verify that the mock's setup was used (handler was registered)
324+ envVarManager . verify ( ( m ) => m . onDidChangeEnvironmentVariables ( typeMoq . It . isAny ( ) ) , typeMoq . Times . once ( ) ) ;
331325 } ) ;
332326 } ) ;
333327} ) ;
0 commit comments