Skip to content

Commit acab5ae

Browse files
fix(core): fix unit tests for middleware module
1 parent 539e281 commit acab5ae

1 file changed

Lines changed: 18 additions & 7 deletions

File tree

packages/core/test/middleware/middleware-module.spec.ts

Lines changed: 18 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)