# Part of Odoo. See LICENSE file for full copyright and licensing details.

from freezegun import freeze_time

from odoo import Command, fields
from odoo.tests.common import new_test_user, tagged, TransactionCase, users
from odoo.addons.mail.tools.discuss import Store


@tagged('post_install', '-at_install')
class TestPartner(TransactionCase):

    @classmethod
    @freeze_time('2024-06-04')
    def setUpClass(cls):
        super().setUpClass()
        # use a single value for today throughout the tests to avoid weird scenarios around midnight
        cls.today = fields.Date.today()
        baseUser = cls.env['res.users'].create({
            'email': 'e.e@example.com',
            'group_ids': [Command.link(cls.env.ref('base.group_user').id)],
            'login': 'emp',
            'name': 'Ernest Employee',
            'notification_type': 'inbox',
            'signature': '--\nErnest',
        })
        cls.partner = baseUser.partner_id
        cls.users = baseUser + cls.env['res.users'].create({
            'name': 'test1',
            'login': 'test1',
            'email': 'test1@example.com',
            'partner_id': cls.partner.id,
        })
        cls.employees = cls.env['hr.employee'].create([{
            'user_id': user.id,
        } for user in cls.users])
        cls.work_entry_type = cls.env['hr.work.entry.type'].create({
            'requires_allocation': False,
            'name': 'Legal Leaves',
            'code': 'Legal Leaves',
            'count_as': 'absence',
            'request_unit': 'day',
            'unit_of_measure': 'day',
        })
        cls.leaves = cls.env['hr.leave'].create([{
            'request_date_from': "2024-06-03",
            'request_date_to': "2024-06-06",
            'employee_id': cls.employees[0].id,
            'work_entry_type_id': cls.work_entry_type.id,
        }, {
            'request_date_from': "2024-06-02",
            'request_date_to': "2024-06-05",
            'employee_id': cls.employees[1].id,
            'work_entry_type_id': cls.work_entry_type.id,
        }])
        cls.user_no_hr_access = new_test_user(
            cls.env, login="user_no_hr_access",
        )

    @freeze_time('2024-06-04')
    def test_store_add_res_partner(self):
        self.leaves.write({'state': 'validate'})
        store_1 = Store().add(self.partner, "_store_partner_fields")
        self.assertEqual(
            store_1.get_result()["hr.employee"][0]["leave_date_to"],
            "2024-06-07",
            "Return date is the return date of the main user of the partner",
        )
        self.leaves[0].action_refuse()
        store_2 = Store().add(self.partner, "_store_partner_fields")
        self.assertEqual(
            store_2.get_result()["hr.employee"][0]["leave_date_to"],
            False,
            "Partner is not considered out of office if their main user is not on holiday",
        )

    @freeze_time("2024-06-04")
    @users("user_no_hr_access")
    def test_store_add_res_partner_no_hr_access(self):
        self.leaves.write({"state": "validate"})
        partner = self.partner.with_user(self.user_no_hr_access)
        store = Store().add(partner, "_store_partner_fields")
        self.assertEqual(
            store.get_result()["hr.employee"][0]["leave_date_to"],
            "2024-06-07",
            "Return date is the return date of the main user of the partner, "
            "even if the user has no access to the company",
        )