fix: Interface & Union type composers on addTypeResolver now automa…

…tically add provided types to schema
graphql-compose · May 10, 2020 · b506bd5 · b506bd5
1 parent f07ebad
commit b506bd5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/InterfaceTypeComposer.js b/src/InterfaceTypeComposer.js
@@ -1092,6 +1092,7 @@ export class InterfaceTypeComposer<TSource, TContext> {
     const typeResolversMap = this.getTypeResolvers();
     typeResolversMap.set(type, checkFn);
     this.setTypeResolvers(typeResolversMap);
+    this.schemaComposer.addSchemaMustHaveType(type);
     return this;
   }
 

diff --git a/src/UnionTypeComposer.js b/src/UnionTypeComposer.js
@@ -621,6 +621,7 @@ export class UnionTypeComposer<TSource, TContext> {
     const typeResolversMap = this.getTypeResolvers();
     const tc = this._convertObjectType(type);
     typeResolversMap.set(tc, checkFn);
+    this.schemaComposer.addSchemaMustHaveType((tc: any));
     this.setTypeResolvers(typeResolversMap);
     return this;
   }