@@ -55,10 +55,15 @@ describe('MiddlewareModule', () => {
5555 configure : configureSpy ,
5656 } ;
5757
58+ const stubContainer = new NestContainer ( ) ;
59+ stubContainer
60+ . getModules ( )
61+ . set ( 'Test' , new Module ( class { } , [ ] , stubContainer ) ) ;
62+
5863 await middlewareModule . loadConfiguration (
59- new MiddlewareContainer ( new NestContainer ( ) ) ,
64+ new MiddlewareContainer ( stubContainer ) ,
6065 mockModule as any ,
61- 'Test' as any ,
66+ 'Test' ,
6267 ) ;
6368
6469 expect ( configureSpy . calledOnce ) . to . be . true ;
@@ -151,7 +156,13 @@ describe('MiddlewareModule', () => {
151156 const app = {
152157 createMiddlewareFactory : createMiddlewareFactoryStub ,
153158 } ;
154- const container = new MiddlewareContainer ( new NestContainer ( ) ) ;
159+
160+ const stubContainer = new NestContainer ( ) ;
161+ stubContainer
162+ . getModules ( )
163+ . set ( 'Test' , new Module ( TestModule , [ ] , stubContainer ) ) ;
164+
165+ const container = new MiddlewareContainer ( stubContainer ) ;
155166 const moduleKey = 'Test' ;
156167 container . insertConfig ( [ configuration ] , moduleKey ) ;
157168
@@ -164,17 +175,17 @@ describe('MiddlewareModule', () => {
164175 } ) ,
165176 ) ;
166177 sinon
167- . stub ( nestContainer , 'getModuleByKey' )
168- . callsFake ( ( ) => new Module ( class { } , [ ] , nestContainer ) ) ;
178+ . stub ( stubContainer , 'getModuleByKey' )
179+ . callsFake ( ( ) => new Module ( class { } , [ ] , stubContainer ) ) ;
169180 // tslint:disable-next-line:no-string-literal
170- middlewareModule [ 'container' ] = nestContainer ;
181+ middlewareModule [ 'container' ] = stubContainer ;
171182
172183 await middlewareModule . registerRouteMiddleware (
173184 container ,
174185 { path : route , method : RequestMethod . ALL } ,
175186 configuration ,
176187 moduleKey ,
177- app as any ,
188+ app ,
178189 ) ;
179190 expect ( createMiddlewareFactoryStub . calledOnce ) . to . be . true ;
180191 } ) ;
0 commit comments