Odoo18-Base/addons/hr_calendar/tests/test_working_hours.py

259 lines
13 KiB
Python
Raw Permalink Normal View History

2025-01-06 10:57:38 +07:00
# Part of Odoo. See LICENSE file for full copyright and licensing details.
from odoo.tests import tagged
from datetime import datetime
from odoo.addons.hr_calendar.tests.common import TestHrCalendarCommon
@tagged('work_hours')
class TestWorkingHours(TestHrCalendarCommon):
@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 test_hr_contract_calendar")
def test_working_hours_2_emp_same_calendar(self):
self.env.user.company_id = self.company_A
self.env.user.company_ids = [self.company_A.id]
work_hours = self.env['res.partner'].get_working_hours_for_all_attendees(
[self.partnerA.id, self.partnerD.id],
datetime(2023, 12, 25).isoformat(),
datetime(2023, 12, 31).isoformat(),
)
self.assertEqual(work_hours, [
{'daysOfWeek': [1], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [1], 'startTime': '13:00', 'endTime': '16:00'},
{'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_working_hours_2_emp_different_calendar(self):
self.env.user.company_id = self.company_A
self.env.user.company_ids = [self.company_A.id]
work_hours = self.env['res.partner'].get_working_hours_for_all_attendees(
[self.partnerA.id, self.partnerB.id],
datetime(2023, 12, 25).isoformat(),
datetime(2023, 12, 31).isoformat(),
)
# Nothing on monday due to partnerB's calendar : calendar_28h
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_working_hours_3_emp_different_calendar(self):
self.env.user.company_id = self.company_A
self.env.user.company_ids = [self.company_A.id]
work_hours = self.env['res.partner'].get_working_hours_for_all_attendees(
[self.partnerA.id, self.partnerB.id, self.partnerC.id],
datetime(2023, 12, 25).isoformat(),
datetime(2023, 12, 31).isoformat(),
)
# Nothing on monday due to partnerB's calendar : calendar_28h
# Nothing before 15:00 due to partnerC's calendar : calendar_35h_night
self.assertEqual(work_hours, [
{'daysOfWeek': [2], 'startTime': '15:00', 'endTime': '16:00'},
{'daysOfWeek': [3], 'startTime': '15:00', 'endTime': '16:00'},
{'daysOfWeek': [4], 'startTime': '15:00', 'endTime': '16:00'},
{'daysOfWeek': [5], 'startTime': '15:00', 'endTime': '16:00'}
])
def test_working_hours_2_emp_same_calendar_hours_different_timezone(self):
self.env.user.company_id = self.company_A
self.env.user.company_ids = [self.company_A.id]
calendar_35h_london_tz = self.calendar_35h.copy()
calendar_35h_london_tz.tz = 'Europe/London'
self.employeeD.resource_calendar_id = calendar_35h_london_tz
work_hours = self.env['res.partner'].get_working_hours_for_all_attendees(
[self.partnerA.id, self.partnerD.id],
datetime(2023, 12, 25).isoformat(),
datetime(2023, 12, 31).isoformat(),
)
# calendar_35h_london_tz.tz = UTC, calendar_35h.tz = UTC +1
self.assertEqual(work_hours, [
{'daysOfWeek': [1], 'startTime': '09:00', 'endTime': '12:00'},
{'daysOfWeek': [1], 'startTime': '14:00', 'endTime': '16:00'},
{'daysOfWeek': [2], 'startTime': '09:00', 'endTime': '12:00'},
{'daysOfWeek': [2], 'startTime': '14:00', 'endTime': '16:00'},
{'daysOfWeek': [3], 'startTime': '09:00', 'endTime': '12:00'},
{'daysOfWeek': [3], 'startTime': '14:00', 'endTime': '16:00'},
{'daysOfWeek': [4], 'startTime': '09:00', 'endTime': '12:00'},
{'daysOfWeek': [4], 'startTime': '14:00', 'endTime': '16:00'},
{'daysOfWeek': [5], 'startTime': '09:00', 'endTime': '12:00'},
{'daysOfWeek': [5], 'startTime': '14:00', 'endTime': '16:00'},
])
def test_multi_companies_2_employees_1_selected_company(self):
"""
INPUT:
======
Employees Companies
employee A company A ---> 35h [ ] A
employee A company B ---> 28h [X] B
OUTPUT:
=======
The schedule will be the 28h's schedule
"""
self.env.user.company_id = self.company_B
self.env.user.company_ids = [self.company_B.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'},
])
def test_multi_companies_2_employees_2_selected_companies(self):
"""
INPUT:
======
Employees Companies
employee A company A ---> 35h [X] A <- main company
employee A company B ---> 28h [X] B
OUTPUT:
=======
The schedule will be the 35h's schedule
"""
self.env.user.company_id = self.company_A
self.env.user.company_ids = [self.company_A.id, self.company_B.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': [1], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [1], 'startTime': '13:00', 'endTime': '16:00'},
{'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_union_between_schedules(self):
"""
INPUT:
======
Employees Companies
employee A company A ---> 35h [X] A <- main company
employee A company B ---> 35h night [X] B
OUTPUT:
=======
The schedule will be the union between 35h's and 35h night's schedule
"""
self.env.user.company_id = self.company_A
self.env.user.company_ids = [self.company_A.id, self.company_B.id]
self.employeeA_company_B.resource_calendar_id = self.calendar_35h_night
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': [1], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [1], 'startTime': '13:00', 'endTime': '22:00'},
{'daysOfWeek': [2], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [2], 'startTime': '13:00', 'endTime': '22:00'},
{'daysOfWeek': [3], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [3], 'startTime': '13:00', 'endTime': '22:00'},
{'daysOfWeek': [4], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [4], 'startTime': '13:00', 'endTime': '22:00'},
{'daysOfWeek': [5], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [5], 'startTime': '13:00', 'endTime': '21:00'},
{'daysOfWeek': [0], 'startTime': '15:00', 'endTime': '16:00'},
])
def test_multi_companies_2_employees_1_partner_1_selected_companies(self):
"""
INPUT:
======
Employees Companies
employee B1 company A ---> 28h [X] A <- main company
employee B2 company A ---> 35h night [ ] B
OUTPUT:
=======
The schedule will be the union between 28h's and 35h night's schedule
"""
self.env.user.company_id = self.company_A
self.env.user.company_ids = [self.company_A.id]
self.env['hr.employee'].create({
'name': "Partner B2 - Calendar 35h night",
'tz': "Europe/Brussels",
'resource_calendar_id': self.calendar_35h_night.id,
'work_contact_id': self.partnerB.id,
'company_id': self.company_A.id,
})
work_hours = self.env['res.partner'].get_working_hours_for_all_attendees(
[self.partnerB.id],
datetime(2023, 12, 25).isoformat(),
datetime(2023, 12, 31).isoformat(),
)
self.assertEqual(work_hours, [
{'daysOfWeek': [1], 'startTime': '15:00', 'endTime': '22:00'},
{'daysOfWeek': [2], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [2], 'startTime': '13:00', 'endTime': '22:00'},
{'daysOfWeek': [3], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [3], 'startTime': '13:00', 'endTime': '22:00'},
{'daysOfWeek': [4], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [4], 'startTime': '13:00', 'endTime': '22:00'},
{'daysOfWeek': [5], 'startTime': '08:00', 'endTime': '12:00'},
{'daysOfWeek': [5], 'startTime': '13:00', 'endTime': '21:00'},
{'daysOfWeek': [0], 'startTime': '15:00', 'endTime': '16:00'},
])
def test_work_hours_of_employee_without_time_zone(self):
self.env.user.tz = False
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': [1], 'startTime': '07:00', 'endTime': '11:00'},
{'daysOfWeek': [1], 'startTime': '12:00', 'endTime': '15:00'},
{'daysOfWeek': [2], 'startTime': '07:00', 'endTime': '11:00'},
{'daysOfWeek': [2], 'startTime': '12:00', 'endTime': '15:00'},
{'daysOfWeek': [3], 'startTime': '07:00', 'endTime': '11:00'},
{'daysOfWeek': [3], 'startTime': '12:00', 'endTime': '15:00'},
{'daysOfWeek': [4], 'startTime': '07:00', 'endTime': '11:00'},
{'daysOfWeek': [4], 'startTime': '12:00', 'endTime': '15:00'},
{'daysOfWeek': [5], 'startTime': '07:00', 'endTime': '11:00'},
{'daysOfWeek': [5], 'startTime': '12:00', 'endTime': '15:00'},
])