def test_create_annual_leave(self):
"""
Test create_annual_leave
"""
user = mommy.make('auth.User', id=56)
StaffProfile.objects.all().delete()
staffprofile = mommy.make('small_small_hr.StaffProfile', user=user)
obj = create_annual_leave(staffprofile, 2016, Leave.REGULAR)
self.assertEqual(staffprofile, obj.staff)
self.assertEqual(2016, obj.year)
self.assertEqual(0, obj.carried_over_days)
self.assertEqual(21, obj.allowed_days)
self.assertEqual(Leave.REGULAR, obj.leave_type)
# 12 days of leave
start = datetime(2016,
6,
5,
0,
0,
0,
tzinfo=pytz.timezone(settings.TIME_ZONE))
end = datetime(2016,
6,
16,
0,
0,
0,
tzinfo=pytz.timezone(settings.TIME_ZONE))
mommy.make('small_small_hr.Leave',
leave_type=Leave.REGULAR,
start=start,
end=end,
status=Leave.APPROVED,
staff=staffprofile)
obj2 = create_annual_leave(staffprofile, 2017, Leave.REGULAR)
self.assertEqual(staffprofile, obj2.staff)
self.assertEqual(2017, obj2.year)
self.assertEqual(9, obj2.carried_over_days)
self.assertEqual(21, obj2.allowed_days)
self.assertEqual(Leave.REGULAR, obj2.leave_type)
obj3 = create_annual_leave(staffprofile, 2018, Leave.SICK)
self.assertEqual(staffprofile, obj3.staff)
self.assertEqual(2018, obj3.year)
self.assertEqual(0, obj3.carried_over_days)
self.assertEqual(10, obj3.allowed_days)
self.assertEqual(Leave.SICK, obj3.leave_type)
def test_get_carry_over(self):
"""
Test get_carry_over
"""
user = mommy.make('auth.User', id=23)
StaffProfile.objects.all().delete()
staffprofile = mommy.make('small_small_hr.StaffProfile', user=user)
self.assertEqual(0, get_carry_over(staffprofile, 2017, Leave.REGULAR))
create_annual_leave(staffprofile, 2017, Leave.REGULAR)
# carry over should be 10 because the balance is 21
self.assertEqual(10, get_carry_over(staffprofile, 2018, Leave.REGULAR))
# 12 days of leave, Sat & Sun not counted
start = datetime(2017,
6,
5,
0,
0,
0,
tzinfo=pytz.timezone(settings.TIME_ZONE))
end = datetime(2017,
6,
20,
0,
0,
0,
tzinfo=pytz.timezone(settings.TIME_ZONE))
mommy.make('small_small_hr.Leave',
leave_type=Leave.REGULAR,
start=start,
end=end,
status=Leave.APPROVED,
staff=staffprofile)
# carry over should be 9 => 21 - 12
self.assertEqual(9, get_carry_over(staffprofile, 2018, Leave.REGULAR))
# no sick leave carry over
create_annual_leave(staffprofile, 2017, Leave.SICK)
self.assertEqual(0, get_carry_over(staffprofile, 2018, Leave.SICK))
def setUp(self):
"""Set up."""
self.user = mommy.make(
"auth.User", first_name="Mosh", last_name="Pitt", email="*****@*****.**"
)
self.staffprofile = mommy.make("small_small_hr.StaffProfile", user=self.user)
self.staffprofile.leave_days = 17
self.staffprofile.sick_days = 9
self.staffprofile.save()
self.staffprofile.refresh_from_db()
create_annual_leave(self.staffprofile, 2017, Leave.REGULAR)
StaffProfile.objects.rebuild()
hr_group = mommy.make("auth.Group", name=settings.SSHR_ADMIN_USER_GROUP_NAME)
self.boss = mommy.make(
"auth.User", first_name="Mother", last_name="Hen", email="*****@*****.**"
)
self.boss.groups.add(hr_group)