Skip to content

Commit 425fbb4

Browse files
committed
Add test case for getDiscriminatorTCs()
1 parent 16a5140 commit 425fbb4

File tree

2 files changed

+15
-1
lines changed

2 files changed

+15
-1
lines changed

src/enhancedDiscriminators/__tests__/eDiscriminatorTypeComposer-test.ts

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,20 @@ describe('EDiscriminatorTypeComposer', () => {
122122
});
123123
});
124124

125+
describe('Get Discriminator TCs', () => {
126+
it('returns discrimTCs with mongooseResolvers present', () => {
127+
Object.values(baseDTC.getDiscriminatorTCs()).forEach((discimTC) => {
128+
expect(discimTC).toHaveProperty('mongooseResolvers');
129+
});
130+
});
131+
it('returns empty object with mongooseResolvers missing', () => {
132+
(baseDTC.discrimTCs[Object.keys(baseDTC.discrimTCs)[0]] as any).mongooseResolvers = undefined;
133+
Object.values(baseDTC.getDiscriminatorTCs()).forEach((discimTC) => {
134+
expect(discimTC).toHaveProperty('mongooseResolvers');
135+
});
136+
});
137+
});
138+
125139
describe('Overridden eDTC Class Methods', () => {
126140
describe('Set Field', () => {
127141
it('updates field on all child TCs', () => {

src/enhancedDiscriminators/eDiscriminatorTypeComposer.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -218,7 +218,7 @@ export class EDiscriminatorTypeComposer<TSource, TContext> extends ObjectTypeCom
218218
mongooseResolvers: GenerateResolverType<any, TContext>;
219219
};
220220
} {
221-
// check if mongooseResolvers are present
221+
// check if mongooseResolvers are present (assume on one = on all)
222222
if ((this.discrimTCs[Object.keys(this.discrimTCs)[0]] as any).mongooseResolvers) {
223223
return this.discrimTCs as any;
224224
} else {

0 commit comments

Comments
 (0)