# Copyright 2019 ACSONE SA/NV
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

from odoo.tests.common import TransactionCase


class TestMailTemplateSubstitute(TransactionCase):
    def setUp(self):
        super(TestMailTemplateSubstitute, self).setUp()
        self.smt2 = self.env["mail.template"].create(
            {
                "name": "substitute_template_2",
                "model_id": self.env.ref("base.model_res_partner").id,
            }
        )
        self.smt1 = self.env["mail.template"].create(
            {
                "name": "substitute_template_1",
                "model_id": self.env.ref("base.model_res_partner").id,
                "mail_template_substitution_rule_ids": [
                    (
                        0,
                        0,
                        {
                            "substitution_mail_template_id": self.smt2.id,
                            "domain": "[('id', '=', False)]",
                        },
                    )
                ],
            }
        )
        self.mt = self.env["mail.template"].create(
            {
                "name": "base_template",
                "model_id": self.env.ref("base.model_res_partner").id,
                "mail_template_substitution_rule_ids": [
                    (0, 0, {"substitution_mail_template_id": self.smt1.id})
                ],
            }
        )
        self.mail_compose = self.env["mail.compose.message"].create(
            {"template_id": self.mt.id, "composition_mode": "mass_mail"}
        )
        self.partners = self.env["res.partner"].search([])

    def test_get_email_template(self):
        self.assertEqual(
            self.mt._get_substitution_template(
                self.env.ref("base.model_res_partner"), self.partners.ids
            ),
            self.smt1,
        )
        self.assertEqual(
            self.mt.get_email_template(self.partners.ids).get(self.partners.ids[0]),
            self.smt1,
        )

    def test_get_substitution_template(self):
        self.assertEqual(
            self.mail_compose.with_context(
                active_ids=self.partners.ids
            )._get_substitution_template("mass_mail", self.mt, None),
            self.smt1,
        )

    def test_onchange_template_id_wrapper(self):
        self.assertEqual(self.mail_compose.template_id, self.mt)
        self.smt1.mail_template_substitution_rule_ids.domain = "[]"
        self.mail_compose.with_context(
            active_ids=self.partners.ids
        ).onchange_template_id_wrapper()
        self.assertEqual(self.mail_compose.template_id, self.smt2)