@apollo/gateway 0.300.0-alpha.2 → 2.0.0-alpha.2

package/src/__tests__/buildQueryPlan.test.ts

@@ -1,43 +1,45 @@
-import { GraphQLSchema, GraphQLError } from 'graphql';
-import gql from 'graphql-tag';
-import { buildQueryPlan, buildOperationContext } from '../buildQueryPlan';
-import { astSerializer, queryPlanSerializer } from '../snapshotSerializers';
+import { astSerializer, queryPlanSerializer } from 'apollo-federation-integration-testsuite';
 import { getFederatedTestingSchema } from './execution-utils';
+import { QueryPlan, QueryPlanner } from '@apollo/query-planner';
+import { Schema, parseOperation } from '@apollo/federation-internals';
 
 expect.addSnapshotSerializer(astSerializer);
 expect.addSnapshotSerializer(queryPlanSerializer);
 
+
 describe('buildQueryPlan', () => {
-  let schema: GraphQLSchema;
-  let errors: GraphQLError[];
+  let schema: Schema;
+  let queryPlanner: QueryPlanner;
+
+  const buildPlan = (operation: string): QueryPlan => {
+    return queryPlanner.buildQueryPlan(parseOperation(schema, operation));
+  }
 
   beforeEach(() => {
-    ({ schema, errors } = getFederatedTestingSchema());
-    expect(errors).toHaveLength(0);
+    expect(
+      () => ({ schema, queryPlanner } = getFederatedTestingSchema()),
+    ).not.toThrow();
   });
 
   it(`should not confuse union types with overlapping field names`, () => {
-    const query = gql`
+    const operationString = `#graphql
       query {
-        body {
-          ... on Image {
-            attributes {
-              url
+          body {
+            ... on Image {
+              attributes {
+                url
+              }
             }
-          }
-          ... on Text {
-            attributes {
-              bold
-              text
+            ... on Text {
+              attributes {
+                bold
+                text
+              }
             }
           }
         }
-      }
     `;
-
-    const queryPlan = buildQueryPlan(
-      buildOperationContext(schema, query, undefined),
-    );
+    const queryPlan = buildPlan(operationString);
 
     expect(queryPlan).toMatchInlineSnapshot(`
       QueryPlan {
@@ -64,22 +66,25 @@ describe('buildQueryPlan', () => {
   });
 
   it(`should use a single fetch when requesting a root field from one service`, () => {
-    const query = gql`
+    const operationString = `#graphql
       query {
         me {
-          name
+          name {
+            first
+          }
         }
       }
     `;
 
-    const queryPlan = buildQueryPlan(buildOperationContext(schema, query));
-
+    const queryPlan = buildPlan(operationString);
     expect(queryPlan).toMatchInlineSnapshot(`
       QueryPlan {
         Fetch(service: "accounts") {
           {
             me {
-              name
+              name {
+                first
+              }
             }
           }
         },
@@ -88,10 +93,12 @@ describe('buildQueryPlan', () => {
   });
 
   it(`should use two independent fetches when requesting root fields from two services`, () => {
-    const query = gql`
+    const operationString = `#graphql
       query {
         me {
-          name
+          name {
+            first
+          }
         }
         topProducts {
           name
@@ -99,7 +106,7 @@ describe('buildQueryPlan', () => {
       }
     `;
 
-    const queryPlan = buildQueryPlan(buildOperationContext(schema, query));
+    const queryPlan = buildPlan(operationString);
 
     expect(queryPlan).toMatchInlineSnapshot(`
       QueryPlan {
@@ -107,7 +114,9 @@ describe('buildQueryPlan', () => {
           Fetch(service: "accounts") {
             {
               me {
-                name
+                name {
+                  first
+                }
               }
             }
           },
@@ -136,8 +145,6 @@ describe('buildQueryPlan', () => {
                 } =>
                 {
                   ... on Book {
-                    __typename
-                    isbn
                     title
                     year
                   }
@@ -149,9 +156,9 @@ describe('buildQueryPlan', () => {
                 {
                   ... on Book {
                     __typename
-                    isbn
                     title
                     year
+                    isbn
                   }
                 } =>
                 {
@@ -168,7 +175,7 @@ describe('buildQueryPlan', () => {
   });
 
   it(`should use a single fetch when requesting multiple root fields from the same service`, () => {
-    const query = gql`
+    const operationString = `#graphql
       query {
         topProducts {
           name
@@ -179,7 +186,7 @@ describe('buildQueryPlan', () => {
       }
     `;
 
-    const queryPlan = buildQueryPlan(buildOperationContext(schema, query));
+    const queryPlan = buildPlan(operationString);
 
     expect(queryPlan).toMatchInlineSnapshot(`
       QueryPlan {
@@ -220,8 +227,6 @@ describe('buildQueryPlan', () => {
                   } =>
                   {
                     ... on Book {
-                      __typename
-                      isbn
                       title
                       year
                     }
@@ -233,9 +238,9 @@ describe('buildQueryPlan', () => {
                   {
                     ... on Book {
                       __typename
-                      isbn
                       title
                       year
+                      isbn
                     }
                   } =>
                   {
@@ -257,8 +262,6 @@ describe('buildQueryPlan', () => {
                   } =>
                   {
                     ... on Book {
-                      __typename
-                      isbn
                       title
                       year
                     }
@@ -270,9 +273,9 @@ describe('buildQueryPlan', () => {
                   {
                     ... on Book {
                       __typename
-                      isbn
                       title
                       year
+                      isbn
                     }
                   } =>
                   {
@@ -290,7 +293,7 @@ describe('buildQueryPlan', () => {
   });
 
   it(`should use a single fetch when requesting relationship subfields from the same service`, () => {
-    const query = gql`
+    const operationString = `#graphql
       query {
         topReviews {
           body
@@ -303,7 +306,7 @@ describe('buildQueryPlan', () => {
       }
     `;
 
-    const queryPlan = buildQueryPlan(buildOperationContext(schema, query));
+    const queryPlan = buildPlan(operationString);
 
     expect(queryPlan).toMatchInlineSnapshot(`
       QueryPlan {
@@ -324,7 +327,7 @@ describe('buildQueryPlan', () => {
   });
 
   it(`should use a single fetch when requesting relationship subfields and provided keys from the same service`, () => {
-    const query = gql`
+    const operationString = `#graphql
       query {
         topReviews {
           body
@@ -338,7 +341,7 @@ describe('buildQueryPlan', () => {
       }
     `;
 
-    const queryPlan = buildQueryPlan(buildOperationContext(schema, query));
+    const queryPlan = buildPlan(operationString);
 
     expect(queryPlan).toMatchInlineSnapshot(`
       QueryPlan {
@@ -361,10 +364,12 @@ describe('buildQueryPlan', () => {
 
   describe(`when requesting an extension field from another service`, () => {
     it(`should add the field's representation requirements to the parent selection set and use a dependent fetch`, () => {
-      const query = gql`
+      const operationString = `#graphql
         query {
           me {
-            name
+            name {
+              first
+            }
             reviews {
               body
             }
@@ -372,7 +377,7 @@ describe('buildQueryPlan', () => {
         }
       `;
 
-      const queryPlan = buildQueryPlan(buildOperationContext(schema, query));
+      const queryPlan = buildPlan(operationString);
 
       expect(queryPlan).toMatchInlineSnapshot(`
         QueryPlan {
@@ -380,9 +385,11 @@ describe('buildQueryPlan', () => {
             Fetch(service: "accounts") {
               {
                 me {
-                  name
                   __typename
                   id
+                  name {
+                    first
+                  }
                 }
               }
             },
@@ -410,7 +417,7 @@ describe('buildQueryPlan', () => {
 
     describe(`when the parent selection set is empty`, () => {
       it(`should add the field's requirements to the parent selection set and use a dependent fetch`, () => {
-        const query = gql`
+        const operationString = `#graphql
           query {
             me {
               reviews {
@@ -420,7 +427,7 @@ describe('buildQueryPlan', () => {
           }
         `;
 
-        const queryPlan = buildQueryPlan(buildOperationContext(schema, query));
+        const queryPlan = buildPlan(operationString);
 
         expect(queryPlan).toMatchInlineSnapshot(`
           QueryPlan {
@@ -458,7 +465,7 @@ describe('buildQueryPlan', () => {
 
     // TODO: Ask martijn about the meaning of this test
     it(`should only add requirements once`, () => {
-      const query = gql`
+      const operationString = `#graphql
         query {
           me {
             reviews {
@@ -469,7 +476,7 @@ describe('buildQueryPlan', () => {
         }
       `;
 
-      const queryPlan = buildQueryPlan(buildOperationContext(schema, query));
+      const queryPlan = buildPlan(operationString);
 
       expect(queryPlan).toMatchInlineSnapshot(`
         QueryPlan {
@@ -508,18 +515,20 @@ describe('buildQueryPlan', () => {
 
   describe(`when requesting a composite field with subfields from another service`, () => {
     it(`should add key fields to the parent selection set and use a dependent fetch`, () => {
-      const query = gql`
+       const operationString = `#graphql
         query {
           topReviews {
             body
             author {
-              name
+              name {
+                first
+              }
             }
           }
         }
       `;
 
-      const queryPlan = buildQueryPlan(buildOperationContext(schema, query));
+      const queryPlan = buildPlan(operationString);
 
       expect(queryPlan).toMatchInlineSnapshot(`
         QueryPlan {
@@ -545,7 +554,9 @@ describe('buildQueryPlan', () => {
                 } =>
                 {
                   ... on User {
-                    name
+                    name {
+                      first
+                    }
                   }
                 }
               },
@@ -557,7 +568,7 @@ describe('buildQueryPlan', () => {
 
     describe(`when requesting a field defined in another service which requires a field in the base service`, () => {
       it(`should add the field provided by base service in first Fetch`, () => {
-        const query = gql`
+        const operationString = `#graphql
           query {
             topCars {
               retailPrice
@@ -565,7 +576,7 @@ describe('buildQueryPlan', () => {
           }
         `;
 
-        const queryPlan = buildQueryPlan(buildOperationContext(schema, query));
+        const queryPlan = buildPlan(operationString);
 
         expect(queryPlan).toMatchInlineSnapshot(`
           QueryPlan {
@@ -603,17 +614,19 @@ describe('buildQueryPlan', () => {
 
     describe(`when the parent selection set is empty`, () => {
       it(`should add key fields to the parent selection set and use a dependent fetch`, () => {
-        const query = gql`
+        const operationString = `#graphql
           query {
             topReviews {
               author {
-                name
+                name {
+                  first
+                }
               }
             }
           }
         `;
 
-        const queryPlan = buildQueryPlan(buildOperationContext(schema, query));
+        const queryPlan = buildPlan(operationString);
 
         expect(queryPlan).toMatchInlineSnapshot(`
           QueryPlan {
@@ -638,7 +651,9 @@ describe('buildQueryPlan', () => {
                   } =>
                   {
                     ... on User {
-                      name
+                      name {
+                        first
+                      }
                     }
                   }
                 },
@@ -651,7 +666,7 @@ describe('buildQueryPlan', () => {
   });
   describe(`when requesting a relationship field with extension subfields from a different service`, () => {
     it(`should first fetch the object using a key from the base service and then pass through the requirements`, () => {
-      const query = gql`
+      const operationString = `#graphql
         query {
           topReviews {
             author {
@@ -661,7 +676,7 @@ describe('buildQueryPlan', () => {
         }
       `;
 
-      const queryPlan = buildQueryPlan(buildOperationContext(schema, query));
+      const queryPlan = buildPlan(operationString);
 
       expect(queryPlan).toMatchInlineSnapshot(`
         QueryPlan {
@@ -702,7 +717,7 @@ describe('buildQueryPlan', () => {
     // Probably an issue with extending / interfaces in composition. None of the fields from the base Book type
     // are showing up in the resulting schema.
     it(`should add __typename when fetching objects of an interface type from a service`, () => {
-      const query = gql`
+      const operationString = `#graphql
         query {
           topProducts {
             price
@@ -710,7 +725,7 @@ describe('buildQueryPlan', () => {
         }
       `;
 
-      const queryPlan = buildQueryPlan(buildOperationContext(schema, query));
+      const queryPlan = buildPlan(operationString);
 
       expect(queryPlan).toMatchInlineSnapshot(`
         QueryPlan {
@@ -718,25 +733,164 @@ describe('buildQueryPlan', () => {
             {
               topProducts {
                 __typename
-                ... on Book {
-                  price
-                }
-                ... on Furniture {
-                  price
-                }
+                price
               }
             }
           },
         }
       `);
     });
+
+    it(`should not get confused by a fragment spread multiple times`, () => {
+      const operationString = `#graphql
+        fragment Price on Product {
+          price
+        }
+
+        query {
+          topProducts {
+            __typename
+            ... on Book {
+              ...Price
+            }
+            ... on Furniture {
+              ...Price
+            }
+          }
+        }
+      `;
+
+      const queryPlan = buildPlan(operationString);
+
+      expect(queryPlan).toMatchInlineSnapshot(`
+                QueryPlan {
+                  Fetch(service: "product") {
+                    {
+                      topProducts {
+                        __typename
+                        ... on Book {
+                          price
+                        }
+                        ... on Furniture {
+                          price
+                        }
+                      }
+                    }
+                  },
+                }
+            `);
+    });
+
+    it(`should not get confused by an inline fragment multiple times`, () => {
+      const operationString = `#graphql
+        query {
+          topProducts {
+            __typename
+            ... on Book {
+              ...on Product {
+                price
+              }
+            }
+            ... on Furniture {
+              ... on Product {
+                price
+              }
+            }
+          }
+        }
+      `;
+
+      const queryPlan = buildPlan(operationString);
+
+      expect(queryPlan).toMatchInlineSnapshot(`
+                QueryPlan {
+                  Fetch(service: "product") {
+                    {
+                      topProducts {
+                        __typename
+                        ... on Book {
+                          price
+                        }
+                        ... on Furniture {
+                          price
+                        }
+                      }
+                    }
+                  },
+                }
+            `);
+    });
+
+    it(`eliminate unecessary type conditions`, () => {
+      const operationString = `#graphql
+        query {
+          body {
+            ... on Image {
+              ... on NamedObject {
+                name
+              }
+            }
+          }
+        }
+      `;
+
+      const queryPlan = buildPlan(operationString);
+
+      expect(queryPlan).toMatchInlineSnapshot(`
+                QueryPlan {
+                  Fetch(service: "documents") {
+                    {
+                      body {
+                        __typename
+                        ... on Image {
+                          name
+                        }
+                      }
+                    }
+                  },
+                }
+            `);
+    });
+
+    it(`should preserve directives on inline fragments even if the fragment is otherwise useless`, () => {
+      const operationString = `#graphql
+        query myQuery($b: Boolean!) {
+          body {
+            ... on Image {
+              ... on NamedObject @include(if: $b) {
+                name
+              }
+            }
+          }
+        }
+      `;
+
+      const queryPlan = buildPlan(operationString);
+
+      expect(queryPlan).toMatchInlineSnapshot(`
+                QueryPlan {
+                  Fetch(service: "documents") {
+                    {
+                      body {
+                        __typename
+                        ... on Image {
+                          ... on NamedObject @include(if: $b) {
+                            name
+                          }
+                        }
+                      }
+                    }
+                  },
+                }
+            `);
+    });
   });
 
   // GraphQLError: Cannot query field "isbn" on type "Book"
   // Probably an issue with extending / interfaces in composition. None of the fields from the base Book type
   // are showing up in the resulting schema.
   it(`should break up when traversing an extension field on an interface type from a service`, () => {
-    const query = gql`
+    const operationString = `#graphql
       query {
         topProducts {
           price
@@ -747,7 +901,7 @@ describe('buildQueryPlan', () => {
       }
     `;
 
-    const queryPlan = buildQueryPlan(buildOperationContext(schema, query));
+    const queryPlan = buildPlan(operationString);
 
     expect(queryPlan).toMatchInlineSnapshot(`
       QueryPlan {
@@ -756,13 +910,12 @@ describe('buildQueryPlan', () => {
             {
               topProducts {
                 __typename
+                price
                 ... on Book {
-                  price
                   __typename
                   isbn
                 }
                 ... on Furniture {
-                  price
                   __typename
                   upc
                 }
@@ -801,7 +954,7 @@ describe('buildQueryPlan', () => {
   });
 
   it(`interface fragments should expand into possible types only`, () => {
-    const query = gql`
+    const operationString = `#graphql
       query {
         books {
           ... on Product {
@@ -814,7 +967,7 @@ describe('buildQueryPlan', () => {
       }
     `;
 
-    const queryPlan = buildQueryPlan(buildOperationContext(schema, query));
+    const queryPlan = buildPlan(operationString);
 
     expect(queryPlan).toMatchInlineSnapshot(`
       QueryPlan {
@@ -851,8 +1004,8 @@ describe('buildQueryPlan', () => {
     `);
   });
 
-  it(`interface inside interface should expand into possible types only`, () => {
-    const query = gql`
+  it(`interface inside interface should not type explode if possible`, () => {
+    const operationString = `#graphql
       query {
         product(upc: "") {
           details {
@@ -862,7 +1015,7 @@ describe('buildQueryPlan', () => {
       }
     `;
 
-    const queryPlan = buildQueryPlan(buildOperationContext(schema, query));
+    const queryPlan = buildPlan(operationString);
 
     expect(queryPlan).toMatchInlineSnapshot(`
       QueryPlan {
@@ -870,15 +1023,9 @@ describe('buildQueryPlan', () => {
           {
             product(upc: "") {
               __typename
-              ... on Book {
-                details {
-                  country
-                }
-              }
-              ... on Furniture {
-                details {
-                  country
-                }
+              details {
+                __typename
+                country
               }
             }
           }
@@ -887,319 +1034,8 @@ describe('buildQueryPlan', () => {
     `);
   });
 
-  describe(`experimental compression to downstream services`, () => {
-    it(`should generate fragments internally to downstream requests`, () => {
-      const query = gql`
-        query {
-          topReviews {
-            body
-            author
-            product {
-              name
-              price
-              details {
-                country
-              }
-            }
-          }
-        }
-      `;
-
-      const queryPlan = buildQueryPlan(
-        buildOperationContext(schema, query, undefined),
-        { autoFragmentization: true },
-      );
-
-      expect(queryPlan).toMatchInlineSnapshot(`
-        QueryPlan {
-          Sequence {
-            Fetch(service: "reviews") {
-              {
-                topReviews {
-                  ...__QueryPlanFragment_1__
-                }
-              }
-              fragment __QueryPlanFragment_1__ on Review {
-                body
-                author
-                product {
-                  ...__QueryPlanFragment_0__
-                }
-              }
-              fragment __QueryPlanFragment_0__ on Product {
-                __typename
-                ... on Book {
-                  __typename
-                  isbn
-                }
-                ... on Furniture {
-                  __typename
-                  upc
-                }
-              }
-            },
-            Parallel {
-              Sequence {
-                Flatten(path: "topReviews.@.product") {
-                  Fetch(service: "books") {
-                    {
-                      ... on Book {
-                        __typename
-                        isbn
-                      }
-                    } =>
-                    {
-                      ... on Book {
-                        __typename
-                        isbn
-                        title
-                        year
-                      }
-                    }
-                  },
-                },
-                Flatten(path: "topReviews.@.product") {
-                  Fetch(service: "product") {
-                    {
-                      ... on Book {
-                        __typename
-                        isbn
-                        title
-                        year
-                      }
-                    } =>
-                    {
-                      ... on Book {
-                        name
-                      }
-                    }
-                  },
-                },
-              },
-              Flatten(path: "topReviews.@.product") {
-                Fetch(service: "product") {
-                  {
-                    ... on Furniture {
-                      __typename
-                      upc
-                    }
-                    ... on Book {
-                      __typename
-                      isbn
-                    }
-                  } =>
-                  {
-                    ... on Furniture {
-                      name
-                      price
-                      details {
-                        country
-                      }
-                    }
-                    ... on Book {
-                      price
-                      details {
-                        country
-                      }
-                    }
-                  }
-                },
-              },
-            },
-          },
-        }
-      `);
-    });
-
-    it(`shouldn't generate fragments for selection sets of length 2 or less`, () => {
-      const query = gql`
-        query {
-          topReviews {
-            body
-            author
-          }
-        }
-      `;
-
-      const queryPlan = buildQueryPlan(
-        buildOperationContext(schema, query, undefined),
-        { autoFragmentization: true },
-      );
-
-      expect(queryPlan).toMatchInlineSnapshot(`
-        QueryPlan {
-          Fetch(service: "reviews") {
-            {
-              topReviews {
-                body
-                author
-              }
-            }
-          },
-        }
-      `);
-    });
-
-    it(`should generate fragments for selection sets of length 3 or greater`, () => {
-      const query = gql`
-        query {
-          topReviews {
-            id
-            body
-            author
-          }
-        }
-      `;
-
-      const queryPlan = buildQueryPlan(
-        buildOperationContext(schema, query, undefined),
-        { autoFragmentization: true },
-      );
-
-      expect(queryPlan).toMatchInlineSnapshot(`
-        QueryPlan {
-          Fetch(service: "reviews") {
-            {
-              topReviews {
-                ...__QueryPlanFragment_0__
-              }
-            }
-            fragment __QueryPlanFragment_0__ on Review {
-              id
-              body
-              author
-            }
-          },
-        }
-      `);
-    });
-
-    it(`should generate fragments correctly when aliases are used`, () => {
-      const query = gql`
-        query {
-          reviews: topReviews {
-            content: body
-            author
-            product {
-              name
-              cost: price
-              details {
-                origin: country
-              }
-            }
-          }
-        }
-      `;
-
-      const queryPlan = buildQueryPlan(
-        buildOperationContext(schema, query, undefined),
-        { autoFragmentization: true },
-      );
-
-      expect(queryPlan).toMatchInlineSnapshot(`
-        QueryPlan {
-          Sequence {
-            Fetch(service: "reviews") {
-              {
-                reviews: topReviews {
-                  ...__QueryPlanFragment_1__
-                }
-              }
-              fragment __QueryPlanFragment_1__ on Review {
-                content: body
-                author
-                product {
-                  ...__QueryPlanFragment_0__
-                }
-              }
-              fragment __QueryPlanFragment_0__ on Product {
-                __typename
-                ... on Book {
-                  __typename
-                  isbn
-                }
-                ... on Furniture {
-                  __typename
-                  upc
-                }
-              }
-            },
-            Parallel {
-              Sequence {
-                Flatten(path: "reviews.@.product") {
-                  Fetch(service: "books") {
-                    {
-                      ... on Book {
-                        __typename
-                        isbn
-                      }
-                    } =>
-                    {
-                      ... on Book {
-                        __typename
-                        isbn
-                        title
-                        year
-                      }
-                    }
-                  },
-                },
-                Flatten(path: "reviews.@.product") {
-                  Fetch(service: "product") {
-                    {
-                      ... on Book {
-                        __typename
-                        isbn
-                        title
-                        year
-                      }
-                    } =>
-                    {
-                      ... on Book {
-                        name
-                      }
-                    }
-                  },
-                },
-              },
-              Flatten(path: "reviews.@.product") {
-                Fetch(service: "product") {
-                  {
-                    ... on Furniture {
-                      __typename
-                      upc
-                    }
-                    ... on Book {
-                      __typename
-                      isbn
-                    }
-                  } =>
-                  {
-                    ... on Furniture {
-                      name
-                      cost: price
-                      details {
-                        origin: country
-                      }
-                    }
-                    ... on Book {
-                      cost: price
-                      details {
-                        origin: country
-                      }
-                    }
-                  }
-                },
-              },
-            },
-          },
-        }
-      `);
-    });
-  });
-
   it(`should properly expand nested unions with inline fragments`, () => {
-    const query = gql`
+    const operationString = `#graphql
       query {
         body {
           ... on Image {
@@ -1226,9 +1062,7 @@ describe('buildQueryPlan', () => {
       }
     `;
 
-    const queryPlan = buildQueryPlan(
-      buildOperationContext(schema, query, undefined),
-    );
+    const queryPlan = buildPlan(operationString);
 
     expect(queryPlan).toMatchInlineSnapshot(`
       QueryPlan {
@@ -1255,16 +1089,9 @@ describe('buildQueryPlan', () => {
 
   describe('deduplicates fields / selections regardless of adjacency and type condition nesting', () => {
     it('for inline fragments', () => {
-      const query = gql`
+      const operationString = `#graphql
         query {
           body {
-            ... on Image {
-              ... on Text {
-                attributes {
-                  bold
-                }
-              }
-            }
             ... on Body {
               ... on Text {
                 attributes {
@@ -1283,9 +1110,7 @@ describe('buildQueryPlan', () => {
         }
       `;
 
-      const queryPlan = buildQueryPlan(
-        buildOperationContext(schema, query, undefined),
-      );
+      const queryPlan = buildPlan(operationString);
 
       expect(queryPlan).toMatchInlineSnapshot(`
         QueryPlan {
@@ -1307,7 +1132,7 @@ describe('buildQueryPlan', () => {
     });
 
     it('for named fragment spreads', () => {
-      const query = gql`
+      const operationString = `#graphql
         fragment TextFragment on Text {
           attributes {
             bold
@@ -1317,9 +1142,6 @@ describe('buildQueryPlan', () => {
 
         query {
           body {
-            ... on Image {
-              ...TextFragment
-            }
             ... on Body {
               ...TextFragment
             }
@@ -1328,9 +1150,7 @@ describe('buildQueryPlan', () => {
         }
       `;
 
-      const queryPlan = buildQueryPlan(
-        buildOperationContext(schema, query, undefined),
-      );
+      const queryPlan = buildPlan(operationString);
 
       expect(queryPlan).toMatchInlineSnapshot(`
         QueryPlan {