Odoo18-Base/addons/hr_holidays/tests/test_working_hours.py
2025-01-06 10:57:38 +07:00

144 lines
6.3 KiB
Python

# Part of Odoo. See LICENSE file for full copyright and licensing details.
from datetime import date, datetime
from odoo.addons.hr_calendar.tests.common import TestHrCalendarCommon
from odoo.tests import tagged
@tagged('work_hours')
class TestWorkingHours(TestHrCalendarCommon):
""" Test global leaves for a whole company, conflict resolutions """
@classmethod
def setUpClass(cls):
super().setUpClass()
if 'hr.contract' in cls.env:
cls.skipTest(cls,
"hr_contract module is installed. To test these features you need to install hr_holidays_contract"
)
cls.leave_type = cls.env['hr.leave.type'].create({
'name': 'Unpaid Time Off',
'requires_allocation': 'no',
'leave_validation_type': 'no_validation',
})
def test_multi_companies_2_employees_2_selected_companies_holidays(self):
"""
INPUT:
======
Employees Companies
employee A company A ---> 35h [X] A <- main company
employee A company B ---> 28h [X] B
employee A company A take a day off for monday and tuesday.
OUTPUT:
=======
The schedule will be : off on monday, following 28h schedule on tuesday and the union for the rest of the week.
"""
self.env.user.company_id = self.company_A
self.env.user.company_ids = [self.company_A.id, self.company_B.id]
self.env['hr.leave'].create({
'name': 'holiday from monday to tuesday',
'employee_id': self.employeeA.id,
'holiday_status_id': self.leave_type.id,
'request_date_from': datetime(2023, 12, 25),
'request_date_to': datetime(2023, 12, 26, 23, 59, 59),
})
work_hours = self.env['res.partner'].get_working_hours_for_all_attendees(
[self.partnerA.id],
datetime(2023, 12, 25).isoformat(),
datetime(2023, 12, 31).isoformat()
)
self.assertEqual(work_hours, [
{'daysOfWeek': [2], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [2], 'startTime': '13:00', 'endTime': '16:00'},
{'daysOfWeek': [3], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [3], 'startTime': '13:00', 'endTime': '16:00'},
{'daysOfWeek': [4], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [4], 'startTime': '13:00', 'endTime': '16:00'},
{'daysOfWeek': [5], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [5], 'startTime': '13:00', 'endTime': '16:00'},
])
def test_multi_companies_2_employees_2_selected_companies_company_holidays(self):
"""
INPUT:
======
Employees Companies
employee A company A ---> 35h [X] A <- main company
employee A company B ---> 28h [X] B
Company A give a day off for everyone on monday and tuesday.
OUTPUT:
=======
The schedule will be : off on monday, following 28h schedule on tuesday and the union for the rest of the week.
"""
self.env.user.company_id = self.company_A
self.env.user.company_ids = [self.company_A.id, self.company_B.id]
company_leave = self.env['hr.leave.generate.multi.wizard'].create({
'name': 'holiday from monday to tuesday',
'allocation_mode': 'company',
'company_id': self.company_A.id,
'holiday_status_id': self.leave_type.id,
'date_from': date(2023, 12, 25),
'date_to': date(2023, 12, 26),
})
company_leave.action_generate_time_off()
work_hours = self.env['res.partner'].get_working_hours_for_all_attendees(
[self.partnerA.id],
datetime(2023, 12, 25).isoformat(),
datetime(2023, 12, 31).isoformat()
)
self.assertEqual(work_hours, [
{'daysOfWeek': [2], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [2], 'startTime': '13:00', 'endTime': '16:00'},
{'daysOfWeek': [3], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [3], 'startTime': '13:00', 'endTime': '16:00'},
{'daysOfWeek': [4], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [4], 'startTime': '13:00', 'endTime': '16:00'},
{'daysOfWeek': [5], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [5], 'startTime': '13:00', 'endTime': '16:00'},
])
def test_multi_companies_2_employees_2_selected_companies_global_holidays(self):
"""
INPUT:
======
Employees Companies
employee A company A ---> 35h [X] A <- main company
employee A company B ---> 28h [X] B
Global leave for calendar 35h on monday and tuesday.
OUTPUT:
=======
The schedule will be : off on monday, following 28h schedule on tuesday and the union for the rest of the week.
"""
self.env.user.company_id = self.company_A
self.env.user.company_ids = [self.company_A.id, self.company_B.id]
self.env['resource.calendar.leaves'].create({
'name': 'Global Time Off',
'date_from': datetime(2023, 12, 25),
'date_to': datetime(2023, 12, 26, 23, 59, 59),
'calendar_id': self.calendar_35h.id,
})
work_hours = self.env['res.partner'].get_working_hours_for_all_attendees(
[self.partnerA.id],
datetime(2023, 12, 25).isoformat(),
datetime(2023, 12, 31).isoformat()
)
self.assertEqual(work_hours, [
{'daysOfWeek': [2], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [2], 'startTime': '13:00', 'endTime': '16:00'},
{'daysOfWeek': [3], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [3], 'startTime': '13:00', 'endTime': '16:00'},
{'daysOfWeek': [4], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [4], 'startTime': '13:00', 'endTime': '16:00'},
{'daysOfWeek': [5], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [5], 'startTime': '13:00', 'endTime': '16:00'},
])