Files
suite/hr_payroll_gamification/tests/test_payroll_badge.py
2022-01-06 13:44:33 -08:00

87 lines
3.4 KiB
Python

from odoo.addons.hr_payroll_hibou.tests import common
from odoo import fields
from datetime import date
class TestPayslip(common.TestPayslip):
def setUp(self):
super(TestPayslip, self).setUp()
self.wage = 2000.00
self.structure = self.env.ref('hr_payroll.structure_002')
self.structure_type = self.structure.type_id
self.structure_type.wage_type = 'monthly'
self.employee = self._createEmployee()
self.employee.user_id = self.env.user
self.contract = self._createContract(self.employee, wage=self.wage)
def test_badge_amounts(self):
badge = self.env['gamification.badge'].create({
'name': 'test',
})
badge.payroll_amount = 5.0
def test_badge_payroll(self):
additional_wage = 5.0
additional_wage_period = 15.0
payslip = self._createPayslip(self.employee,
'2018-01-01',
'2018-01-31',
)
payslip.compute_sheet()
cats = self._getCategories(payslip)
self.assertEqual(cats['BASIC'], self.wage)
badge = self.env['gamification.badge'].create({
'name': 'test',
'payroll_type': 'fixed',
'payroll_amount': additional_wage,
})
badge_user = self.env['gamification.badge.user'].create({
'badge_id': badge.id,
'employee_id': self.employee.id,
'user_id': self.env.user.id,
})
badge_period = self.env['gamification.badge'].create({
'name': 'test period',
'payroll_type': 'period',
'payroll_amount': additional_wage_period,
})
# Need a specific 'create_date' to test.
badge_user_period = self.env['gamification.badge.user'].create({
'badge_id': badge_period.id,
'employee_id': self.employee.id,
'user_id': self.env.user.id,
})
self.env.cr.execute('update gamification_badge_user set create_date = \'2018-02-10\' where id = %d;' % (badge_user_period.id, ))
badge_user_period = self.env['gamification.badge.user'].browse(badge_user_period.id)
self.assertEqual(self.employee.badge_ids, badge_user + badge_user_period)
# Includes only one badge
payslip = self._createPayslip(self.employee,
'2018-01-01',
'2018-01-31',)
self.assertTrue(payslip.input_line_ids)
input_line = payslip.input_line_ids.filtered(lambda l: l.name == 'Badges')
self.assertTrue(input_line)
self.assertEqual(input_line.amount, additional_wage)
payslip.compute_sheet()
cats = self._getCategories(payslip)
self.assertEqual(cats['BASIC'], self.wage + additional_wage)
# Include both Badges
payslip = self._createPayslip(self.employee,
'2018-02-01',
'2018-02-25',)
input_line = payslip.input_line_ids.filtered(lambda l: l.name == 'Badges')
self.assertTrue(input_line)
self.assertEqual(input_line.amount, additional_wage + additional_wage_period)
payslip.compute_sheet()
cats = self._getCategories(payslip)
self.assertEqual(cats['BASIC'], self.wage + additional_wage + additional_wage_period)