mail-addons/mail_multi_website/tests/test_send.py

# Copyright 2018 Ivan Yelizariev <https://it-projects.info/team/yelizariev>
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl.html).
from odoo.tests.common import TransactionCase


class TestSendMail(TransactionCase):
    at_install = True
    post_install = True

    def setUp(self):
        super(TestSendMail, self).setUp()
        self.website = self.env.ref("website.website2")
        self.company = self.env["res.company"].create({"name": "New Test Website"})
        self.original_email = self.env.user.email
        self.original_company = self.env.user.company_id
        self.email = "superadmin@second-website.example"
        # Check that current email is set and differs
        self.assertTrue(self.email)
        self.assertNotEqual(self.original_email, self.email)
        self.website.company_id = self.company

    def switch_user_website(self):
        # add website to allowed
        self.env.user.write(
            dict(
                backend_website_ids=[(4, self.website.id)],
                backend_website_id=self.website.id,
                company_id=self.company.id,
                company_ids=[(4, self.company.id)],
            )
        )

    def test_multi_email(self):
        """User has email addresses per website"""
        self.switch_user_website()
        # update user's email
        self.env.user.email = self.email
        # Check that writing works
        self.env.user.invalidate_cache()
        self.assertEqual(
            self.env.user.email,
            self.email,
            "Write methods doesn't work (Field is not in registry?)",
        )

        # changing company will automatically update website value to empty value
        self.env.user.company_id = self.original_company
        self.env.user.invalidate_cache()
        self.assertEqual(
            self.env.user.email,
            self.original_email,
            "Multi-email doesn't work on switching websites",
        )

    def test_multi_email_partner(self):
        """Partner doesn't have email addresses per website"""
        original_email = "original@email1"
        new_email = "new@email2"
        partner = self.env["res.partner"].create(
            {"name": "test", "email": original_email}
        )
        self.switch_user_website()
        # update partner's email
        partner.email = new_email
        self.assertEqual(partner.email, new_email)
        # changing company will automatically update website value to empty value
        self.env.user.company_id = self.original_company
        self.env.user.invalidate_cache()
        self.assertEqual(
            partner.email, new_email, "Partner's email must not be Multi-website"
        )