From 163faf983e326c2679c54d1993c0daeade61b7b4 Mon Sep 17 00:00:00 2001
From: sergiodt <sergiodt@verdnatura.es>
Date: Tue, 6 Aug 2024 16:34:46 +0200
Subject: [PATCH] feat workerActivity refs #6078

---
 back/methods/workerActivity/add.js            | 10 +++++---
 back/methods/workerActivity/specs/add.spec.js | 25 +++++++------------
 2 files changed, 15 insertions(+), 20 deletions(-)

diff --git a/back/methods/workerActivity/add.js b/back/methods/workerActivity/add.js
index 587ebfae7a..bba05dafd5 100644
--- a/back/methods/workerActivity/add.js
+++ b/back/methods/workerActivity/add.js
@@ -24,8 +24,12 @@ module.exports = Self => {
 
     Self.add = async(ctx, code, model, options) => {
         const userId = ctx.req.accessToken.userId;
+        const myOptions = {};
 
-        const result = await await Self.rawSql(`
+        if (typeof options == 'object')
+            Object.assign(myOptions, options);
+
+        return await Self.rawSql(`
             INSERT INTO workerActivity (workerFk, workerActivityTypeFk, model)
             SELECT ?,
                 ?, 
@@ -44,8 +48,6 @@ module.exports = Self => {
             WHERE sub.workerFk IS NULL
                 OR sub.code <> ? 
                 OR TIMESTAMPDIFF(SECOND, sub.created, util.VN_NOW()) > wtcp.dayBreak;`
-        , [userId, code, model, userId, code]);
-
-        return result;
+        , [userId, code, model, userId, code], myOptions);
     };
 };
diff --git a/back/methods/workerActivity/specs/add.spec.js b/back/methods/workerActivity/specs/add.spec.js
index 3f9657c676..67a85cb7d0 100644
--- a/back/methods/workerActivity/specs/add.spec.js
+++ b/back/methods/workerActivity/specs/add.spec.js
@@ -1,36 +1,29 @@
 const {models} = require('vn-loopback');
 
 describe('workerActivity insert()', () => {
-    beforeAll(async() => {
-        ctx = {
-            req: {
-                accessToken: {},
-                headers: {origin: 'http://localhost'},
-                __: value => value
-            }
-        };
-    });
+    const ctx = beforeAll.getCtx(1106);
 
     it('should insert in workerActivity', async() => {
         const tx = await models.WorkerActivity.beginTransaction({});
+        let count = 0;
+        const options = {transaction: tx};
 
         try {
             await models.WorkerActivityType.create(
-                {'code': 'STOP', 'description': 'STOP'}
+                {'code': 'STOP', 'description': 'STOP'}, options
             );
-            const options = {transaction: tx};
-            ctx.req.accessToken.userId = 1106;
 
-            models.WorkerActivity.add(ctx, 'STOP', 'APP', options);
+            result = await models.WorkerActivity.add(ctx, 'STOP', 'APP', options);
+
+            count = await models.WorkerActivity.count(
+                {'workerFK': 1106}, options
+            );
 
             await tx.rollback();
         } catch (e) {
             await tx.rollback();
             throw e;
         }
-        const count = await models.WorkerActivity.count(
-            {'workerFK': 1106}
-        );
 
         expect(count).toEqual(1);
     });