Adds support for resolving a union type when using a generated schema

Author: jonbretman

src/utilities/__tests__/buildASTSchema-test.js

@@ -394,6 +394,137 @@ describe('Schema Builder', () => {
     expect(output).to.equal(body);
   });
 
+  it('Specifying Union type using __typename', async () => {
+    const schema = buildSchema(`
+      schema {
+        query: Root
+      }
+
+      type Root {
+        fruits: [Fruit]
+      }
+
+      union Fruit = Apple | Banana
+
+      type Apple {
+        color: String
+      }
+
+      type Banana {
+        length: Int
+      }
+    `);
+
+    const query = `
+      {
+        fruits {
+          ... on Apple {
+            color
+          }
+          ... on Banana {
+            length
+          }
+        }
+      }
+    `;
+
+    const root = {
+      fruits: [
+        {
+          color: 'green',
+          __typename: 'Apple',
+        },
+        {
+          length: 5,
+          __typename: 'Banana',
+        }
+      ]
+    };
+
+    expect(await graphql(schema, query, root)).to.deep.equal({
+      data: {
+        fruits: [
+          {
+            color: 'green',
+          },
+          {
+            length: 5,
+          }
+        ]
+      }
+    });
+  });
+
+  it('Specifying Interface type using __typename', async () => {
+    const schema = buildSchema(`
+      schema {
+        query: Root
+      }
+
+      type Root {
+        characters: [Character]
+      }
+
+      interface Character {
+        name: String!
+      }
+
+      type Human implements Character {
+        name: String!
+        totalCredits: Int
+      }
+
+      type Droid implements Character {
+        name: String!
+        primaryFunction: String
+      }
+    `);
+
+    const query = `
+      {
+        characters {
+          name
+          ... on Human {
+            totalCredits
+          }
+          ... on Droid {
+            primaryFunction
+          }
+        }
+      }
+    `;
+
+    const root = {
+      characters: [
+        {
+          name: 'Han Solo',
+          totalCredits: 10,
+          __typename: 'Human',
+        },
+        {
+          name: 'R2-D2',
+          primaryFunction: 'Astromech',
+          __typename: 'Droid',
+        }
+      ]
+    };
+
+    expect(await graphql(schema, query, root)).to.deep.equal({
+      data: {
+        characters: [
+          {
+            name: 'Han Solo',
+            totalCredits: 10,
+          },
+          {
+            name: 'R2-D2',
+            primaryFunction: 'Astromech',
+          }
+        ]
+      }
+    });
+  });
+
   it('Custom Scalar', () => {
     const body = dedent`
       schema {

src/utilities/buildASTSchema.js

@@ -409,7 +409,7 @@ export function buildASTSchema(ast: DocumentNode): GraphQLSchema {
       description: getDescription(def),
       fields: () => makeFieldDefMap(def),
       astNode: def,
-      resolveType: cannotExecuteSchema,
+      resolveType,
     });
   }
 
@@ -437,7 +437,7 @@ export function buildASTSchema(ast: DocumentNode): GraphQLSchema {
       name: def.name.value,
       description: getDescription(def),
       types: def.types.map(t => produceObjectType(t)),
-      resolveType: cannotExecuteSchema,
+      resolveType,
       astNode: def,
     });
   }
@@ -530,8 +530,12 @@ function leadingSpaces(str) {
   return i;
 }
 
-function cannotExecuteSchema() {
+function resolveType(result) {
+  if (result && result.__typename) {
+    return result.__typename;
+  }
   throw new Error(
-    'Generated Schema cannot use Interface or Union types for execution.'
+    'To resolve a Union or Interface type with a generated schema the result ' +
+    'must have a __typename property corresponding to the exact type used.'
   );
 }