test: improve descriminator tests

related #78

Author: nodkz

src/__tests__/github-issues-test.js

@@ -1,13 +1,34 @@
 /* @flow */
 
 import mongoose from 'mongoose';
+import MongodbMemoryServer from 'mongodb-memory-server';
+import { graphql, GQC } from 'graphql-compose';
 import { composeWithMongoose } from '../index';
 
+// May require additional time for downloading MongoDB binaries
+jasmine.DEFAULT_TIMEOUT_INTERVAL = 60000;
+
+let mongoServer;
+const opts = { useMongoClient: true };
+
+beforeAll(async () => {
+  mongoServer = new MongodbMemoryServer();
+  const mongoUri = await mongoServer.getConnectionString();
+  mongoose.connect(mongoUri, opts, err => {
+    if (err) console.error(err);
+  });
+});
+
+afterAll(() => {
+  mongoose.disconnect();
+  mongoServer.stop();
+});
+
 describe('github issues checks', () => {
-  it('#78 Mongoose and Discriminators', () => {
+  describe('#78 Mongoose and Discriminators', () => {
     const options = { discriminatorKey: 'kind' };
 
-    const eventSchema = new mongoose.Schema({ time: Date }, options);
+    const eventSchema = new mongoose.Schema({ refId: String }, options);
     const Event = mongoose.model('GenericEvent', eventSchema);
 
     const clickedLinkSchema = new mongoose.Schema({ url: String }, options);
@@ -16,7 +37,63 @@ describe('github issues checks', () => {
     const EventTC = composeWithMongoose(Event);
     const ClickedLinkEventTC = composeWithMongoose(ClickedLinkEvent);
 
-    expect(EventTC.getFieldNames()).toEqual(['time', '_id', 'kind']);
-    expect(ClickedLinkEventTC.getFieldNames()).toEqual(['url', '_id', 'time', 'kind']);
+    it('creating Types from models', () => {
+      expect(EventTC.getFieldNames()).toEqual(['refId', '_id', 'kind']);
+      expect(ClickedLinkEventTC.getFieldNames()).toEqual(['url', '_id', 'refId', 'kind']);
+    });
+
+    it('manually override resolver output type for findMany', async () => {
+      const EventDescriminatorType = new graphql.GraphQLUnionType({
+        name: 'EventDescriminator',
+        types: [EventTC.getType(), ClickedLinkEventTC.getType()],
+        resolveType: value => {
+          if (value.kind === 'ClickedLinkEvent') {
+            return ClickedLinkEventTC.getType();
+          }
+          return EventTC.getType();
+        },
+      });
+
+      EventTC.getResolver('findMany').setType(new graphql.GraphQLList(EventDescriminatorType));
+
+      // let's check graphql response
+
+      await Event.create({ refId: 'aaa' });
+      await Event.create({ refId: 'bbb' });
+      await ClickedLinkEvent.create({ refId: 'ccc', url: 'url1' });
+      await ClickedLinkEvent.create({ refId: 'ddd', url: 'url2' });
+
+      GQC.rootQuery().addFields({
+        eventFindMany: EventTC.getResolver('findMany'),
+      });
+      const schema = GQC.buildSchema();
+
+      const res = await graphql.graphql(
+        schema,
+        `{
+        eventFindMany {
+          __typename
+          ... on GenericEvent {
+            refId
+          }
+          ... on ClickedLinkEvent {
+            refId
+            url
+          }
+        }
+      }`
+      );
+
+      expect(res).toEqual({
+        data: {
+          eventFindMany: [
+            { __typename: 'GenericEvent', refId: 'aaa' },
+            { __typename: 'GenericEvent', refId: 'bbb' },
+            { __typename: 'ClickedLinkEvent', refId: 'ccc', url: 'url1' },
+            { __typename: 'ClickedLinkEvent', refId: 'ddd', url: 'url2' },
+          ],
+        },
+      });
+    });
   });
 });