# -*- coding: utf-8 -*- # Part of Odoo, Flectra. See LICENSE file for full copyright and licensing details. from psycopg2 import IntegrityError from flectra.addons.mail.tests.common import TestMail from flectra.tools.misc import mute_logger class TestMailFollowers(TestMail): def setUp(self): super(TestMailFollowers, self).setUp() Subtype = self.env['mail.message.subtype'] self.mt_mg_def = Subtype.create({'name': 'mt_mg_def', 'default': True, 'res_model': 'mail.test'}) self.mt_cl_def = Subtype.create({'name': 'mt_cl_def', 'default': True, 'res_model': 'crm.lead'}) self.mt_al_def = Subtype.create({'name': 'mt_al_def', 'default': True, 'res_model': False}) self.mt_mg_nodef = Subtype.create({'name': 'mt_mg_nodef', 'default': False, 'res_model': 'mail.test'}) self.mt_al_nodef = Subtype.create({'name': 'mt_al_nodef', 'default': False, 'res_model': False}) self.default_group_subtypes = Subtype.search([('default', '=', True), '|', ('res_model', '=', 'mail.test'), ('res_model', '=', False)]) def test_m2o_command_new(self): test_channel = self.env['mail.channel'].create({'name': 'Test'}) groups = self.test_pigs | self.test_public generic, specific = self.env['mail.followers']._add_follower_command( 'mail.test', groups.ids, {self.user_employee.partner_id.id: [self.mt_mg_nodef.id]}, {test_channel.id: [self.mt_al_nodef.id]}) self.assertFalse(specific) self.assertEqual(len(generic), 2) items = [it[2] for it in generic] self.assertEqual({item['res_model'] for item in items}, {'mail.test'}) self.assertEqual({item['channel_id'] for item in items if item.get('channel_id')}, {test_channel.id}) self.assertEqual({item['partner_id'] for item in items if item.get('partner_id')}, {self.user_employee.partner_id.id}) self.assertEqual(set(items[0]['subtype_ids'][0][2] + items[1]['subtype_ids'][0][2]), {self.mt_mg_nodef.id, self.mt_al_nodef.id}) def test_m2o_command_update_selective(self): test_channel = self.env['mail.channel'].create({'name': 'Test'}) groups = self.test_pigs | self.test_public self.env['mail.followers'].create({'partner_id': self.user_employee.partner_id.id, 'res_model': 'mail.test', 'res_id': self.test_pigs.id}) generic, specific = self.env['mail.followers']._add_follower_command( 'mail.test', groups.ids, {self.user_employee.partner_id.id: [self.mt_mg_nodef.id]}, {test_channel.id: False}, force=False) self.assertEqual(len(generic), 1) self.assertEqual(len(specific), 1) self.assertEqual(generic[0][2]['res_model'], 'mail.test') self.assertEqual(generic[0][2]['channel_id'], test_channel.id) self.assertEqual(set(generic[0][2]['subtype_ids'][0][2]), set(self.default_group_subtypes.ids)) self.assertEqual(list(specific), [self.test_public.id]) self.assertEqual(specific[self.test_public.id][0][2]['res_model'], 'mail.test') self.assertEqual(specific[self.test_public.id][0][2]['partner_id'], self.user_employee.partner_id.id) self.assertEqual(set(specific[self.test_public.id][0][2]['subtype_ids'][0][2]), set([self.mt_mg_nodef.id])) def test_message_is_follower(self): qty_followed_before = len(self.test_pigs.sudo(self.user_employee).search([('message_is_follower', '=', True)])) self.assertFalse(self.test_pigs.sudo(self.user_employee).message_is_follower) self.test_pigs.message_subscribe_users(user_ids=[self.user_employee.id]) qty_followed_after = len(self.test_pigs.sudo(self.user_employee).search([('message_is_follower', '=', True)])) self.assertTrue(self.test_pigs.sudo(self.user_employee).message_is_follower) self.assertEqual(qty_followed_before + 1, qty_followed_after) def test_followers_subtypes_default(self): self.test_pigs.message_subscribe_users(user_ids=[self.user_employee.id]) self.assertEqual(self.test_pigs.message_follower_ids.mapped('partner_id'), self.user_employee.partner_id) self.assertEqual(self.test_pigs.message_follower_ids.mapped('channel_id'), self.env['mail.channel']) follower = self.env['mail.followers'].search([ ('res_model', '=', 'mail.test'), ('res_id', '=', self.test_pigs.id), ('partner_id', '=', self.user_employee.partner_id.id)]) self.assertEqual(len(follower), 1) self.assertEqual(follower.subtype_ids, self.default_group_subtypes) def test_followers_subtypes_default_internal(self): mt_mg_def_int = self.env['mail.message.subtype'].create({'name': 'mt_mg_def', 'default': True, 'res_model': 'mail.test', 'internal': True}) self.test_pigs.message_subscribe_users(user_ids=[self.user_employee.id]) follower = self.env['mail.followers'].search([ ('res_model', '=', 'mail.test'), ('res_id', '=', self.test_pigs.id), ('partner_id', '=', self.user_employee.partner_id.id)]) self.assertEqual(follower.subtype_ids, self.default_group_subtypes | mt_mg_def_int) self.test_pigs.message_subscribe_users(user_ids=[self.user_portal.id]) follower = self.env['mail.followers'].search([ ('res_model', '=', 'mail.test'), ('res_id', '=', self.test_pigs.id), ('partner_id', '=', self.user_portal.partner_id.id)]) self.assertEqual(follower.subtype_ids, self.default_group_subtypes.filtered(lambda subtype: not subtype.internal)) def test_followers_subtypes_specified(self): self.test_pigs.sudo(self.user_employee).message_subscribe_users(subtype_ids=[self.mt_mg_nodef.id]) self.assertEqual(self.test_pigs.message_follower_ids.mapped('partner_id'), self.user_employee.partner_id) self.assertEqual(self.test_pigs.message_follower_ids.mapped('channel_id'), self.env['mail.channel']) follower = self.env['mail.followers'].search([ ('res_model', '=', 'mail.test'), ('res_id', '=', self.test_pigs.id), ('partner_id', '=', self.user_employee.partner_id.id)]) self.assertEqual(len(follower), 1) self.assertEqual(follower.subtype_ids, self.mt_mg_nodef) def test_followers_multiple_subscription(self): self.test_pigs.sudo(self.user_employee).message_subscribe_users(subtype_ids=[self.mt_mg_nodef.id]) self.assertEqual(self.test_pigs.message_follower_ids.mapped('partner_id'), self.user_employee.partner_id) self.assertEqual(self.test_pigs.message_follower_ids.mapped('channel_id'), self.env['mail.channel']) follower = self.env['mail.followers'].search([ ('res_model', '=', 'mail.test'), ('res_id', '=', self.test_pigs.id), ('partner_id', '=', self.user_employee.partner_id.id)]) self.assertEqual(len(follower), 1) self.assertEqual(follower.subtype_ids, self.mt_mg_nodef) self.test_pigs.sudo(self.user_employee).message_subscribe_users(subtype_ids=[self.mt_mg_nodef.id, self.mt_al_nodef.id]) self.assertEqual(self.test_pigs.message_follower_ids.mapped('partner_id'), self.user_employee.partner_id) self.assertEqual(self.test_pigs.message_follower_ids.mapped('channel_id'), self.env['mail.channel']) follower = self.env['mail.followers'].search([ ('res_model', '=', 'mail.test'), ('res_id', '=', self.test_pigs.id), ('partner_id', '=', self.user_employee.partner_id.id)]) self.assertEqual(len(follower), 1) self.assertEqual(follower.subtype_ids, self.mt_mg_nodef | self.mt_al_nodef) def test_no_DID(self): """Test that a follower cannot suffer from dissociative identity disorder. It cannot be both a partner and a channel. """ test_record = self.env['mail.channel'].create({ 'name': 'I used to be schizo, but now we are alright.' }) test_channel = self.env['mail.channel'].create({'name': 'Follower Channel'}) with self.assertRaises(IntegrityError), mute_logger('flectra.sql_db'): self.env['mail.followers'].create({ 'res_model': test_record._name, 'res_id': test_record.id, 'partner_id': self.user_employee.partner_id.id, 'channel_id': test_channel.id, })