133 lines
7.3 KiB
Python
133 lines
7.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
|
from datetime import date, datetime
|
|
from pytz import utc, timezone
|
|
|
|
from odoo.addons.resource.models.utils import Intervals, sum_intervals
|
|
from odoo.fields import Date
|
|
|
|
from .common import TestContractCommon
|
|
|
|
class TestResource(TestContractCommon):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super(TestResource, cls).setUpClass()
|
|
cls.calendar_richard = cls.env['resource.calendar'].create({'name': 'Calendar of Richard'})
|
|
cls.employee.resource_calendar_id = cls.calendar_richard
|
|
|
|
cls.calendar_35h = cls.env['resource.calendar'].create({
|
|
'name': '35h calendar',
|
|
'attendance_ids': [
|
|
(0, 0, {'name': 'Monday Morning', 'dayofweek': '0', 'hour_from': 8, 'hour_to': 12, 'day_period': 'morning'}),
|
|
(0, 0, {'name': 'Monday Lunch', 'dayofweek': '0', 'hour_from': 12, 'hour_to': 13, 'day_period': 'lunch'}),
|
|
(0, 0, {'name': 'Monday Evening', 'dayofweek': '0', 'hour_from': 13, 'hour_to': 16, 'day_period': 'afternoon'}),
|
|
(0, 0, {'name': 'Tuesday Morning', 'dayofweek': '1', 'hour_from': 8, 'hour_to': 12, 'day_period': 'morning'}),
|
|
(0, 0, {'name': 'Tuesday Lunch', 'dayofweek': '1', 'hour_from': 12, 'hour_to': 13, 'day_period': 'lunch'}),
|
|
(0, 0, {'name': 'Tuesday Evening', 'dayofweek': '1', 'hour_from': 13, 'hour_to': 16, 'day_period': 'afternoon'}),
|
|
(0, 0, {'name': 'Wednesday Morning', 'dayofweek': '2', 'hour_from': 8, 'hour_to': 12, 'day_period': 'morning'}),
|
|
(0, 0, {'name': 'Wednesday Lunch', 'dayofweek': '2', 'hour_from': 12, 'hour_to': 13, 'day_period': 'lunch'}),
|
|
(0, 0, {'name': 'Wednesday Evening', 'dayofweek': '2', 'hour_from': 13, 'hour_to': 16, 'day_period': 'afternoon'}),
|
|
(0, 0, {'name': 'Thursday Morning', 'dayofweek': '3', 'hour_from': 8, 'hour_to': 12, 'day_period': 'morning'}),
|
|
(0, 0, {'name': 'Thursday Lunch', 'dayofweek': '3', 'hour_from': 12, 'hour_to': 13, 'day_period': 'lunch'}),
|
|
(0, 0, {'name': 'Thursday Evening', 'dayofweek': '3', 'hour_from': 13, 'hour_to': 16, 'day_period': 'afternoon'}),
|
|
(0, 0, {'name': 'Friday Morning', 'dayofweek': '4', 'hour_from': 8, 'hour_to': 12, 'day_period': 'morning'}),
|
|
(0, 0, {'name': 'Friday Lunch', 'dayofweek': '4', 'hour_from': 12, 'hour_to': 13, 'day_period': 'lunch'}),
|
|
(0, 0, {'name': 'Friday Evening', 'dayofweek': '4', 'hour_from': 13, 'hour_to': 16, 'day_period': 'afternoon'})
|
|
],
|
|
})
|
|
|
|
cls.contract_cdd = cls.env['hr.contract'].create({
|
|
'date_start': Date.to_date('2021-09-01'),
|
|
'date_end': Date.to_date('2021-10-31'),
|
|
'name': 'First CDD Contract for Richard',
|
|
'resource_calendar_id': cls.calendar_35h.id,
|
|
'wage': 5000.0,
|
|
'employee_id': cls.employee.id,
|
|
'state': 'open',
|
|
})
|
|
cls.contract_cdi = cls.env['hr.contract'].create({
|
|
'date_start': Date.to_date('2021-11-01'),
|
|
'name': 'CDI Contract for Richard',
|
|
'resource_calendar_id': cls.calendar_richard.id,
|
|
'wage': 5000.0,
|
|
'employee_id': cls.employee.id,
|
|
'state': 'draft',
|
|
'kanban_state': 'done',
|
|
})
|
|
|
|
def test_calendars_validity_within_period(self):
|
|
tz = timezone(self.employee.tz)
|
|
calendars = self.employee.resource_id._get_calendars_validity_within_period(
|
|
tz.localize(datetime(2021, 10, 1, 0, 0, 0)),
|
|
tz.localize(datetime(2021, 12, 1, 0, 0, 0)),
|
|
)
|
|
interval_35h = Intervals([(
|
|
tz.localize(datetime(2021, 10, 1, 0, 0, 0)),
|
|
tz.localize(datetime.combine(date(2021, 10, 31), datetime.max.time())),
|
|
self.env['resource.calendar.attendance']
|
|
)])
|
|
interval_40h = Intervals([(
|
|
tz.localize(datetime(2021, 11, 1, 0, 0, 0)),
|
|
tz.localize(datetime(2021, 12, 1, 0, 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(2, len(calendars[self.employee.resource_id.id]), "Jean should only have one calendar")
|
|
richard_entries = calendars[self.employee.resource_id.id]
|
|
for calendar in richard_entries:
|
|
self.assertTrue(calendar in (self.calendar_35h | self.calendar_richard), "Each calendar should be listed")
|
|
if calendar == self.calendar_35h:
|
|
self.assertFalse(richard_entries[calendar] - interval_35h, "Interval 35h should cover all calendar 35h validity")
|
|
self.assertFalse(interval_35h - richard_entries[calendar], "Calendar 35h validity should cover all interval 35h")
|
|
elif calendar == self.calendar_richard:
|
|
self.assertFalse(richard_entries[calendar] - interval_40h, "Interval 40h should cover all calendar 40h validity")
|
|
self.assertFalse(interval_40h - richard_entries[calendar], "Calendar 40h validity should cover all interval 40h")
|
|
|
|
def test_queries(self):
|
|
employees_test = self.env['hr.employee'].create([{
|
|
'name': 'Employee ' + str(i),
|
|
} for i in range(0, 50)])
|
|
for emp in employees_test:
|
|
new_contract = self.contract_cdd.copy()
|
|
new_contract.employee_id = emp
|
|
new_contract.state = 'open'
|
|
new_contract = self.contract_cdi.copy()
|
|
new_contract.employee_id = emp
|
|
new_contract.state = 'draft'
|
|
new_contract.kanban_state = 'done'
|
|
|
|
start = utc.localize(datetime(2021, 9, 1, 0, 0, 0))
|
|
end = utc.localize(datetime(2021, 11, 30, 23, 59, 59))
|
|
with self.assertQueryCount(15):
|
|
work_intervals, _ = (employees_test | self.employee).resource_id._get_valid_work_intervals(start, end)
|
|
|
|
self.assertEqual(len(work_intervals), 51)
|
|
|
|
def test_get_valid_work_intervals(self):
|
|
start = timezone(self.employee.tz).localize(datetime(2021, 10, 24, 2, 0, 0))
|
|
end = timezone(self.employee.tz).localize(datetime(2021, 11, 6, 23, 59, 59))
|
|
work_intervals, _ = self.employee.resource_id._get_valid_work_intervals(start, end)
|
|
sum_work_intervals = sum_intervals(work_intervals[self.employee.resource_id.id])
|
|
self.assertEqual(75, sum_work_intervals, "Sum of the work intervals for the employee should be 35h+40h = 75h")
|
|
|
|
def test_multi_contract_attendance(self):
|
|
""" Verify whether retrieving an employee's calendar attendances can
|
|
handle multiple contracts with different calendars.
|
|
"""
|
|
|
|
date_from = utc.localize(datetime(2021, 10, 1, 0, 0, 0))
|
|
date_to = utc.localize(datetime(2021, 11, 30, 0, 0, 0))
|
|
|
|
attendances = self.employee._get_calendar_attendances(date_from, date_to)
|
|
self.assertEqual(21 * 7, attendances['hours'],
|
|
"Attendances should only include running or finished contracts.")
|
|
|
|
self.contract_cdd.state = 'close'
|
|
self.contract_cdi.state = 'open'
|
|
|
|
attendances = self.employee._get_calendar_attendances(date_from, date_to)
|
|
self.assertEqual(21 * 7 + 21 * 8, attendances['hours'],
|
|
"Attendances should add up multiple contracts with varying work weeks.")
|