oca-ocb-hr/odoo-bringout-oca-ocb-hr_fleet/hr_fleet/tests/test_mail_activity_plan.py
Ernad Husremovic e1d89e11e3 19.0 vanilla
2026-03-09 09:31:00 +01:00

84 lines
3.6 KiB
Python

# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.
from odoo import Command
from odoo.addons.hr.tests.test_mail_activity_plan import ActivityScheduleHRCase
from odoo.exceptions import ValidationError
from odoo.tests import tagged, users
@tagged('mail_activity', 'mail_activity_plan')
class TestActivitySchedule(ActivityScheduleHRCase):
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.plan_fleet = cls.env['mail.activity.plan'].create({
'name': 'Car return plan',
'res_model': 'hr.employee',
'template_ids': [Command.create({
'activity_type_id': cls.activity_type_todo.id,
'responsible_type': 'fleet_manager',
'summary': 'Car return',
})]
})
cls.brand = cls.env["fleet.vehicle.model.brand"].create({"name": "Audi"})
for employee in (cls.employee_1, cls.employee_2):
car = cls.env["fleet.vehicle"].create({
"driver_id": employee.user_id.partner_id.id,
"model_id": cls.env["fleet.vehicle.model"].create({
"brand_id": cls.brand.id,
"name": "A3",
}).id,
"manager_id": cls.user_manager.id,
"plan_to_change_car": False,
})
employee.car_ids = car
@users('admin')
def test_responsible(self):
""" Check that the responsible is correctly configured. """
for employees in (self.employee_1, self.employee_1 + self.employee_2):
employees = employees.with_env(self.env)
# Happy case
form = self._instantiate_activity_schedule_wizard(employees)
form.plan_id = self.plan_fleet
schedule_lines = form.plan_schedule_line_ids._records
self.assertEqual(len(schedule_lines), 1)
self.assertEqual(schedule_lines[0]['line_description'], 'Car return')
if len(employees) == 1:
self.assertEqual(schedule_lines[0]['responsible_user_id'], self.user_manager.id)
else:
self.assertEqual(schedule_lines[0]['responsible_user_id'], False)
self.assertFalse(form.has_error)
wizard = form.save()
wizard.action_schedule_plan()
for employee in employees:
activities = self.get_last_activities(employee, 1)
self.assertEqual(len(activities), 1)
self.assertEqual(activities[0].user_id, self.user_manager)
employees = (self.employee_1 + self.employee_2).with_env(self.env)
# Cases with errors
self.employee_1.car_ids[0].manager_id = False
form = self._instantiate_activity_schedule_wizard(employees)
form.plan_id = self.plan_fleet
self.assertTrue(form.has_warning)
n_warning = form.warning.count('<li>')
self.assertEqual(n_warning, 1)
self.assertIn(f"The vehicle of employee {self.employee_1.name} is not linked to a fleet manager, assigning to you.", form.warning)
# assert form can now be saved without raising an error
form.save()
self.employee_1.car_ids = self.env["fleet.vehicle"]
form = self._instantiate_activity_schedule_wizard(employees)
form.plan_id = self.plan_fleet
self.assertTrue(form.has_error)
n_error = form.error.count('<li>')
self.assertEqual(n_error, 1)
self.assertIn(f"Employee {self.employee_1.name} is not linked to a vehicle.", form.error)
with self.assertRaises(ValidationError):
form.save()