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

import odoo
from odoo.addons.point_of_sale.tests.common import CommonPosTest


class CommonPosAppointmentTest(CommonPosTest):

    @classmethod
    def setUpClass(cls):
        super().setUpClass()
        cls.env.user.group_ids += cls.env.ref('appointment.group_appointment_manager')

        cls.reservation_appointment = cls.env['appointment.type'].create({
            'appointment_tz': 'UTC',
            'is_auto_assign': True,
            'min_schedule_hours': 1.0,
            'max_schedule_days': 8,
            'name': 'Test',
            'manage_capacity': True,
            'schedule_based_on': 'resources',
        })
        cls.resource_1, cls.resource_2 = cls.env['appointment.resource'].create([
            {
                'capacity': 4,
                'name': 'Table of 4',
                'appointment_type_ids': cls.reservation_appointment.ids,
            },
            {
                'capacity': 2,
                'name': 'Table of 2',
                'appointment_type_ids': cls.reservation_appointment.ids,
            }
        ])
        cls.env.user.group_ids -= cls.env.ref('appointment.group_appointment_manager')


@odoo.tests.tagged('post_install', '-at_install')
class TestPosAppointmentFlow(CommonPosAppointmentTest):

    def test_appointment_reservation_count(self):
        """ Test that bookings are created with the correct number of reservation. """

        # Reservation should depend on resource_total_capacity_reserved
        capacity_reservation = self.env['calendar.event'].create({
            'name': 'event without partner',
            'resource_ids': self.resource_1.ids,
            'total_capacity_reserved': 3,
            'appointment_type_id': self.reservation_appointment.id,
        })
        self.assertEqual(capacity_reservation.waiting_list_capacity, 3)

        # resource_total_capacity_reserved has priority over resource_ids
        capacity_reservation_2 = self.env['calendar.event'].create({
            'name': 'event without partner',
            'total_capacity_reserved': 5,
            'appointment_type_id': self.reservation_appointment.id,
            'resource_ids': [(4, self.resource_1.id), (4, self.resource_2.id)],
        })
        self.assertEqual(capacity_reservation_2.waiting_list_capacity, 5)

        # Reservation should depend on resource_ids
        resource_reservation = self.env['calendar.event'].create({
            'name': 'event without partner',
            'appointment_type_id': self.reservation_appointment.id,
            'resource_ids': [(4, self.resource_1.id), (4, self.resource_2.id)],
        })
        self.assertEqual(resource_reservation.waiting_list_capacity, 6)

    def test_prepare_calendar_event_values_phone_number(self):
        """ Test that the phone number of the customer is added to the calendar event when an appointment is booked
        from the POS. """
        partner = self.env['res.partner'].create({
            'name': 'Customer A',
            'phone': '123456789',
        })
        appointment_invite = self.env['appointment.invite'].create({
            'appointment_type_ids': self.reservation_appointment.ids,
        })
        event_values = self.reservation_appointment._prepare_calendar_event_values(
            asked_capacity=2,
            booking_line_values={},
            description='Test Description',
            duration=2,
            allday=False,
            appointment_invite=appointment_invite,
            guests=0,
            name='Test Event',
            customer=partner,
            staff_user=None,
            start=odoo.fields.Datetime.now(),
            stop=odoo.fields.Datetime.now(),
        )
        self.assertEqual(event_values['phone_number'], '123456789')

        # Create calendar event
        event = self.env['calendar.event'].create(event_values)
        self.assertEqual(partner.phone, '123456789')
        self.assertEqual(event.phone_number, '123456789')

        # Phone number is not related to the partner
        partner.phone = '987654321'
        self.assertEqual(event.phone_number, '123456789')

        # And reversely, changing the phone number on the event does not change it on the partner
        event.phone_number = '111222333'
        self.assertEqual(partner.phone, '987654321')