# -*- coding: utf-8 -*- # Part of Odoo. See LICENSE file for full copyright and licensing details. import base64 import datetime from freezegun import freeze_time from odoo.addons.mail.tests.common import MailCommon from odoo.addons.test_mail.tests.common import TestRecipients from odoo.tests import tagged, users from odoo.tools import mute_logger class TestMailTemplateCommon(MailCommon, TestRecipients): @classmethod def setUpClass(cls): super(TestMailTemplateCommon, cls).setUpClass() cls.test_record = cls.env['mail.test.lang'].with_context(cls._test_context).create({ 'email_from': 'ignasse@example.com', 'name': 'Test', }) cls.user_employee.write({ 'groups_id': [(4, cls.env.ref('base.group_partner_manager').id)], }) cls._attachments = [{ 'name': 'first.txt', 'datas': base64.b64encode(b'My first attachment'), 'res_model': 'res.partner', 'res_id': cls.user_admin.partner_id.id }, { 'name': 'second.txt', 'datas': base64.b64encode(b'My second attachment'), 'res_model': 'res.partner', 'res_id': cls.user_admin.partner_id.id }] cls.email_1 = 'test1@example.com' cls.email_2 = 'test2@example.com' cls.email_3 = cls.partner_1.email # create a complete test template cls.test_template = cls._create_template('mail.test.lang', { 'attachment_ids': [(0, 0, cls._attachments[0]), (0, 0, cls._attachments[1])], 'body_html': '

EnglishBody for

', 'lang': '{{ object.customer_id.lang or object.lang }}', 'email_to': '%s, %s' % (cls.email_1, cls.email_2), 'email_cc': '%s' % cls.email_3, 'partner_to': '%s,%s' % (cls.partner_2.id, cls.user_admin.partner_id.id), 'subject': 'EnglishSubject for {{ object.name }}', }) # activate translations cls._activate_multi_lang( layout_arch_db=' English Layout for ', test_record=cls.test_record, test_template=cls.test_template ) # admin should receive emails cls.user_admin.write({'notification_type': 'email'}) # Force the attachments of the template to be in the natural order. cls.test_template.invalidate_recordset(['attachment_ids']) @tagged('mail_template') class TestMailTemplate(TestMailTemplateCommon): def test_template_add_context_action(self): self.test_template.create_action() # check template act_window has been updated self.assertTrue(bool(self.test_template.ref_ir_act_window)) # check those records action = self.test_template.ref_ir_act_window self.assertEqual(action.name, 'Send Mail (%s)' % self.test_template.name) self.assertEqual(action.binding_model_id.model, 'mail.test.lang') @mute_logger('odoo.addons.mail.models.mail_mail') @users('employee') def test_template_schedule_email(self): """ Test scheduling email sending from template. """ now = datetime.datetime.now() test_template = self.test_template.with_env(self.env) # schedule the mail in 3 days test_template.scheduled_date = '{{datetime.datetime.now() + datetime.timedelta(days=3)}}' with freeze_time(now): mail_id = test_template.send_mail(self.test_record.id) mail = self.env['mail.mail'].sudo().browse(mail_id) self.assertEqual( mail.scheduled_date.replace(second=0, microsecond=0), (now + datetime.timedelta(days=3)).replace(second=0, microsecond=0), ) self.assertEqual(mail.state, 'outgoing') # check a wrong format test_template.scheduled_date = '{{"test " * 5}}' with freeze_time(now): mail_id = test_template.send_mail(self.test_record.id) mail = self.env['mail.mail'].sudo().browse(mail_id) self.assertFalse(mail.scheduled_date) self.assertEqual(mail.state, 'outgoing') @mute_logger('odoo.addons.mail.models.mail_mail') def test_template_send_mail_body(self): """ Test that the body and body_html is set correctly in 'mail.mail' when sending an email from mail.template """ mail_id = self.test_template.send_mail(self.test_record.id) mail = self.env['mail.mail'].sudo().browse(mail_id) body_result = '

EnglishBody for %s

' % self.test_record.name self.assertEqual(mail.body_html, body_result) self.assertEqual(mail.body, body_result) @tagged('mail_template', 'multi_lang', 'post_install', '-at_install') class TestMailTemplateLanguages(TestMailTemplateCommon): @classmethod def setUpClass(cls): """ Create lang-based records and templates, to test batch and performances with language involved. """ super().setUpClass() # use test notification layout cls.test_template.write({ 'email_layout_xmlid': 'mail.test_layout', }) # double record, one in each lang cls.test_records = cls.test_record + cls.env['mail.test.lang'].create({ 'email_from': 'ignasse.es@example.com', 'lang': 'es_ES', 'name': 'Test Record 2', }) # pure batch, 100 records cls.test_records_batch, test_partners = cls._create_records_for_batch( 'mail.test.lang', 100, ) test_partners[:50].lang = 'es_ES' # have a template with dynamic templates to check impact cls.test_template_wreports = cls.test_template.copy({ 'email_layout_xmlid': 'mail.test_layout', }) cls.test_reports = cls.env['ir.actions.report'].create([ { 'name': f'Test Report on {cls.test_record._name}', 'model': cls.test_record._name, 'print_report_name': "f'TestReport for {object.name}'", 'report_type': 'qweb-pdf', 'report_name': 'test_mail.mail_test_ticket_test_template', }, { 'name': f'Test Report 2 on {cls.test_record._name}', 'model': cls.test_record._name, 'print_report_name': "f'TestReport2 for {object.name}'", 'report_type': 'qweb-pdf', 'report_name': 'test_mail.mail_test_ticket_test_template_2', } ]) cls.test_template_wreports.report_template_ids = cls.test_reports cls.env.flush_all() @mute_logger('odoo.addons.mail.models.mail_mail') def test_template_send_email(self): """ Test 'send_email' on template on a given record, used notably as contextual action. """ self.env.invalidate_all() with self.with_user(self.user_employee.login), self.assertQueryCount(28): # test_mail: 28 mail_id = self.test_template.with_env(self.env).send_mail(self.test_record.id) mail = self.env['mail.mail'].sudo().browse(mail_id) self.assertEqual(sorted(mail.attachment_ids.mapped('name')), ['first.txt', 'second.txt']) self.assertEqual(mail.body_html, f'

EnglishBody for {self.test_record.name}

English Layout for Lang Chatter Model') self.assertEqual(mail.email_cc, self.test_template.email_cc) self.assertEqual(mail.email_to, self.test_template.email_to) self.assertEqual(mail.recipient_ids, self.partner_2 | self.user_admin.partner_id) self.assertEqual(mail.subject, f'EnglishSubject for {self.test_record.name}') @mute_logger('odoo.addons.mail.models.mail_mail') def test_template_send_email_nolayout(self): """ Test without layout, just to check impact """ self.test_template.email_layout_xmlid = False self.env.invalidate_all() with self.with_user(self.user_employee.login), self.assertQueryCount(21): # test_mail: 21 mail_id = self.test_template.with_env(self.env).send_mail(self.test_record.id) mail = self.env['mail.mail'].sudo().browse(mail_id) self.assertEqual(sorted(mail.attachment_ids.mapped('name')), ['first.txt', 'second.txt']) self.assertEqual(mail.body_html, f'

EnglishBody for {self.test_record.name}

') self.assertEqual(mail.email_cc, self.test_template.email_cc) self.assertEqual(mail.email_to, self.test_template.email_to) self.assertEqual(mail.recipient_ids, self.partner_2 | self.user_admin.partner_id) self.assertEqual(mail.subject, f'EnglishSubject for {self.test_record.name}') @mute_logger('odoo.addons.mail.models.mail_mail') def test_template_send_email_batch(self): """ Test 'send_email' on template in batch """ self.env.invalidate_all() mails = self.env['mail.mail'].sudo() with self.with_user(self.user_employee.login), self.assertQueryCount(934): # test_mail: 934 template = self.test_template.with_env(self.env) for record in self.test_records_batch: mails += mails.browse(template.send_mail(record.id)) self.assertEqual(len(mails), 100) for idx, (mail, record) in enumerate(zip(mails, self.test_records_batch)): self.assertEqual(sorted(mail.attachment_ids.mapped('name')), ['first.txt', 'second.txt']) self.assertEqual(mail.email_cc, self.test_template.email_cc) self.assertEqual(mail.email_to, self.test_template.email_to) self.assertEqual(mail.recipient_ids, self.partner_2 | self.user_admin.partner_id) if idx >= 50: self.assertEqual(mail.subject, f'EnglishSubject for {record.name}') else: self.assertEqual(mail.subject, f'SpanishSubject for {record.name}') @mute_logger('odoo.addons.mail.models.mail_mail') def test_template_send_email_wreport(self): """ Test 'send_email' on template on a given record, used notably as contextual action, with dynamic reports involved """ self.env.invalidate_all() with self.with_user(self.user_employee.login), self.assertQueryCount(107): # test_mail: 106 mail_id = self.test_template_wreports.with_env(self.env).send_mail(self.test_record.id) mail = self.env['mail.mail'].sudo().browse(mail_id) self.assertEqual( sorted(mail.attachment_ids.mapped('name')), [f'TestReport for {self.test_record.name}.html', f'TestReport2 for {self.test_record.name}.html', 'first.txt', 'second.txt'] ) self.assertEqual(mail.recipient_ids, self.partner_2 | self.user_admin.partner_id) self.assertEqual(mail.subject, f'EnglishSubject for {self.test_record.name}') @mute_logger('odoo.addons.mail.models.mail_mail') def test_template_send_email_wreport_batch(self): """ Test 'send_email' on template in batch with dynamic reports """ self.env.invalidate_all() mails = self.env['mail.mail'].sudo() with self.with_user(self.user_employee.login), self.assertQueryCount(1645): # test_mail: 1645 template = self.test_template_wreports.with_env(self.env) for record in self.test_records_batch: mails += mails.browse(template.send_mail(record.id)) self.assertEqual(len(mails), 100) for idx, (mail, record) in enumerate(zip(mails, self.test_records_batch)): self.assertEqual( sorted(mail.attachment_ids.mapped('name')), [f'TestReport for {record.name}.html', f'TestReport2 for {record.name}.html', 'first.txt', 'second.txt'] ) self.assertEqual(mail.email_cc, self.test_template.email_cc) self.assertEqual(mail.email_to, self.test_template.email_to) self.assertEqual(mail.recipient_ids, self.partner_2 | self.user_admin.partner_id) if idx >= 50: self.assertEqual(mail.subject, f'EnglishSubject for {record.name}') self.assertEqual(mail.body_html, f'

EnglishBody for {record.name}

English Layout for Lang Chatter Model') else: self.assertEqual(mail.subject, f'SpanishSubject for {record.name}') self.assertEqual(mail.body_html, f'

SpanishBody for {record.name}

Spanish Layout para Spanish Model Description') @mute_logger('odoo.addons.mail.models.mail_mail') def test_template_translation_lang(self): """ Test template rendering using lang defined directly on the record """ test_record = self.test_record.with_env(self.env) test_record.write({ 'lang': 'es_ES', }) test_template = self.test_template.with_env(self.env) mail_id = test_template.send_mail(test_record.id) mail = self.env['mail.mail'].sudo().browse(mail_id) self.assertEqual(mail.body_html, f'

SpanishBody for {self.test_record.name}

Spanish Layout para Spanish Model Description') self.assertEqual(mail.subject, f'SpanishSubject for {self.test_record.name}') @mute_logger('odoo.addons.mail.models.mail_mail') def test_template_translation_partner_lang(self): """ Test template rendering using lang defined on a sub-record aka 'partner_id.lang' """ test_records = self.env['mail.test.lang'].browse(self.test_records.ids) customers = self.env['res.partner'].create([ { 'email': 'roberto.carlos@test.example.com', 'lang': 'es_ES', 'name': 'Roberto Carlos', }, { 'email': 'rob.charly@test.example.com', 'lang': 'en_US', 'name': 'Rob Charly', } ]) test_records[0].write({'customer_id': customers[0].id}) test_records[1].write({'customer_id': customers[1].id}) self.env.invalidate_all() mails = self.env['mail.mail'].sudo() with self.with_user(self.user_employee.login), self.assertQueryCount(52): template = self.test_template.with_env(self.env) for record in self.test_records: mails += mails.browse( template.send_mail(record.id, email_layout_xmlid='mail.test_layout') ) self.assertEqual(mails[0].body_html, f'

SpanishBody for {test_records[0].name}

Spanish Layout para Spanish Model Description') self.assertEqual(mails[0].subject, f'SpanishSubject for {test_records[0].name}') self.assertEqual(mails[1].body_html, f'

EnglishBody for {test_records[1].name}

English Layout for Lang Chatter Model') self.assertEqual(mails[1].subject, f'EnglishSubject for {test_records[1].name}')