# Part of Hibou Suite Professional. See LICENSE_PROFESSIONAL file for full copyright and licensing details.


import odoo.tests
from odoo.addons.hr_payroll_account.tests.test_hr_payroll_account import TestHrPayrollAccount as TestBase

@odoo.tests.tagged('post_install', '-at_install')
class TestHrPayrollAccount(TestBase):

    def setUp(self):
        super().setUp()
        # Two employees, but in stock tests they share the same partner...
        self.hr_employee_mark.address_home_id = self.env['res.partner'].create({
            'name': 'employee_mark',
        })
        # This rule has a partner, and is the only one with any accounting side effects.
        # Remove partner to use the home address...
        self.rule = self.env.ref('hr_payroll.hr_salary_rule_houserentallowance1')
        self.rule.partner_id = False

    def _setup_fiscal_position(self):
        account_rule_debit = self.rule.account_debit
        self.assertTrue(account_rule_debit)
        account_other = self.env['account.account'].search([('id', '!=', account_rule_debit.id)], limit=1)
        self.assertTrue(account_other)
        fp = self.env['account.fiscal.position'].create({
            'name': 'Salary Remap 1',
            'account_ids': [(0, 0, {
                'account_src_id': account_rule_debit.id,
                'account_dest_id': account_other.id,
            })]
        })
        self.hr_contract_john.payroll_fiscal_position_id = fp

    def _setup_fiscal_position_empty(self):
        self._setup_fiscal_position()
        self.hr_contract_john.payroll_fiscal_position_id.write({'account_ids': [(5, 0, 0)]})

    def test_00_hr_payslip_run(self):
        # Original method groups but has no partners.
        self.account_journal.payroll_entry_type = 'original'
        super().test_00_hr_payslip_run()
        self.assertEqual(len(self.payslip_run.slip_ids), 3)
        self.assertEqual(len(self.payslip_run.slip_ids.mapped('move_id')), 1)
        self.assertEqual(len(self.payslip_run.slip_ids.mapped('move_id.line_ids.partner_id')), 0)

    def test_00_fiscal_position(self):
        self._setup_fiscal_position()
        self.test_00_hr_payslip_run()
        self.assertEqual(len(self.payslip_run.slip_ids.mapped('move_id.line_ids.account_id')), 2)

    def test_00_fiscal_position_empty(self):
        self._setup_fiscal_position_empty()
        self.test_00_hr_payslip_run()
        self.assertEqual(len(self.payslip_run.slip_ids.mapped('move_id.line_ids.account_id')), 1)

    def test_01_hr_payslip_run(self):
        # Grouped method groups but has partners.
        self.account_journal.payroll_entry_type = 'grouped'
        super().test_01_hr_payslip_run()
        self.assertEqual(len(self.payslip_run.slip_ids), 3)
        self.assertEqual(len(self.payslip_run.slip_ids.mapped('move_id')), 1)
        self.assertEqual(len(self.payslip_run.slip_ids.mapped('move_id.line_ids.partner_id')), 2)

    def test_01_fiscal_position(self):
        self._setup_fiscal_position()
        self.test_01_hr_payslip_run()
        self.assertEqual(len(self.payslip_run.slip_ids.mapped('move_id.line_ids.account_id')), 2)

    def test_01_fiscal_position_empty(self):
        self._setup_fiscal_position_empty()
        self.test_01_hr_payslip_run()
        self.assertEqual(len(self.payslip_run.slip_ids.mapped('move_id.line_ids.account_id')), 1)

    def test_01_2_hr_payslip_run(self):
        # Payslip method makes an entry per payslip
        self.account_journal.payroll_entry_type = 'slip'
        # Call 'other' implementation.
        super().test_01_hr_payslip_run()
        self.assertEqual(len(self.payslip_run.slip_ids), 3)
        self.assertEqual(len(self.payslip_run.slip_ids.mapped('move_id')), 3)
        self.assertEqual(len(self.payslip_run.slip_ids.mapped('move_id.line_ids.partner_id')), 2)

    def test_01_2_fiscal_position(self):
        self._setup_fiscal_position()
        self.test_01_2_hr_payslip_run()
        self.assertEqual(len(self.payslip_run.slip_ids.mapped('move_id.line_ids.account_id')), 2)

    def test_01_2_fiscal_position_empty(self):
        self._setup_fiscal_position_empty()
        self.test_01_2_hr_payslip_run()
        self.assertEqual(len(self.payslip_run.slip_ids.mapped('move_id.line_ids.account_id')), 1)