Fix namespaces module

AEB-labs · Aug 17, 2019 · fdc9d4e · fdc9d4e
1 parent e3e0d6e
commit fdc9d4e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/pipeline/namespaces.ts b/src/pipeline/namespaces.ts
@@ -2,7 +2,7 @@ import { PipelineModule } from './pipeline-module';
 import {
     DocumentNode, FragmentDefinitionNode, GraphQLObjectType, GraphQLSchema, SelectionSetNode, visit
 } from 'graphql';
-import { maybeDo } from '../utils/utils';
+import { maybeDo, objectValues } from '../utils/utils';
 import { Query } from '../graphql/common';
 
 /**
@@ -22,7 +22,8 @@ export class NamespaceModule implements PipelineModule {
             directives: Array.from(schema.getDirectives()),
             query: maybeDo(schema.getQueryType(), type => this.wrap(type, 'Query')),
             mutation: maybeDo(schema.getMutationType(), type => this.wrap(type, 'Mutation')),
-            subscription: maybeDo(schema.getSubscriptionType(), type => this.wrap(type, 'Subscription'))
+            subscription: maybeDo(schema.getSubscriptionType(), type => this.wrap(type, 'Subscription')),
+            types: objectValues(schema.getTypeMap())
         });
         this.schema = newSchema;
         return newSchema;