test: Mocking mongoose connect/disconnect with MongodbMemoryServer

Author: nodkz

src/__mocks__/mongooseCommon.js

@@ -6,17 +6,19 @@ import MongodbMemoryServer from 'mongodb-memory-server';
 
 mongoose.Promise = Promise;
 
-const mongoServer = new MongodbMemoryServer();
-
 jasmine.DEFAULT_TIMEOUT_INTERVAL = 60000;
 
-mongoServer.getConnectionString().then(mongoUri => {
-  mongoose.connect(mongoUri);
+const originalConnect = mongoose.connect;
+mongoose.connect = async () => {
+  const mongoServer = new MongodbMemoryServer();
+
+  const mongoUri = await mongoServer.getConnectionString();
+
+  originalConnect.bind(mongoose)(mongoUri);
 
   mongoose.connection.on('error', e => {
     if (e.message.code === 'ETIMEDOUT') {
       console.error(e);
-      mongoose.connect(mongoUri);
     } else {
       throw e;
     }
@@ -25,6 +27,11 @@ mongoServer.getConnectionString().then(mongoUri => {
   mongoose.connection.once('open', () => {
     // console.log(`MongoDB successfully connected to ${mongoUri}`);
   });
-});
+
+  mongoose.connection.once('disconnected', () => {
+    // console.log('MongoDB disconnected!');
+    mongoServer.stop();
+  });
+};
 
 export { mongoose, Schema };

src/__tests__/composeWithMongoose-test.js

@@ -7,6 +7,9 @@ import { UserModel } from '../__mocks__/userModel';
 import { composeWithMongoose } from '../composeWithMongoose';
 import typeStorage from '../typeStorage';
 
+beforeAll(() => UserModel.base.connect());
+afterAll(() => UserModel.base.disconnect());
+
 describe('composeWithMongoose ->', () => {
   beforeEach(() => {
     typeStorage.clear();

src/__tests__/fieldConverter-test.js

@@ -19,6 +19,9 @@ import {
 import { composeWithMongoose } from '../composeWithMongoose';
 import GraphQLMongoID from '../types/mongoid';
 
+beforeAll(() => UserModel.base.connect());
+afterAll(() => UserModel.base.disconnect());
+
 const {
   GraphQLString,
   GraphQLFloat,

src/resolvers/__tests__/count-test.js

@@ -8,6 +8,9 @@ import typeStorage from '../../typeStorage';
 
 const { GraphQLInt } = graphql;
 
+beforeAll(() => UserModel.base.connect());
+afterAll(() => UserModel.base.disconnect());
+
 describe('count() ->', () => {
   let UserTypeComposer;
 

src/resolvers/__tests__/createOne-test.js

@@ -10,6 +10,9 @@ import typeStorage from '../../typeStorage';
 
 const { GraphQLNonNull, GraphQLObjectType } = graphql;
 
+beforeAll(() => UserModel.base.connect());
+afterAll(() => UserModel.base.disconnect());
+
 describe('createOne() ->', () => {
   let UserTypeComposer;
 

src/resolvers/__tests__/findById-test.js

@@ -10,6 +10,9 @@ import typeStorage from '../../typeStorage';
 
 const { GraphQLNonNull } = graphql;
 
+beforeAll(() => UserModel.base.connect());
+afterAll(() => UserModel.base.disconnect());
+
 describe('findById() ->', () => {
   let UserTypeComposer;
   let PostTypeComposer;

src/resolvers/__tests__/findByIds-test.js

@@ -10,6 +10,9 @@ import typeStorage from '../../typeStorage';
 
 const { GraphQLNonNull, GraphQLList } = graphql;
 
+beforeAll(() => UserModel.base.connect());
+afterAll(() => UserModel.base.disconnect());
+
 describe('findByIds() ->', () => {
   let UserTypeComposer;
   let PostTypeComposer;

src/resolvers/__tests__/findMany-test.js

@@ -6,6 +6,9 @@ import findMany from '../findMany';
 import { composeWithMongoose } from '../../composeWithMongoose';
 import typeStorage from '../../typeStorage';
 
+beforeAll(() => UserModel.base.connect());
+afterAll(() => UserModel.base.disconnect());
+
 describe('findMany() ->', () => {
   let UserTypeComposer;
 

src/resolvers/__tests__/findOne-test.js

@@ -8,6 +8,9 @@ import typeStorage from '../../typeStorage';
 
 const { GraphQLNonNull } = graphql;
 
+beforeAll(() => UserModel.base.connect());
+afterAll(() => UserModel.base.disconnect());
+
 describe('findOne() ->', () => {
   let UserTypeComposer;
 

src/resolvers/__tests__/removeById-test.js

@@ -10,6 +10,9 @@ import typeStorage from '../../typeStorage';
 
 const { GraphQLNonNull, GraphQLObjectType } = graphql;
 
+beforeAll(() => UserModel.base.connect());
+afterAll(() => UserModel.base.disconnect());
+
 describe('removeById() ->', () => {
   let UserTypeComposer;