back/methods/collection/setSaleQuantity.js

@@ -26,11 +26,30 @@ module.exports = Self => {
 
     Self.setSaleQuantity = async(saleId, quantity) => {
         const models = Self.app.models;
+        const myOptions = {};
+        let tx;
 
-        const sale = await models.Sale.findById(saleId);
+        if (typeof options == 'object')
-        return await sale.updateAttributes({
+            Object.assign(myOptions, options);
+
+        if (!myOptions.transaction) {
+            tx = await Self.beginTransaction({});
+            myOptions.transaction = tx;
+        }
+
+        try {
+            const sale = await models.Sale.findById(saleId, null, myOptions);
+            const saleUpdated = await sale.updateAttributes({
                 originalQuantity: sale.quantity,
                 quantity: quantity
-        });
+            }, myOptions);
+
+            if (tx) await tx.commit();
+
+            return saleUpdated;
+        } catch (e) {
+            if (tx) await tx.rollback();
+            throw e;
+        }
     };
 };

back/methods/collection/spec/setSaleQuantity.spec.js

@@ -2,15 +2,26 @@ const models = require('vn-loopback/server/server').models;
 
 describe('setSaleQuantity()', () => {
     it('should change quantity sale', async() => {
+        const tx = await models.Ticket.beginTransaction({});
+
+        try {
+            const options = {transaction: tx};
+
             const saleId = 30;
             const newQuantity = 10;
 
-        const originalSale = await models.Sale.findById(saleId);
+            const originalSale = await models.Sale.findById(saleId, null, options);
 
-        await models.Collection.setSaleQuantity(saleId, newQuantity);
+            await models.Collection.setSaleQuantity(saleId, newQuantity, options);
-        const updateSale = await models.Sale.findById(saleId);
+            const updateSale = await models.Sale.findById(saleId, null, options);
 
             expect(updateSale.originalQuantity).toEqual(originalSale.quantity);
             expect(updateSale.quantity).toEqual(newQuantity);
+
+            await tx.rollback();
+        } catch (e) {
+            await tx.rollback();
+            throw e;
+        }
     });
 });