68 lines
2.9 KiB
Python
68 lines
2.9 KiB
Python
# Part of Odoo. See LICENSE file for full copyright and licensing details
|
|
|
|
from datetime import date, datetime
|
|
import pytz
|
|
|
|
from odoo import fields
|
|
from .common import TestCommonForecast
|
|
|
|
|
|
class TestUnavailabilityForForecasts(TestCommonForecast):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super().setUpClass()
|
|
|
|
cls.setUpEmployees()
|
|
cls.setUpProjects()
|
|
cls.employee_bert.resource_calendar_id.tz = 'UTC'
|
|
# extra employee to test _gantt_unavailability grouped by resource_id
|
|
cls.employee_lionel = cls.env['hr.employee'].create({
|
|
'name': 'lionel',
|
|
'work_email': '[email protected]',
|
|
'tz': 'UTC'
|
|
})
|
|
cls.resource_lionel = cls.employee_lionel.resource_id
|
|
|
|
# employee leaves
|
|
leave_values = {
|
|
'name': 'leave 1',
|
|
'date_from': fields.Datetime.to_string(date(2019, 5, 5)),
|
|
'date_to': fields.Datetime.to_string(date(2019, 5, 18)),
|
|
'resource_id': cls.employee_bert.resource_id.id,
|
|
'calendar_id': cls.employee_bert.resource_calendar_id.id,
|
|
}
|
|
cls.bert_leave = cls.env['resource.calendar.leaves'].create(leave_values)
|
|
|
|
def test__gantt_unavailability_correctly_update_gantt_object(self):
|
|
# create a few forecasts in the opera project
|
|
values = {
|
|
'project_id': self.project_opera.id,
|
|
'allocated_hours': 8,
|
|
'start_datetime': datetime(2019, 8, 6, 0, 0),
|
|
'end_datetime': datetime(2019, 8, 8, 0, 0),
|
|
}
|
|
generated = {}
|
|
for resource in [self.resource_bert, self.resource_lionel, self.resource_joseph]:
|
|
generated[resource.id] = self.env['planning.slot'].create({'resource_id': resource.id, **values})
|
|
|
|
unavailabilities = self.env['planning.slot']._gantt_unavailability(
|
|
'resource_id',
|
|
[self.resource_bert.id, self.resource_joseph.id, self.resource_lionel.id],
|
|
datetime(2019, 1, 1),
|
|
datetime(2019, 1, 7),
|
|
'month',
|
|
)
|
|
|
|
expected_unavailabilities = [
|
|
{'start': datetime(2019, 1, 1, 17, 0, tzinfo=pytz.utc), 'stop': datetime(2019, 1, 2, 8, 0, tzinfo=pytz.utc)},
|
|
{'start': datetime(2019, 1, 2, 17, 0, tzinfo=pytz.utc), 'stop': datetime(2019, 1, 3, 8, 0, tzinfo=pytz.utc)},
|
|
{'start': datetime(2019, 1, 3, 17, 0, tzinfo=pytz.utc), 'stop': datetime(2019, 1, 4, 8, 0, tzinfo=pytz.utc)},
|
|
{'start': datetime(2019, 1, 4, 17, 0, tzinfo=pytz.utc), 'stop': datetime(2019, 1, 7, 0, 0, tzinfo=pytz.utc)},
|
|
]
|
|
|
|
bert_unavailabilities = unavailabilities[self.resource_bert.id]
|
|
lionel_unavailabilities = unavailabilities[self.resource_lionel.id]
|
|
|
|
self.assertEqual(bert_unavailabilities, expected_unavailabilities, 'the gantt object was tranformed for bert')
|
|
self.assertEqual(lionel_unavailabilities, expected_unavailabilities, 'the gantt object was tranformed for lionel')
|