Shiloh/brotherton-erpnext
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix Leave Application test cases

Browse Source
This commit is contained in:
Shreya 2018-06-07 13:13:17 +05:30
parent 88fe0d59d9
commit 88591314b0
1 changed files with 54 additions and 85 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

107
erpnext/hr/doctype/leave_application/test_leave_application.py
View File

@ -47,15 +47,16 @@ class TestLeaveApplication(unittest.TestCase):
for dt in ["Leave Application", "Leave Allocation", "Salary Slip"]: for dt in ["Leave Application", "Leave Allocation", "Salary Slip"]:
frappe.db.sql("delete from `tab%s`" % dt) frappe.db.sql("delete from `tab%s`" % dt)
@classmethod
def setUpClass(cls):
set_leave_approver()
def tearDown(self): def tearDown(self):
frappe.set_user("Administrator") frappe.set_user("Administrator")
def _clear_roles(self):
frappe.db.sql("""delete from `tabHas Role` where parent in
("test@example.com", "test1@example.com", "test2@example.com")""")
def _clear_applications(self): def _clear_applications(self):
frappe.db.sql("""delete from `tabLeave Application`""") frappe.db.sql("""delete from `tabLeave Application`""")
frappe.db.sql("""delete from `tabDepartment Approver` where parentfield = 'Leave Approver'""")
def get_application(self, doc): def get_application(self, doc):
application = frappe.copy_doc(doc) application = frappe.copy_doc(doc)
@ -64,11 +65,6 @@ class TestLeaveApplication(unittest.TestCase):
return application return application
def test_block_list(self): def test_block_list(self):
self._clear_roles()
from frappe.utils.user import add_role
add_role("test1@example.com", "HR User")
add_role("test1@example.com", "Leave Approver")
clear_user_permissions_for_doctype("Employee") clear_user_permissions_for_doctype("Employee")
frappe.db.set_value("Department", "_Test Department - _TC", frappe.db.set_value("Department", "_Test Department - _TC",
@ -81,7 +77,7 @@ class TestLeaveApplication(unittest.TestCase):
application.status = "Approved" application.status = "Approved"
self.assertRaises(LeaveDayBlockedError, application.submit) self.assertRaises(LeaveDayBlockedError, application.submit)
frappe.set_user("test1@example.com") frappe.set_user("test@example.com")
# clear other applications # clear other applications
frappe.db.sql("delete from `tabLeave Application`") frappe.db.sql("delete from `tabLeave Application`")
@ -90,13 +86,8 @@ class TestLeaveApplication(unittest.TestCase):
self.assertTrue(application.insert()) self.assertTrue(application.insert())
def test_overlap(self): def test_overlap(self):
self._clear_roles()
self._clear_applications() self._clear_applications()
from frappe.utils.user import add_role
add_role("test@example.com", "Employee")
add_role("test2@example.com", "Leave Approver")
frappe.set_user("test@example.com") frappe.set_user("test@example.com")
make_allocation_record() make_allocation_record()
@ -108,13 +99,8 @@ class TestLeaveApplication(unittest.TestCase):
self.assertRaises(OverlapError, application.insert) self.assertRaises(OverlapError, application.insert)
def test_overlap_with_half_day_1(self): def test_overlap_with_half_day_1(self):
self._clear_roles()
self._clear_applications() self._clear_applications()
from frappe.utils.user import add_role
add_role("test@example.com", "Employee")
add_role("test2@example.com", "Leave Approver")
frappe.set_user("test@example.com") frappe.set_user("test@example.com")
make_allocation_record() make_allocation_record()
@ -143,13 +129,8 @@ class TestLeaveApplication(unittest.TestCase):
self.assertRaises(OverlapError, application.insert) self.assertRaises(OverlapError, application.insert)
def test_overlap_with_half_day_2(self): def test_overlap_with_half_day_2(self):
self._clear_roles()
self._clear_applications() self._clear_applications()
from frappe.utils.user import add_role
add_role("test@example.com", "Employee")
add_role("test2@example.com", "Leave Approver")
frappe.set_user("test@example.com") frappe.set_user("test@example.com")
make_allocation_record() make_allocation_record()
@ -166,13 +147,8 @@ class TestLeaveApplication(unittest.TestCase):
self.assertRaises(OverlapError, application.insert) self.assertRaises(OverlapError, application.insert)
def test_overlap_with_half_day_3(self): def test_overlap_with_half_day_3(self):
self._clear_roles()
self._clear_applications() self._clear_applications()
from frappe.utils.user import add_role
add_role("test@example.com", "Employee")
add_role("test2@example.com", "Leave Approver")
frappe.set_user("test@example.com") frappe.set_user("test@example.com")
make_allocation_record() make_allocation_record()
@ -200,61 +176,42 @@ class TestLeaveApplication(unittest.TestCase):
application.half_day_date = "2013-01-05" application.half_day_date = "2013-01-05"
application.insert() application.insert()
def test_global_block_list(self):
self._clear_roles()
from frappe.utils.user import add_role
add_role("test1@example.com", "Employee")
add_role("test@example.com", "Leave Approver")
make_allocation_record(employee="_T-Employee-00002")
application = self.get_application(_test_records[1])
frappe.db.set_value("Leave Block List", "_Test Leave Block List",
"applies_to_all_departments", 1)
frappe.db.set_value("Employee", "_T-Employee-00002", "department",
"_Test Department - _TC")
frappe.set_user("test1@example.com")
application.insert()
application.status = "Approved"
frappe.set_user("test@example.com")
self.assertRaises(LeaveDayBlockedError, application.submit)
frappe.db.set_value("Leave Block List", "_Test Leave Block List",
"applies_to_all_departments", 0)
def test_optional_leave(self): def test_optional_leave(self):
leave_period = get_leave_period() leave_period = get_leave_period()
today = nowdate() today = nowdate()
from datetime import date from datetime import date
holiday_list = frappe.get_doc(dict( holiday_list = 'Test Holiday List for Optional Holiday'
if not frappe.db.exists('Holiday List', holiday_list):
frappe.get_doc(dict(
doctype = 'Holiday List', doctype = 'Holiday List',
holiday_list_name = 'test holiday list for optional holiday', holiday_list_name = holiday_list,
from_date = date(date.today().year, 1, 1), from_date = date(date.today().year, 1, 1),
to_date = date(date.today().year, 12, 31), to_date = date(date.today().year, 12, 31),
holidays = [ holidays = [
dict(holiday_date = today, description = 'test') dict(holiday_date = today, description = 'Test')
] ]
)).insert() )).insert()
employee = get_employee() employee = get_employee()
frappe.db.set_value('Leave Period', leave_period.name, 'optional_holiday_list', holiday_list.name) frappe.db.set_value('Leave Period', leave_period.name, 'optional_holiday_list', holiday_list)
leave_type = frappe.get_doc(dict( leave_type = 'Test Optional Type'
leave_type_name = 'Test Optional Type', if not frappe.db.exists('Leave Type', leave_type):
frappe.get_doc(dict(
leave_type_name = leave_type,
doctype = 'Leave Type', doctype = 'Leave Type',
is_optional_leave = 1 is_optional_leave = 1
)).insert() )).insert()
allocate_leaves(employee, leave_period, leave_type.name, 10) allocate_leaves(employee, leave_period, leave_type, 10)
date = add_days(today, - 1) date = add_days(today, - 1)
leave_application = frappe.get_doc(dict( leave_application = frappe.get_doc(dict(
doctype = 'Leave Application', doctype = 'Leave Application',
employee = employee.name, employee = employee.name,
leave_type = leave_type.name, company = '_Test Company',
leave_type = leave_type,
from_date = date, from_date = date,
to_date = date, to_date = date,
)) ))
@ -269,7 +226,7 @@ class TestLeaveApplication(unittest.TestCase):
leave_application.submit() leave_application.submit()
# check leave balance is reduced # check leave balance is reduced
self.assertEqual(get_leave_balance_on(employee.name, leave_type.name, today), 9) self.assertEqual(get_leave_balance_on(employee.name, leave_type, today), 9)
def test_leaves_allowed(self): def test_leaves_allowed(self):
employee = get_employee() employee = get_employee()
@ -389,9 +346,10 @@ class TestLeaveApplication(unittest.TestCase):
def test_earned_leave(self): def test_earned_leave(self):
leave_period = get_leave_period() leave_period = get_leave_period()
employee = get_employee() employee = get_employee()
leave_type = 'Test Earned Leave Type'
if not frappe.db.exists('Leave Type', leave_type):
leave_type = frappe.get_doc(dict( leave_type = frappe.get_doc(dict(
leave_type_name = 'Test Earned Leave Type', leave_type_name = leave_type,
doctype = 'Leave Type', doctype = 'Leave Type',
is_earned_leave = 1, is_earned_leave = 1,
earned_leave_frequency = 'Monthly', earned_leave_frequency = 'Monthly',
@ -400,18 +358,18 @@ class TestLeaveApplication(unittest.TestCase):
)).insert() )).insert()
leave_policy = frappe.get_doc({ leave_policy = frappe.get_doc({
"doctype": "Leave Policy", "doctype": "Leave Policy",
"leave_policy_details": [{"leave_type": leave_type.name, "annual_allocation": 6}] "leave_policy_details": [{"leave_type": leave_type, "annual_allocation": 6}]
}).insert() }).insert()
frappe.db.set_value("Employee", employee.name, "leave_policy", leave_policy.name) frappe.db.set_value("Employee", employee.name, "leave_policy", leave_policy.name)
allocate_leaves(employee, leave_period, leave_type.name, 0, eligible_leaves = 12) allocate_leaves(employee, leave_period, leave_type, 0, eligible_leaves = 12)
from erpnext.hr.utils import allocate_earned_leaves from erpnext.hr.utils import allocate_earned_leaves
i = 0 i = 0
while(i<14): while(i<14):
allocate_earned_leaves() allocate_earned_leaves()
i += 1 i += 1
self.assertEqual(get_leave_balance_on(employee.name, leave_type.name, nowdate()), 6) self.assertEqual(get_leave_balance_on(employee.name, leave_type, nowdate()), 6)
def make_allocation_record(employee=None, leave_type=None): def make_allocation_record(employee=None, leave_type=None):
frappe.db.sql("delete from `tabLeave Allocation`") frappe.db.sql("delete from `tabLeave Allocation`")
@ -431,6 +389,14 @@ def make_allocation_record(employee=None, leave_type=None):
def get_employee(): def get_employee():
return frappe.get_doc("Employee", "_T-Employee-00001") return frappe.get_doc("Employee", "_T-Employee-00001")
def set_leave_approver():
employee = get_employee()
dept_doc = frappe.get_doc("Department", employee.department)
dept_doc.append('leave_approver', {
'approver': 'test@example.com'
})
dept_doc.save(ignore_permissions=True)
def get_leave_period(): def get_leave_period():
leave_period_name = frappe.db.exists({ leave_period_name = frappe.db.exists({
"doctype": "Leave Period", "doctype": "Leave Period",
@ -449,7 +415,7 @@ def get_leave_period():
)).insert() )).insert()
def allocate_leaves(employee, leave_period, leave_type, new_leaves_allocated, eligible_leaves=0): def allocate_leaves(employee, leave_period, leave_type, new_leaves_allocated, eligible_leaves=0):
frappe.get_doc({ allocate_leave = frappe.get_doc({
"doctype": "Leave Allocation", "doctype": "Leave Allocation",
"__islocal": 1, "__islocal": 1,
"employee": employee.name, "employee": employee.name,
@ -460,3 +426,6 @@ def allocate_leaves(employee, leave_period, leave_type, new_leaves_allocated, el
"new_leaves_allocated": new_leaves_allocated, "new_leaves_allocated": new_leaves_allocated,
"docstatus": 1 "docstatus": 1
}).insert() }).insert()
allocate_leave.submit()