feat(resolvers): Add callback beforeQuery to resolveParams for updateMany, removeById, removeMany,

For `wrapResolve` you may provide `resolveParams.beforeQuery` callback, which will be called in

`updateMany`, `removeById`, `removeMany`, `removeOne` resolvers. To this callback will be provided

non-executed mongoose query, which you may change. Or you may return Promise.reject('Msg') in this

callback, if you want prohibit operation.

Author: nodkz

src/resolvers/__tests__/removeById-test.js

@@ -2,6 +2,7 @@
 
 import { expect } from 'chai';
 import { GraphQLNonNull } from 'graphql';
+import { Query } from 'mongoose';
 import { Resolver, TypeComposer } from 'graphql-compose';
 import { UserModel } from '../../__mocks__/userModel.js';
 import removeById from '../removeById';
@@ -113,6 +114,20 @@ describe('removeById() ->', () => {
       });
       expect(result.record).instanceof(UserModel);
     });
+
+    it('should call `beforeQuery` method with non-executed `query` as arg', async () => {
+      let beforeQueryCalled = false;
+      const result = await removeById(UserModel, UserTypeComposer).resolve({
+        args: { _id: user.id },
+        beforeQuery: (query) => {
+          expect(query).instanceof(Query);
+          beforeQueryCalled = true;
+          return query;
+        }
+      });
+      expect(beforeQueryCalled).to.be.true;
+      expect(result.record).instanceof(UserModel);
+    });
   });
 
   describe('Resolver.getOutputType()', () => {

src/resolvers/__tests__/removeMany-test.js

@@ -2,6 +2,7 @@
 
 import { expect } from 'chai';
 import { GraphQLInt, GraphQLNonNull } from 'graphql';
+import { Query } from 'mongoose';
 import Resolver from 'graphql-compose/lib/resolver/resolver';
 import TypeComposer from 'graphql-compose/lib/typeComposer';
 import { UserModel } from '../../__mocks__/userModel.js';
@@ -117,6 +118,22 @@ describe('removeMany() ->', () => {
       });
       expect(result).have.property('numAffected', 2);
     });
+
+    it('should call `beforeQuery` method with non-executed `query` as arg', async () => {
+      let beforeQueryCalled = false;
+      const result = await removeMany(UserModel, UserTypeComposer).resolve({
+        args: {
+          filter: { gender: 'female' },
+        },
+        beforeQuery: (query) => {
+          expect(query).instanceof(Query);
+          beforeQueryCalled = true;
+          return query;
+        }
+      });
+      expect(beforeQueryCalled).to.be.true;
+      expect(result).have.property('numAffected', 2);
+    });
   });
 
   describe('Resolver.getOutputType()', () => {

src/resolvers/__tests__/removeOne-test.js

@@ -2,6 +2,7 @@
 
 import { expect } from 'chai';
 import { Resolver, TypeComposer } from 'graphql-compose';
+import { Query } from 'mongoose';
 import { UserModel } from '../../__mocks__/userModel.js';
 import removeOne from '../removeOne';
 import GraphQLMongoID from '../../types/mongoid';
@@ -143,6 +144,20 @@ describe('removeOne() ->', () => {
       });
       expect(result.record).instanceof(UserModel);
     });
+
+    it('should call `beforeQuery` method with non-executed `query` as arg', async () => {
+      let beforeQueryCalled = false;
+      const result = await removeOne(UserModel, UserTypeComposer).resolve({
+        args: { },
+        beforeQuery: (query) => {
+          expect(query).instanceof(Query);
+          beforeQueryCalled = true;
+          return query;
+        }
+      });
+      expect(beforeQueryCalled).to.be.true;
+      expect(result.record).instanceof(UserModel);
+    });
   });
 
   describe('Resolver.getOutputType()', () => {

src/resolvers/__tests__/updateMany-test.js

@@ -2,6 +2,7 @@
 
 import { expect } from 'chai';
 import { GraphQLInt, GraphQLNonNull } from 'graphql';
+import { Query } from 'mongoose';
 import { Resolver, TypeComposer } from 'graphql-compose';
 import { UserModel } from '../../__mocks__/userModel.js';
 import updateMany from '../updateMany';
@@ -115,6 +116,23 @@ describe('updateMany() ->', () => {
       });
       expect(result).have.deep.property('numAffected', 2);
     });
+
+    it('should call `beforeQuery` method with non-executed `query` as arg', async () => {
+      let beforeQueryCalled = false;
+      const result = await updateMany(UserModel, UserTypeComposer).resolve({
+        args: {
+          record: { gender: 'female' },
+        },
+        beforeQuery: (query) => {
+          expect(query).instanceof(Query);
+          beforeQueryCalled = true;
+          // modify query before execution
+          return query.where({ _id: user1.id })
+        }
+      });
+      expect(beforeQueryCalled).to.be.true;
+      expect(result).have.deep.property('numAffected', 1);
+    });
   });
 
   describe('Resolver.getOutputType()', () => {

src/resolvers/removeById.js

@@ -76,7 +76,11 @@ export default function removeById(
       resolveParams.query = model.findByIdAndRemove(args._id);
       projectionHelper(resolveParams);
 
-      return resolveParams.query.exec()
+      return (
+        resolveParams.beforeQuery
+          ? Promise.resolve(resolveParams.beforeQuery(resolveParams.query))
+          : resolveParams.query.exec()
+        )
         .then(record => {
           if (record) {
             return {

src/resolvers/removeMany.js

@@ -75,8 +75,11 @@ export default function removeMany(
       filterHelper(resolveParams);
       resolveParams.query = resolveParams.query.remove();
 
-      return resolveParams.query
-        .exec()
+      return (
+        resolveParams.beforeQuery
+          ? Promise.resolve(resolveParams.beforeQuery(resolveParams.query))
+          : resolveParams.query.exec()
+        )
         .then(res => {
           if (res.result && res.result.ok) {
             return {

src/resolvers/removeOne.js

@@ -74,8 +74,11 @@ export default function removeOne(
       sortHelper(resolveParams);
       projectionHelper(resolveParams);
 
-      return resolveParams.query
-        .exec()
+      return (
+        resolveParams.beforeQuery
+          ? Promise.resolve(resolveParams.beforeQuery(resolveParams.query))
+          : resolveParams.query.exec()
+        )
         .then(record => {
           if (record) {
             return {

src/resolvers/updateMany.js

@@ -96,8 +96,11 @@ export default function updateMany(
       resolveParams.query = resolveParams.query.setOptions({ multi: true }); // eslint-disable-line
       resolveParams.query.update({ $set: toDottedObject(recordData) });
 
-      return resolveParams.query
-        .exec()
+      return (
+        resolveParams.beforeQuery
+          ? Promise.resolve(resolveParams.beforeQuery(resolveParams.query))
+          : resolveParams.query.exec()
+        )
         .then(res => {
           if (res.ok) {
             return {