134 lines
5.0 KiB
Python
134 lines
5.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import pytz
|
|
from datetime import datetime
|
|
from unittest.mock import patch
|
|
|
|
from odoo import fields
|
|
from odoo.tests import new_test_user
|
|
from odoo.tests.common import tagged, TransactionCase, freeze_time
|
|
|
|
|
|
@tagged('attendance_process')
|
|
class TestHrAttendance(TransactionCase):
|
|
"""Test for presence validity"""
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super(TestHrAttendance, cls).setUpClass()
|
|
cls.user = new_test_user(cls.env, login='fru', groups='base.group_user')
|
|
cls.user_no_pin = new_test_user(cls.env, login='gru', groups='base.group_user')
|
|
cls.test_employee = cls.env['hr.employee'].create({
|
|
'name': "François Russie",
|
|
'user_id': cls.user.id,
|
|
'pin': '1234',
|
|
})
|
|
cls.employee_kiosk = cls.env['hr.employee'].create({
|
|
'name': "Machiavel",
|
|
'pin': '5678',
|
|
})
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
# Cache error if not done during setup
|
|
(self.test_employee | self.employee_kiosk).last_attendance_id.unlink()
|
|
|
|
def test_employee_state(self):
|
|
# Make sure the attendance of the employee will display correctly
|
|
assert self.test_employee.attendance_state == 'checked_out'
|
|
self.test_employee._attendance_action_change()
|
|
assert self.test_employee.attendance_state == 'checked_in'
|
|
self.test_employee._attendance_action_change()
|
|
assert self.test_employee.attendance_state == 'checked_out'
|
|
|
|
def test_employee_group_id(self):
|
|
# Create attendance for one of them
|
|
self.env['hr.attendance'].create({
|
|
'employee_id': self.employee_kiosk.id,
|
|
'check_in': '2025-08-01 08:00:00',
|
|
'check_out': '2025-08-01 17:00:00',
|
|
})
|
|
context = self.env.context.copy()
|
|
# Specific to gantt view.
|
|
context['gantt_start_date'] = fields.Datetime.now()
|
|
context['allowed_company_ids'] = [self.env.company.id]
|
|
|
|
groups = self.env['hr.attendance'].read_group(
|
|
domain=[],
|
|
fields=['employee_id'],
|
|
groupby=['employee_id']
|
|
)
|
|
|
|
grouped_employee_ids = [g['employee_id'][0] for g in groups]
|
|
|
|
# Check that only the employee with attendance appears
|
|
self.assertNotIn(self.test_employee.id, grouped_employee_ids)
|
|
self.assertIn(self.employee_kiosk.id, grouped_employee_ids)
|
|
|
|
# Check that no group has a count of 0
|
|
for group in groups:
|
|
self.assertGreater(group['employee_id_count'], 0)
|
|
|
|
groups = self.env['hr.attendance'].with_context(**context).read_group(
|
|
domain=[],
|
|
fields=['employee_id'],
|
|
groupby=['employee_id']
|
|
)
|
|
|
|
grouped_employee_ids = [g['employee_id'][0] for g in groups]
|
|
|
|
# Check that both employees appears
|
|
self.assertIn(self.test_employee.id, grouped_employee_ids)
|
|
self.assertIn(self.employee_kiosk.id, grouped_employee_ids)
|
|
|
|
def test_hours_today(self):
|
|
""" Test day start is correctly computed according to the employee's timezone """
|
|
|
|
def tz_datetime(year, month, day, hour, minute):
|
|
tz = pytz.timezone('Europe/Brussels')
|
|
return tz.localize(datetime(year, month, day, hour, minute)).astimezone(pytz.utc).replace(tzinfo=None)
|
|
|
|
employee = self.env['hr.employee'].create({'name': 'Cunégonde', 'tz': 'Europe/Brussels'})
|
|
self.env['hr.attendance'].create({
|
|
'employee_id': employee.id,
|
|
'check_in': tz_datetime(2019, 3, 1, 22, 0), # should count from midnight in the employee's timezone (=the previous day in utc!)
|
|
'check_out': tz_datetime(2019, 3, 2, 2, 0),
|
|
})
|
|
self.env['hr.attendance'].create({
|
|
'employee_id': employee.id,
|
|
'check_in': tz_datetime(2019, 3, 2, 11, 0),
|
|
})
|
|
|
|
# now = 2019/3/2 14:00 in the employee's timezone
|
|
with patch.object(fields.Datetime, 'now', lambda: tz_datetime(2019, 3, 2, 14, 0).astimezone(pytz.utc).replace(tzinfo=None)):
|
|
self.assertEqual(employee.hours_today, 5, "It should have counted 5 hours")
|
|
|
|
@freeze_time("2024-02-1")
|
|
def test_change_in_out_mode_when_manual_modification(self):
|
|
company = self.env['res.company'].create({
|
|
'name': 'Monsters, Inc.',
|
|
'absence_management': True,
|
|
})
|
|
|
|
employee = self.env['hr.employee'].create({
|
|
'name': "James P. Sullivan",
|
|
'company_id': company.id,
|
|
})
|
|
|
|
self.env['hr.attendance']._cron_absence_detection()
|
|
|
|
attendance = self.env['hr.attendance'].search([('employee_id', '=', employee.id)])
|
|
|
|
self.assertEqual(attendance.in_mode, 'technical')
|
|
self.assertEqual(attendance.out_mode, 'technical')
|
|
self.assertEqual(attendance.color, 1)
|
|
|
|
attendance.write({
|
|
'check_in': datetime(2021, 1, 4, 8, 0),
|
|
'check_out': datetime(2021, 1, 4, 17, 0),
|
|
})
|
|
|
|
self.assertEqual(attendance.in_mode, 'manual')
|
|
self.assertEqual(attendance.out_mode, 'manual')
|
|
self.assertEqual(attendance.color, 0)
|