Yaltik/social
2
0
Fork 0
Code Pull Requests Releases Activity

[IMP] mail_restrict_follower_selection: Adding tests and improving pre-commit

Browse Source
This commit is contained in:
angelmoya 2020-12-22 10:18:50 +01:00 committed by Aungkokolin1997
parent d6cc4a8aa9
commit 9b95ddd50c
3 changed files with 30 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mail_restrict_follower_selection/models/mail_followers.py
View File

@ -35,11 +35,10 @@ class MailFollowers(models.Model):
existing_policy=existing_policy,
)
domain = self.env[
'mail.wizard.invite'
"mail.wizard.invite"
]._mail_restrict_follower_selection_get_domain(res_model=res_model)
partners = self.env['res.partner'].search(
[('id', 'in', partner_ids)] +
safe_eval(domain)
partners = self.env["res.partner"].search(
[("id", "in", partner_ids)] + safe_eval(domain)
)
_res_ids = res_ids.copy() or [0]
new, update = super()._add_followers(

18
mail_restrict_follower_selection/models/mail_wizard_invite.py
View File

@ -13,13 +13,17 @@ class MailWizardInvite(models.TransientModel):
@api.model
def _mail_restrict_follower_selection_get_domain(self, res_model=None):
if not res_model:
res_model = self.env.context.get('default_res_model')
parameter_name = 'mail_restrict_follower_selection.domain'
return self.env['ir.config_parameter'].sudo().get_param(
"{0}.{1}".format(parameter_name,
res_model),
self.env['ir.config_parameter'].sudo().get_param(
parameter_name, default='[]')
res_model = self.env.context.get("default_res_model")
parameter_name = "mail_restrict_follower_selection.domain"
return (
self.env["ir.config_parameter"]
.sudo()
.get_param(
"{}.{}".format(parameter_name, res_model),
self.env["ir.config_parameter"]
.sudo()
.get_param(parameter_name, default="[]"),
)
)
@api.model

16
mail_restrict_follower_selection/tests/test_mail_restrict_follower_selection.py
View File

@ -64,3 +64,19 @@ class TestMailRestrictFollowerSelection(TransactionCase):
self.assertNotIn(
self.partner, self.partner.message_follower_ids.mapped("partner_id")
)
def test_message_add_suggested_recipient(self):
res = self.partner.with_context(
test_restrict_follower=True
)._message_add_suggested_recipient({self.partner.id: []}, partner=self.partner)
self.assertEqual(res[self.partner.id][0][0], self.partner.id)
self.env["ir.config_parameter"].create(
{
"key": "mail_restrict_follower_selection.domain.res.partner",
"value": "[('category_id.name', '!=', 'Employees')]",
}
)
new_res = self.partner.with_context(
test_restrict_follower=True
)._message_add_suggested_recipient({self.partner.id: []})
self.assertFalse(new_res[self.partner.id][0][0])