# -*- coding: utf-8 -*- # © 2016 Antonio Espinosa - # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html). import mock import base64 import time from openerp import http from openerp.tests.common import TransactionCase from ..controllers.main import MailTrackingController, BLANK mock_send_email = ('openerp.addons.base.ir.ir_mail_server.' 'ir_mail_server.send_email') class FakeUserAgent(object): browser = 'Test browser' platform = 'Test platform' def __str__(self): """Return name""" return 'Test suite' class TestMailTracking(TransactionCase): def setUp(self, *args, **kwargs): super(TestMailTracking, self).setUp(*args, **kwargs) self.sender = self.env['res.partner'].create({ 'name': 'Test sender', 'email': 'sender@example.com', 'notify_email': 'always', }) self.recipient = self.env['res.partner'].create({ 'name': 'Test recipient', 'email': 'recipient@example.com', 'notify_email': 'always', }) self.last_request = http.request http.request = type('obj', (object,), { 'db': self.env.cr.dbname, 'env': self.env, 'endpoint': type('obj', (object,), { 'routing': [], }), 'httprequest': type('obj', (object,), { 'remote_addr': '123.123.123.123', 'user_agent': FakeUserAgent(), }), }) def tearDown(self, *args, **kwargs): http.request = self.last_request return super(TestMailTracking, self).tearDown(*args, **kwargs) def test_message_post(self): # This message will generate a notification for recipient message = self.env['mail.message'].create({ 'subject': 'Message test', 'author_id': self.sender.id, 'email_from': self.sender.email, 'type': 'comment', 'model': 'res.partner', 'res_id': self.recipient.id, 'partner_ids': [(4, self.recipient.id)], 'body': '

This is a test message

', }) # Search tracking created tracking_email = self.env['mail.tracking.email'].search([ ('mail_message_id', '=', message.id), ('partner_id', '=', self.recipient.id), ]) # The tracking email must be sent self.assertTrue(tracking_email) self.assertEqual(tracking_email.state, 'sent') # message_dict read by web interface message_dict = self.env['mail.message'].message_read(message.id) # First item is message content self.assertTrue(len(message_dict) > 0) message_dict = message_dict[0] self.assertTrue(len(message_dict['partner_ids']) > 0) # First partner is recipient partner_id = message_dict['partner_ids'][0][0] self.assertEqual(partner_id, self.recipient.id) status = message_dict['partner_trackings'][str(partner_id)] # Tracking status must be sent and # mail tracking must be the one search before self.assertEqual(status[0], 'sent') self.assertEqual(status[1], tracking_email.id) # And now open the email metadata = { 'ip': '127.0.0.1', 'user_agent': 'Odoo Test/1.0', 'os_family': 'linux', 'ua_family': 'odoo', } tracking_email.event_create('open', metadata) self.assertEqual(tracking_email.state, 'opened') def mail_send(self, recipient): mail = self.env['mail.mail'].create({ 'subject': 'Test subject', 'email_from': 'from@domain.com', 'email_to': recipient, 'body_html': '

This is a test message

', }) mail.send() # Search tracking created tracking_email = self.env['mail.tracking.email'].search([ ('mail_id', '=', mail.id), ]) return mail, tracking_email def test_mail_send(self): controller = MailTrackingController() db = self.env.cr.dbname image = base64.decodestring(BLANK) mail, tracking = self.mail_send(self.recipient.email) self.assertEqual(mail.email_to, tracking.recipient) self.assertEqual(mail.email_from, tracking.sender) res = controller.mail_tracking_open(db, tracking.id) self.assertEqual(image, res.response[0]) # Two events: sent and open self.assertEqual(2, len(tracking.tracking_event_ids)) # Fake event: tracking_email_id = False res = controller.mail_tracking_open(db, False) self.assertEqual(image, res.response[0]) # Two events again because no tracking_email_id found for False self.assertEqual(2, len(tracking.tracking_event_ids)) def test_concurrent_open(self): mail, tracking = self.mail_send(self.recipient.email) ts = time.time() metadata = { 'ip': '127.0.0.1', 'user_agent': 'Odoo Test/1.0', 'os_family': 'linux', 'ua_family': 'odoo', 'timestamp': ts, } # First open event tracking.event_create('open', metadata) opens = tracking.tracking_event_ids.filtered( lambda r: r.event_type == 'open' ) self.assertEqual(len(opens), 1) # Concurrent open event metadata['timestamp'] = ts + 2 tracking.event_create('open', metadata) opens = tracking.tracking_event_ids.filtered( lambda r: r.event_type == 'open' ) self.assertEqual(len(opens), 1) # Second open event metadata['timestamp'] = ts + 350 tracking.event_create('open', metadata) opens = tracking.tracking_event_ids.filtered( lambda r: r.event_type == 'open' ) self.assertEqual(len(opens), 2) def test_concurrent_click(self): mail, tracking = self.mail_send(self.recipient.email) ts = time.time() metadata = { 'ip': '127.0.0.1', 'user_agent': 'Odoo Test/1.0', 'os_family': 'linux', 'ua_family': 'odoo', 'timestamp': ts, 'url': 'https://www.example.com/route/1', } # First click event (URL 1) tracking.event_create('click', metadata) opens = tracking.tracking_event_ids.filtered( lambda r: r.event_type == 'click' ) self.assertEqual(len(opens), 1) # Concurrent click event (URL 1) metadata['timestamp'] = ts + 2 tracking.event_create('click', metadata) opens = tracking.tracking_event_ids.filtered( lambda r: r.event_type == 'click' ) self.assertEqual(len(opens), 1) # Second click event (URL 1) metadata['timestamp'] = ts + 350 tracking.event_create('click', metadata) opens = tracking.tracking_event_ids.filtered( lambda r: r.event_type == 'click' ) self.assertEqual(len(opens), 2) # Concurrent click event (URL 2) metadata['timestamp'] = ts + 2 metadata['url'] = 'https://www.example.com/route/2' tracking.event_create('click', metadata) opens = tracking.tracking_event_ids.filtered( lambda r: r.event_type == 'click' ) self.assertEqual(len(opens), 3) def test_smtp_error(self): with mock.patch(mock_send_email) as mock_func: mock_func.side_effect = Warning('Test error') mail, tracking = self.mail_send(self.recipient.email) self.assertEqual('error', tracking.state) self.assertEqual('Warning', tracking.error_type) self.assertEqual('Test error', tracking.error_description) self.assertTrue(self.recipient.email_bounced) def test_partner_email_change(self): mail, tracking = self.mail_send(self.recipient.email) tracking.event_create('open', {}) orig_score = self.recipient.email_score orig_count = self.recipient.tracking_emails_count orig_email = self.recipient.email self.recipient.email = orig_email + '2' self.assertEqual(50.0, self.recipient.email_score) self.assertEqual(0, self.recipient.tracking_emails_count) self.recipient.email = orig_email self.assertEqual(orig_score, self.recipient.email_score) self.assertEqual(orig_count, self.recipient.tracking_emails_count) def test_process_hard_bounce(self): mail, tracking = self.mail_send(self.recipient.email) tracking.event_create('hard_bounce', {}) self.assertEqual('bounced', tracking.state) self.assertTrue(self.recipient.email_score < 50.0) def test_process_soft_bounce(self): mail, tracking = self.mail_send(self.recipient.email) tracking.event_create('soft_bounce', {}) self.assertEqual('soft-bounced', tracking.state) self.assertTrue(self.recipient.email_score < 50.0) def test_process_delivered(self): mail, tracking = self.mail_send(self.recipient.email) tracking.event_create('delivered', {}) self.assertEqual('delivered', tracking.state) self.assertTrue(self.recipient.email_score > 50.0) def test_process_deferral(self): mail, tracking = self.mail_send(self.recipient.email) tracking.event_create('deferral', {}) self.assertEqual('deferred', tracking.state) def test_process_spam(self): mail, tracking = self.mail_send(self.recipient.email) tracking.event_create('spam', {}) self.assertEqual('spam', tracking.state) self.assertTrue(self.recipient.email_score < 50.0) def test_process_unsub(self): mail, tracking = self.mail_send(self.recipient.email) tracking.event_create('unsub', {}) self.assertEqual('unsub', tracking.state) self.assertTrue(self.recipient.email_score < 50.0) def test_process_reject(self): mail, tracking = self.mail_send(self.recipient.email) tracking.event_create('reject', {}) self.assertEqual('rejected', tracking.state) self.assertTrue(self.recipient.email_score < 50.0) def test_process_open(self): mail, tracking = self.mail_send(self.recipient.email) tracking.event_create('open', {}) self.assertEqual('opened', tracking.state) self.assertTrue(self.recipient.email_score > 50.0) def test_process_click(self): mail, tracking = self.mail_send(self.recipient.email) tracking.event_create('click', {}) self.assertEqual('opened', tracking.state) self.assertTrue(self.recipient.email_score > 50.0) def test_process_several_bounce(self): for i in range(1, 10): mail, tracking = self.mail_send(self.recipient.email) tracking.event_create('hard_bounce', {}) self.assertEqual('bounced', tracking.state) self.assertEqual(0.0, self.recipient.email_score) def test_db(self): db = self.env.cr.dbname controller = MailTrackingController() not_found = controller.mail_tracking_all('not_found_db') self.assertEqual('NOT FOUND', not_found.response[0]) none = controller.mail_tracking_all(db) self.assertEqual('NONE', none.response[0]) none = controller.mail_tracking_event(db, 'open') self.assertEqual('NONE', none.response[0])