[ADD] b_shift : attendance sheet tests setUp

beesdoo_shift/tests/__init__.py

@@ -0,0 +1,2 @@
+# -*- coding: utf-8 -*-
+import test_attendance_sheet

beesdoo_shift/tests/test_attendance_sheet.py

@@ -0,0 +1,133 @@
+# -*- coding: utf-8 -*-
+# Copyright 2019 - Today Coop IT Easy SCRLfs (<http://www.coopiteasy.be>)
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
+
+from datetime import date, datetime, timedelta
+
+from openerp.tests.common import TransactionCase
+
+
+class TestAttendanceSheet(TransactionCase):
+    def setUp(self):
+        super(TestAttendanceSheet, self).setUp()
+        self.shift_model = self.env["beesdoo.shift.shift"]
+        self.shift_template_model = self.env["beesdoo.shift.template"]
+        self.attendance_sheet_model = self.env["beesdoo.shift.sheet"]
+
+        self.current_time = datetime.now()
+
+        self.user_admin = self.env.ref("base.partner_root")
+        self.user_permanent = self.env.ref(
+            "beesdoo_shift.beesdoo_shift_partner_2_demo"
+        )
+        self.user_generic = self.env.ref(
+            "beesdoo_shift.beesdoo_shift_partner_1_demo"
+        )
+
+        self.worker_regular_1 = self.env.ref(
+            "beesdoo_base.res_partner_cooperator_6_demo"
+        )
+        self.worker_regular_2 = self.env.ref(
+            "beesdoo_base.res_partner_cooperator_5_demo"
+        )
+        self.worker_regular_3 = self.env.ref(
+            "beesdoo_base.res_partner_cooperator_4_demo"
+        )
+        self.worker_regular_super_1 = self.env.ref(
+            "beesdoo_base.res_partner_cooperator_1_demo"
+        )
+        self.worker_irregular_1 = self.env.ref(
+            "beesdoo_base.res_partner_cooperator_2_demo"
+        )
+        self.worker_irregular_2 = self.env.ref(
+            "beesdoo_base.res_partner_cooperator_4_demo"
+        )
+
+        self.task_type_1 = self.env.ref(
+            "beesdoo_shift.beesdoo_shift_task_type_1_demo"
+        )
+        self.task_type_2 = self.env.ref(
+            "beesdoo_shift.beesdoo_shift_task_type_2_demo"
+        )
+        self.task_type_3 = self.env.ref(
+            "beesdoo_shift.beesdoo_shift_task_type_3_demo"
+        )
+
+        self.task_template_1 = self.env.ref(
+            "beesdoo.shift.beesdoo_shift_task_template_1_demo"
+        )
+        self.task_template_2 = self.env.ref(
+            "beesdoo.shift.beesdoo_shift_task_template_2_demo"
+        )
+
+        self.shift_regular_regular_1 = self.shift_model.create(
+            {
+                "task_template_id": self.task_template_1.id,
+                "task_type_id": self.task_type_1.id,
+                "worker_id": self.worker_regular_1.id,
+                "start_time": self.current_time + timedelta(minutes=5),
+                "end_time": self.current_time + timedelta(minutes=10),
+                "is_regular": True,
+                "is_compensation": False,
+            }
+        )
+        self.shift_regular_regular_2 = self.shift_model.create(
+            {
+                "task_template_id": self.task_template_2.id,
+                "task_type_id": self.task_type_2.id,
+                "worker_id": self.worker_regular_2.id,
+                "start_time": self.current_time - timedelta(minutes=50),
+                "end_time": self.current_time - timedelta(minutes=20),
+                "is_regular": True,
+                "is_compensation": False,
+            }
+        )
+        self.shift_regular_regular_replaced_1 = self.shift_model.create(
+            {
+                "task_template_id": self.task_template_1.id,
+                "task_type_id": self.task_type_3.id,
+                "worker_id": self.worker_regular_3.id,
+                "start_time": self.current_time + timedelta(minutes=5),
+                "end_time": self.current_time + timedelta(minutes=10),
+                "is_regular": True,
+                "is_compensation": False,
+                "replaced_id": self.worker_regular_1.id,
+            }
+        )
+        self.shift_regular_compensation_1 = self.shift_model.create(
+            {
+                "task_template_id": self.task_template_2.id,
+                "task_type_id": self.task_type_1.id,
+                "worker_id": self.worker_regular_super_1.id,
+                "start_time": self.current_time + timedelta(minutes=9),
+                "end_time": self.current_time + timedelta(minutes=21),
+                "is_regular": False,
+                "is_compensation": True,
+            }
+        )
+        self.shift_irregular_1 = self.shift_model.create(
+            {
+                "task_template_id": self.task_template_1.id,
+                "task_type_id": self.task_type_2.id,
+                "worker_id": self.worker_irregular_1.id,
+                "start_time": self.current_time + timedelta(minutes=5),
+                "end_time": self.current_time + timedelta(minutes=10),
+            }
+        )
+        self.shift_irregular_2 = self.shift_model.create(
+            {
+                "task_template_id": self.task_template_2.id,
+                "task_type_id": self.task_type_3.id,
+                "worker_id": self.worker_irregular_2.id,
+                "start_time": self.current_time + timedelta(minutes=40),
+                "end_time": self.current_time + timedelta(minutes=50),
+            }
+        )
+        self.shift_empty_1 = self.shift_model.create(
+            {
+                "task_template_id": self.task_template_1.id,
+                "task_type_id": self.task_type_1.id,
+                "start_time": self.current_time + timedelta(minutes=5),
+                "end_time": self.current_time + timedelta(minutes=10),
+            }
+        )