1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165
|
# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.
import base64
import datetime
from odoo.addons.mail.tests.common import TestMail
from odoo.tools import mute_logger
class TestMailTemplate(TestMail):
def setUp(self):
super(TestMailTemplate, self).setUp()
self._attachments = [{
'name': '_Test_First',
'datas_fname':
'first.txt',
'datas': base64.b64encode(b'My first attachment'),
'res_model': 'res.partner',
'res_id': self.user_admin.partner_id.id
}, {
'name': '_Test_Second',
'datas_fname': 'second.txt',
'datas': base64.b64encode(b'My second attachment'),
'res_model': 'res.partner',
'res_id': self.user_admin.partner_id.id
}]
self.email_1 = 'test1@example.com'
self.email_2 = 'test2@example.com'
self.email_3 = self.partner_1.email
self.email_template = self.env['mail.template'].create({
'model_id': self.env['ir.model']._get('mail.test').id,
'name': 'Pigs Template',
'subject': '${object.name}',
'body_html': '${object.description}',
'user_signature': False,
'attachment_ids': [(0, 0, self._attachments[0]), (0, 0, self._attachments[1])],
'partner_to': '%s,%s' % (self.partner_2.id, self.user_employee.partner_id.id),
'email_to': '%s, %s' % (self.email_1, self.email_2),
'email_cc': '%s' % self.email_3})
def test_composer_template_onchange(self):
composer = self.env['mail.compose.message'].with_context({
'default_composition_mode': 'comment',
'default_model': 'mail.test',
'default_res_id': self.test_pigs.id,
'default_use_template': False,
'default_template_id': False
}).create({'subject': 'Forget me subject', 'body': 'Dummy body'})
values = composer.onchange_template_id(self.email_template.id, 'comment', 'mail.test', self.test_pigs.id)['value']
# use _convert_to_cache to return a browse record list from command list or id list for x2many fields
values = composer._convert_to_record(composer._convert_to_cache(values))
recipients = values['partner_ids']
attachments = values['attachment_ids']
test_recipients = self.env['res.partner'].search([('email', 'in', ['test1@example.com', 'test2@example.com'])]) | self.partner_1 | self.partner_2 | self.user_employee.partner_id
test_attachments = self.env['ir.attachment'].search([('name', 'in', ['_Test_First', '_Test_Second'])])
self.assertEqual(values['subject'], self.test_pigs.name)
self.assertEqual(values['body'], '<p>%s</p>' % self.test_pigs.description)
self.assertEqual(recipients, test_recipients)
self.assertEqual(set(recipients.mapped('email')), set([self.email_1, self.email_2, self.partner_1.email, self.partner_2.email, self.user_employee.email]))
self.assertEqual(attachments, test_attachments)
self.assertEqual(set(attachments.mapped('res_model')), set(['res.partner']))
self.assertEqual(set(attachments.mapped('res_id')), set([self.user_admin.partner_id.id]))
@mute_logger('odoo.addons.mail.models.mail_mail')
def test_composer_template_send(self):
self.test_pigs.with_context(use_template=False).message_post_with_template(self.email_template.id, composition_mode='comment')
message = self.test_pigs.message_ids[0]
test_recipients = self.env['res.partner'].search([('email', 'in', ['test1@example.com', 'test2@example.com'])]) | self.partner_1 | self.partner_2 | self.user_employee.partner_id
self.assertEqual(message.subject, self.test_pigs.name)
self.assertEqual(message.body, '<p>%s</p>' % self.test_pigs.description)
self.assertEqual(message.partner_ids, test_recipients)
self.assertEqual(set(message.attachment_ids.mapped('res_model')), set(['mail.test']))
self.assertEqual(set(message.attachment_ids.mapped('res_id')), set([self.test_pigs.id]))
# self.assertIn((attach.datas_fname, base64.b64decode(attach.datas)), _attachments_test,
# 'mail.message attachment name / data incorrect')
@mute_logger('odoo.addons.mail.models.mail_mail')
def test_composer_template_mass_mailing(self):
composer = self.env['mail.compose.message'].with_context({
'default_composition_mode': 'mass_mail',
'default_notify': True,
'default_model': 'mail.test',
'default_res_id': self.test_pigs.id,
'default_template_id': self.email_template.id,
'active_ids': [self.test_pigs.id, self.test_public.id]
}).create({})
values = composer.onchange_template_id(self.email_template.id, 'mass_mail', 'mail.test', self.test_pigs.id)['value']
composer.write(values)
composer.send_mail()
message_1 = self.test_pigs.message_ids[0]
message_2 = self.test_public.message_ids[0]
self.assertEqual(message_1.subject, self.test_pigs.name, 'mail.message subject on Pigs incorrect')
self.assertEqual(message_2.subject, self.test_public.name, 'mail.message subject on Bird incorrect')
self.assertIn(self.test_pigs.description, message_1.body, 'mail.message body on Pigs incorrect')
self.assertIn(self.test_public.description, message_2.body, 'mail.message body on Bird incorrect')
# todo for JDC: ! (False -> <p>False</p>)
def test_mail_template(self):
mail_id = self.email_template.send_mail(self.test_pigs.id)
mail = self.env['mail.mail'].browse(mail_id)
self.assertEqual(mail.subject, self.test_pigs.name)
self.assertEqual(mail.email_to, self.email_template.email_to)
self.assertEqual(mail.email_cc, self.email_template.email_cc)
self.assertEqual(mail.recipient_ids, self.partner_2 | self.user_employee.partner_id)
def test_message_compose_template_save(self):
self.env['mail.compose.message'].with_context(
{'default_composition_mode': 'comment',
'default_model': 'mail.test',
'default_res_id': self.test_pigs.id,
'active_ids': [self.test_pigs.id, self.test_public.id]
}).create({
'subject': 'Forget me subject',
'body': '<p>Dummy body</p>'
}).with_context({'default_model': 'mail.test'}).save_as_template()
# Test: email_template subject, body_html, model
last_template = self.env['mail.template'].search([('model', '=', 'mail.test'), ('subject', '=', 'Forget me subject')], limit=1)
self.assertEqual(last_template.body_html, '<p>Dummy body</p>', 'email_template incorrect body_html')
def test_add_context_action(self):
self.email_template.create_action()
# check template act_window has been updated
self.assertTrue(bool(self.email_template.ref_ir_act_window))
# check those records
action = self.email_template.ref_ir_act_window
self.assertEqual(action.name, 'Send Mail (%s)' % self.email_template.name)
self.assertEqual(action.binding_model_id.model, 'mail.test')
def test_set_scheduled_date_on_a_template(self):
self.email_template_in_2_days = self.email_template.copy()
self.email_template_in_2_days.write({'scheduled_date': "${(datetime.datetime.now() + relativedelta(days=2)).strftime('%Y-%m-%d %H:%M')}"})
mail_now_id = self.email_template.send_mail(self.test_pigs.id)
mail_in_2_days_id = self.email_template_in_2_days.send_mail(self.test_pigs.id)
mail_now = self.env['mail.mail'].browse(mail_now_id)
mail_in_2_days = self.env['mail.mail'].browse(mail_in_2_days_id)
# assert scheduled date are correct
self.assertEqual(bool(mail_now.scheduled_date), False)
scheduled_date = datetime.datetime.strptime(mail_in_2_days.scheduled_date, '%Y-%m-%d %H:%M')
date_in_2_days = datetime.datetime.today() + datetime.timedelta(days = 2)
self.assertEqual(scheduled_date.day, date_in_2_days.day)
self.assertEqual(scheduled_date.month, date_in_2_days.month)
self.assertEqual(scheduled_date.year, date_in_2_days.year)
# Launch the scheduler on the first mail, it should be reported in self.mails
# and the mail_mail is now deleted
self.env['mail.mail'].process_email_queue(ids=[mail_now.id])
self.assertTrue(len(self._mails) > 0)
# Launch the scheduler on the first mail, it's still in 'outgoing' state
self.env['mail.mail'].process_email_queue(ids=[mail_in_2_days.id])
self.assertEqual(mail_in_2_days.state, 'outgoing')
|