diff --git a/src/enhancedDiscriminators/__tests__/eDiscriminatorTypeComposer-test.ts b/src/enhancedDiscriminators/__tests__/eDiscriminatorTypeComposer-test.ts index bdcff830..82917b37 100644 --- a/src/enhancedDiscriminators/__tests__/eDiscriminatorTypeComposer-test.ts +++ b/src/enhancedDiscriminators/__tests__/eDiscriminatorTypeComposer-test.ts @@ -122,6 +122,20 @@ describe('EDiscriminatorTypeComposer', () => { }); }); + describe('Get Discriminator TCs', () => { + it('returns discrimTCs with mongooseResolvers present', () => { + Object.values(baseDTC.getDiscriminatorTCs()).forEach((discimTC) => { + expect(discimTC).toHaveProperty('mongooseResolvers'); + }); + }); + it('returns empty object with mongooseResolvers missing', () => { + (baseDTC.discrimTCs[Object.keys(baseDTC.discrimTCs)[0]] as any).mongooseResolvers = undefined; + Object.values(baseDTC.getDiscriminatorTCs()).forEach((discimTC) => { + expect(discimTC).toHaveProperty('mongooseResolvers'); + }); + }); + }); + describe('Overridden eDTC Class Methods', () => { describe('Set Field', () => { it('updates field on all child TCs', () => { diff --git a/src/enhancedDiscriminators/eDiscriminatorTypeComposer.ts b/src/enhancedDiscriminators/eDiscriminatorTypeComposer.ts index de743090..758abab5 100644 --- a/src/enhancedDiscriminators/eDiscriminatorTypeComposer.ts +++ b/src/enhancedDiscriminators/eDiscriminatorTypeComposer.ts @@ -218,7 +218,7 @@ export class EDiscriminatorTypeComposer extends ObjectTypeCom mongooseResolvers: GenerateResolverType; }; } { - // check if mongooseResolvers are present + // check if mongooseResolvers are present (assume on one = on all) if ((this.discrimTCs[Object.keys(this.discrimTCs)[0]] as any).mongooseResolvers) { return this.discrimTCs as any; } else {