# -*- coding: utf-8 -*-
# Copyright 2018 Camptocamp
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).

from odoo.addons.mail.tests.common import TestMail


class TestMailTemplate(TestMail):

    @classmethod
    def setUpClass(cls):
        super(TestMailTemplate, cls).setUpClass()
        cls.env = cls.env(context=dict(cls.env.context, tracking_disable=True))

        cls.Users = cls.env['res.users']
        cls.res_partner = cls.env['res.partner']
        cls.mail_template = cls.env['mail.template']
        cls.mail_comp_msg = cls.env['mail.compose.message']

        cls.env.user.company_id.use_parent_mail_address = True
        # Company
        company_vals = {
            'name': 'company_name_test',
            'email': 'company.mail.test@company'
        }

        cls.company_test = cls.res_partner.create(company_vals)

        # Partners test 1 without email
        partner_no_mail_vals = {
            'name': 'partner_1',
            'parent_id': cls.company_test.id,
        }

        # Partners test 2 with email
        partner_with_mail_vals = {
            'name': 'partner_2',
            'email': 'partner.2.mail.test@company',
            'parent_id': cls.company_test.id,
        }

        cls.partner_no_mail = cls.res_partner.create(partner_no_mail_vals)
        cls.partner_with_mail = cls.res_partner.create(
            partner_with_mail_vals
        )

    def create_mail_composer(self, partner_to_send_ids):
        email_template = self.env[
            'mail.template'
        ].with_context(test_parent_mail_recipient=True).create({
            'model_id': self.env['ir.model'].search([
                ('model', '=', 'mail.channel')
            ], limit=1).id,
            'name': 'Pigs Template',
            'subject': '${object.name}',
            'body_html': '${object.description}',
            'user_signature': False,
            'partner_to': ",".join(partner_to_send_ids),
        })

        composer = self.mail_comp_msg.with_context({
            'test_parent_mail_recipient': True,
            'default_composition_mode': 'comment',
            'default_model': 'mail.channel',
            'default_use_template': True,
        }).create({
            'subject': 'Forget me subject',
            'body': 'Dummy body',
            'template_id': email_template.id
        })
        values = composer.onchange_template_id(
            email_template.id,
            'comment', 'mail.channel', self.group_pigs.id
        )['value']

        # use _convert_to_cache to return a browse record list from command
        # list or id list for x2many fields
        values = composer._convert_to_record(
            composer._convert_to_cache(values)
        )
        recipients = values['partner_ids']

        return recipients

    def test_1_mail_send_to_partner_no_mail(self):
        """Use company mail if recipient partner has no email."""
        recipients = self.create_mail_composer([str(self.partner_no_mail.id)])

        self.assertEqual(recipients.email, self.company_test.email)
        self.assertNotEqual(recipients.email, self.partner_no_mail.email)
        self.assertNotEqual(recipients.email, self.partner_with_mail.email)

    def test_2_mail_send_to_partner_with_mail(self):
        """Use partner mail if recipient partner has an email."""
        recipients = self.create_mail_composer(
            [str(self.partner_with_mail.id)]
        )

        self.assertNotEqual(recipients.email, self.company_test.email)
        self.assertNotEqual(recipients.email, self.partner_no_mail.email)
        self.assertEqual(recipients.email, self.partner_with_mail.email)

    def test_3_mail_send_to_company_test(self):
        """Use company mail if recipient is the company."""
        recipients = self.create_mail_composer([str(self.company_test.id)])

        self.assertEqual(recipients.email, self.company_test.email)
        self.assertNotEqual(recipients.email, self.partner_no_mail.email)
        self.assertNotEqual(recipients.email, self.partner_with_mail.email)

    def test_4_mail_send_to_company_and_partner_no_mail(self):
        """ Use only one time company mail if recipient is the company
            and partner without mail.
        """
        recipients = self.create_mail_composer([
            str(self.partner_no_mail.id),
            str(self.company_test.id)
        ])

        self.assertEqual(recipients.email, self.company_test.email)
        self.assertNotEqual(recipients.email, self.partner_no_mail.email)
        self.assertNotEqual(recipients.email, self.partner_with_mail.email)