class AttendanceTest(APITestCase):
"""
Testing class for attendance
"""
def setUp(self):
"""
Set up the Testing clas:
"""
super(AttendanceTest, self).setUp()
#self.client.client.login(username='******', password='******')
self.shift = Shift(start_time=time(8, 0), end_time=time(17, 0))
self.shift.save()
self.employee = Employee(shift=self.shift, **employee2_data)
self.employee.save()
# Regular attendance
self.attendance = Attendance(
date=date(2014, 7, 1),
start_time=timezone('Asia/Bangkok').localize(
datetime(2014, 7, 1, 7, 30, 0)),
end_time=timezone('Asia/Bangkok').localize(
datetime(2014, 7, 1, 23, 33, 0)),
employee=self.employee,
shift=self.shift)
self.attendance.save()
# Sunday attendance
self.sunday_attendance = Attendance(
date=date(2016, 2, 7),
start_time=timezone('Asia/Bangkok').localize(
datetime(2016, 2, 7, 8, 02, 0)),
end_time=timezone('Asia/Bangkok').localize(
datetime(2016, 2, 7, 23, 15, 0)),
employee=self.employee,
shift=self.shift)
self.sunday_attendance.save()
self.tz = timezone('Asia/Bangkok')
def test_start_time_property(self):
"""Test that the getter and setter for start time work properly
"""
a = Attendance(employee=self.employee, date=date(2016, 3, 21))
d1 = datetime(2016, 3, 21, 8, 11, 0)
# Test start time without timezone
a.start_time = d1
self.assertEqual(a._start_time, self.tz.localize(d1))
self.assertEqual(a.start_time, self.tz.localize(d1))
# Test start time with timezone
a.start_time = self.tz.localize(d1)
self.assertEqual(a._start_time, self.tz.localize(d1))
self.assertEqual(a.start_time, self.tz.localize(d1))
def test_end_time_property(self):
"""Test that the getter and setter for start time work properly
"""
a = Attendance(employee=self.employee, date=date(2016, 3, 21))
d1 = datetime(2016, 3, 21, 15, 29, 0)
# Test start time without timezone
a.end_time = d1
self.assertEqual(a._end_time, self.tz.localize(d1))
self.assertEqual(a.end_time, self.tz.localize(d1))
# Test start time with timezone
a.end_time = self.tz.localize(d1)
self.assertEqual(a._end_time, self.tz.localize(d1))
self.assertEqual(a.end_time, self.tz.localize(d1))
def test_regular_attedance_regular_hours(self):
"""Test the regular hours of a regular attedance
"""
self.attendance.calculate_times()
self.assertEqual(self.attendance.regular_time, Decimal('8.0'))
self.assertEqual(self.attendance.overtime, Decimal('0'))
def test_regular_attedance_with_overtime_enabled(self):
"""Test the regular hours of a regular attedance
"""
self.attendance.enable_overtime = True
self.attendance.calculate_times()
self.assertEqual(self.attendance.regular_time, Decimal('8.0'))
self.assertEqual(self.attendance.overtime, Decimal('6.5'))
def test_regular_attendance_gross_wage(self):
"""Test the gross wage of a regular attendance
"""
self.attendance.calculate_times()
self.attendance.calculate_gross_wage()
self.assertEqual(self.attendance.regular_pay, Decimal('550'))
self.assertEqual(self.attendance.overtime_pay, Decimal('0'))
self.assertEqual(self.attendance.lunch_pay, Decimal('0'))
self.assertEqual(self.attendance.gross_wage, Decimal('550'))
def test_regular_attendance_gross_wage_with_overtime_enabled(self):
"""Test the gross wage of a regular attendance with overtime enabled
"""
self.attendance.enable_overtime = True
self.attendance.calculate_times()
self.attendance.calculate_gross_wage()
self.assertEqual(self.attendance.regular_pay, Decimal('550'))
# Calculate expected overtime pay
ot_rate = (Decimal('550') / Decimal('8')) * Decimal('1.5')
self.assertEqual(self.attendance.overtime_pay,
ot_rate * Decimal('6.5'))
self.assertEqual(self.attendance.lunch_pay, Decimal('0'))
# Test gross wage
self.assertEqual(self.attendance.gross_wage,
Decimal('550') + (ot_rate * Decimal('6.5')))
def test_regular_attendance_net_wage(self):
"""Test the gross wage of a regular attendance
"""
self.attendance.calculate_times()
self.attendance.calculate_net_wage()
self.assertEqual(self.attendance.gross_wage, Decimal('550'))
self.assertEqual(self.attendance.reimbursement, Decimal('30'))
self.assertEqual(self.attendance.net_wage, Decimal('580'))
def test_regular_attendance_net_wage_with_lunch(self):
"""Test the gross wage of a regular attendance
"""
self.attendance.receive_lunch_overtime = True
self.attendance.calculate_times()
self.attendance.calculate_net_wage()
self.assertEqual(self.attendance.gross_wage, Decimal('653.125'))
self.assertEqual(self.attendance.reimbursement, Decimal('60'))
self.assertEqual(self.attendance.net_wage, Decimal('713.125'))
def test_regular_attendance_net_wage_where_clockin_late(self):
"""Test the net wage where an employee is late
"""
logger.debug(
"\n\n\n\nTesting late clocking for regular attendance\n\n\n")
# Change start time so employee is late
self.attendance.start_time = self.tz.localize(
datetime(2014, 7, 1, 8, 15, 0))
self.attendance.calculate_times()
self.attendance.calculate_net_wage()
self.assertEqual(self.attendance.regular_time, Decimal('7.5'))
self.assertEqual(self.attendance.reimbursement, Decimal('0'))
self.assertEqual(self.attendance.net_wage, Decimal('515.625'))
def test_sunday_attendance_regular_hours(self):
"""Test the regular hours of a regular attedance
"""
logger.debug("\n\n\n\nTesting Sunday Regular attendance hours\n\n\n")
self.sunday_attendance.calculate_times()
self.assertEqual(self.sunday_attendance.regular_time, Decimal('8.0'))
self.assertEqual(self.sunday_attendance.overtime, Decimal('0'))
def test_sunday_attendance_with_overtime_enabled(self):
"""Test the regular hours of a regular attedance
"""
self.sunday_attendance.enable_overtime = True
self.sunday_attendance.calculate_times()
self.assertEqual(self.sunday_attendance.regular_time, Decimal('8.0'))
self.assertEqual(self.sunday_attendance.overtime, Decimal('6'))
def test_sunday_attendance_gross_wage(self):
"""Test the gross wage of a sunday attendance
"""
self.sunday_attendance.calculate_times()
self.sunday_attendance.calculate_gross_wage()
self.assertEqual(self.sunday_attendance.regular_pay, Decimal('1100'))
self.assertEqual(self.sunday_attendance.overtime_pay, Decimal('0'))
self.assertEqual(self.sunday_attendance.lunch_pay, Decimal('0'))
self.assertEqual(self.sunday_attendance.gross_wage, Decimal('1100'))
def test_sunday_attendance_gross_wage_with_overtime_enabled(self):
"""Test the gross wage of a sunday attendance
"""
self.sunday_attendance.enable_overtime = True
self.sunday_attendance.calculate_times()
self.sunday_attendance.calculate_gross_wage()
self.assertEqual(self.sunday_attendance.regular_pay, Decimal('1100'))
# Calculate expected overtime pay
ot_rate = (Decimal('550') / Decimal('8')) * Decimal('3')
self.assertEqual(self.sunday_attendance.overtime_pay,
ot_rate * Decimal('6'))
self.assertEqual(self.sunday_attendance.lunch_pay, Decimal('0'))
# Test gross wage
self.assertEqual(self.sunday_attendance.gross_wage,
Decimal('1100') + (ot_rate * Decimal('6')))
class PayrollTest(APITestCase):
def setUp(self):
"""Setup for testing
Employees:
- 1. Daily employee
- 1.1 Attendances
- 2. Salaried employee
- 2.1 Attendances
- 3. Daily Manager
- 3.1 Attendances
"""
self.shift = Shift(start_time=time(8, 0), end_time=time(17, 0))
self.shift.save()
self.employee1 = Employee.objects.create(**employee2_data)
self.employee2 = Employee.objects.create(**employee1_data)
self.employee3 = Employee.objects.create(**employee4_data)
self.manager = Employee.objects.create(**manager_data)
# Create attendances for first half of the month
for i in xrange(0, 6):
a_date = date(2016, 2, 1 + i)
# Create attendances for daily employee
a = Attendance.objects.create(
date=a_date,
start_time=datetime(2016,
2,
1 + i,
7,
30,
0,
tzinfo=timezone('Asia/Bangkok')),
end_time=datetime(2016,
2,
1 + i,
17,
15,
0,
tzinfo=timezone('Asia/Bangkok')),
employee=self.employee1,
shift=self.shift)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for monthly employee
a = Attendance.objects.create(
date=a_date,
start_time=datetime(2016,
2,
1 + i,
7,
30,
0,
tzinfo=timezone('Asia/Bangkok')),
end_time=datetime(2016,
2,
1 + i,
17,
15,
0,
tzinfo=timezone('Asia/Bangkok')),
employee=self.employee2,
shift=self.shift)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for daily employee in cambodia
a = Attendance.objects.create(
date=a_date,
start_time=datetime(2016,
2,
1 + i,
7,
30,
0,
tzinfo=timezone('Asia/Bangkok')),
end_time=datetime(2016,
2,
1 + i,
17,
15,
0,
tzinfo=timezone('Asia/Bangkok')),
employee=self.employee3,
shift=self.shift)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for daily manager
a = Attendance.objects.create(
date=a_date,
start_time=datetime(2016,
2,
1 + i,
7,
30,
0,
tzinfo=timezone('Asia/Bangkok')),
end_time=datetime(2016,
2,
1 + i,
17,
15,
0,
tzinfo=timezone('Asia/Bangkok')),
employee=self.manager,
shift=self.shift)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create the attendances for the latter half of the month
for i in xrange(0, 6):
a_date = date(2016, 2, 15 + i)
# Create attendances for daily employee
a = Attendance.objects.create(
date=a_date,
start_time=datetime(2016,
2,
15 + i,
7,
30,
0,
tzinfo=timezone('Asia/Bangkok')),
end_time=datetime(2016,
2,
15 + i,
17,
15,
0,
tzinfo=timezone('Asia/Bangkok')),
employee=self.employee1,
shift=self.shift)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for monthly employee
a = Attendance.objects.create(
date=a_date,
start_time=datetime(2016,
2,
15 + i,
7,
30,
0,
tzinfo=timezone('Asia/Bangkok')),
end_time=datetime(2016,
2,
15 + i,
17,
15,
0,
tzinfo=timezone('Asia/Bangkok')),
employee=self.employee2,
shift=self.shift)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for daily employee in cambodia
a = Attendance.objects.create(
date=a_date,
start_time=datetime(2016,
2,
15 + i,
7,
30,
0,
tzinfo=timezone('Asia/Bangkok')),
end_time=datetime(2016,
2,
15 + i,
17,
15,
0,
tzinfo=timezone('Asia/Bangkok')),
employee=self.employee3,
shift=self.shift)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for daily manager
a = Attendance.objects.create(
date=a_date,
start_time=datetime(2016,
2,
15 + i,
7,
30,
0,
tzinfo=timezone('Asia/Bangkok')),
end_time=datetime(2016,
2,
15 + i,
17,
15,
0,
tzinfo=timezone('Asia/Bangkok')),
employee=self.manager,
shift=self.shift)
a.calculate_times()
a.calculate_net_wage()
a.save()
def test_document_creation(self):
"""Test the creation of documents for this payroll
"""
start_date = date(2016, 1, 26)
end_date = date(2016, 2, 10)
payroll = Payroll.objects.create(start_date, end_date)
def test_document_creation_end_month(self):
"""Test the creation of documents for this payroll
"""
start_date = date(2016, 2, 11)
end_date = date(2016, 2, 25)
payroll = Payroll.objects.create(start_date, end_date)
class PayRecordTest(APITestCase):
"""Test class for Payrecord"""
def setUp(self):
"""Setup for testing
Employees:
- 1. Daily employee
- 1.1 Attendances
- 2. Salaried employee
- 2.1 Attendances
- 3. Daily Manager
- 3.1 Attendances
"""
logger.debug("\n\nBegin Setup\n\n")
self.shift = Shift(start_time=time(8,
0,
tzinfo=timezone('Asia/Bangkok')),
end_time=time(17,
0,
tzinfo=timezone('Asia/Bangkok')))
self.shift.save()
self.employee1 = Employee.objects.create(**employee2_data)
self.employee2 = Employee.objects.create(**employee1_data)
self.employee3 = Employee.objects.create(shift=self.shift,
**employee4_data)
self.manager = Employee.objects.create(**manager_data)
logger.warn(self.shift)
# Create attendances for first half of the month
for i in xrange(0, 7):
a_date = date(2016, 2, 1 + i)
# Create attendances for daily employee
a = Attendance.objects.create(
date=a_date,
start_time=timezone('Asia/Bangkok').localize(
datetime(2016, 2, 1 + i, 7, 30, 0)),
end_time=timezone('Asia/Bangkok').localize(
datetime(2016, 2, 1 + i, 17, 15, 0)),
employee=self.employee1,
shift=self.shift)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for monthly employee
a = Attendance.objects.create(
date=a_date,
start_time=timezone('Asia/Bangkok').localize(
datetime(2016, 2, 1 + i, 7, 30, 0)),
end_time=timezone('Asia/Bangkok').localize(
datetime(2016, 2, 1 + i, 17, 15, 0)),
employee=self.employee2,
shift=self.shift)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for daily employee in cambodia
a = Attendance.objects.create(
date=a_date,
start_time=timezone('Asia/Bangkok').localize(
datetime(2016, 2, 1 + i, 7, 30, 0)),
end_time=timezone('Asia/Bangkok').localize(
datetime(2016, 2, 1 + i, 17, 15, 0)),
employee=self.employee3,
shift=self.shift,
cambodia=True)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for daily manager
a = Attendance.objects.create(
date=a_date,
start_time=timezone('Asia/Bangkok').localize(
datetime(2016, 2, 1 + i, 7, 30, 0)),
end_time=timezone('Asia/Bangkok').localize(
datetime(2016, 2, 1 + i, 17, 15, 0)),
employee=self.manager,
shift=self.shift)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create the attendances for the latter half of the month
for i in xrange(0, 6):
a_date = date(2016, 2, 15 + i)
# Create attendances for daily employee
a = Attendance.objects.create(
date=a_date,
start_time=timezone('Asia/Bangkok').localize(
datetime(2016, 2, 1 + i, 7, 30, 0)),
end_time=timezone('Asia/Bangkok').localize(
datetime(2016, 2, 1 + i, 17, 15, 0)),
employee=self.employee1,
shift=self.shift)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for monthly employee
a = Attendance.objects.create(
date=a_date,
start_time=timezone('Asia/Bangkok').localize(
datetime(2016, 2, 1 + i, 7, 30, 0)),
end_time=timezone('Asia/Bangkok').localize(
datetime(2016, 2, 1 + i, 17, 15, 0)),
employee=self.employee2,
shift=self.shift)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for daily employee in cambodia
a = Attendance.objects.create(
date=a_date,
start_time=timezone('Asia/Bangkok').localize(
datetime(2016, 2, 1 + i, 7, 30, 0)),
end_time=timezone('Asia/Bangkok').localize(
datetime(2016, 2, 1 + i, 17, 15, 0)),
employee=self.employee3,
shift=self.shift,
cambodia=True)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for daily manager
a = Attendance.objects.create(
date=a_date,
start_time=timezone('Asia/Bangkok').localize(
datetime(2016, 2, 1 + i, 7, 30, 0)),
end_time=timezone('Asia/Bangkok').localize(
datetime(2016, 2, 1 + i, 17, 15, 0)),
employee=self.manager,
shift=self.shift)
a.calculate_times()
a.calculate_net_wage()
a.save()
logger.debug("\n\nEnd Setup\n\n")
def test_gross_wage_hourly_employee(self):
"""Test calculate the gross wage of a pay record
"""
record = PayRecord.objects.create(self.employee1,
start_date=date(2016, 2, 1),
end_date=date(2016, 2, 10))
gw = record.calculate_gross_wage()
self.assertEqual(gw, Decimal('4400'))
def test_net_wage_hourly_employee(self):
"""Test calculate the net wage of a pay record
"""
record = PayRecord.objects.create(self.employee1,
start_date=date(2016, 2, 1),
end_date=date(2016, 2, 10))
nw = record.calculate_net_wage()
self.assertEqual(record.stipend, Decimal('210'))
self.assertEqual(record.manager_stipend, Decimal('0'))
self.assertEqual(record.deductions, Decimal('0'))
self.assertEqual(record.social_security_withholding, Decimal('0'))
self.assertEqual(nw, Decimal('4610'))
def test_net_wage_hourly_employee_end_month(self):
"""Test calculate the net wage of a pay record
"""
record = PayRecord.objects.create(self.employee1,
start_date=date(2016, 2, 11),
end_date=date(2016, 2, 25))
nw = record.calculate_net_wage()
self.assertEqual(record.stipend, Decimal('180'))
self.assertEqual(record.manager_stipend, Decimal('0'))
self.assertEqual(record.deductions, Decimal('0'))
self.assertEqual(record.social_security_withholding, Decimal('385'))
self.assertEqual(nw, Decimal('3095'))
def test_gross_wage_hourly_employee_in_cambodia(self):
"""Test calculate the gross wage of a pay record
"""
record = PayRecord.objects.create(self.employee3,
start_date=date(2016, 2, 1),
end_date=date(2016, 2, 10))
gw = record.calculate_gross_wage()
self.assertEqual(gw, Decimal('9050'))
def test_net_wage_hourly_employee_in_cambodia(self):
"""Test calculate the net wage of a pay record
"""
record = PayRecord.objects.create(self.employee3,
start_date=date(2016, 2, 1),
end_date=date(2016, 2, 10))
nw = record.calculate_net_wage()
self.assertEqual(record.stipend, Decimal('300'))
self.assertEqual(record.manager_stipend, Decimal('0'))
self.assertEqual(record.deductions, Decimal('0'))
self.assertEqual(record.social_security_withholding, Decimal('0'))
self.assertEqual(nw, Decimal('9350'))
def test_gross_wage_hourly_employee_in_cambodia_end_month(self):
"""Test calculate the net wage of a pay record
"""
record = PayRecord.objects.create(self.employee3,
start_date=date(2016, 2, 11),
end_date=date(2016, 2, 25))
gw = record.calculate_gross_wage()
self.assertEqual(gw, Decimal('11050'))
def test_net_wage_hourly_employee_in_cambodia_end_month(self):
"""Test calculate the net wage of a pay record
"""
record = PayRecord.objects.create(self.employee3,
start_date=date(2016, 2, 11),
end_date=date(2016, 2, 25))
nw = record.calculate_net_wage()
self.assertEqual(record.stipend, Decimal('390'))
self.assertEqual(record.manager_stipend, Decimal('0'))
self.assertEqual(record.deductions, Decimal('0'))
self.assertEqual(record.social_security_withholding, Decimal('750'))
self.assertEqual(nw, Decimal('10690'))
def test_gross_wage_salaried_employee(self):
"""Test calculate the gross wage of a pay record
"""
record = PayRecord.objects.create(self.employee2,
start_date=date(2016, 2, 1),
end_date=date(2016, 2, 10))
gw = record.calculate_gross_wage()
self.assertEqual(gw, Decimal('10200'))
def test_net_wage_salaried_employee(self):
"""Test calculate the net wage of a pay record
"""
record = PayRecord.objects.create(self.employee2,
start_date=date(2016, 2, 1),
end_date=date(2016, 2, 10))
nw = record.calculate_net_wage()
self.assertEqual(nw, Decimal('10550'))
self.assertEqual(record.reimbursements, Decimal('350'))
self.assertEqual(record.social_security_withholding, Decimal('0'))
def test_gross_wage_manager(self):
"""Test calculate the gross wage of a pay record
"""
record = PayRecord.objects.create(self.manager,
start_date=date(2016, 2, 1),
end_date=date(2016, 2, 10))
gw = record.calculate_gross_wage()
self.assertEqual(gw, Decimal('5200'))
def test_net_wage_manager_employee(self):
"""Test calculate the net wage of a pay record
"""
record = PayRecord.objects.create(self.manager,
start_date=date(2016, 2, 1),
end_date=date(2016, 2, 10))
nw = record.calculate_net_wage()
self.assertEqual(nw, Decimal('5550.00'))
self.assertEqual(record.reimbursements, Decimal('350'))
self.assertEqual(record.social_security_withholding, Decimal('0'))
def test_net_wage_manager_end_month(self):
"""Test calculate the net wage of a pay record
"""
record = PayRecord.objects.create(self.manager,
start_date=date(2016, 2, 11),
end_date=date(2016, 2, 25))
nw = record.calculate_net_wage()
self.assertEqual(record.reimbursements, Decimal('300'))
self.assertEqual(record.social_security_withholding, Decimal('455'))
self.assertEqual(nw, Decimal('5245'))
class AttendanceTest(APITestCase):
"""
Testing class for attendance
"""
def setUp(self):
"""
Set up the Testing clas:
"""
super(AttendanceTest, self).setUp()
# self.client.client.login(username='******', password='******')
self.shift = Shift(start_time=time(8, 0), end_time=time(17, 0))
self.shift.save()
self.employee = Employee(shift=self.shift, **employee2_data)
self.employee.save()
# Regular attendance
self.attendance = Attendance(
date=date(2014, 7, 1),
start_time=timezone("Asia/Bangkok").localize(datetime(2014, 7, 1, 7, 30, 0)),
end_time=timezone("Asia/Bangkok").localize(datetime(2014, 7, 1, 23, 33, 0)),
employee=self.employee,
shift=self.shift,
)
self.attendance.save()
# Sunday attendance
self.sunday_attendance = Attendance(
date=date(2016, 2, 7),
start_time=timezone("Asia/Bangkok").localize(datetime(2016, 2, 7, 8, 02, 0)),
end_time=timezone("Asia/Bangkok").localize(datetime(2016, 2, 7, 23, 15, 0)),
employee=self.employee,
shift=self.shift,
)
self.sunday_attendance.save()
self.tz = timezone("Asia/Bangkok")
def test_start_time_property(self):
"""Test that the getter and setter for start time work properly
"""
a = Attendance(employee=self.employee, date=date(2016, 3, 21))
d1 = datetime(2016, 3, 21, 8, 11, 0)
# Test start time without timezone
a.start_time = d1
self.assertEqual(a._start_time, self.tz.localize(d1))
self.assertEqual(a.start_time, self.tz.localize(d1))
# Test start time with timezone
a.start_time = self.tz.localize(d1)
self.assertEqual(a._start_time, self.tz.localize(d1))
self.assertEqual(a.start_time, self.tz.localize(d1))
def test_end_time_property(self):
"""Test that the getter and setter for start time work properly
"""
a = Attendance(employee=self.employee, date=date(2016, 3, 21))
d1 = datetime(2016, 3, 21, 15, 29, 0)
# Test start time without timezone
a.end_time = d1
self.assertEqual(a._end_time, self.tz.localize(d1))
self.assertEqual(a.end_time, self.tz.localize(d1))
# Test start time with timezone
a.end_time = self.tz.localize(d1)
self.assertEqual(a._end_time, self.tz.localize(d1))
self.assertEqual(a.end_time, self.tz.localize(d1))
def test_regular_attedance_regular_hours(self):
"""Test the regular hours of a regular attedance
"""
self.attendance.calculate_times()
self.assertEqual(self.attendance.regular_time, Decimal("8.0"))
self.assertEqual(self.attendance.overtime, Decimal("0"))
def test_regular_attedance_with_overtime_enabled(self):
"""Test the regular hours of a regular attedance
"""
self.attendance.enable_overtime = True
self.attendance.calculate_times()
self.assertEqual(self.attendance.regular_time, Decimal("8.0"))
self.assertEqual(self.attendance.overtime, Decimal("6.5"))
def test_regular_attendance_gross_wage(self):
"""Test the gross wage of a regular attendance
"""
self.attendance.calculate_times()
self.attendance.calculate_gross_wage()
self.assertEqual(self.attendance.regular_pay, Decimal("550"))
self.assertEqual(self.attendance.overtime_pay, Decimal("0"))
self.assertEqual(self.attendance.lunch_pay, Decimal("0"))
self.assertEqual(self.attendance.gross_wage, Decimal("550"))
def test_regular_attendance_gross_wage_with_overtime_enabled(self):
"""Test the gross wage of a regular attendance with overtime enabled
"""
self.attendance.enable_overtime = True
self.attendance.calculate_times()
self.attendance.calculate_gross_wage()
self.assertEqual(self.attendance.regular_pay, Decimal("550"))
# Calculate expected overtime pay
ot_rate = (Decimal("550") / Decimal("8")) * Decimal("1.5")
self.assertEqual(self.attendance.overtime_pay, ot_rate * Decimal("6.5"))
self.assertEqual(self.attendance.lunch_pay, Decimal("0"))
# Test gross wage
self.assertEqual(self.attendance.gross_wage, Decimal("550") + (ot_rate * Decimal("6.5")))
def test_regular_attendance_net_wage(self):
"""Test the gross wage of a regular attendance
"""
self.attendance.calculate_times()
self.attendance.calculate_net_wage()
self.assertEqual(self.attendance.gross_wage, Decimal("550"))
self.assertEqual(self.attendance.reimbursement, Decimal("30"))
self.assertEqual(self.attendance.net_wage, Decimal("580"))
def test_regular_attendance_net_wage_with_lunch(self):
"""Test the gross wage of a regular attendance
"""
self.attendance.receive_lunch_overtime = True
self.attendance.calculate_times()
self.attendance.calculate_net_wage()
self.assertEqual(self.attendance.gross_wage, Decimal("653.125"))
self.assertEqual(self.attendance.reimbursement, Decimal("60"))
self.assertEqual(self.attendance.net_wage, Decimal("713.125"))
def test_regular_attendance_net_wage_where_clockin_late(self):
"""Test the net wage where an employee is late
"""
logger.debug("\n\n\n\nTesting late clocking for regular attendance\n\n\n")
# Change start time so employee is late
self.attendance.start_time = self.tz.localize(datetime(2014, 7, 1, 8, 15, 0))
self.attendance.calculate_times()
self.attendance.calculate_net_wage()
self.assertEqual(self.attendance.regular_time, Decimal("7.5"))
self.assertEqual(self.attendance.reimbursement, Decimal("0"))
self.assertEqual(self.attendance.net_wage, Decimal("515.625"))
def test_sunday_attendance_regular_hours(self):
"""Test the regular hours of a regular attedance
"""
logger.debug("\n\n\n\nTesting Sunday Regular attendance hours\n\n\n")
self.sunday_attendance.calculate_times()
self.assertEqual(self.sunday_attendance.regular_time, Decimal("8.0"))
self.assertEqual(self.sunday_attendance.overtime, Decimal("0"))
def test_sunday_attendance_with_overtime_enabled(self):
"""Test the regular hours of a regular attedance
"""
self.sunday_attendance.enable_overtime = True
self.sunday_attendance.calculate_times()
self.assertEqual(self.sunday_attendance.regular_time, Decimal("8.0"))
self.assertEqual(self.sunday_attendance.overtime, Decimal("6"))
def test_sunday_attendance_gross_wage(self):
"""Test the gross wage of a sunday attendance
"""
self.sunday_attendance.calculate_times()
self.sunday_attendance.calculate_gross_wage()
self.assertEqual(self.sunday_attendance.regular_pay, Decimal("1100"))
self.assertEqual(self.sunday_attendance.overtime_pay, Decimal("0"))
self.assertEqual(self.sunday_attendance.lunch_pay, Decimal("0"))
self.assertEqual(self.sunday_attendance.gross_wage, Decimal("1100"))
def test_sunday_attendance_gross_wage_with_overtime_enabled(self):
"""Test the gross wage of a sunday attendance
"""
self.sunday_attendance.enable_overtime = True
self.sunday_attendance.calculate_times()
self.sunday_attendance.calculate_gross_wage()
self.assertEqual(self.sunday_attendance.regular_pay, Decimal("1100"))
# Calculate expected overtime pay
ot_rate = (Decimal("550") / Decimal("8")) * Decimal("3")
self.assertEqual(self.sunday_attendance.overtime_pay, ot_rate * Decimal("6"))
self.assertEqual(self.sunday_attendance.lunch_pay, Decimal("0"))
# Test gross wage
self.assertEqual(self.sunday_attendance.gross_wage, Decimal("1100") + (ot_rate * Decimal("6")))
class PayrollTest(APITestCase):
def setUp(self):
"""Setup for testing
Employees:
- 1. Daily employee
- 1.1 Attendances
- 2. Salaried employee
- 2.1 Attendances
- 3. Daily Manager
- 3.1 Attendances
"""
self.shift = Shift(start_time=time(8, 0), end_time=time(17, 0))
self.shift.save()
self.employee1 = Employee.objects.create(**employee2_data)
self.employee2 = Employee.objects.create(**employee1_data)
self.employee3 = Employee.objects.create(**employee4_data)
self.manager = Employee.objects.create(**manager_data)
# Create attendances for first half of the month
for i in xrange(0, 6):
a_date = date(2016, 2, 1 + i)
# Create attendances for daily employee
a = Attendance.objects.create(
date=a_date,
start_time=datetime(2016, 2, 1 + i, 7, 30, 0, tzinfo=timezone("Asia/Bangkok")),
end_time=datetime(2016, 2, 1 + i, 17, 15, 0, tzinfo=timezone("Asia/Bangkok")),
employee=self.employee1,
shift=self.shift,
)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for monthly employee
a = Attendance.objects.create(
date=a_date,
start_time=datetime(2016, 2, 1 + i, 7, 30, 0, tzinfo=timezone("Asia/Bangkok")),
end_time=datetime(2016, 2, 1 + i, 17, 15, 0, tzinfo=timezone("Asia/Bangkok")),
employee=self.employee2,
shift=self.shift,
)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for daily employee in cambodia
a = Attendance.objects.create(
date=a_date,
start_time=datetime(2016, 2, 1 + i, 7, 30, 0, tzinfo=timezone("Asia/Bangkok")),
end_time=datetime(2016, 2, 1 + i, 17, 15, 0, tzinfo=timezone("Asia/Bangkok")),
employee=self.employee3,
shift=self.shift,
)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for daily manager
a = Attendance.objects.create(
date=a_date,
start_time=datetime(2016, 2, 1 + i, 7, 30, 0, tzinfo=timezone("Asia/Bangkok")),
end_time=datetime(2016, 2, 1 + i, 17, 15, 0, tzinfo=timezone("Asia/Bangkok")),
employee=self.manager,
shift=self.shift,
)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create the attendances for the latter half of the month
for i in xrange(0, 6):
a_date = date(2016, 2, 15 + i)
# Create attendances for daily employee
a = Attendance.objects.create(
date=a_date,
start_time=datetime(2016, 2, 15 + i, 7, 30, 0, tzinfo=timezone("Asia/Bangkok")),
end_time=datetime(2016, 2, 15 + i, 17, 15, 0, tzinfo=timezone("Asia/Bangkok")),
employee=self.employee1,
shift=self.shift,
)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for monthly employee
a = Attendance.objects.create(
date=a_date,
start_time=datetime(2016, 2, 15 + i, 7, 30, 0, tzinfo=timezone("Asia/Bangkok")),
end_time=datetime(2016, 2, 15 + i, 17, 15, 0, tzinfo=timezone("Asia/Bangkok")),
employee=self.employee2,
shift=self.shift,
)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for daily employee in cambodia
a = Attendance.objects.create(
date=a_date,
start_time=datetime(2016, 2, 15 + i, 7, 30, 0, tzinfo=timezone("Asia/Bangkok")),
end_time=datetime(2016, 2, 15 + i, 17, 15, 0, tzinfo=timezone("Asia/Bangkok")),
employee=self.employee3,
shift=self.shift,
)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for daily manager
a = Attendance.objects.create(
date=a_date,
start_time=datetime(2016, 2, 15 + i, 7, 30, 0, tzinfo=timezone("Asia/Bangkok")),
end_time=datetime(2016, 2, 15 + i, 17, 15, 0, tzinfo=timezone("Asia/Bangkok")),
employee=self.manager,
shift=self.shift,
)
a.calculate_times()
a.calculate_net_wage()
a.save()
def test_document_creation(self):
"""Test the creation of documents for this payroll
"""
start_date = date(2016, 1, 26)
end_date = date(2016, 2, 10)
payroll = Payroll.objects.create(start_date, end_date)
def test_document_creation_end_month(self):
"""Test the creation of documents for this payroll
"""
start_date = date(2016, 2, 11)
end_date = date(2016, 2, 25)
payroll = Payroll.objects.create(start_date, end_date)
class PayRecordTest(APITestCase):
"""Test class for Payrecord"""
def setUp(self):
"""Setup for testing
Employees:
- 1. Daily employee
- 1.1 Attendances
- 2. Salaried employee
- 2.1 Attendances
- 3. Daily Manager
- 3.1 Attendances
"""
logger.debug("\n\nBegin Setup\n\n")
self.shift = Shift(
start_time=time(8, 0, tzinfo=timezone("Asia/Bangkok")),
end_time=time(17, 0, tzinfo=timezone("Asia/Bangkok")),
)
self.shift.save()
self.employee1 = Employee.objects.create(**employee2_data)
self.employee2 = Employee.objects.create(**employee1_data)
self.employee3 = Employee.objects.create(shift=self.shift, **employee4_data)
self.manager = Employee.objects.create(**manager_data)
logger.warn(self.shift)
# Create attendances for first half of the month
for i in xrange(0, 7):
a_date = date(2016, 2, 1 + i)
# Create attendances for daily employee
a = Attendance.objects.create(
date=a_date,
start_time=timezone("Asia/Bangkok").localize(datetime(2016, 2, 1 + i, 7, 30, 0)),
end_time=timezone("Asia/Bangkok").localize(datetime(2016, 2, 1 + i, 17, 15, 0)),
employee=self.employee1,
shift=self.shift,
)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for monthly employee
a = Attendance.objects.create(
date=a_date,
start_time=timezone("Asia/Bangkok").localize(datetime(2016, 2, 1 + i, 7, 30, 0)),
end_time=timezone("Asia/Bangkok").localize(datetime(2016, 2, 1 + i, 17, 15, 0)),
employee=self.employee2,
shift=self.shift,
)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for daily employee in cambodia
a = Attendance.objects.create(
date=a_date,
start_time=timezone("Asia/Bangkok").localize(datetime(2016, 2, 1 + i, 7, 30, 0)),
end_time=timezone("Asia/Bangkok").localize(datetime(2016, 2, 1 + i, 17, 15, 0)),
employee=self.employee3,
shift=self.shift,
cambodia=True,
)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for daily manager
a = Attendance.objects.create(
date=a_date,
start_time=timezone("Asia/Bangkok").localize(datetime(2016, 2, 1 + i, 7, 30, 0)),
end_time=timezone("Asia/Bangkok").localize(datetime(2016, 2, 1 + i, 17, 15, 0)),
employee=self.manager,
shift=self.shift,
)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create the attendances for the latter half of the month
for i in xrange(0, 6):
a_date = date(2016, 2, 15 + i)
# Create attendances for daily employee
a = Attendance.objects.create(
date=a_date,
start_time=timezone("Asia/Bangkok").localize(datetime(2016, 2, 1 + i, 7, 30, 0)),
end_time=timezone("Asia/Bangkok").localize(datetime(2016, 2, 1 + i, 17, 15, 0)),
employee=self.employee1,
shift=self.shift,
)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for monthly employee
a = Attendance.objects.create(
date=a_date,
start_time=timezone("Asia/Bangkok").localize(datetime(2016, 2, 1 + i, 7, 30, 0)),
end_time=timezone("Asia/Bangkok").localize(datetime(2016, 2, 1 + i, 17, 15, 0)),
employee=self.employee2,
shift=self.shift,
)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for daily employee in cambodia
a = Attendance.objects.create(
date=a_date,
start_time=timezone("Asia/Bangkok").localize(datetime(2016, 2, 1 + i, 7, 30, 0)),
end_time=timezone("Asia/Bangkok").localize(datetime(2016, 2, 1 + i, 17, 15, 0)),
employee=self.employee3,
shift=self.shift,
cambodia=True,
)
a.calculate_times()
a.calculate_net_wage()
a.save()
# Create attendances for daily manager
a = Attendance.objects.create(
date=a_date,
start_time=timezone("Asia/Bangkok").localize(datetime(2016, 2, 1 + i, 7, 30, 0)),
end_time=timezone("Asia/Bangkok").localize(datetime(2016, 2, 1 + i, 17, 15, 0)),
employee=self.manager,
shift=self.shift,
)
a.calculate_times()
a.calculate_net_wage()
a.save()
logger.debug("\n\nEnd Setup\n\n")
def test_gross_wage_hourly_employee(self):
"""Test calculate the gross wage of a pay record
"""
record = PayRecord.objects.create(self.employee1, start_date=date(2016, 2, 1), end_date=date(2016, 2, 10))
gw = record.calculate_gross_wage()
self.assertEqual(gw, Decimal("4400"))
def test_net_wage_hourly_employee(self):
"""Test calculate the net wage of a pay record
"""
record = PayRecord.objects.create(self.employee1, start_date=date(2016, 2, 1), end_date=date(2016, 2, 10))
nw = record.calculate_net_wage()
self.assertEqual(record.stipend, Decimal("210"))
self.assertEqual(record.manager_stipend, Decimal("0"))
self.assertEqual(record.deductions, Decimal("0"))
self.assertEqual(record.social_security_withholding, Decimal("0"))
self.assertEqual(nw, Decimal("4610"))
def test_net_wage_hourly_employee_end_month(self):
"""Test calculate the net wage of a pay record
"""
record = PayRecord.objects.create(self.employee1, start_date=date(2016, 2, 11), end_date=date(2016, 2, 25))
nw = record.calculate_net_wage()
self.assertEqual(record.stipend, Decimal("180"))
self.assertEqual(record.manager_stipend, Decimal("0"))
self.assertEqual(record.deductions, Decimal("0"))
self.assertEqual(record.social_security_withholding, Decimal("385"))
self.assertEqual(nw, Decimal("3095"))
def test_gross_wage_hourly_employee_in_cambodia(self):
"""Test calculate the gross wage of a pay record
"""
record = PayRecord.objects.create(self.employee3, start_date=date(2016, 2, 1), end_date=date(2016, 2, 10))
gw = record.calculate_gross_wage()
self.assertEqual(gw, Decimal("9050"))
def test_net_wage_hourly_employee_in_cambodia(self):
"""Test calculate the net wage of a pay record
"""
record = PayRecord.objects.create(self.employee3, start_date=date(2016, 2, 1), end_date=date(2016, 2, 10))
nw = record.calculate_net_wage()
self.assertEqual(record.stipend, Decimal("300"))
self.assertEqual(record.manager_stipend, Decimal("0"))
self.assertEqual(record.deductions, Decimal("0"))
self.assertEqual(record.social_security_withholding, Decimal("0"))
self.assertEqual(nw, Decimal("9350"))
def test_gross_wage_hourly_employee_in_cambodia_end_month(self):
"""Test calculate the net wage of a pay record
"""
record = PayRecord.objects.create(self.employee3, start_date=date(2016, 2, 11), end_date=date(2016, 2, 25))
gw = record.calculate_gross_wage()
self.assertEqual(gw, Decimal("11050"))
def test_net_wage_hourly_employee_in_cambodia_end_month(self):
"""Test calculate the net wage of a pay record
"""
record = PayRecord.objects.create(self.employee3, start_date=date(2016, 2, 11), end_date=date(2016, 2, 25))
nw = record.calculate_net_wage()
self.assertEqual(record.stipend, Decimal("390"))
self.assertEqual(record.manager_stipend, Decimal("0"))
self.assertEqual(record.deductions, Decimal("0"))
self.assertEqual(record.social_security_withholding, Decimal("750"))
self.assertEqual(nw, Decimal("10690"))
def test_gross_wage_salaried_employee(self):
"""Test calculate the gross wage of a pay record
"""
record = PayRecord.objects.create(self.employee2, start_date=date(2016, 2, 1), end_date=date(2016, 2, 10))
gw = record.calculate_gross_wage()
self.assertEqual(gw, Decimal("10200"))
def test_net_wage_salaried_employee(self):
"""Test calculate the net wage of a pay record
"""
record = PayRecord.objects.create(self.employee2, start_date=date(2016, 2, 1), end_date=date(2016, 2, 10))
nw = record.calculate_net_wage()
self.assertEqual(nw, Decimal("10550"))
self.assertEqual(record.reimbursements, Decimal("350"))
self.assertEqual(record.social_security_withholding, Decimal("0"))
def test_gross_wage_manager(self):
"""Test calculate the gross wage of a pay record
"""
record = PayRecord.objects.create(self.manager, start_date=date(2016, 2, 1), end_date=date(2016, 2, 10))
gw = record.calculate_gross_wage()
self.assertEqual(gw, Decimal("5200"))
def test_net_wage_manager_employee(self):
"""Test calculate the net wage of a pay record
"""
record = PayRecord.objects.create(self.manager, start_date=date(2016, 2, 1), end_date=date(2016, 2, 10))
nw = record.calculate_net_wage()
self.assertEqual(nw, Decimal("5550.00"))
self.assertEqual(record.reimbursements, Decimal("350"))
self.assertEqual(record.social_security_withholding, Decimal("0"))
def test_net_wage_manager_end_month(self):
"""Test calculate the net wage of a pay record
"""
record = PayRecord.objects.create(self.manager, start_date=date(2016, 2, 11), end_date=date(2016, 2, 25))
nw = record.calculate_net_wage()
self.assertEqual(record.reimbursements, Decimal("300"))
self.assertEqual(record.social_security_withholding, Decimal("455"))
self.assertEqual(nw, Decimal("5245"))