# Part of Odoo. See LICENSE file for full copyright and licensing details. from datetime import datetime, date, UTC from odoo.tests.common import tagged, TransactionCase @tagged('at_install', '-post_install') # LEGACY at_install class TestFlexibleResourceCalendar(TransactionCase): @classmethod def setUpClass(cls): super().setUpClass() cls.flex_resource, cls.fully_flex_resource = cls.env['resource.resource'].create([{ 'name': 'Flex', 'tz': 'UTC', 'calendar_id': False, 'hours_per_week': 40, 'hours_per_day': 8.0, }, { 'name': 'fully flex', 'tz': 'UTC', 'calendar_id': False, }]) cls.flex_employee, cls.fully_flex_employee = cls.env['hr.employee'].create([{ 'name': "flexible employee", 'date_version': date(2025, 1, 1), 'contract_date_start': date(2025, 1, 1), 'wage': 10, 'resource_calendar_id': False, 'hours_per_week': 40, 'hours_per_day': 8.0, 'resource_id': cls.flex_resource.id, }, { 'name': "fully flexible employee", 'date_version': date(2025, 1, 1), 'contract_date_start': date(2025, 1, 1), 'wage': 10, 'resource_calendar_id': False, 'resource_id': cls.fully_flex_resource.id, }]) def test_flexible_resource_work_intervals_with_leaves(self): self.env['resource.calendar.leaves'].create([{ 'resource_id': self.flex_resource.id, 'date_from': datetime(2025, 7, 31, 8), 'date_to': datetime(2025, 8, 1, 17), }, { 'resource_id': self.fully_flex_resource.id, 'date_from': datetime(2025, 7, 31, 8), 'date_to': datetime(2025, 8, 1, 17), }]) custom_leave, half_day_leave = self.env['hr.work.entry.type'].create([{ 'name': 'Custom Leave', 'code': 'Custom Leave', 'requires_allocation': False, 'request_unit': 'hour', 'unit_of_measure': 'hour', }, { 'name': 'Half day', 'code': 'Half day', 'requires_allocation': False, 'request_unit': 'half_day', 'unit_of_measure': 'day', }]) self.env['hr.leave'].with_context(mail_create_nolog=True, mail_notrack=True).create([{ 'name': 'Half 1', 'work_entry_type_id': half_day_leave.id, 'employee_id': self.flex_employee.id, 'request_date_from': date(2025, 7, 28), 'request_date_to': date(2025, 7, 28), 'request_date_from_period': 'am', 'request_date_to_period': 'am', }, { 'name': 'Half 2', 'work_entry_type_id': half_day_leave.id, 'employee_id': self.flex_employee.id, 'request_date_from': date(2025, 7, 30), 'request_date_to': date(2025, 7, 30), 'request_date_from_period': 'pm', 'request_date_to_period': 'pm', }, { 'name': 'Custom', 'work_entry_type_id': custom_leave.id, 'employee_id': self.flex_employee.id, 'request_date_from': date(2025, 7, 29), 'request_date_to': date(2025, 7, 29), 'request_hour_from': 11.0, 'request_hour_to': 16.0, }, { 'name': 'Half 1', 'work_entry_type_id': half_day_leave.id, 'employee_id': self.fully_flex_employee.id, 'request_date_from': date(2025, 7, 28), 'request_date_to': date(2025, 7, 28), 'request_date_from_period': 'am', 'request_date_to_period': 'am', }, { 'name': 'Half 2', 'work_entry_type_id': half_day_leave.id, 'employee_id': self.fully_flex_employee.id, 'request_date_from': date(2025, 7, 30), 'request_date_to': date(2025, 7, 30), 'request_date_from_period': 'pm', 'request_date_to_period': 'pm', }, { 'name': 'Custom', 'work_entry_type_id': custom_leave.id, 'employee_id': self.fully_flex_employee.id, 'request_date_from': date(2025, 7, 29), 'request_date_to': date(2025, 7, 29), 'request_hour_from': 11.0, 'request_hour_to': 16.0, }]).action_approve() start_dt = datetime(2025, 7, 28).astimezone(UTC) end_dt = datetime(2025, 8, 3, 17).astimezone(UTC) resources = self.flex_resource | self.fully_flex_resource work_intervals, hours_per_day, hours_per_week = resources._get_flexible_resource_valid_work_intervals(start_dt, end_dt) self.maxDiff = None for resource in resources: self.assertEqual(work_intervals[resource.id]._items, [ (datetime(2025, 7, 28, 12, 0, tzinfo=UTC), datetime(2025, 7, 28, 23, 59, 59, 999999, tzinfo=UTC), self.env['resource.calendar.attendance']), (datetime(2025, 7, 29, 0, 0, tzinfo=UTC), datetime(2025, 7, 29, 11, 0, tzinfo=UTC), self.env['resource.calendar.attendance']), (datetime(2025, 7, 29, 16, 0, tzinfo=UTC), datetime(2025, 7, 29, 23, 59, 59, 999999, tzinfo=UTC), self.env['resource.calendar.attendance']), (datetime(2025, 7, 30, 0, 0, tzinfo=UTC), datetime(2025, 7, 30, 12, 0, tzinfo=UTC), self.env['resource.calendar.attendance']), (datetime(2025, 8, 2, 0, 0, tzinfo=UTC), datetime(2025, 8, 2, 23, 59, 59, 999999, tzinfo=UTC), self.env['resource.calendar.attendance']), (datetime(2025, 8, 3, 0, 0, tzinfo=UTC), datetime(2025, 8, 3, 17, 0, tzinfo=UTC), self.env['resource.calendar.attendance']), ], "resource not available on 29, 31, 01, 28 morning and 30 afternoon. for other days, resource can do his hours at any moment of the day (from 00:00:00 to 23:59:59)") self.assertDictEqual(hours_per_day[self.flex_resource.id], { date(2025, 7, 28): 4.0, date(2025, 7, 29): 3.0, date(2025, 7, 30): 4.0, date(2025, 7, 31): 0.0, date(2025, 8, 1): 0.0, date(2025, 8, 2): 8.0, date(2025, 8, 3): 8.0, }, "hours_per_day/2 for half days off, and hours_per_day - number_of_hoursfor custom time off") self.assertTrue(self.fully_flex_resource.id not in hours_per_day) self.assertDictEqual(hours_per_week[self.flex_resource.id], { (2025, 31): 11.0, (2025, 32): 40.0, }, "week 31 (27/07 -> 02/08): 2 days off 31 & 01 (-16 hours), half day on 28 and 30 (-8 hours), 5 hours off on day 29 / hours = 40-(16+8+5) = 11 hours, no timeoff on week 32") self.assertTrue(self.fully_flex_resource.id not in hours_per_week) def test_get_unusal_days_for_fully_flexible_employees(self): """ Test that _get_unusual_days return correct value for fully flexible employees. """ self.env['resource.calendar.leaves'].create({ 'name': '2 day holiday', 'calendar_id': False, 'date_from': datetime(2025, 1, 2), 'date_to': datetime(2025, 1, 3), }) start = str(datetime(2025, 1, 1)) end = str(datetime(2025, 1, 7)) unusual_days = self.fully_flex_employee._get_unusual_days(start, end) self.assertDictEqual( unusual_days, { '2025-01-01': False, '2025-01-02': True, '2025-01-03': True, '2025-01-04': False, '2025-01-05': False, '2025-01-06': False, '2025-01-07': False, }, "Employee should have unusual days on 2nd and 3rd January", ) def test_create_leave_fully_flexible_employee(self): """ Test that creating a leave for a fully flexible employee works as expected. """ test_leave_type = self.env['hr.work.entry.type'].create({ 'name': 'Test Leave Type', 'code': "FLEXITEST", 'requires_allocation': False, 'request_unit': 'day', 'unit_of_measure': 'day', }) self.env['hr.leave'].with_context(mail_create_nolog=True, mail_notrack=True).create({ 'name': 'Test Leave', 'work_entry_type_id': test_leave_type.id, 'employee_id': self.fully_flex_employee.id, 'request_date_from': date(2025, 2, 4), 'request_date_to': date(2025, 2, 6), }).action_approve() start = str(datetime(2025, 2, 1)) end = str(datetime(2025, 2, 7)) unusual_days = self.fully_flex_employee._get_unusual_days(start, end) self.assertDictEqual( unusual_days, { '2025-02-01': False, '2025-02-02': False, '2025-02-03': False, '2025-02-04': True, '2025-02-05': True, '2025-02-06': True, '2025-02-07': False, }, "Employee should have unusual days on 4th, 5th and 6th February", )