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

test: create leave allocation check

Browse Source
This commit is contained in:
Mangesh-Khairnar 2019-04-11 00:21:11 +05:30
parent bd3b3ea12c
commit c28d2e4b2a
1 changed files with 65 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
erpnext/hr/doctype/leave_allocation/test_leave_allocation.py
View File

@ -1,7 +1,8 @@
from __future__ import unicode_literals from __future__ import unicode_literals
import frappe import frappe
import unittest import unittest
from frappe.utils import getdate from frappe.utils import nowdate, add_months, getdate
from erpnext.hr.doctype.leave_type.test_leave_type import create_leave_type
class TestLeaveAllocation(unittest.TestCase): class TestLeaveAllocation(unittest.TestCase):
def test_overlapping_allocation(self): def test_overlapping_allocation(self):
@ -38,7 +39,7 @@ class TestLeaveAllocation(unittest.TestCase):
def test_invalid_period(self): def test_invalid_period(self):
employee = frappe.get_doc("Employee", frappe.db.sql_list("select name from tabEmployee limit 1")[0]) employee = frappe.get_doc("Employee", frappe.db.sql_list("select name from tabEmployee limit 1")[0])
d = frappe.get_doc({ doc = frappe.get_doc({
"doctype": "Leave Allocation", "doctype": "Leave Allocation",
"__islocal": 1, "__islocal": 1,
"employee": employee.name, "employee": employee.name,
@ -50,11 +51,11 @@ class TestLeaveAllocation(unittest.TestCase):
}) })
#invalid period #invalid period
self.assertRaises(frappe.ValidationError, d.save) self.assertRaises(frappe.ValidationError, doc.save)
def test_allocated_leave_days_over_period(self): def test_allocated_leave_days_over_period(self):
employee = frappe.get_doc("Employee", frappe.db.sql_list("select name from tabEmployee limit 1")[0]) employee = frappe.get_doc("Employee", frappe.db.sql_list("select name from tabEmployee limit 1")[0])
d = frappe.get_doc({ doc = frappe.get_doc({
"doctype": "Leave Allocation", "doctype": "Leave Allocation",
"__islocal": 1, "__islocal": 1,
"employee": employee.name, "employee": employee.name,
@ -64,8 +65,54 @@ class TestLeaveAllocation(unittest.TestCase):
"to_date": getdate("2015-09-30"), "to_date": getdate("2015-09-30"),
"new_leaves_allocated": 35 "new_leaves_allocated": 35
}) })
#allocated leave more than period #allocated leave more than period
self.assertRaises(frappe.ValidationError, d.save) self.assertRaises(frappe.ValidationError, doc.save)
def test_carry_forward_allocation(self):
frappe.db.sql("delete from `tabLeave Allocation`")
employee = frappe.get_doc("Employee", frappe.db.sql_list("select name from tabEmployee limit 1")[0])
doc = frappe.get_doc({
"doctype": "Leave Allocation",
"__islocal": 1,
"employee": employee.name,
"employee_name": employee.employee_name,
"leave_type": "_Test Leave Type Carry Forward",
"from_date": nowdate(),
"to_date": add_months(nowdate(),-12),
"new_leaves_allocated": 10
})
doc.save()
doc = frappe.get_doc({
"doctype": "Leave Allocation",
"__islocal": 1,
"employee": employee.name,
"employee_name": employee.employee_name,
"leave_type": "_Test Leave Type Carry Forward",
"from_date": nowdate(),
"to_date": add_months(now_date(),12),
"carry_forward": 1
})
doc.save()
self.assertEquals(doc.total_leaves_allocated, 10)
def create_leave_allocation(**args):
args = frappe._dict(args)
if not frappe.db.exists("Leave Type", "_Test Leave Type"):
leave_type = create_leave_type(args.leave_type)
leave_type.insert()
employee = frappe.get_doc("Employee", frappe.db.sql_list("select name from tabEmployee limit 1")[0])
leave_allocation = frappe.get_doc({
"doctype": "Leave Allocation",
"__islocal": 1,
"employee": employee.name,
"employee_name": employee.employee_name,
"leave_type": args.leave_type or "_Test Leave Type",
"from_date": args.from_date or nowdate(),
"to_date": args.to_date or add_months(nowdate(), 12),
"new_leaves_allocated": args.new_leaves_allocated or 20
})
return leave_allocation
test_dependencies = ["Employee", "Leave Type"] test_dependencies = ["Employee", "Leave Type"]