# -*- coding: utf-8 -*-

import datetime
from dateutil.relativedelta import relativedelta

from flectra import fields, tools
from flectra.addons.event.tests.common import TestEventCommon
from flectra.tools import mute_logger


class TestMailSchedule(TestEventCommon):

    @mute_logger('flectra.addons.base.ir.ir_model', 'flectra.models')
    def test_00_event_mail_schedule(self):
        """ Test mail scheduling for events """
        now = fields.datetime.now()
        event_date_begin = now + relativedelta(days=1)
        event_date_end = now + relativedelta(days=3)

        test_event = self.Event.sudo(self.user_eventmanager).create({
            'name': 'TestEventMail',
            'auto_confirm': True,
            'date_begin': event_date_begin,
            'date_end': event_date_end,
            'seats_max': 10,
            'event_mail_ids': [
                (0, 0, {  # right at subscription
                    'interval_unit': 'now',
                    'interval_type': 'after_sub',
                    'template_id': self.env['ir.model.data'].xmlid_to_res_id('event.event_subscription')}),
                (0, 0, {  # 1 days before event
                    'interval_nbr': 1,
                    'interval_unit': 'days',
                    'interval_type': 'before_event',
                    'template_id': self.env['ir.model.data'].xmlid_to_res_id('event.event_reminder')}),
            ]
        })

        # create some registrations
        self.Registration.sudo(self.user_eventuser).create({
            'event_id': test_event.id,
            'name': 'Reg0',
            'email': 'reg0@example.com',
        })
        self.Registration.sudo(self.user_eventuser).create({
            'event_id': test_event.id,
            'name': 'Reg1',
            'email': 'reg1@example.com',
        })

        # check subscription scheduler
        schedulers = self.EventMail.search([('event_id', '=', test_event.id), ('interval_type', '=', 'after_sub')])
        self.assertEqual(len(schedulers), 1, 'event: wrong scheduler creation')
        self.assertEqual(schedulers[0].scheduled_date, test_event.create_date, 'event: incorrect scheduled date for checking controller')

        # verify that subscription scheduler was auto-executed after each registration
        self.assertEqual(len(schedulers[0].mail_registration_ids), 2, 'event: incorrect number of mail scheduled date')

        mails = self.env['mail.mail'].search([('subject', 'ilike', 'registration'), ('date', '>=', datetime.datetime.strftime(now, tools.DEFAULT_SERVER_DATETIME_FORMAT))], order='date DESC', limit=3)
        self.assertEqual(len(mails), 2, 'event: wrong number of registration mail sent')

        for registration in schedulers[0].mail_registration_ids:
            self.assertTrue(registration.mail_sent, 'event: wrongly confirmed mailing on registration')

        # check before event scheduler
        schedulers = self.EventMail.search([('event_id', '=', test_event.id), ('interval_type', '=', 'before_event')])
        self.assertEqual(len(schedulers), 1, 'event: wrong scheduler creation')
        self.assertEqual(schedulers[0].scheduled_date, datetime.datetime.strftime(event_date_begin + relativedelta(days=-1), tools.DEFAULT_SERVER_DATETIME_FORMAT), 'event: incorrect scheduled date')

        # execute event reminder scheduler explicitly
        schedulers[0].execute()

        self.assertTrue(schedulers[0].mail_sent, 'event: reminder scheduler should have sent an email')
        self.assertTrue(schedulers[0].done, 'event: reminder scheduler should be done')

        mails = self.env['mail.mail'].search([('subject', 'ilike', 'TestEventMail'), ('date', '>=', datetime.datetime.strftime(now, tools.DEFAULT_SERVER_DATETIME_FORMAT))], order='date DESC', limit=3)
        self.assertEqual(len(mails), 3, 'event: wrong number of reminders in outgoing mail queue')