test: add demo for issue 135

Author: nodkz

src/__tests__/guthub_issues/135-test.js

@@ -0,0 +1,82 @@
+/* @flow */
+/* eslint-disable no-await-in-loop */
+
+import mongoose from 'mongoose';
+import MongodbMemoryServer from 'mongodb-memory-server';
+import { schemaComposer, graphql } from 'graphql-compose';
+import { composeWithMongoose } from '../../index';
+
+let mongoServer;
+beforeAll(async () => {
+  mongoServer = new MongodbMemoryServer();
+  const mongoUri = await mongoServer.getConnectionString();
+  await mongoose.connect(
+    mongoUri,
+    { useNewUrlParser: true }
+  );
+  // mongoose.set('debug', true);
+});
+
+afterAll(() => {
+  mongoose.disconnect();
+  mongoServer.stop();
+});
+
+// May require additional time for downloading MongoDB binaries
+jasmine.DEFAULT_TIMEOUT_INTERVAL = 60000;
+
+describe('issue #135 - Mongoose virtuals', () => {
+  const RecordSchema = new mongoose.Schema({ id: String, title: String });
+
+  // ADD VIRTUAL FIELDS VIA loadClass METHOD
+  // see https://mongoosejs.com/docs/api.html#schema_Schema-loadClass
+  class RecordDoc {
+    get virtualField123() {
+      // $FlowFixMe
+      return `Improved ${this.title}`;
+    }
+  }
+  RecordSchema.loadClass(RecordDoc);
+
+  // ADD MOCK DATA TO DB
+  const Record = mongoose.model('Record', RecordSchema);
+  beforeAll(async () => {
+    for (let i = 1; i <= 3; i++) {
+      // $FlowFixMe
+      await Record.create({ _id: `10000000000000000000000${i}`, title: `Title ${i}` });
+    }
+  });
+
+  // ADD VIRTUAL FIELD DEFINITION <-------------------   JUST ADD FIELD DEFINITION 🛑🛑🛑
+  // no need to define resolve method explicitly
+  const RecordTC = composeWithMongoose(Record);
+  RecordTC.addFields({
+    virtualField123: {
+      type: 'String',
+    },
+  });
+
+  // INIT GRAPHQL SCHEMA
+  schemaComposer.Query.addFields({
+    findMany: RecordTC.getResolver('findMany'),
+  });
+
+  const schema = schemaComposer.buildSchema();
+
+  it('check that virtual field works', async () => {
+    const res = await graphql.graphql({
+      schema,
+      source: 'query { findMany { id title virtualField123 } }',
+    });
+
+    expect(res).toEqual({
+      data: {
+        findMany: [
+          { id: null, title: 'Title 1', virtualField123: 'Improved Title 1' },
+          { id: null, title: 'Title 2', virtualField123: 'Improved Title 2' },
+          { id: null, title: 'Title 3', virtualField123: 'Improved Title 3' },
+        ],
+      },
+    });
+  });
+});