# -*- coding: utf-8 -*- # Copyright 2016 ACSONE SA/NV # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl). import json from odoo.tests import common from odoo.addons.bus.models.bus import json_dump import mock class TestResUsers(common.TransactionCase): def test_notify_info(self): bus_bus = self.env['bus.bus'] domain = [ ('channel', '=', json_dump(self.env.user.notify_info_channel_name)) ] existing = bus_bus.search(domain) self.env.user.notify_info( message='message', title='title', sticky=True, show_reload=True, foo="bar" ) news = bus_bus.search(domain) - existing self.assertEqual(1, len(news)) expected = ('{"message":"message","sticky":true,"title":"title",' '"show_reload":true,"action":null,' '"action_link_name":null,"foo":"bar"}') self.assertDictEqual( json.loads(expected), json.loads(news.message) ) def test_notify_warning(self): bus_bus = self.env['bus.bus'] domain = [ ('channel', '=', json_dump(self.env.user.notify_warning_channel_name)) ] existing = bus_bus.search(domain) self.env.user.notify_warning( message='message', title='title', sticky=True, show_reload=True, foo="bar" ) news = bus_bus.search(domain) - existing self.assertEqual(1, len(news)) expected = ('{"message":"message","sticky":true,"title":"title",' '"show_reload":true,"action":null,' '"action_link_name":null,"foo":"bar"}') self.assertDictEqual( json.loads(expected), json.loads(news.message) ) def test_notify_with_action(self): bus_bus = self.env['bus.bus'] domain = [ ('channel', '=', json_dump(self.env.user.notify_info_channel_name)) ] existing = bus_bus.search(domain) self.env.user.notify_info( message='message', title='title', sticky=True, action={ "type": "ir.actions.act_window", "view_mode": "form", }, action_link_name="Open" ) news = bus_bus.search(domain) - existing self.assertEqual(1, len(news)) # the action should be transformed by Odoo (clean_action) expected = ('{"message":"message","sticky":true,"title":"title",' '"show_reload":false,"action":' '{"type": "ir.actions.act_window", "view_mode":"form",' '"flags":{},"views":[[false, "form"]]},' '"action_link_name":"Open"}') self.assertDictEqual( json.loads(expected), json.loads(news.message) ) def test_notify_many(self): # check that the notification of a list of users is done with # a single call to the bus with mock.patch('odoo.addons.bus.models.bus.ImBus.sendmany' ) as mockedSendMany: users = self.env.user.search([(1, "=", 1)]) self.assertTrue(len(users) > 1) users.notify_warning('message') self.assertEqual(1, mockedSendMany.call_count) # call_args is a tuple with args (tuple) & kwargs (dict). We check # positional arguments (args), hence the [0]. pos_call_args = mockedSendMany.call_args[0] # Ensure the first positional argument is a list with as many # elements as we had users. first_pos_call_args = pos_call_args[0] self.assertIsInstance(first_pos_call_args, list) self.assertEqual(len(users), len(first_pos_call_args))