fix(wrap): get the original schema correctly

ardatan · Nov 16, 2022 · e3f81ea · e3f81ea
1 parent 23664a2
commit e3f81ea
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/.changeset/heavy-lobsters-argue.md b/.changeset/heavy-lobsters-argue.md
@@ -0,0 +1,5 @@
+---
+'@graphql-tools/wrap': patch
+---
+
+Select the original schema correctly
diff --git a/packages/wrap/src/transforms/MapLeafValues.ts b/packages/wrap/src/transforms/MapLeafValues.ts
@@ -22,7 +22,7 @@ import {
   transformInputValue,
 } from '@graphql-tools/utils';
 
-import { Transform, DelegationContext, SubschemaConfig, Subschema } from '@graphql-tools/delegate';
+import { Transform, DelegationContext, SubschemaConfig } from '@graphql-tools/delegate';
 
 import { LeafValueTransformer } from '../types.js';
 
@@ -107,12 +107,9 @@ export default class MapLeafValues<TContext = Record<string, any>>
     delegationContext: DelegationContext<TContext>,
     transformationContext: MapLeafValuesTransformationContext
   ): ExecutionResult {
-    return visitResult(
-      originalResult,
-      transformationContext.transformedRequest,
-      (delegationContext.subschema as Subschema<any, any, any, TContext>).schema,
-      this.resultVisitorMap
-    );
+    const schema =
+      'schema' in delegationContext.subschema ? delegationContext.subschema.schema : delegationContext.subschema;
+    return visitResult(originalResult, transformationContext.transformedRequest, schema, this.resultVisitorMap);
   }
 
   private transformOperations(