# Part of Odoo. See LICENSE file for full copyright and licensing details. from datetime import datetime from odoo.tests import new_test_user from odoo.tests.common import TransactionCase, tagged from odoo.exceptions import AccessError, ValidationError import time @tagged('post_install', '-at_install', 'holidays_attendance') class TestHolidaysOvertime(TransactionCase): @classmethod def setUpClass(cls): super().setUpClass() cls.company = cls.env['res.company'].create({ 'name': 'SweatChipChop Inc.', 'hr_attendance_overtime': True, 'overtime_start_date': datetime(2021, 1, 1), }) cls.user = new_test_user(cls.env, login='user', groups='base.group_user,hr_attendance.group_hr_attendance', company_id=cls.company.id).with_company(cls.company) cls.user_manager = new_test_user(cls.env, login='manager', groups='base.group_user,hr_holidays.group_hr_holidays_user', company_id=cls.company.id).with_company(cls.company) cls.manager = cls.env['hr.employee'].create({ 'name': 'Dominique', 'user_id': cls.user_manager.id, 'company_id': cls.company.id, }) cls.employee = cls.env['hr.employee'].create({ 'name': 'Barnabé', 'user_id': cls.user.id, 'parent_id': cls.manager.id, 'company_id': cls.company.id, }) cls.leave_type_no_alloc = cls.env['hr.leave.type'].create({ 'name': 'Overtime Compensation No Allocation', 'company_id': cls.company.id, 'requires_allocation': 'no', 'overtime_deductible': True, }) cls.leave_type_employee_allocation = cls.env['hr.leave.type'].create({ 'name': 'Overtime Compensation Employee Allocation', 'company_id': cls.company.id, 'requires_allocation': 'yes', 'employee_requests': 'yes', 'overtime_deductible': True, }) cls.leave_type_manager_alloc = cls.env['hr.leave.type'].create({ 'name': 'Overtime Compensation Manager Allocation', 'company_id': cls.company.id, 'requires_allocation': 'yes', 'employee_requests': 'no', 'allocation_validation_type': 'set', 'overtime_deductible': True, }) def new_attendance(self, check_in, check_out=False): return self.env['hr.attendance'].create({ 'employee_id': self.employee.id, 'check_in': check_in, 'check_out': check_out, }) def test_deduct_button_visibility(self): with self.with_user('user'): self.assertFalse(self.user.request_overtime, 'Button should not be visible') self.new_attendance(check_in=datetime(2021, 1, 2, 8), check_out=datetime(2021, 1, 2, 18)) self.assertEqual(self.user.total_overtime, 10, 'Should have 10 hours of overtime') self.assertTrue(self.user.request_overtime, 'Button should be visible') def test_check_overtime(self): with self.with_user('user'): self.assertEqual(self.user.total_overtime, 0, 'No overtime') with self.assertRaises(ValidationError), self.cr.savepoint(): self.env['hr.leave'].create({ 'name': 'no overtime', 'employee_id': self.employee.id, 'holiday_status_id': self.leave_type_no_alloc.id, 'number_of_days': 1, 'date_from': datetime(2021, 1, 4), 'date_to': datetime(2021, 1, 5), 'state': 'draft', }) self.new_attendance(check_in=datetime(2021, 1, 2, 8), check_out=datetime(2021, 1, 2, 16)) self.assertEqual(self.employee.total_overtime, 8, 'Should have 8 hours of overtime') leave = self.env['hr.leave'].create({ 'name': 'no overtime', 'employee_id': self.employee.id, 'holiday_status_id': self.leave_type_no_alloc.id, 'number_of_days': 1, 'date_from': datetime(2021, 1, 4), 'date_to': datetime(2021, 1, 5), }) # The employee doesn't have the right to read the overtime from the leave overtime = leave.sudo().overtime_id.with_user(self.user) # An employee cannot delete an overtime adjustment with self.assertRaises(AccessError), self.cr.savepoint(): overtime.unlink() # ... nor change its duration with self.assertRaises(AccessError), self.cr.savepoint(): overtime.duration = 8 def test_leave_adjust_overtime(self): self.new_attendance(check_in=datetime(2021, 1, 2, 8), check_out=datetime(2021, 1, 2, 16)) self.assertEqual(self.employee.total_overtime, 8, 'Should have 8 hours of overtime') leave = self.env['hr.leave'].create({ 'name': 'no overtime', 'employee_id': self.employee.id, 'holiday_status_id': self.leave_type_no_alloc.id, 'number_of_days': 1, 'date_from': datetime(2021, 1, 4), 'date_to': datetime(2021, 1, 5), }) self.assertTrue(leave.overtime_id.adjustment, "An adjustment overtime should be created") self.assertEqual(leave.overtime_id.duration, -8) self.assertEqual(self.employee.total_overtime, 0) leave.action_refuse() self.assertFalse(leave.overtime_id.exists(), "Overtime should be deleted") self.assertEqual(self.employee.total_overtime, 8) leave.action_draft() self.assertTrue(leave.overtime_id.exists(), "Overtime should be created") self.assertEqual(self.employee.total_overtime, 0) overtime = leave.overtime_id leave.unlink() self.assertFalse(overtime.exists(), "Overtime should be deleted along with the leave") self.assertEqual(self.employee.total_overtime, 8) def test_leave_check_overtime_write(self): self.new_attendance(check_in=datetime(2021, 1, 2, 8), check_out=datetime(2021, 1, 2, 16)) self.new_attendance(check_in=datetime(2021, 1, 3, 8), check_out=datetime(2021, 1, 3, 16)) self.assertEqual(self.employee.total_overtime, 16) leave = self.env['hr.leave'].create({ 'name': 'no overtime', 'employee_id': self.employee.id, 'holiday_status_id': self.leave_type_no_alloc.id, 'number_of_days': 1, 'date_from': datetime(2021, 1, 4), 'date_to': datetime(2021, 1, 5), }) self.assertEqual(self.employee.total_overtime, 8) leave.date_to = datetime(2021, 1, 6) self.assertEqual(self.employee.total_overtime, 0) with self.assertRaises(ValidationError), self.cr.savepoint(): leave.date_to = datetime(2021, 1, 7) leave.date_to = datetime(2021, 1, 5) self.assertEqual(self.employee.total_overtime, 8) def test_employee_create_allocation(self): with self.with_user('user'): self.assertEqual(self.employee.total_overtime, 0) with self.assertRaises(ValidationError), self.cr.savepoint(): self.env['hr.leave.allocation'].create({ 'name': 'test allocation', 'holiday_status_id': self.leave_type_employee_allocation.id, 'employee_id': self.employee.id, 'number_of_days': 1, 'state': 'draft', 'date_from': time.strftime('%Y-1-1'), 'date_to': time.strftime('%Y-12-31'), }) self.new_attendance(check_in=datetime(2021, 1, 2, 8), check_out=datetime(2021, 1, 2, 16)) self.assertAlmostEqual(self.employee.total_overtime, 8, 'Should have 8 hours of overtime') allocation = self.env['hr.leave.allocation'].create({ 'name': 'test allocation', 'holiday_status_id': self.leave_type_employee_allocation.id, 'employee_id': self.employee.id, 'number_of_days': 1, 'state': 'draft', 'date_from': time.strftime('%Y-1-1'), 'date_to': time.strftime('%Y-12-31'), }) allocation.action_confirm() self.assertEqual(self.employee.total_overtime, 0) leave_type = self.env['hr.leave.type'].sudo().create({ 'name': 'Overtime Compensation Employee Allocation', 'company_id': self.company.id, 'requires_allocation': 'yes', 'employee_requests': 'yes', 'overtime_deductible': False, }) # User can request another allocation even without overtime allocation2 = self.env['hr.leave.allocation'].create({ 'name': 'test allocation', 'holiday_status_id': leave_type.id, 'employee_id': self.employee.id, 'number_of_days': 1, 'state': 'draft', 'date_from': time.strftime('%Y-1-1'), 'date_to': time.strftime('%Y-12-31'), }) allocation2.action_confirm() def test_allocation_check_overtime_write(self): self.new_attendance(check_in=datetime(2021, 1, 2, 8), check_out=datetime(2021, 1, 2, 16)) self.new_attendance(check_in=datetime(2021, 1, 3, 8), check_out=datetime(2021, 1, 3, 16)) self.assertEqual(self.employee.total_overtime, 16, 'Should have 16 hours of overtime') alloc = self.env['hr.leave.allocation'].create({ 'name': 'test allocation', 'holiday_status_id': self.leave_type_employee_allocation.id, 'employee_id': self.employee.id, 'number_of_days': 1, 'state': 'draft', 'date_from': time.strftime('%Y-1-1'), 'date_to': time.strftime('%Y-12-31'), }) self.assertEqual(self.employee.total_overtime, 8) with self.assertRaises(ValidationError), self.cr.savepoint(): alloc.number_of_days = 3 alloc.number_of_days = 2 self.assertEqual(self.employee.total_overtime, 0)