173 lines
10 KiB
Python
173 lines
10 KiB
Python
# -*- coding: utf-8 -*-
|
|
##############################################################################
|
|
#
|
|
# OpenERP, Open Source Business Applications
|
|
# Copyright (c) 2012-TODAY OpenERP S.A. <http://openerp.com>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
##############################################################################
|
|
|
|
import base64
|
|
from openerp.addons.mail.tests import test_mail
|
|
|
|
|
|
class test_message_compose(test_mail.TestMailMockups):
|
|
|
|
def setUp(self):
|
|
super(test_message_compose, self).setUp()
|
|
self.mail_group = self.registry('mail.group')
|
|
self.mail_mail = self.registry('mail.mail')
|
|
self.mail_message = self.registry('mail.message')
|
|
self.res_users = self.registry('res.users')
|
|
self.res_partner = self.registry('res.partner')
|
|
|
|
# create a 'pigs' and 'bird' groups that will be used through the various tests
|
|
self.group_pigs_id = self.mail_group.create(self.cr, self.uid,
|
|
{'name': 'Pigs', 'description': 'Fans of Pigs, unite !'})
|
|
self.group_bird_id = self.mail_group.create(self.cr, self.uid,
|
|
{'name': 'Bird', 'description': 'I am angry !'})
|
|
|
|
def test_00_message_compose_wizard(self):
|
|
""" Tests designed for the mail.compose.message wizard updated by email_template. """
|
|
cr, uid = self.cr, self.uid
|
|
mail_compose = self.registry('mail.compose.message')
|
|
self.res_users.write(cr, uid, [uid], {'signature': 'Admin', 'email': 'a@a.a'})
|
|
user_admin = self.res_users.browse(cr, uid, uid)
|
|
p_a_id = user_admin.partner_id.id
|
|
group_pigs = self.mail_group.browse(cr, uid, self.group_pigs_id)
|
|
group_bird = self.mail_group.browse(cr, uid, self.group_bird_id)
|
|
|
|
# Mail data
|
|
_subject1 = 'Pigs'
|
|
_subject2 = 'Bird'
|
|
_body_html1 = 'Fans of Pigs, unite !\n<pre>Admin</pre>\n'
|
|
_body_html2 = 'I am angry !\n<pre>Admin</pre>\n'
|
|
_attachments = [
|
|
{'name': 'First', 'datas_fname': 'first.txt', 'datas': base64.b64encode('My first attachment')},
|
|
{'name': 'Second', 'datas_fname': 'second.txt', 'datas': base64.b64encode('My second attachment')}
|
|
]
|
|
_attachments_test = [('first.txt', 'My first attachment'), ('second.txt', 'My second attachment')]
|
|
|
|
# Create template on mail.group, with attachments
|
|
group_model_id = self.registry('ir.model').search(cr, uid, [('model', '=', 'mail.group')])[0]
|
|
email_template = self.registry('email.template')
|
|
email_template_id = email_template.create(cr, uid, {'model_id': group_model_id,
|
|
'name': 'Pigs Template', 'subject': '${object.name}',
|
|
'body_html': '${object.description}', 'user_signature': True,
|
|
'attachment_ids': [(0, 0, _attachments[0]), (0, 0, _attachments[1])],
|
|
'email_to': 'b@b.b c@c.c', 'email_cc': 'd@d.d'})
|
|
|
|
# ----------------------------------------
|
|
# CASE1: comment and save as template
|
|
# ----------------------------------------
|
|
|
|
# 1. Comment on pigs
|
|
compose_id = mail_compose.create(cr, uid,
|
|
{'subject': 'Forget me subject', 'body': '<p>Dummy body</p>'},
|
|
{'default_composition_mode': 'comment', 'default_model': 'mail.group',
|
|
'default_res_id': self.group_pigs_id,
|
|
'default_template_id': email_template_id,
|
|
'active_ids': [self.group_pigs_id, self.group_bird_id]})
|
|
compose = mail_compose.browse(cr, uid, compose_id)
|
|
|
|
# 2. Save current composition form as a template
|
|
mail_compose.save_as_template(cr, uid, [compose_id], context={'default_model': 'mail.group'})
|
|
# Test: email_template subject, body_html, model
|
|
last_template_id = email_template.search(cr, uid, [('model', '=', 'mail.group'), ('subject', '=', 'Forget me subject')], limit=1)[0]
|
|
self.assertTrue(last_template_id, 'email_template not found for model mail.group, subject Forget me subject')
|
|
last_template = email_template.browse(cr, uid, last_template_id)
|
|
self.assertEqual(last_template.body_html, '<p>Dummy body</p>', 'email_template incorrect body_html')
|
|
|
|
# ----------------------------------------
|
|
# CASE2: comment with template, save as template
|
|
# ----------------------------------------
|
|
|
|
# 1. Comment on pigs
|
|
compose_id = mail_compose.create(cr, uid,
|
|
{'subject': 'Forget me subject', 'body': 'Dummy body'},
|
|
{'default_composition_mode': 'comment', 'default_model': 'mail.group',
|
|
'default_res_id': self.group_pigs_id,
|
|
'default_template_id': email_template_id,
|
|
'active_ids': [self.group_pigs_id, self.group_bird_id]})
|
|
compose = mail_compose.browse(cr, uid, compose_id)
|
|
|
|
# 2. Perform 'toggle_template', to set use_template and use template_id
|
|
mail_compose.toggle_template(cr, uid, [compose_id], {'default_composition_mode': 'comment', 'default_model': 'mail.group'})
|
|
compose.refresh()
|
|
message_pids = [partner.id for partner in compose.partner_ids]
|
|
partner_ids = self.res_partner.search(cr, uid, [('email', 'in', ['b@b.b', 'c@c.c', 'd@d.d'])])
|
|
# Test: mail.compose.message: subject, body, content_subtype, partner_ids
|
|
self.assertEqual(compose.subject, _subject1, 'mail.compose.message subject incorrect')
|
|
self.assertEqual(compose.body, _body_html1, 'mail.compose.message body incorrect')
|
|
self.assertEqual(compose.content_subtype, 'html', 'mail.compose.message content_subtype incorrect')
|
|
self.assertEqual(set(message_pids), set(partner_ids), 'mail.compose.message partner_ids incorrect')
|
|
# Test: mail.compose.message: attachments
|
|
# Test: mail.message: attachments
|
|
for attach in compose.attachment_ids:
|
|
self.assertEqual(attach.res_model, 'mail.group', 'mail.message attachment res_model incorrect')
|
|
self.assertEqual(attach.res_id, self.group_pigs_id, 'mail.message attachment res_id incorrect')
|
|
self.assertIn((attach.name, base64.b64decode(attach.datas)), _attachments_test,
|
|
'mail.message attachment name / data incorrect')
|
|
|
|
# 3. Perform 'toggle_template': template is not set anymore
|
|
mail_compose.toggle_template(cr, uid, [compose_id], {'default_composition_mode': 'comment', 'default_model': 'mail.group'})
|
|
compose.refresh()
|
|
# Test: subject, body, partner_ids
|
|
self.assertEqual(compose.subject, False, 'mail.compose.message subject incorrect')
|
|
self.assertEqual(compose.body, '', 'mail.compose.message body incorrect')
|
|
|
|
# ----------------------------------------
|
|
# CASE3: mass_mail with template
|
|
# ----------------------------------------
|
|
|
|
# 1. Mass_mail on pigs and bird, with a default_partner_ids set to check he is correctly added
|
|
compose_id = mail_compose.create(cr, uid,
|
|
{'subject': 'Forget me subject', 'body': 'Dummy body'},
|
|
{'default_composition_mode': 'mass_mail', 'default_model': 'mail.group',
|
|
'default_res_id': self.group_pigs_id,
|
|
'default_template_id': email_template_id,
|
|
'default_partner_ids': [p_a_id],
|
|
'active_ids': [self.group_pigs_id, self.group_bird_id]})
|
|
compose = mail_compose.browse(cr, uid, compose_id)
|
|
|
|
# 2. Perform 'toggle_template', to set use_template and use template_id
|
|
mail_compose.toggle_template(cr, uid, [compose_id], {'default_composition_mode': 'comment', 'default_model': 'mail.group'})
|
|
compose.refresh()
|
|
message_pids = [partner.id for partner in compose.partner_ids]
|
|
partner_ids = [p_a_id]
|
|
# Test: mail.compose.message: subject, body, content_subtype, partner_ids
|
|
self.assertEqual(compose.subject, '${object.name}', 'mail.compose.message subject incorrect')
|
|
self.assertEqual(compose.body, '${object.description}', 'mail.compose.message body incorrect')
|
|
self.assertEqual(compose.content_subtype, 'html', 'mail.compose.message content_subtype incorrect')
|
|
self.assertEqual(set(message_pids), set(partner_ids), 'mail.compose.message partner_ids incorrect')
|
|
|
|
# 3. Post the comment, get created message
|
|
mail_compose.send_mail(cr, uid, [compose_id], {'default_res_id': -1, 'active_ids': [self.group_pigs_id, self.group_bird_id]})
|
|
group_pigs.refresh()
|
|
group_bird.refresh()
|
|
message_pigs = group_pigs.message_ids[0]
|
|
message_bird = group_bird.message_ids[0]
|
|
# Test: subject, body
|
|
self.assertEqual(message_pigs.subject, _subject1, 'mail.message subject on Pigs incorrect')
|
|
self.assertEqual(message_bird.subject, _subject2, 'mail.message subject on Bird incorrect')
|
|
self.assertEqual(message_pigs.body, _body_html1, 'mail.message body on Pigs incorrect')
|
|
self.assertEqual(message_bird.body, _body_html2, 'mail.message body on Bird incorrect')
|
|
# Test: partner_ids: p_a_id (default) + 3 newly created partners
|
|
message_pigs_pids = [partner.id for partner in message_pigs.notified_partner_ids]
|
|
message_bird_pids = [partner.id for partner in message_bird.notified_partner_ids]
|
|
partner_ids = self.res_partner.search(cr, uid, [('email', 'in', ['b@b.b', 'c@c.c', 'd@d.d'])])
|
|
self.assertEqual(set(message_pigs_pids), set(partner_ids), 'mail.message on pigs incorrect number of notified_partner_ids')
|
|
self.assertEqual(set(message_bird_pids), set(partner_ids), 'mail.message on bird notified_partner_ids incorrect')
|