Thibault Libioulle
74624f43ee
This commit adds methods to easily obtain calendar validity per resource for a given search period and get the work intervals per resource, taking into account the validity of their calendars. In hr, the create and departure date of an employee are taken into account to compute the calendars validity. In hr_contract, it looks for every contract of human resources with employee of type student or employee existing in the search period and consider the specific calendar of the contract as valid during the contract-lifetime. PR: #77362 task-2646630
87 lines
4.5 KiB
Python
87 lines
4.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
|
from datetime import datetime
|
|
from pytz import timezone, utc
|
|
|
|
from odoo.addons.resource.models.resource import Intervals, sum_intervals
|
|
|
|
from .common import TestHrCommon
|
|
|
|
|
|
class TestResource(TestHrCommon):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super(TestResource, cls).setUpClass()
|
|
cls.calendar_40h = cls.env['resource.calendar'].create({'name': 'Default calendar'})
|
|
cls.employee_niv = cls.env['hr.employee'].create({
|
|
'name': 'Sharlene Rhodes',
|
|
'departure_date': '2022-06-01',
|
|
'resource_calendar_id': cls.calendar_40h.id,
|
|
})
|
|
cls.employee_niv_create_date = '2021-01-01 10:00:00'
|
|
cls.env.cr.execute("UPDATE hr_employee SET create_date=%s WHERE id=%s",
|
|
(cls.employee_niv_create_date, cls.employee_niv.id))
|
|
|
|
def test_calendars_validity_within_period_default(self):
|
|
calendars = self.employee_niv.resource_id._get_calendars_validity_within_period(
|
|
utc.localize(datetime(2021, 7, 1, 8, 0, 0)),
|
|
utc.localize(datetime(2021, 7, 30, 17, 0, 0)),
|
|
)
|
|
interval = Intervals([(
|
|
utc.localize(datetime(2021, 7, 1, 8, 0, 0)),
|
|
utc.localize(datetime(2021, 7, 30, 17, 0, 0)),
|
|
self.env['resource.calendar.attendance']
|
|
)])
|
|
|
|
self.assertEqual(1, len(calendars), "The dict returned by calendars validity should only have 1 entry")
|
|
self.assertEqual(1, len(calendars[self.employee_niv.resource_id.id]), "Niv should only have one calendar")
|
|
niv_entry = calendars[self.employee_niv.resource_id.id]
|
|
niv_calendar = next(iter(niv_entry))
|
|
self.assertEqual(niv_calendar, self.calendar_40h, "It should be Niv's Calendar")
|
|
self.assertFalse(niv_entry[niv_calendar] - interval, "Interval should cover all calendar's validity")
|
|
self.assertFalse(interval - niv_entry[niv_calendar], "Calendar validity should cover all interval")
|
|
|
|
def test_calendars_validity_within_period_creation(self):
|
|
calendars = self.employee_niv.resource_id._get_calendars_validity_within_period(
|
|
utc.localize(datetime(2020, 12, 1, 8, 0, 0)),
|
|
utc.localize(datetime(2021, 1, 31, 17, 0, 0)),
|
|
)
|
|
interval = Intervals([(
|
|
utc.localize(datetime(2021, 1, 1, 10, 0, 0)),
|
|
utc.localize(datetime(2021, 1, 31, 17, 0, 0)),
|
|
self.env['resource.calendar.attendance']
|
|
)])
|
|
niv_entry = calendars[self.employee_niv.resource_id.id]
|
|
self.assertFalse(niv_entry[self.calendar_40h] - interval, "Interval should cover all calendar's validity")
|
|
self.assertFalse(interval - niv_entry[self.calendar_40h], "Calendar validity should cover all interval")
|
|
|
|
def test_calendars_validity_within_period_before_creation(self):
|
|
calendars = self.employee_niv.resource_id._get_calendars_validity_within_period(
|
|
utc.localize(datetime(2020, 12, 1, 8, 0, 0)),
|
|
utc.localize(datetime(2020, 12, 31, 17, 0, 0)),
|
|
)
|
|
niv_entry = calendars[self.employee_niv.resource_id.id]
|
|
self.assertFalse(niv_entry[self.calendar_40h], "Interval should be empty")
|
|
|
|
def test_calendars_validity_within_period_before_departure(self):
|
|
calendars = self.employee_niv.resource_id._get_calendars_validity_within_period(
|
|
utc.localize(datetime(2022, 5, 1, 8, 0, 0)),
|
|
utc.localize(datetime(2022, 6, 30, 17, 0, 0)),
|
|
)
|
|
interval = Intervals([(
|
|
utc.localize(datetime(2022, 5, 1, 8, 0, 0)),
|
|
timezone(self.employee_niv.tz).localize(datetime(2022, 6, 1, 23, 59, 59, 999999)),
|
|
self.env['resource.calendar.attendance']
|
|
)])
|
|
niv_entry = calendars[self.employee_niv.resource_id.id]
|
|
self.assertFalse(niv_entry[self.calendar_40h] - interval, "Interval should cover all calendar's validity")
|
|
self.assertFalse(interval - niv_entry[self.calendar_40h], "Calendar validity should cover all interval")
|
|
|
|
def test_get_valid_work_intervals(self):
|
|
start = utc.localize(datetime(2022, 5, 29, 0, 0, 0))
|
|
end = utc.localize(datetime(2022, 6, 4, 23, 59, 59))
|
|
work_intervals, _ = self.employee_niv.resource_id._get_valid_work_intervals(start, end)
|
|
sum_work_intervals = sum_intervals(work_intervals[self.employee_niv.resource_id.id])
|
|
self.assertEqual(24, sum_work_intervals, "Sum of the work intervals for the resource niv should be 24h as he left his work after wednesday.")
|