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

chore: clean-up leave policy assignment tests

Browse Source
This commit is contained in:
Rucha Mahabal 2022-02-11 17:40:20 +05:30
parent c7be9ef5d2
commit 51e6086829
2 changed files with 36 additions and 81 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
erpnext/hr/doctype/leave_policy_assignment/leave_policy_assignment.py
View File

@ -12,7 +12,6 @@ from frappe.utils import date_diff, flt, formatdate, get_last_day, getdate
class LeavePolicyAssignment(Document): class LeavePolicyAssignment(Document):
def validate(self): def validate(self):
self.validate_policy_assignment_overlap() self.validate_policy_assignment_overlap()
self.set_dates() self.set_dates()

116
erpnext/hr/doctype/leave_policy_assignment/test_leave_policy_assignment.py
View File

@ -22,34 +22,27 @@ class TestLeavePolicyAssignment(unittest.TestCase):
for doctype in ["Leave Period", "Leave Application", "Leave Allocation", "Leave Policy Assignment", "Leave Ledger Entry"]: for doctype in ["Leave Period", "Leave Application", "Leave Allocation", "Leave Policy Assignment", "Leave Ledger Entry"]:
frappe.db.sql("delete from `tab{0}`".format(doctype)) #nosec frappe.db.sql("delete from `tab{0}`".format(doctype)) #nosec
self.employee = get_employee()
def test_grant_leaves(self): def test_grant_leaves(self):
leave_period = get_leave_period() leave_period = get_leave_period()
employee = get_employee() # allocation = 10
# create the leave policy with leave type "_Test Leave Type", allocation = 10
leave_policy = create_leave_policy() leave_policy = create_leave_policy()
leave_policy.submit() leave_policy.submit()
data = { data = {
"assignment_based_on": "Leave Period", "assignment_based_on": "Leave Period",
"leave_policy": leave_policy.name, "leave_policy": leave_policy.name,
"leave_period": leave_period.name "leave_period": leave_period.name
} }
leave_policy_assignments = create_assignment_for_multiple_employees([self.employee.name], frappe._dict(data))
leave_policy_assignments = create_assignment_for_multiple_employees([employee.name], frappe._dict(data)) self.assertEqual(frappe.db.get_value("Leave Policy Assignment", leave_policy_assignments[0], "leaves_allocated"), 1)
leave_policy_assignment_doc = frappe.get_doc("Leave Policy Assignment", leave_policy_assignments[0])
leave_policy_assignment_doc.reload()
self.assertEqual(leave_policy_assignment_doc.leaves_allocated, 1)
leave_allocation = frappe.get_list("Leave Allocation", filters={ leave_allocation = frappe.get_list("Leave Allocation", filters={
"employee": employee.name, "employee": self.employee.name,
"leave_policy":leave_policy.name, "leave_policy":leave_policy.name,
"leave_policy_assignment": leave_policy_assignments[0], "leave_policy_assignment": leave_policy_assignments[0],
"docstatus": 1})[0] "docstatus": 1})[0]
leave_alloc_doc = frappe.get_doc("Leave Allocation", leave_allocation) leave_alloc_doc = frappe.get_doc("Leave Allocation", leave_allocation)
self.assertEqual(leave_alloc_doc.new_leaves_allocated, 10) self.assertEqual(leave_alloc_doc.new_leaves_allocated, 10)
@ -61,49 +54,32 @@ class TestLeavePolicyAssignment(unittest.TestCase):
def test_allow_to_grant_all_leave_after_cancellation_of_every_leave_allocation(self): def test_allow_to_grant_all_leave_after_cancellation_of_every_leave_allocation(self):
leave_period = get_leave_period() leave_period = get_leave_period()
employee = get_employee()
# create the leave policy with leave type "_Test Leave Type", allocation = 10 # create the leave policy with leave type "_Test Leave Type", allocation = 10
leave_policy = create_leave_policy() leave_policy = create_leave_policy()
leave_policy.submit() leave_policy.submit()
data = { data = {
"assignment_based_on": "Leave Period", "assignment_based_on": "Leave Period",
"leave_policy": leave_policy.name, "leave_policy": leave_policy.name,
"leave_period": leave_period.name "leave_period": leave_period.name
} }
leave_policy_assignments = create_assignment_for_multiple_employees([self.employee.name], frappe._dict(data))
leave_policy_assignments = create_assignment_for_multiple_employees([employee.name], frappe._dict(data))
leave_policy_assignment_doc = frappe.get_doc("Leave Policy Assignment", leave_policy_assignments[0])
leave_policy_assignment_doc.reload()
# every leave is allocated no more leave can be granted now # every leave is allocated no more leave can be granted now
self.assertEqual(leave_policy_assignment_doc.leaves_allocated, 1) self.assertEqual(frappe.db.get_value("Leave Policy Assignment", leave_policy_assignments[0], "leaves_allocated"), 1)
leave_allocation = frappe.get_list("Leave Allocation", filters={ leave_allocation = frappe.get_list("Leave Allocation", filters={
"employee": employee.name, "employee": self.employee.name,
"leave_policy":leave_policy.name, "leave_policy":leave_policy.name,
"leave_policy_assignment": leave_policy_assignments[0], "leave_policy_assignment": leave_policy_assignments[0],
"docstatus": 1})[0] "docstatus": 1})[0]
leave_alloc_doc = frappe.get_doc("Leave Allocation", leave_allocation) leave_alloc_doc = frappe.get_doc("Leave Allocation", leave_allocation)
# User all allowed to grant leave when there is no allocation against assignment
leave_alloc_doc.cancel() leave_alloc_doc.cancel()
leave_alloc_doc.delete() leave_alloc_doc.delete()
self.assertEqual(frappe.db.get_value("Leave Policy Assignment", leave_policy_assignments[0], "leaves_allocated"), 0)
leave_policy_assignment_doc.reload()
# User are now allowed to grant leave
self.assertEqual(leave_policy_assignment_doc.leaves_allocated, 0)
def test_earned_leave_allocation(self): def test_earned_leave_allocation(self):
leave_period = create_leave_period("Test Earned Leave Period") leave_period = create_leave_period("Test Earned Leave Period")
employee = get_employee()
leave_type = create_earned_leave_type("Test Earned Leave") leave_type = create_earned_leave_type("Test Earned Leave")
leave_policy = frappe.get_doc({ leave_policy = frappe.get_doc({
@ -117,7 +93,7 @@ class TestLeavePolicyAssignment(unittest.TestCase):
"leave_policy": leave_policy.name, "leave_policy": leave_policy.name,
"leave_period": leave_period.name "leave_period": leave_period.name
} }
leave_policy_assignments = create_assignment_for_multiple_employees([employee.name], frappe._dict(data)) leave_policy_assignments = create_assignment_for_multiple_employees([self.employee.name], frappe._dict(data))
# leaves allocated should be 0 since it is an earned leave and allocation happens via scheduler based on set frequency # leaves allocated should be 0 since it is an earned leave and allocation happens via scheduler based on set frequency
leaves_allocated = frappe.db.get_value("Leave Allocation", { leaves_allocated = frappe.db.get_value("Leave Allocation", {
@ -125,16 +101,8 @@ class TestLeavePolicyAssignment(unittest.TestCase):
}, "total_leaves_allocated") }, "total_leaves_allocated")
self.assertEqual(leaves_allocated, 0) self.assertEqual(leaves_allocated, 0)
def test_earned_leave_allocation_for_passed_months(self): def test_earned_leave_alloc_for_passed_months_based_on_leave_period(self):
employee = get_employee() leave_period, leave_policy = setup_leave_period_and_policy(get_first_day(add_months(getdate(), -1)))
leave_type = create_earned_leave_type("Test Earned Leave")
leave_period = create_leave_period("Test Earned Leave Period",
start_date=get_first_day(add_months(getdate(), -1)))
leave_policy = frappe.get_doc({
"doctype": "Leave Policy",
"title": "Test Leave Policy",
"leave_policy_details": [{"leave_type": leave_type.name, "annual_allocation": 12}]
}).insert()
# Case 1: assignment created one month after the leave period, should allocate 1 leave # Case 1: assignment created one month after the leave period, should allocate 1 leave
frappe.flags.current_date = get_first_day(getdate()) frappe.flags.current_date = get_first_day(getdate())
@ -143,24 +111,15 @@ class TestLeavePolicyAssignment(unittest.TestCase):
"leave_policy": leave_policy.name, "leave_policy": leave_policy.name,
"leave_period": leave_period.name "leave_period": leave_period.name
} }
leave_policy_assignments = create_assignment_for_multiple_employees([employee.name], frappe._dict(data)) leave_policy_assignments = create_assignment_for_multiple_employees([self.employee.name], frappe._dict(data))
leaves_allocated = frappe.db.get_value("Leave Allocation", { leaves_allocated = frappe.db.get_value("Leave Allocation", {
"leave_policy_assignment": leave_policy_assignments[0] "leave_policy_assignment": leave_policy_assignments[0]
}, "total_leaves_allocated") }, "total_leaves_allocated")
self.assertEqual(leaves_allocated, 1) self.assertEqual(leaves_allocated, 1)
def test_earned_leave_allocation_for_passed_months_on_month_end(self): def test_earned_leave_alloc_for_passed_months_based_on_leave_period(self):
employee = get_employee() leave_period, leave_policy = setup_leave_period_and_policy(get_first_day(add_months(getdate(), -2)))
leave_type = create_earned_leave_type("Test Earned Leave")
leave_period = create_leave_period("Test Earned Leave Period",
start_date=get_first_day(add_months(getdate(), -2)))
leave_policy = frappe.get_doc({
"doctype": "Leave Policy",
"title": "Test Leave Policy",
"leave_policy_details": [{"leave_type": leave_type.name, "annual_allocation": 12}]
}).insert()
# Case 2: assignment created on the last day of the leave period's latter month # Case 2: assignment created on the last day of the leave period's latter month
# should allocate 1 leave for current month even though the month has not ended # should allocate 1 leave for current month even though the month has not ended
# since the daily job might have already executed # since the daily job might have already executed
@ -171,7 +130,7 @@ class TestLeavePolicyAssignment(unittest.TestCase):
"leave_policy": leave_policy.name, "leave_policy": leave_policy.name,
"leave_period": leave_period.name "leave_period": leave_period.name
} }
leave_policy_assignments = create_assignment_for_multiple_employees([employee.name], frappe._dict(data)) leave_policy_assignments = create_assignment_for_multiple_employees([self.employee.name], frappe._dict(data))
leaves_allocated = frappe.db.get_value("Leave Allocation", { leaves_allocated = frappe.db.get_value("Leave Allocation", {
"leave_policy_assignment": leave_policy_assignments[0] "leave_policy_assignment": leave_policy_assignments[0]
@ -188,33 +147,17 @@ class TestLeavePolicyAssignment(unittest.TestCase):
}, "total_leaves_allocated") }, "total_leaves_allocated")
self.assertEqual(leaves_allocated, 3) self.assertEqual(leaves_allocated, 3)
def test_earned_leave_allocation_for_passed_months_with_carry_forwarded_leaves(self): def test_earned_leave_alloc_for_passed_months_with_cf_leaves_based_on_leave_period(self):
from erpnext.hr.doctype.leave_allocation.test_leave_allocation import create_leave_allocation from erpnext.hr.doctype.leave_allocation.test_leave_allocation import create_leave_allocation
employee = get_employee() leave_period, leave_policy = setup_leave_period_and_policy(get_first_day(add_months(getdate(), -2)))
leave_type = create_earned_leave_type("Test Earned Leave")
leave_period = create_leave_period("Test Earned Leave Period",
start_date=get_first_day(add_months(getdate(), -2)))
leave_policy = frappe.get_doc({
"doctype": "Leave Policy",
"title": "Test Leave Policy",
"leave_policy_details": [{"leave_type": leave_type.name, "annual_allocation": 12}]
}).insert()
# initial leave allocation = 5 # initial leave allocation = 5
leave_allocation = create_leave_allocation( leave_allocation = create_leave_allocation(employee=self.employee.name, employee_name=self.employee.employee_name, leave_type="Test Earned Leave",
employee=employee.name, from_date=add_months(getdate(), -12), to_date=add_months(getdate(), -3), new_leaves_allocated=5, carry_forward=0)
employee_name=employee.employee_name,
leave_type=leave_type.name,
from_date=add_months(getdate(), -12),
to_date=add_months(getdate(), -3),
new_leaves_allocated=5,
carry_forward=0)
leave_allocation.submit() leave_allocation.submit()
# Case 3: assignment created on the last day of the leave period's latter month with carry forwarding # Case 3: assignment created on the last day of the leave period's latter month with carry forwarding
frappe.flags.current_date = get_last_day(add_months(getdate(), -1)) frappe.flags.current_date = get_last_day(add_months(getdate(), -1))
data = { data = {
"assignment_based_on": "Leave Period", "assignment_based_on": "Leave Period",
"leave_policy": leave_policy.name, "leave_policy": leave_policy.name,
@ -222,7 +165,7 @@ class TestLeavePolicyAssignment(unittest.TestCase):
"carry_forward": 1 "carry_forward": 1
} }
# carry forwarded leaves = 5, 3 leaves allocated for passed months # carry forwarded leaves = 5, 3 leaves allocated for passed months
leave_policy_assignments = create_assignment_for_multiple_employees([employee.name], frappe._dict(data)) leave_policy_assignments = create_assignment_for_multiple_employees([self.employee.name], frappe._dict(data))
details = frappe.db.get_value("Leave Allocation", { details = frappe.db.get_value("Leave Allocation", {
"leave_policy_assignment": leave_policy_assignments[0] "leave_policy_assignment": leave_policy_assignments[0]
@ -269,4 +212,17 @@ def create_leave_period(name, start_date=None):
to_date=add_months(start_date, 12), to_date=add_months(start_date, 12),
company="_Test Company", company="_Test Company",
is_active=1 is_active=1
)).insert() )).insert()
def setup_leave_period_and_policy(start_date):
leave_type = create_earned_leave_type("Test Earned Leave")
leave_period = create_leave_period("Test Earned Leave Period",
start_date=start_date)
leave_policy = frappe.get_doc({
"doctype": "Leave Policy",
"title": "Test Leave Policy",
"leave_policy_details": [{"leave_type": leave_type.name, "annual_allocation": 12}]
}).insert()
return leave_period, leave_policy