modules/worker/back/methods/worker-time-control/specs/getMailStates.spec.js

@@ -0,0 +1,29 @@
+const models = require('vn-loopback/server/server').models;
+
+describe('workerTimeControl getMailStates()', () => {
+    const workerId = 9;
+    const ctx = {args: {
+        month: 12,
+        year: 2000
+    }};
+
+    it('should get the states of a month about time control mail', async() => {
+        const tx = await models.WorkerTimeControl.beginTransaction({});
+
+        try {
+            const options = {transaction: tx};
+
+            const response = await models.WorkerTimeControl.getMailStates(ctx, workerId, options);
+
+            expect(response[0].state).toEqual('REVISE');
+            expect(response[1].state).toEqual('SENDED');
+            expect(response[2].state).toEqual('CONFIRMED');
+
+            await tx.rollback();
+        } catch (e) {
+            await tx.rollback();
+            throw e;
+        }
+    });
+});
+