test: monthly attendance sheet
This commit is contained in:
Rucha Mahabal
parent
41cfcdba44
commit
acb27430ac
@ -1,18 +1,27 @@
|
||||
import frappe
|
||||
from dateutil.relativedelta import relativedelta
|
||||
from frappe.tests.utils import FrappeTestCase
|
||||
from frappe.utils import now_datetime
|
||||
from frappe.utils import get_year_ending, get_year_start, getdate, now_datetime
|
||||
|
||||
from erpnext.hr.doctype.attendance.attendance import mark_attendance
|
||||
from erpnext.hr.doctype.employee.test_employee import make_employee
|
||||
from erpnext.hr.report.monthly_attendance_sheet.monthly_attendance_sheet import execute
|
||||
from erpnext.hr.doctype.holiday_list.test_holiday_list import set_holiday_list
|
||||
from erpnext.hr.doctype.leave_application.test_leave_application import make_allocation_record
|
||||
from erpnext.hr.report.monthly_attendance_sheet.monthly_attendance_sheet import (
|
||||
execute,
|
||||
get_total_days_in_month,
|
||||
)
|
||||
from erpnext.payroll.doctype.salary_slip.test_salary_slip import make_leave_application
|
||||
|
||||
test_dependencies = ["Shift Type"]
|
||||
|
||||
|
||||
class TestMonthlyAttendanceSheet(FrappeTestCase):
|
||||
def setUp(self):
|
||||
self.employee = make_employee("test_employee@example.com")
|
||||
frappe.db.delete("Attendance", {"employee": self.employee})
|
||||
self.employee = make_employee("test_employee@example.com", company="_Test Company")
|
||||
frappe.db.delete("Attendance")
|
||||
|
||||
@set_holiday_list("Salary Slip Test Holiday List", "_Test Company")
|
||||
def test_monthly_attendance_sheet_report(self):
|
||||
now = now_datetime()
|
||||
previous_month = now.month - 1
|
||||
@ -35,13 +44,144 @@ class TestMonthlyAttendanceSheet(FrappeTestCase):
|
||||
report = execute(filters=filters)
|
||||
|
||||
record = report[1][0]
|
||||
datasets = report[3]['data']['datasets']
|
||||
absent = datasets[0]['values']
|
||||
present = datasets[1]['values']
|
||||
leaves = datasets[2]['values']
|
||||
datasets = report[3]["data"]["datasets"]
|
||||
absent = datasets[0]["values"]
|
||||
present = datasets[1]["values"]
|
||||
leaves = datasets[2]["values"]
|
||||
|
||||
# ensure correct attendance is reflect on the report
|
||||
self.assertEqual(self.employee, record.get('employee'))
|
||||
self.assertEqual(self.employee, record.get("employee"))
|
||||
self.assertEqual(absent[0], 1)
|
||||
self.assertEqual(present[1], 1)
|
||||
self.assertEqual(leaves[2], 1)
|
||||
|
||||
@set_holiday_list("Salary Slip Test Holiday List", "_Test Company")
|
||||
def test_monthly_attendance_sheet_with_detailed_view(self):
|
||||
now = now_datetime()
|
||||
previous_month = now.month - 1
|
||||
previous_month_first = now.replace(day=1).replace(month=previous_month).date()
|
||||
|
||||
company = frappe.db.get_value("Employee", self.employee, "company")
|
||||
|
||||
# attendance with shift
|
||||
mark_attendance(self.employee, previous_month_first, "Absent", "Day Shift")
|
||||
mark_attendance(
|
||||
self.employee, previous_month_first + relativedelta(days=1), "Present", "Day Shift"
|
||||
)
|
||||
|
||||
# attendance without shift
|
||||
mark_attendance(self.employee, previous_month_first + relativedelta(days=2), "On Leave")
|
||||
mark_attendance(self.employee, previous_month_first + relativedelta(days=3), "Present")
|
||||
|
||||
filters = frappe._dict(
|
||||
{
|
||||
"month": previous_month,
|
||||
"year": now.year,
|
||||
"company": company,
|
||||
}
|
||||
)
|
||||
report = execute(filters=filters)
|
||||
|
||||
day_shift_row = report[1][0]
|
||||
row_without_shift = report[1][1]
|
||||
|
||||
self.assertEqual(day_shift_row["shift"], "Day Shift")
|
||||
self.assertEqual(day_shift_row[1], "A") # absent on the 1st day of the month
|
||||
self.assertEqual(day_shift_row[2], "P") # present on the 2nd day
|
||||
|
||||
self.assertEqual(row_without_shift["shift"], None)
|
||||
self.assertEqual(row_without_shift[3], "L") # on leave on the 3rd day
|
||||
self.assertEqual(row_without_shift[4], "P") # present on the 4th day
|
||||
|
||||
@set_holiday_list("Salary Slip Test Holiday List", "_Test Company")
|
||||
def test_monthly_attendance_sheet_with_summarized_view(self):
|
||||
now = now_datetime()
|
||||
previous_month = now.month - 1
|
||||
previous_month_first = now.replace(day=1).replace(month=previous_month).date()
|
||||
|
||||
company = frappe.db.get_value("Employee", self.employee, "company")
|
||||
|
||||
# attendance with shift
|
||||
mark_attendance(self.employee, previous_month_first, "Absent", "Day Shift")
|
||||
mark_attendance(
|
||||
self.employee, previous_month_first + relativedelta(days=1), "Present", "Day Shift"
|
||||
)
|
||||
|
||||
mark_attendance(
|
||||
self.employee, previous_month_first + relativedelta(days=3), "Present"
|
||||
) # attendance without shift
|
||||
mark_attendance(
|
||||
self.employee, previous_month_first + relativedelta(days=4), "Present", late_entry=1
|
||||
) # late entry
|
||||
mark_attendance(
|
||||
self.employee, previous_month_first + relativedelta(days=5), "Present", early_exit=1
|
||||
) # early exit
|
||||
|
||||
leave_application = get_leave_application(self.employee)
|
||||
|
||||
filters = frappe._dict(
|
||||
{"month": previous_month, "year": now.year, "company": company, "summarized_view": 1}
|
||||
)
|
||||
report = execute(filters=filters)
|
||||
|
||||
row = report[1][0]
|
||||
self.assertEqual(row["employee"], self.employee)
|
||||
self.assertEqual(row["total_present"], 4)
|
||||
self.assertEqual(row["total_absent"], 1)
|
||||
self.assertEqual(row["total_leaves"], leave_application.total_leave_days)
|
||||
|
||||
self.assertEqual(row["_test_leave_type"], leave_application.total_leave_days)
|
||||
self.assertEqual(row["total_late_entries"], 1)
|
||||
self.assertEqual(row["total_early_exits"], 1)
|
||||
|
||||
@set_holiday_list("Salary Slip Test Holiday List", "_Test Company")
|
||||
def test_attendance_with_group_by_filter(self):
|
||||
now = now_datetime()
|
||||
previous_month = now.month - 1
|
||||
previous_month_first = now.replace(day=1).replace(month=previous_month).date()
|
||||
|
||||
company = frappe.db.get_value("Employee", self.employee, "company")
|
||||
|
||||
# attendance with shift
|
||||
mark_attendance(self.employee, previous_month_first, "Absent", "Day Shift")
|
||||
mark_attendance(
|
||||
self.employee, previous_month_first + relativedelta(days=1), "Present", "Day Shift"
|
||||
)
|
||||
|
||||
# attendance without shift
|
||||
mark_attendance(self.employee, previous_month_first + relativedelta(days=2), "On Leave")
|
||||
mark_attendance(self.employee, previous_month_first + relativedelta(days=3), "Present")
|
||||
|
||||
filters = frappe._dict(
|
||||
{"month": previous_month, "year": now.year, "company": company, "group_by": "Department"}
|
||||
)
|
||||
report = execute(filters=filters)
|
||||
|
||||
department = frappe.db.get_value("Employee", self.employee, "department")
|
||||
department_row = report[1][0]
|
||||
self.assertIn(department, department_row["department"])
|
||||
|
||||
day_shift_row = report[1][1]
|
||||
row_without_shift = report[1][2]
|
||||
|
||||
self.assertEqual(day_shift_row["shift"], "Day Shift")
|
||||
self.assertEqual(day_shift_row[1], "A") # absent on the 1st day of the month
|
||||
self.assertEqual(day_shift_row[2], "P") # present on the 2nd day
|
||||
|
||||
self.assertEqual(row_without_shift["shift"], None)
|
||||
self.assertEqual(row_without_shift[3], "L") # on leave on the 3rd day
|
||||
self.assertEqual(row_without_shift[4], "P") # present on the 4th day
|
||||
|
||||
|
||||
def get_leave_application(employee):
|
||||
now = now_datetime()
|
||||
previous_month = now.month - 1
|
||||
|
||||
date = getdate()
|
||||
year_start = getdate(get_year_start(date))
|
||||
year_end = getdate(get_year_ending(date))
|
||||
make_allocation_record(employee=employee, from_date=year_start, to_date=year_end)
|
||||
|
||||
from_date = now.replace(day=7).replace(month=previous_month).date()
|
||||
to_date = now.replace(day=8).replace(month=previous_month).date()
|
||||
return make_leave_application(employee, from_date, to_date, "_Test Leave Type")
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user