social/mail_optional_autofollow/tests/test_mail_optional_autofollow.py

# Copyright 2016 ACSONE SA/NV (<http://acsone.eu>)
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).

from odoo.tests import common


class TestAttachExistingAttachment(common.TransactionCase):

    def setUp(self):
        super(TestAttachExistingAttachment, self).setUp()
        self.partner_obj = self.env['res.partner']
        self.partner_01 = self.env.ref('base.res_partner_10')
        self.partner_02 = self.env.ref('base.res_partner_address_17')

    def test_send_email_attachment(self):
        ctx = self.env.context.copy()
        ctx.update({
            'default_model': 'res.partner',
            'default_res_id': self.partner_01.id,
            'default_composition_mode': 'comment',
        })
        mail_compose = self.env['mail.compose.message']
        values = mail_compose.with_context(ctx)\
            .onchange_template_id(False, 'comment', 'res.partner',
                                  self.partner_01.id)['value']
        values['partner_ids'] = [(4, self.partner_02.id)]
        compose_id = mail_compose.with_context(ctx).create(values)
        compose_id.autofollow_recipients = False
        compose_id.with_context(ctx).send_mail()
        res = self.env["mail.followers"].search(
            [('res_model', '=', 'res.partner'),
             ('res_id', '=', self.partner_01.id),
             ('partner_id', '=', self.partner_02.id)])
        # I check if the recipient isn't a follower
        self.assertEqual(len(res.ids), 0)
        compose_id = mail_compose.with_context(ctx).create(values)
        compose_id.autofollow_recipients = True
        compose_id.with_context(ctx).send_mail()
        res = self.env["mail.followers"].search(
            [('res_model', '=', 'res.partner'),
             ('res_id', '=', self.partner_01.id),
             ('partner_id', '=', self.partner_02.id)])
        # I check if the recipient is a follower
        self.assertEqual(len(res.ids), 1)