Xavier Morel
c56e8c9f5a
* fix generation of preferences markup so everything's properly safe (and flag as markup) * tips should be the result of templates rendering and thus safe * same for body
75 lines
2.8 KiB
Python
75 lines
2.8 KiB
Python
import itertools
|
|
import random
|
|
|
|
from dateutil.relativedelta import relativedelta
|
|
from lxml import html
|
|
|
|
from odoo import fields
|
|
from odoo.addons.mail.tests import common as mail_test
|
|
|
|
|
|
class TestDigest(mail_test.MailCommon):
|
|
def test_digest_numbers(self):
|
|
self._setup_messages()
|
|
|
|
digest = self.env['digest.digest'].create({
|
|
'name': "My Digest",
|
|
'kpi_mail_message_total': True
|
|
})
|
|
|
|
digest_user = digest.with_user(self.user_employee)
|
|
# subscribe a user so at least one mail gets sent
|
|
digest_user.action_subscribe()
|
|
self.assertTrue(
|
|
digest_user.is_subscribed,
|
|
"check the user was subscribed as action_subscribe will silently "
|
|
"ignore subs of non-employees"
|
|
)
|
|
|
|
# digest creates its mails in auto_delete mode so we need to capture
|
|
# the formatted body during the sending process
|
|
with self.mock_mail_gateway():
|
|
digest.action_send()
|
|
|
|
self.assertEqual(len(self._mails), 1, "a mail has been created for the digest")
|
|
body = self._mails[0]['body']
|
|
|
|
kpi_message_values = html.fromstring(body).xpath('//div[@data-field="kpi_mail_message_total"]//*[hasclass("kpi_value")]/text()')
|
|
self.assertEqual(
|
|
[t.strip() for t in kpi_message_values],
|
|
['3', '8', '15']
|
|
)
|
|
|
|
def _setup_messages(self):
|
|
""" Remove all existing messages, then create a bunch of them on random
|
|
partners with the correct types in correct time-bucket:
|
|
|
|
- 3 in the previous 24h
|
|
- 5 more in the 6 days before that for a total of 8 in the previous week
|
|
- 7 more in the 20 days before *that* (because digest doc lies and is
|
|
based around weeks and months not days), for a total of 15 in the
|
|
previous month
|
|
"""
|
|
self.env['mail.message'].search([]).unlink()
|
|
now = fields.Datetime.now()
|
|
# regular employee can't necessarily access "private" addresses
|
|
partners = self.env['res.partner'].search([('type', '!=', 'private')])
|
|
counter = itertools.count()
|
|
|
|
# pylint: disable=bad-whitespace
|
|
for count, (low, high) in [
|
|
(3, (0 * 24, 1 * 24)),
|
|
(5, (1 * 24, 7 * 24)),
|
|
(7, (7 * 24, 27 * 24)),
|
|
]:
|
|
for _ in range(count):
|
|
create_date = now - relativedelta(hours=random.randint(low + 1, high - 1))
|
|
random.choice(partners).message_post(
|
|
body=f"Awesome Partner! ({next(counter)})",
|
|
message_type='comment',
|
|
subtype_xmlid='mail.mt_comment',
|
|
# adjust top and bottom by 1h to avoid overlapping with the
|
|
# range limit and dropping out of the digest's selection thing
|
|
create_date=create_date
|
|
)
|