You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
57 lines
2.2 KiB
57 lines
2.2 KiB
# Copyright 2016 ACSONE SA/NV
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
|
|
|
|
from odoo.tests import common
|
|
from odoo.addons.bus.models.bus import json_dump
|
|
import json
|
|
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)
|
|
test_msg = {'message': 'message', 'title': 'title', 'sticky': True}
|
|
self.env.user.notify_info(**test_msg)
|
|
news = bus_bus.search(domain) - existing
|
|
self.assertEqual(1, len(news))
|
|
self.assertEqual(test_msg, 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)
|
|
test_msg = {'message': 'message', 'title': 'title', 'sticky': True}
|
|
self.env.user.notify_warning(**test_msg)
|
|
news = bus_bus.search(domain) - existing
|
|
self.assertEqual(1, len(news))
|
|
self.assertEqual(test_msg, 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))
|