@@ -466,31 +466,45 @@ describe('DependenciesScanner', () => {
466
466
scope : Scope . REQUEST ,
467
467
} ;
468
468
469
- it ( 'should add enhancers metadata to every controller' , ( ) => {
469
+ it ( 'should add enhancers metadata to every controller and every entry provider ' , ( ) => {
470
470
( scanner as any ) . applicationProvidersApplyMap = [ provider ] ;
471
471
472
472
const instance = new InstanceWrapper ( { name : 'test' } ) ;
473
473
const controllers = new Map ( ) ;
474
- const mockController = new InstanceWrapper ( ) ;
474
+ const providers = new Map ( ) ;
475
+
476
+ const fakeController = new InstanceWrapper ( ) ;
477
+ const fakeProvider = new InstanceWrapper ( ) ;
478
+
479
+ const providerToken = 'entryProvider' ;
480
+ controllers . set ( 'test' , fakeController ) ;
481
+ providers . set ( providerToken , fakeProvider ) ;
475
482
476
- controllers . set ( 'test' , mockController ) ;
477
483
mockContainer . expects ( 'getModules' ) . callsFake ( ( ) => ( {
478
484
get : ( ) => ( {
479
485
injectables : { get : ( ) => instance } ,
480
486
controllers,
487
+ entryProviders : Array . from ( providers . values ( ) ) ,
481
488
} ) ,
482
489
values ( ) {
483
490
return [ this . get ( ) ] ;
484
491
} ,
485
492
} ) ) ;
486
- const addEnhancerMetadataSpy = sinon . spy (
487
- mockController ,
493
+
494
+ const addEnhancerMetadataControllerSpy = sinon . spy (
495
+ fakeController ,
496
+ 'addEnhancerMetadata' ,
497
+ ) ;
498
+ const addEnhancerMetadataProviderSpy = sinon . spy (
499
+ fakeProvider ,
488
500
'addEnhancerMetadata' ,
489
501
) ;
490
502
scanner . addScopedEnhancersMetadata ( ) ;
491
503
492
- expect ( addEnhancerMetadataSpy . called ) . to . be . true ;
493
- expect ( addEnhancerMetadataSpy . calledWith ( instance ) ) . to . be . true ;
504
+ expect ( addEnhancerMetadataControllerSpy . called ) . to . be . true ;
505
+ expect ( addEnhancerMetadataControllerSpy . calledWith ( instance ) ) . to . be . true ;
506
+ expect ( addEnhancerMetadataProviderSpy . called ) . to . be . true ;
507
+ expect ( addEnhancerMetadataProviderSpy . calledWith ( instance ) ) . to . be . true ;
494
508
} ) ;
495
509
} ) ;
496
510
0 commit comments