test(updateById, updateOne): Add test when beforeRecordMutate rejects operation due context

Author: nodkz

src/resolvers/__tests__/updateById-test.js

@@ -168,6 +168,22 @@ describe('updateById() ->', () => {
       expect(result).have.deep.property('record.someDynamic', '1.1.1.1');
       expect(beforeMutationId).to.equal(user1.id);
     });
+
+    it('`beforeRecordMutate` may reject operation', async () => {
+      const result = updateById(UserModel, UserTypeComposer).resolve({
+        args: { record: { _id: user1.id, name: 'new name' } },
+        context: { readOnly: true },
+        beforeRecordMutate: (record, rp) => {
+          if (rp.context.readOnly) {
+            return Promise.reject(new Error('Denied due context ReadOnly'));
+          }
+          return record;
+        },
+      });
+      await expect(result).be.rejectedWith(Error, 'Denied due context ReadOnly');
+      const exist = await UserModel.collection.findOne({ _id: user1._id });
+      expect(exist.name).to.equal(user1.name);
+    });
   });
 
   describe('Resolver.getOutputType()', () => {

src/resolvers/__tests__/updateOne-test.js

@@ -193,6 +193,22 @@ describe('updateOne() ->', () => {
       expect(result).have.deep.property('record.someDynamic', '1.1.1.1');
       expect(beforeMutationId).to.equal(user1.id);
     });
+
+    it('`beforeRecordMutate` may reject operation', async () => {
+      const result = updateOne(UserModel, UserTypeComposer).resolve({
+        args: { filter: { _id: user1.id }, record: { name: 'new name' } },
+        context: { readOnly: true },
+        beforeRecordMutate: (record, rp) => {
+          if (rp.context.readOnly) {
+            return Promise.reject(new Error('Denied due context ReadOnly'));
+          }
+          return record;
+        },
+      });
+      await expect(result).be.rejectedWith(Error, 'Denied due context ReadOnly');
+      const exist = await UserModel.collection.findOne({ _id: user1._id });
+      expect(exist.name).to.equal(user1.name);
+    });
   });
 
   describe('Resolver.getOutputType()', () => {

src/resolvers/updateById.js

@@ -107,7 +107,7 @@ export default function updateById(
         // save changes to DB
         .then((doc) => {
           if (!doc) {
-            return Promise.reject('Document not found');
+            return Promise.reject(new Error('Document not found'));
           }
           if (recordData) {
             doc.set(recordData);