# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.

from unittest.mock import patch

from odoo.tests.common import TransactionCase, tagged
from odoo.addons.hr_contract_salary.models.hr_contract import HrContract


@tagged('benefits')
class TestBenefits(TransactionCase):

    @classmethod
    def setUpClass(cls):
        super().setUpClass()
        cls.employee = cls.env['hr.employee'].create({'name': "John"})
        cls.structure_type = cls.env['hr.payroll.structure.type'].create({'name': 'struct'})
        cls.contract = cls.env['hr.contract'].create({
            'name': "Contract",
            'employee_id': cls.employee.id,
            'wage': 6500,
            'structure_type_id': cls.structure_type.id,
        })

    def test_yearly_cost_new_benefit(self):
        fieldname = 'x_test_field'
        model = self.env.ref('hr_contract.model_hr_contract')
        field = self.env['ir.model.fields'].create({
            'name': fieldname,
            'model': model.name,
            'ttype': 'float',
            'model_id': model.id,
        })
        self.contract.write({fieldname: 50})
        self.assertEqual(self.contract.final_yearly_costs, 12 * self.contract.wage)
        benefit_costs = HrContract._get_benefits_costs
        with patch.object(HrContract, '_get_benefits_costs', lambda self: benefit_costs(self) + self[fieldname]):
            atype = self.env['hr.contract.salary.benefit.type'].create({})
            self.env['hr.contract.salary.benefit'].create({
                'impacts_net_salary': True,
                'benefit_type_id': atype.id,
                'res_field_id': field.id,
                'cost_res_field_id': field.id,
                'structure_type_id': self.structure_type.id,
            })
        self.assertEqual(self.contract.final_yearly_costs, 12 * (self.contract.wage + 50), "The new benefit should have updated the yearly cost")

    def test_holidays_yearly_cost(self):
        # yearly cost should not change even if the number of extra time off changes
        with patch.object(HrContract, '_get_benefits_costs', lambda self: 250):
            self.contract._compute_final_yearly_costs()
            base_yearly_cost = self.contract.final_yearly_costs
            self.contract.holidays = 15
            # this is triggered when configuring/signing a contract
            # and recomputes the final_yearly_costs field
            self.contract.wage_with_holidays = 6076.57
            self.assertAlmostEqual(base_yearly_cost, self.contract.final_yearly_costs, 2,
                'Yearly costs should stay the same')

    def test_final_yearly_costs(self):
        # Yearly costs should not change when set manually on the interface
        # 100000 € / 12 = 8333.33333333
        # Wage is rounded, then 8333.33 * 12 = 99999.96
        self.contract.final_yearly_costs = 100000
        self.contract._onchange_final_yearly_costs()
        self.assertAlmostEqual(self.contract.final_yearly_costs, 100000, 2) # And not 99999.96

        self.contract.holidays = 10
        self.contract.final_yearly_costs = 100000
        self.contract._onchange_final_yearly_costs()
        self.contract._onchange_wage_with_holidays()
        self.assertAlmostEqual(self.contract.final_yearly_costs, 100000, 2) # And not 99999.96