135-test.js

/* @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' },
        ],
      },
    });
  });
});