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

from ast import literal_eval

from odoo.addons.website_event.tests.common import TestEventOnlineCommon
from odoo.addons.base.tests.common import TransactionCaseWithUserDemo


class TestTrackPush(TestEventOnlineCommon, TransactionCaseWithUserDemo):
    def test_track_push(self):
        """" Test 'Send Push to Attendees' action and verify that it correctly
        targets all visitors that are registered to the event """

        registered_parent_visitor = self.env['website.visitor'].create({
            'access_token': self.user_demo.partner_id.id,
            'push_subscription_ids': [(0, 0, {'push_token': 'AAAAA1'})],
            'event_registration_ids': [(0, 0, {
                'event_id': self.event_0.id
            })]
        })
        register_child_visitor = self.env['website.visitor'].create({
            'access_token': 'f9d254aa8344a31e3120ef5ac2d8f87b',
            'push_subscription_ids': [(0, 0, {'push_token': 'AAAAA2'})],
        })
        register_child_visitor._merge_visitor(registered_parent_visitor)

        registered_visitor = self.env['website.visitor'].create({
            'access_token': 'f9d2277006a7f10496c311860070df6e',
            'push_subscription_ids': [(0, 0, {'push_token': 'BBBBB'})],
            'event_registration_ids': [(0, 0, {
                'event_id': self.event_0.id
            })]
        })

        # unregistered attendee that should not appear in results
        self.env['website.visitor'].create({
            'access_token': 'f9d2e3cc29e8175daf79c6f5beb1d5ce',
            'push_subscription_ids': [(0, 0, {'push_token': 'CCCCC'})],
        })

        action = self.event_0.action_send_push()
        social_post = self.env['social.post'] \
            .with_context(action.get('context', {})) \
            .create({'message': 'Hello Attendees!'})

        targeted_visitors = self.env['website.visitor'].search(literal_eval(social_post.visitor_domain))
        # the result should show the 2 active visitors that are registered for the event
        self.assertEqual(targeted_visitors, registered_parent_visitor + registered_visitor)

        # however, we should have 3 push_tokens to push to
        # (the 1 from the registered_visitor and the 2 from the registered_parent_visitor)
        self.assertEqual(len(targeted_visitors.push_subscription_ids.mapped('push_token')), 3)
        self.assertEqual(
            targeted_visitors.push_subscription_ids,
            (registered_parent_visitor + registered_visitor + register_child_visitor).push_subscription_ids
        )