# -*- coding: utf-8 -*- # Part of Odoo. See LICENSE file for full copyright and licensing details. from collections import defaultdict from datetime import datetime, timedelta from freezegun import freeze_time from odoo import Command from odoo.tests import common class TestTimesheetGlobalTimeOff(common.TransactionCase): def setUp(self): super(TestTimesheetGlobalTimeOff, self).setUp() # Creates 1 test company and a calendar for employees that # work part time. Then creates an employee per calendar (one # for the standard calendar and one for the one we created) self.test_company = self.env['res.company'].create({ 'name': 'My Test Company', }) attendance_ids = [ Command.create({'name': 'Monday Morning', 'dayofweek': '0', 'hour_from': 9, 'hour_to': 12, 'day_period': 'morning'}), Command.create({'name': 'Monday Afternoon', 'dayofweek': '0', 'hour_from': 13, 'hour_to': 16, 'day_period': 'afternoon'}), Command.create({'name': 'Tuesday Morning', 'dayofweek': '1', 'hour_from': 9, 'hour_to': 12, 'day_period': 'morning'}), Command.create({'name': 'Tuesday Afternoon', 'dayofweek': '1', 'hour_from': 13, 'hour_to': 16, 'day_period': 'afternoon'}), Command.create({'name': 'Thursday Morning', 'dayofweek': '3', 'hour_from': 9, 'hour_to': 12, 'day_period': 'morning'}), Command.create({'name': 'Thursday Afternoon', 'dayofweek': '3', 'hour_from': 13, 'hour_to': 16, 'day_period': 'afternoon'}), Command.create({'name': 'Friday Morning', 'dayofweek': '4', 'hour_from': 9, 'hour_to': 12, 'day_period': 'morning'}), Command.create({'name': 'Friday Afternoon', 'dayofweek': '4', 'hour_from': 13, 'hour_to': 16, 'day_period': 'afternoon'}) ] self.part_time_calendar = self.env['resource.calendar'].create({ 'name': 'Part Time Calendar', 'company_id': self.test_company.id, 'hours_per_day': 6, 'attendance_ids': attendance_ids, }) self.full_time_employee = self.env['hr.employee'].create({ 'name': 'John Doe', 'company_id': self.test_company.id, 'resource_calendar_id': self.test_company.resource_calendar_id.id, }) self.full_time_employee_2 = self.env['hr.employee'].create({ 'name': 'John Smith', 'company_id': self.test_company.id, 'resource_calendar_id': self.test_company.resource_calendar_id.id, }) self.part_time_employee = self.env['hr.employee'].create({ 'name': 'Jane Doe', 'company_id': self.test_company.id, 'resource_calendar_id': self.part_time_calendar.id, }) # This tests that timesheets are created for every employee with the same calendar # when a global time off is created. # This also tests that timesheets are deleted when global time off is deleted. def test_timesheet_creation_and_deletion_for_time_off(self): leave_start_datetime = datetime(2021, 1, 4, 7, 0, 0, 0) # This is a monday leave_end_datetime = datetime(2021, 1, 8, 18, 0, 0, 0) # This is a friday global_time_off = self.env['resource.calendar.leaves'].create({ 'name': 'Test', 'calendar_id': self.test_company.resource_calendar_id.id, 'date_from': leave_start_datetime, 'date_to': leave_end_datetime, }) # 5 Timesheets should have been created for full_time_employee and full_time_employee_2 # but none for part_time_employee leave_task = self.test_company.leave_timesheet_task_id timesheets_by_employee = defaultdict(lambda: self.env['account.analytic.line']) for timesheet in leave_task.timesheet_ids: timesheets_by_employee[timesheet.employee_id] |= timesheet self.assertFalse(timesheets_by_employee.get(self.part_time_employee, False)) self.assertEqual(len(timesheets_by_employee.get(self.full_time_employee)), 5) self.assertEqual(len(timesheets_by_employee.get(self.full_time_employee_2)), 5) # The standard calendar is for 8 hours/day from 8 to 12 and from 13 to 17. # So we need to check that the timesheets don't have more than 8 hours per day. self.assertEqual(leave_task.effective_hours, 80) # Now we delete the global time off. The timesheets should be deleted too. global_time_off.unlink() self.assertFalse(leave_task.timesheet_ids.ids) @freeze_time('2022-01-01 08:00:00') def test_timesheet_creation_and_deletion_on_employee_archive(self): """ Test the timesheets linked to the global time off in the future when the employee is archived """ today = datetime.today() leave_start_datetime = today + timedelta(days=-today.weekday(), weeks=1) # Next monday leave_end_datetime = leave_start_datetime + timedelta(days=5) # Next friday self.env['resource.calendar.leaves'].create({ 'name': 'Test', 'calendar_id': self.test_company.resource_calendar_id.id, 'date_from': leave_start_datetime, 'date_to': leave_end_datetime, }) # 5 Timesheets should have been created for full_time_employee timesheets_full_time_employee = self.env['account.analytic.line'].search([('employee_id', '=', self.full_time_employee.id)]) self.assertEqual(len(timesheets_full_time_employee), 5) # All timesheets should have been deleted for full_time_employee when he is archived self.full_time_employee.active = False timesheets_full_time_employee = self.env['account.analytic.line'].search([('employee_id', '=', self.full_time_employee.id)]) self.assertEqual(len(timesheets_full_time_employee), 0) # 5 Timesheets should have been created for full_time_employee when he is unarchived self.full_time_employee.active = True timesheets_full_time_employee = self.env['account.analytic.line'].search([('employee_id', '=', self.full_time_employee.id)]) self.assertEqual(len(timesheets_full_time_employee), 5) # This tests that no timesheet are created for days when the employee is not supposed to work def test_no_timesheet_on_off_days(self): leave_start_datetime = datetime(2021, 1, 4, 7, 0, 0, 0) # This is a monday leave_end_datetime = datetime(2021, 1, 8, 18, 0, 0, 0) # This is a friday day_off = datetime(2021, 1, 6, 0, 0, 0) # part_time_employee does not work on wednesday self.env['resource.calendar.leaves'].create({ 'name': 'Test', 'calendar_id': self.part_time_calendar.id, 'date_from': leave_start_datetime, 'date_to': leave_end_datetime, }) # The total number of hours for the timesheet created should be equal to the # hours_per_day of the calendar leave_task = self.test_company.leave_timesheet_task_id self.assertEqual(leave_task.effective_hours, 4 * self.part_time_calendar.hours_per_day) # No timesheet should have been created on the day off timesheet = self.env['account.analytic.line'].search([('date', '=', day_off), ('task_id', '=', leave_task.id)]) self.assertFalse(timesheet.id) # This tests that timesheets are created/deleted for every employee with the same calendar # when a global time off has a calendar_id set/remove def test_timesheet_creation_and_deletion_for_calendar_set_and_remove(self): leave_start_datetime = datetime(2021, 1, 4, 7, 0, 0, 0) # This is a monday leave_end_datetime = datetime(2021, 1, 8, 18, 0, 0, 0) # This is a friday global_time_off = self.env['resource.calendar.leaves'].create({ 'name': 'Test', 'calendar_id': self.test_company.resource_calendar_id.id, 'date_from': leave_start_datetime, 'date_to': leave_end_datetime, }) # 5 Timesheets should have been created for full_time_employee and full_time_employee_2 # but none for part_time_employee leave_task = self.test_company.leave_timesheet_task_id # Now we delete the calendar_id. The timesheets should be deleted too. global_time_off.calendar_id = False self.assertFalse(leave_task.timesheet_ids.ids) # Now we reset the calendar_id. The timesheets should be created and have the right value. global_time_off.calendar_id = self.test_company.resource_calendar_id.id timesheets_by_employee = defaultdict(lambda: self.env['account.analytic.line']) for timesheet in leave_task.timesheet_ids: timesheets_by_employee[timesheet.employee_id] |= timesheet self.assertFalse(timesheets_by_employee.get(self.part_time_employee, False)) self.assertEqual(len(timesheets_by_employee.get(self.full_time_employee)), 5) self.assertEqual(len(timesheets_by_employee.get(self.full_time_employee_2)), 5) # The standard calendar is for 8 hours/day from 8 to 12 and from 13 to 17. # So we need to check that the timesheets don't have more than 8 hours per day. self.assertEqual(leave_task.effective_hours, 80)