Spaces:
Sleeping
Sleeping
File size: 2,494 Bytes
2a1c46d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 | "use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.TestingInjector = void 0;
const constants_1 = require("@nestjs/core/injector/constants");
const injector_1 = require("@nestjs/core/injector/injector");
const instance_wrapper_1 = require("@nestjs/core/injector/instance-wrapper");
/**
* @publicApi
*/
class TestingInjector extends injector_1.Injector {
setMocker(mocker) {
this.mocker = mocker;
}
setContainer(container) {
this.container = container;
}
async resolveComponentWrapper(moduleRef, name, dependencyContext, wrapper, resolutionContext = { contextId: constants_1.STATIC_CONTEXT }, keyOrIndex) {
try {
const existingProviderWrapper = await super.resolveComponentWrapper(moduleRef, name, dependencyContext, wrapper, resolutionContext, keyOrIndex);
return existingProviderWrapper;
}
catch (err) {
return this.mockWrapper(err, moduleRef, name, wrapper);
}
}
async resolveComponentHost(moduleRef, instanceWrapper, resolutionContext = { contextId: constants_1.STATIC_CONTEXT }) {
try {
const existingProviderWrapper = await super.resolveComponentHost(moduleRef, instanceWrapper, resolutionContext);
return existingProviderWrapper;
}
catch (err) {
return this.mockWrapper(err, moduleRef, instanceWrapper.name, instanceWrapper);
}
}
async mockWrapper(err, moduleRef, name, wrapper) {
if (!this.mocker) {
throw err;
}
const mockedInstance = this.mocker(name);
if (!mockedInstance) {
throw err;
}
const newWrapper = new instance_wrapper_1.InstanceWrapper({
name,
isAlias: false,
scope: wrapper.scope,
instance: mockedInstance,
isResolved: true,
host: moduleRef,
metatype: wrapper.metatype,
});
const internalCoreModule = this.container.getInternalCoreModuleRef();
if (!internalCoreModule) {
throw new Error('Expected to have internal core module reference at this point.');
}
internalCoreModule.addCustomProvider({
provide: name,
useValue: mockedInstance,
}, internalCoreModule.providers);
internalCoreModule.addExportedProviderOrModule(name);
return newWrapper;
}
}
exports.TestingInjector = TestingInjector;
|