paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-06 15:03:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

messages: Add unit test for get_raw_unread_data.

Browse Source 
Add test with 2 messages to the same group PM thread.
This commit is contained in:
Nikhil Kumar Mishra
2018-03-19 01:12:30 +05:30
committed by Tim Abbott
parent 3b4ff4f75c
commit 1579f8cb4b
1 changed files with 26 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
zerver/tests/test_messages.py
View File

@@ -34,6 +34,7 @@ from zerver.lib.message import (
get_first_visible_message_id,
update_first_visible_message_id,
maybe_update_first_visible_message_id,
get_raw_unread_data,
)
from zerver.lib.test_helpers import (
@@ -754,6 +755,31 @@ class StreamMessagesTest(ZulipTestCase):
self.assert_stream_message(non_ascii_stream_name, topic_name=u"hümbüǵ",
content=u"hümbüǵ")
def test_get_raw_unread_data_for_huddle_messages(self) -> None:
users = [
self.example_user('hamlet'),
self.example_user('cordelia'),
self.example_user('iago'),
self.example_user('prospero'),
self.example_user('othello'),
]
message1_id = self.send_huddle_message(users[0].email,
[user.email for user in users],
"test content 1")
message2_id = self.send_huddle_message(users[0].email,
[user.email for user in users],
"test content 2")
msg_data = get_raw_unread_data(users[1])
# both the messages are present in msg_data
self.assertIn(message1_id, msg_data["huddle_dict"].keys())
self.assertIn(message2_id, msg_data["huddle_dict"].keys())
# only these two messages are present in msg_data
self.assertEqual(len(msg_data["huddle_dict"].keys()), 2)
class MessageDictTest(ZulipTestCase):
@slow('builds lots of messages')
def test_bulk_message_fetching(self) -> None: