# -*- coding: utf-8 -*- # Copyright 2017 Simone Orsi # License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl). from odoo.tests.common import SavepointCase class PartnerDomainCase(SavepointCase): @classmethod def setUpClass(cls): super(PartnerDomainCase, cls).setUpClass() cls.partner_model = cls.env['res.partner'] cls.message_model = cls.env['mail.message'] cls.subtype_model = cls.env['mail.message.subtype'] cls.partner1 = cls.partner_model.with_context( tracking_disable=1).create({ 'name': 'Partner 1', 'email': 'partner1@test.foo.com', }) cls.partner2 = cls.partner_model.with_context( tracking_disable=1).create({ 'name': 'Partner 2', 'email': 'partner2@test.foo.com', }) cls.partner3 = cls.partner_model.with_context( tracking_disable=1).create({ 'name': 'Partner 3', 'email': 'partner3@test.foo.com', }) cls.subtype1 = cls.subtype_model.create({'name': 'Type 1'}) cls.subtype2 = cls.subtype_model.create({'name': 'Type 2'}) def _assert_found(self, domain, not_found=False, partner=None): partner = partner or self.partner1 if not_found: self.assertNotIn(partner, self.partner_model.search(domain)) else: self.assertIn(partner, self.partner_model.search(domain)) def test_notify_domains_always(self): # we don't set recipients # because we call `_get_notify_by_email_domain` directly message = self.message_model.create({'body': 'My Body', }) partner = self.partner1 partner.notify_email = 'always' domain = partner._get_notify_by_email_domain(message) self._assert_found(domain) domain = partner._get_notify_by_email_domain(message, digest=1) self._assert_found(domain, not_found=1) def test_notify_domains_only_recipients(self): # we don't set recipients # because we call `_get_notify_by_email_domain` directly self.partner1.notify_email = 'always' self.partner2.notify_email = 'always' partners = self.partner1 + self.partner2 # followers self.partner3.message_subscribe(self.partner2.ids) # partner1 is the only recipient message = self.message_model.create({ 'body': 'My Body', 'res_id': self.partner3.id, 'model': 'res.partner', 'partner_ids': [(4, self.partner1.id)] }) domain = partners._get_notify_by_email_domain(message) # we find both of them since partner2 is a follower self._assert_found(domain) self._assert_found(domain, partner=self.partner2) # no one here in digest mode domain = partners._get_notify_by_email_domain(message, digest=1) self._assert_found(domain, not_found=1) self._assert_found(domain, not_found=1, partner=self.partner2) # include only recipients domain = partners.with_context( notify_only_recipients=1)._get_notify_by_email_domain(message) self._assert_found(domain) self._assert_found(domain, partner=self.partner2, not_found=1) def test_notify_domains_digest(self): # we don't set recipients # because we call `_get_notify_by_email_domain` directly message = self.message_model.create({'body': 'My Body', }) partner = self.partner1 partner.notify_email = 'digest' domain = partner._get_notify_by_email_domain(message) self._assert_found(domain, not_found=1) domain = partner._get_notify_by_email_domain(message, digest=1) self._assert_found(domain) def test_notify_domains_digest_force_send(self): # when `force_send` is true, digest machinery is bypassed message = self.message_model.create({'body': 'My Body', }) partner = self.partner1 partner.notify_email = 'digest' # even if we have digest mode on, we find the guy domain = partner._get_notify_by_email_domain(message, force_send=True) self._assert_found(domain) # when asking for digest domain we don't get digest-related leaves # as digest domain part is bypassed domain = partner._get_notify_by_email_domain( message, force_send=True, digest=True) self.assertNotIn('notify_email', [x[0] for x in domain]) def test_notify_domains_none(self): message = self.message_model.create({'body': 'My Body', }) partner = self.partner1 partner.notify_email = 'none' domain = partner._get_notify_by_email_domain(message) self._assert_found(domain, not_found=1) domain = partner._get_notify_by_email_domain(message, digest=1) self._assert_found(domain, not_found=1) def test_notify_domains_match_type_digest(self): # Test message subtype matches partner settings. # The partner can have several `partner.notification.conf` records. # Each records establish notification rules by type. # If you don't have any record in it, you allow all subtypes. # Record `typeX` with `enable=True` enables notification for `typeX`. # Record `typeX` with `enable=False` disables notification for `typeX`. partner = self.partner1 # enable digest partner.notify_email = 'digest' message_t1 = self.message_model.create({ 'body': 'My Body', 'subtype_id': self.subtype1.id, }) message_t2 = self.message_model.create({ 'body': 'My Body', 'subtype_id': self.subtype2.id, }) # enable subtype on partner partner._notify_enable_subtype(self.subtype1) domain = partner._get_notify_by_email_domain( message_t1, digest=True) # notification enabled: we find the partner. self._assert_found(domain) # for subtype2 we don't have any explicit rule: we find the partner domain = partner._get_notify_by_email_domain( message_t2, digest=True) self._assert_found(domain) # enable subtype2: find the partner anyway partner._notify_enable_subtype(self.subtype2) domain = partner._get_notify_by_email_domain( message_t2, digest=True) self._assert_found(domain) # disable subtype2: we don't find the partner anymore partner._notify_disable_subtype(self.subtype2) domain = partner._get_notify_by_email_domain( message_t2, digest=True) self._assert_found(domain, not_found=1) def test_notify_domains_match_type_always(self): message_t1 = self.message_model.create({ 'body': 'My Body', 'subtype_id': self.subtype1.id, }) message_t2 = self.message_model.create({ 'body': 'My Body', 'subtype_id': self.subtype2.id, }) # enable always partner = self.partner1 partner.notify_email = 'always' # enable subtype on partner partner._notify_enable_subtype(self.subtype1) domain = partner._get_notify_by_email_domain(message_t1) # notification enabled: we find the partner. self._assert_found(domain) # for subtype2 we don't have any explicit rule: we find the partner domain = partner._get_notify_by_email_domain(message_t2) self._assert_found(domain) # enable subtype2: find the partner anyway partner._notify_enable_subtype(self.subtype2) domain = partner._get_notify_by_email_domain(message_t2) self._assert_found(domain) # disable subtype2: we don't find the partner anymore partner._notify_disable_subtype(self.subtype2) domain = partner._get_notify_by_email_domain(message_t2) self._assert_found(domain, not_found=1)