222 lines
11 KiB
Python
222 lines
11 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
||
|
|
|
||
|
|
from datetime import datetime
|
||
|
|
from dateutil.relativedelta import relativedelta
|
||
|
|
from freezegun import freeze_time
|
||
|
|
|
||
|
|
from odoo import fields, SUPERUSER_ID
|
||
|
|
|
||
|
|
from odoo.exceptions import UserError
|
||
|
|
from odoo.tests import common, new_test_user
|
||
|
|
from odoo.addons.hr_timesheet.tests.test_timesheet import TestCommonTimesheet
|
||
|
|
import time
|
||
|
|
|
||
|
|
|
||
|
|
class TestTimesheetHolidaysCreate(common.TransactionCase):
|
||
|
|
|
||
|
|
def test_status_create(self):
|
||
|
|
"""Ensure that when a status is created, it fullfills the project and task constrains"""
|
||
|
|
status = self.env['hr.leave.type'].create({
|
||
|
|
'name': 'A nice Time Off Type',
|
||
|
|
'requires_allocation': 'no'
|
||
|
|
})
|
||
|
|
|
||
|
|
self.assertEqual(status.timesheet_project_id, status.company_id.internal_project_id, 'The default project linked to the status should be the same as the company')
|
||
|
|
self.assertEqual(status.timesheet_task_id, status.company_id.leave_timesheet_task_id, 'The default task linked to the status should be the same as the company')
|
||
|
|
|
||
|
|
def test_company_create(self):
|
||
|
|
main_company = self.env.ref('base.main_company')
|
||
|
|
user = new_test_user(self.env, login='fru',
|
||
|
|
groups='base.group_user,base.group_erp_manager,base.group_partner_manager',
|
||
|
|
company_id=main_company.id,
|
||
|
|
company_ids=[(6, 0, main_company.ids)])
|
||
|
|
Company = self.env['res.company']
|
||
|
|
Company = Company.with_user(user)
|
||
|
|
Company = Company.with_company(main_company)
|
||
|
|
company = Company.create({'name': "Wall Company"})
|
||
|
|
self.assertEqual(company.internal_project_id.sudo().company_id, company, "It should have created a project for the company")
|
||
|
|
|
||
|
|
class TestTimesheetHolidays(TestCommonTimesheet):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
super(TestTimesheetHolidays, self).setUp()
|
||
|
|
|
||
|
|
self.employee_working_calendar = self.empl_employee.resource_calendar_id
|
||
|
|
# leave dates : from next monday to next wednesday (to avoid crashing tests on weekend, when
|
||
|
|
# there is no work days in working calendar)
|
||
|
|
# NOTE: second and millisecond can add a working days
|
||
|
|
self.leave_start_datetime = datetime(2018, 2, 5) # this is monday
|
||
|
|
self.leave_end_datetime = self.leave_start_datetime + relativedelta(days=2)
|
||
|
|
|
||
|
|
# all company have those internal project/task (created by default)
|
||
|
|
self.internal_project = self.env.company.internal_project_id
|
||
|
|
self.internal_task_leaves = self.env.company.leave_timesheet_task_id
|
||
|
|
|
||
|
|
self.hr_leave_type_with_ts = self.env['hr.leave.type'].create({
|
||
|
|
'name': 'Time Off Type with timesheet generation',
|
||
|
|
'requires_allocation': 'no',
|
||
|
|
'timesheet_generate': True,
|
||
|
|
'timesheet_project_id': self.internal_project.id,
|
||
|
|
'timesheet_task_id': self.internal_task_leaves.id,
|
||
|
|
})
|
||
|
|
self.hr_leave_type_no_ts = self.env['hr.leave.type'].create({
|
||
|
|
'name': 'Time Off Type without timesheet generation',
|
||
|
|
'requires_allocation': 'no',
|
||
|
|
'timesheet_generate': False,
|
||
|
|
'timesheet_project_id': False,
|
||
|
|
'timesheet_task_id': False,
|
||
|
|
})
|
||
|
|
|
||
|
|
# HR Officer allocates some leaves to the employee 1
|
||
|
|
self.Requests = self.env['hr.leave'].with_context(mail_create_nolog=True, mail_notrack=True)
|
||
|
|
self.Allocations = self.env['hr.leave.allocation'].with_context(mail_create_nolog=True, mail_notrack=True)
|
||
|
|
self.hr_leave_allocation_with_ts = self.Allocations.sudo().create({
|
||
|
|
'name': 'Days for limited category with timesheet',
|
||
|
|
'employee_id': self.empl_employee.id,
|
||
|
|
'holiday_status_id': self.hr_leave_type_with_ts.id,
|
||
|
|
'number_of_days': 10,
|
||
|
|
'state': 'confirm',
|
||
|
|
'date_from': time.strftime('%Y-01-01'),
|
||
|
|
'date_to': time.strftime('%Y-12-31'),
|
||
|
|
})
|
||
|
|
self.hr_leave_allocation_no_ts = self.Allocations.sudo().create({
|
||
|
|
'name': 'Days for limited category without timesheet',
|
||
|
|
'employee_id': self.empl_employee.id,
|
||
|
|
'holiday_status_id': self.hr_leave_type_no_ts.id,
|
||
|
|
'number_of_days': 10,
|
||
|
|
'state': 'confirm',
|
||
|
|
'date_from': time.strftime('%Y-01-01'),
|
||
|
|
'date_to': time.strftime('%Y-12-31'),
|
||
|
|
})
|
||
|
|
|
||
|
|
def test_validate_with_timesheet(self):
|
||
|
|
# employee creates a leave request
|
||
|
|
holiday = self.Requests.with_user(self.user_employee).create({
|
||
|
|
'name': 'Time Off 1',
|
||
|
|
'employee_id': self.empl_employee.id,
|
||
|
|
'holiday_status_id': self.hr_leave_type_with_ts.id,
|
||
|
|
'request_date_from': self.leave_start_datetime,
|
||
|
|
'request_date_to': self.leave_end_datetime,
|
||
|
|
})
|
||
|
|
holiday.with_user(SUPERUSER_ID).action_validate()
|
||
|
|
|
||
|
|
# The leave type and timesheet are linked to the same project and task of hr_leave_type_with_ts as the company is set
|
||
|
|
self.assertEqual(holiday.timesheet_ids.project_id.id, self.hr_leave_type_with_ts.timesheet_project_id.id)
|
||
|
|
self.assertEqual(holiday.timesheet_ids.task_id.id, self.hr_leave_type_with_ts.timesheet_task_id.id)
|
||
|
|
|
||
|
|
self.assertEqual(len(holiday.timesheet_ids), holiday.number_of_days, 'Number of generated timesheets should be the same as the leave duration (1 per day between %s and %s)' % (fields.Datetime.to_string(self.leave_start_datetime), fields.Datetime.to_string(self.leave_end_datetime)))
|
||
|
|
|
||
|
|
# manager refuse the leave
|
||
|
|
holiday.with_user(SUPERUSER_ID).action_refuse()
|
||
|
|
self.assertEqual(len(holiday.timesheet_ids), 0, 'Number of linked timesheets should be zero, since the leave is refused.')
|
||
|
|
|
||
|
|
company = self.env['res.company'].create({"name": "new company"})
|
||
|
|
self.empl_employee.write({
|
||
|
|
"company_id": company.id,
|
||
|
|
})
|
||
|
|
|
||
|
|
hr_leave_type_with_ts_without_company = self.hr_leave_type_with_ts.copy()
|
||
|
|
hr_leave_type_with_ts_without_company.write({
|
||
|
|
'company_id': False,
|
||
|
|
})
|
||
|
|
self.assertTrue(hr_leave_type_with_ts_without_company.timesheet_generate)
|
||
|
|
|
||
|
|
holiday = self.Requests.create({
|
||
|
|
'name': 'Time Off 2',
|
||
|
|
'employee_id': self.empl_employee.id,
|
||
|
|
'holiday_status_id': hr_leave_type_with_ts_without_company.id,
|
||
|
|
'request_date_from': self.leave_start_datetime,
|
||
|
|
'request_date_to': self.leave_end_datetime,
|
||
|
|
})
|
||
|
|
holiday.with_user(SUPERUSER_ID).action_validate()
|
||
|
|
|
||
|
|
# The leave type and timesheet are linked to the same project and task of the employee company as the company is not set
|
||
|
|
self.assertEqual(holiday.timesheet_ids.project_id.id, company.internal_project_id.id)
|
||
|
|
self.assertEqual(holiday.timesheet_ids.task_id.id, company.leave_timesheet_task_id.id)
|
||
|
|
|
||
|
|
def test_validate_without_timesheet(self):
|
||
|
|
# employee creates a leave request
|
||
|
|
holiday = self.Requests.with_user(self.user_employee).create({
|
||
|
|
'name': 'Time Off 1',
|
||
|
|
'employee_id': self.empl_employee.id,
|
||
|
|
'holiday_status_id': self.hr_leave_type_no_ts.id,
|
||
|
|
'request_date_from': self.leave_start_datetime,
|
||
|
|
'request_date_to': self.leave_end_datetime,
|
||
|
|
})
|
||
|
|
holiday.with_user(SUPERUSER_ID).action_validate()
|
||
|
|
self.assertEqual(len(holiday.timesheet_ids), 0, 'Number of generated timesheets should be zero since the leave type does not generate timesheet')
|
||
|
|
|
||
|
|
@freeze_time('2018-02-05') # useful to be able to cancel the validated time off
|
||
|
|
def test_cancel_validate_holidays(self):
|
||
|
|
holiday = self.Requests.with_user(self.user_employee).create({
|
||
|
|
'name': 'Time Off 1',
|
||
|
|
'employee_id': self.empl_employee.id,
|
||
|
|
'holiday_status_id': self.hr_leave_type_with_ts.id,
|
||
|
|
'request_date_from': self.leave_start_datetime,
|
||
|
|
'request_date_to': self.leave_end_datetime,
|
||
|
|
})
|
||
|
|
holiday.with_user(self.env.user).action_validate()
|
||
|
|
self.assertEqual(len(holiday.timesheet_ids), holiday.number_of_days, 'Number of generated timesheets should be the same as the leave duration (1 per day between %s and %s)' % (fields.Datetime.to_string(self.leave_start_datetime), fields.Datetime.to_string(self.leave_end_datetime)))
|
||
|
|
|
||
|
|
self.env['hr.holidays.cancel.leave'].with_user(self.user_employee).with_context(default_leave_id=holiday.id) \
|
||
|
|
.new({'reason': 'Test remove holiday'}) \
|
||
|
|
.action_cancel_leave()
|
||
|
|
self.assertFalse(holiday.active, 'The time off should be archived')
|
||
|
|
self.assertEqual(len(holiday.timesheet_ids), 0, 'The timesheets generated should be unlink.')
|
||
|
|
|
||
|
|
def test_timesheet_time_off_including_public_holiday(self):
|
||
|
|
""" Generate one timesheet for the public holiday and 4 timesheets for the time off.
|
||
|
|
Test Case:
|
||
|
|
=========
|
||
|
|
1) Create a public time off on Wednesday
|
||
|
|
2) In the same week, create a time off during one week for an employee
|
||
|
|
3) Check if there are five timesheets generated for time off and public
|
||
|
|
holiday.4 timesheets should be linked to the time off and 1 for
|
||
|
|
the public one.
|
||
|
|
"""
|
||
|
|
|
||
|
|
leave_start_datetime = datetime(2022, 1, 24, 7, 0, 0, 0) # Monday
|
||
|
|
leave_end_datetime = datetime(2022, 1, 28, 18, 0, 0, 0)
|
||
|
|
|
||
|
|
# Create a public holiday
|
||
|
|
self.env['resource.calendar.leaves'].create({
|
||
|
|
'name': 'Test',
|
||
|
|
'calendar_id': self.employee_working_calendar.id,
|
||
|
|
'date_from': datetime(2022, 1, 26, 7, 0, 0, 0), # This is Wednesday and India Independence
|
||
|
|
'date_to': datetime(2022, 1, 26, 18, 0, 0, 0),
|
||
|
|
})
|
||
|
|
|
||
|
|
holiday = self.Requests.with_user(self.user_employee).create({
|
||
|
|
'name': 'Time Off 1',
|
||
|
|
'employee_id': self.empl_employee.id,
|
||
|
|
'holiday_status_id': self.hr_leave_type_with_ts.id,
|
||
|
|
'request_date_from': leave_start_datetime,
|
||
|
|
'request_date_to': leave_end_datetime,
|
||
|
|
})
|
||
|
|
holiday.with_user(SUPERUSER_ID).action_validate()
|
||
|
|
self.assertEqual(len(holiday.timesheet_ids), 4, '4 timesheets should be generated for this time off.')
|
||
|
|
|
||
|
|
timesheets = self.env['account.analytic.line'].search([
|
||
|
|
('date', '>=', leave_start_datetime),
|
||
|
|
('date', '<=', leave_end_datetime),
|
||
|
|
('employee_id', '=', self.empl_employee.id),
|
||
|
|
])
|
||
|
|
|
||
|
|
# should not able to update timeoff timesheets
|
||
|
|
with self.assertRaises(UserError):
|
||
|
|
timesheets.with_user(self.empl_employee).write({'task_id': 4})
|
||
|
|
|
||
|
|
# should not able to create timesheet in timeoff task
|
||
|
|
with self.assertRaises(UserError):
|
||
|
|
self.env['account.analytic.line'].with_user(self.empl_employee).create({
|
||
|
|
'name': "my timesheet",
|
||
|
|
'project_id': self.internal_project.id,
|
||
|
|
'task_id': self.internal_task_leaves.id,
|
||
|
|
'date': '2021-10-04',
|
||
|
|
'unit_amount': 8.0,
|
||
|
|
})
|
||
|
|
|
||
|
|
self.assertEqual(len(timesheets.filtered('holiday_id')), 4, "4 timesheet should be linked to employee's timeoff")
|
||
|
|
self.assertEqual(len(timesheets.filtered('global_leave_id')), 1, '1 timesheet should be linked to global leave')
|