flectra/addons/hr_attendance/tests/test_hr_attendance_constraints.py
2018-01-16 02:34:37 -08:00

97 lines
3.6 KiB
Python

# -*- coding: utf-8 -*-
from flectra.tests.common import TransactionCase
import time
class TestHrAttendance(TransactionCase):
"""Tests for attendance date ranges validity"""
def setUp(self):
super(TestHrAttendance, self).setUp()
self.attendance = self.env['hr.attendance']
self.test_employee = self.browse_ref('hr.employee_qdp')
# demo data contains set up for self.test_employee
def test_attendance_in_before_out(self):
# Make sure check_out is before check_in
with self.assertRaises(Exception):
self.my_attend = self.attendance.create({
'employee_id': self.test_employee.id,
'check_in': time.strftime('%Y-%m-10 12:00'),
'check_out': time.strftime('%Y-%m-10 11:00'),
})
def test_attendance_no_check_out(self):
# Make sure no second attandance without check_out can be created
self.attendance.create({
'employee_id': self.test_employee.id,
'check_in': time.strftime('%Y-%m-10 10:00'),
})
with self.assertRaises(Exception):
self.attendance.create({
'employee_id': self.test_employee.id,
'check_in': time.strftime('%Y-%m-10 11:00'),
})
# 5 next tests : Make sure that when attendances overlap an error is raised
def test_attendance_1(self):
with self.assertRaises(Exception):
self.attendance.create({
'employee_id': self.test_employee.id,
'check_in': time.strftime('%Y-%m-10 08:30'),
'check_out': time.strftime('%Y-%m-10 09:30'),
})
def test_attendance_2(self):
with self.assertRaises(Exception):
self.attendance.create({
'employee_id': self.test_employee.id,
'check_in': time.strftime('%Y-%m-10 07:30'),
'check_out': time.strftime('%Y-%m-10 08:30'),
})
def test_attendance_3(self):
with self.assertRaises(Exception):
self.attendance.create({
'employee_id': self.test_employee.id,
'check_in': time.strftime('%Y-%m-10 07:30'),
'check_out': time.strftime('%Y-%m-10 09:30'),
})
def test_attendance_4(self):
with self.assertRaises(Exception):
self.attendance.create({
'employee_id': self.test_employee.id,
'check_in': time.strftime('%Y-%m-10 08:15'),
'check_out': time.strftime('%Y-%m-10 08:45'),
})
def test_attendance_5(self):
self.attendance.create({
'employee_id': self.test_employee.id,
'check_in': time.strftime('%Y-%m-10 10:00'),
})
with self.assertRaises(Exception):
self.attendance.create({
'employee_id': self.test_employee.id,
'check_in': time.strftime('%Y-%m-10 09:30'),
'check_out': time.strftime('%Y-%m-10 10:30'),
})
def test_new_attendances(self):
# Make sure attendance modification raises an error when it causes an overlap
self.attendance.create({
'employee_id': self.test_employee.id,
'check_in': time.strftime('%Y-%m-10 11:00'),
'check_out': time.strftime('%Y-%m-10 12:00'),
})
open_attendance = self.attendance.create({
'employee_id': self.test_employee.id,
'check_in': time.strftime('%Y-%m-10 10:00'),
})
with self.assertRaises(Exception):
open_attendance.write({
'check_out': time.strftime('%Y-%m-10 11:30'),
})