Add test case for getDiscriminatorTCs()

graphql-compose · May 26, 2021 · 425fbb4 · 425fbb4
1 parent 16a5140
commit 425fbb4
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git 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
@@ -218,7 +218,7 @@ export class EDiscriminatorTypeComposer<TSource, TContext> extends ObjectTypeCom
       mongooseResolvers: GenerateResolverType<any, TContext>;
     };
   } {
-    // 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 {