(fix) - respect alternative rootkeys in query (#87)

urql-graphql · Sep 24, 2019 · 9a51cbb · 9a51cbb
1 parent 0c2229a
commit 9a51cbb
Show file tree

Hide file tree

Showing 3 changed files with 1,109 additions and 2 deletions.
diff --git a/src/operations/query.test.ts b/src/operations/query.test.ts
@@ -22,11 +22,12 @@ const TODO_QUERY = gql`
 `;
 
 describe('Query', () => {
-  let schema, store;
+  let schema, store, alteredRoot;
   const spy: { console?: any } = {};
 
   beforeAll(() => {
     schema = require('../test-utils/simple_schema.json');
+    alteredRoot = require('../test-utils/altered_root_schema.json');
   });
 
   afterEach(() => {
@@ -140,4 +141,35 @@ describe('Query', () => {
       todos: [{ __typename: 'Todo', id: '0', text: 'Solve bug' }],
     });
   });
+
+  it('should respect altered root types', () => {
+    const QUERY = gql`
+      query getTodos {
+        todos {
+          id
+          text
+        }
+      }
+    `;
+
+    const store = new Store(new SchemaPredicates(alteredRoot));
+
+    let { data } = query(store, { query: QUERY });
+    expect(data).toEqual(null);
+
+    write(
+      store,
+      { query: QUERY },
+      {
+        todos: [{ __typename: 'Todo', id: '0', text: 'Solve bug' }],
+        __typename: 'query_root',
+      }
+    );
+
+    ({ data } = query(store, { query: QUERY }));
+    expect(data).toEqual({
+      __typename: 'query_root',
+      todos: [{ __typename: 'Todo', id: '0', text: 'Solve bug' }],
+    });
+  });
 });
diff --git a/src/operations/query.ts b/src/operations/query.ts
@@ -79,7 +79,7 @@ export const read = (
 
   let data = input || Object.create(null);
   data =
-    rootKey !== 'Query'
+    rootKey !== ctx.store.getRootKey('query')
       ? readRoot(ctx, rootKey, rootSelect, data)
       : readSelection(ctx, rootKey, rootSelect, data);