2014-01-31 16:44:45 +01:00
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from __future__ import absolute_import
|
2017-07-13 16:39:01 +02:00
|
|
|
from django.db.models import Q, Max
|
2015-07-30 10:42:04 +02:00
|
|
|
from django.conf import settings
|
2016-07-30 06:01:08 +02:00
|
|
|
from django.http import HttpResponse
|
2016-11-02 23:48:47 +01:00
|
|
|
from django.test import TestCase, override_settings
|
2017-04-15 04:03:56 +02:00
|
|
|
from django.utils.timezone import now as timezone_now
|
2014-01-31 16:44:45 +01:00
|
|
|
from zerver.lib import bugdown
|
|
|
|
from zerver.decorator import JsonableError
|
|
|
|
from zerver.lib.test_runner import slow
|
2017-05-30 09:53:42 +02:00
|
|
|
from zerver.lib.cache import get_stream_cache_key, cache_delete
|
2017-07-18 16:03:47 +02:00
|
|
|
from zerver.lib.str_utils import force_text
|
2014-01-31 16:44:45 +01:00
|
|
|
from zilencer.models import Deployment
|
|
|
|
|
2017-04-27 22:48:06 +02:00
|
|
|
from zerver.lib.addressee import Addressee
|
|
|
|
|
2017-08-18 12:26:43 +02:00
|
|
|
from zerver.lib.actions import (
|
2017-09-09 16:55:26 +02:00
|
|
|
do_send_messages,
|
2017-09-05 20:50:25 +02:00
|
|
|
get_userids_for_missed_messages,
|
2017-08-18 12:26:43 +02:00
|
|
|
internal_send_private_message,
|
|
|
|
)
|
|
|
|
|
2016-10-04 15:52:26 +02:00
|
|
|
from zerver.lib.message import (
|
|
|
|
MessageDict,
|
|
|
|
message_to_dict,
|
|
|
|
)
|
|
|
|
|
2014-01-31 16:44:45 +01:00
|
|
|
from zerver.lib.test_helpers import (
|
|
|
|
get_user_messages,
|
2016-09-13 23:32:35 +02:00
|
|
|
make_client,
|
2017-03-19 04:40:28 +01:00
|
|
|
message_stream_count,
|
2014-01-31 16:44:45 +01:00
|
|
|
most_recent_message,
|
2016-10-03 18:29:29 +02:00
|
|
|
most_recent_usermessage,
|
2014-01-31 16:44:45 +01:00
|
|
|
queries_captured,
|
|
|
|
)
|
|
|
|
|
2016-11-10 19:30:09 +01:00
|
|
|
from zerver.lib.test_classes import (
|
|
|
|
ZulipTestCase,
|
|
|
|
)
|
|
|
|
|
2017-07-16 09:41:38 +02:00
|
|
|
from zerver.lib.soft_deactivation import add_missing_messages, do_soft_deactivate_users, \
|
|
|
|
maybe_catch_up_soft_deactivated_user
|
2017-07-13 16:39:01 +02:00
|
|
|
|
2014-01-31 16:44:45 +01:00
|
|
|
from zerver.models import (
|
|
|
|
MAX_MESSAGE_LENGTH, MAX_SUBJECT_LENGTH,
|
2017-07-13 16:39:01 +02:00
|
|
|
Message, Realm, Recipient, Stream, UserMessage, UserProfile, Attachment,
|
2017-09-09 16:55:26 +02:00
|
|
|
RealmAuditLog, RealmDomain, get_realm, UserPresence, Subscription,
|
2017-07-13 16:39:01 +02:00
|
|
|
get_stream, get_recipient, get_system_bot, get_user, Reaction,
|
|
|
|
sew_messages_and_reactions, flush_per_request_caches
|
2014-01-31 16:44:45 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
from zerver.lib.actions import (
|
2017-03-08 12:48:35 +01:00
|
|
|
check_message,
|
|
|
|
check_send_message,
|
2017-07-29 00:52:27 +02:00
|
|
|
do_deactivate_user,
|
2017-07-16 11:00:44 +02:00
|
|
|
do_set_realm_property,
|
2017-03-08 12:48:35 +01:00
|
|
|
extract_recipients,
|
2016-10-20 00:14:44 +02:00
|
|
|
do_create_user,
|
2016-07-27 03:31:17 +02:00
|
|
|
get_client,
|
2017-07-18 16:03:47 +02:00
|
|
|
do_add_alert_words,
|
2014-01-31 16:44:45 +01:00
|
|
|
)
|
|
|
|
|
2016-03-24 20:24:01 +01:00
|
|
|
from zerver.lib.upload import create_attachment
|
|
|
|
|
2016-07-27 03:31:17 +02:00
|
|
|
from zerver.views.messages import create_mirrored_message_users
|
|
|
|
|
2014-01-31 16:44:45 +01:00
|
|
|
import datetime
|
2016-06-01 05:30:10 +02:00
|
|
|
import DNS
|
|
|
|
import mock
|
2014-01-31 16:44:45 +01:00
|
|
|
import time
|
|
|
|
import ujson
|
2015-11-01 17:15:05 +01:00
|
|
|
from six.moves import range
|
2017-09-05 20:50:25 +02:00
|
|
|
from typing import Any, Dict, List, Optional, Text
|
2014-01-31 16:44:45 +01:00
|
|
|
|
2017-08-05 22:10:34 +02:00
|
|
|
from collections import namedtuple
|
|
|
|
|
2016-10-27 15:54:49 +02:00
|
|
|
class TopicHistoryTest(ZulipTestCase):
|
|
|
|
def test_topics_history(self):
|
|
|
|
# type: () -> None
|
|
|
|
# verified: int(UserMessage.flags.read) == 1
|
2017-05-07 19:39:30 +02:00
|
|
|
user_profile = self.example_user('iago')
|
|
|
|
email = user_profile.email
|
2016-10-27 15:54:49 +02:00
|
|
|
stream_name = 'Verona'
|
|
|
|
self.login(email)
|
|
|
|
|
2017-01-13 15:50:17 +01:00
|
|
|
stream = get_stream(stream_name, user_profile.realm)
|
2016-10-27 15:54:49 +02:00
|
|
|
recipient = get_recipient(Recipient.STREAM, stream.id)
|
|
|
|
|
2017-07-20 21:40:11 +02:00
|
|
|
def create_test_message(topic):
|
|
|
|
# type: (str) -> int
|
2016-10-27 15:54:49 +02:00
|
|
|
|
2017-05-07 17:21:26 +02:00
|
|
|
hamlet = self.example_user('hamlet')
|
2016-10-27 15:54:49 +02:00
|
|
|
message = Message.objects.create(
|
|
|
|
sender=hamlet,
|
|
|
|
recipient=recipient,
|
|
|
|
subject=topic,
|
|
|
|
content='whatever',
|
2017-04-15 04:03:56 +02:00
|
|
|
pub_date=timezone_now(),
|
2016-10-27 15:54:49 +02:00
|
|
|
sending_client=get_client('whatever'),
|
|
|
|
)
|
|
|
|
|
|
|
|
UserMessage.objects.create(
|
|
|
|
user_profile=user_profile,
|
|
|
|
message=message,
|
2017-07-20 21:40:11 +02:00
|
|
|
flags=0,
|
2016-10-27 15:54:49 +02:00
|
|
|
)
|
|
|
|
|
2017-07-20 21:40:11 +02:00
|
|
|
return message.id
|
|
|
|
|
|
|
|
# our most recent topics are topic0, topic1, topic2
|
|
|
|
|
|
|
|
# Create old messages with strange spellings.
|
|
|
|
create_test_message('topic2')
|
|
|
|
create_test_message('toPIc1')
|
|
|
|
create_test_message('toPIc0')
|
|
|
|
create_test_message('topic2')
|
|
|
|
create_test_message('topic2')
|
|
|
|
create_test_message('Topic2')
|
|
|
|
|
|
|
|
# Create new messages
|
|
|
|
topic2_msg_id = create_test_message('topic2')
|
|
|
|
create_test_message('topic1')
|
|
|
|
create_test_message('topic1')
|
|
|
|
topic1_msg_id = create_test_message('topic1')
|
|
|
|
topic0_msg_id = create_test_message('topic0')
|
2016-10-27 15:54:49 +02:00
|
|
|
|
|
|
|
endpoint = '/json/users/me/%d/topics' % (stream.id,)
|
|
|
|
result = self.client_get(endpoint, dict())
|
|
|
|
self.assert_json_success(result)
|
2017-08-17 08:40:54 +02:00
|
|
|
history = result.json()['topics']
|
2016-10-27 15:54:49 +02:00
|
|
|
|
|
|
|
# We only look at the most recent three topics, because
|
|
|
|
# the prior fixture data may be unreliable.
|
2017-07-20 21:40:11 +02:00
|
|
|
history = history[:3]
|
|
|
|
|
|
|
|
self.assertEqual([topic['name'] for topic in history], [
|
|
|
|
'topic0',
|
|
|
|
'topic1',
|
|
|
|
'topic2',
|
|
|
|
])
|
|
|
|
|
|
|
|
self.assertEqual([topic['max_id'] for topic in history], [
|
|
|
|
topic0_msg_id,
|
|
|
|
topic1_msg_id,
|
|
|
|
topic2_msg_id,
|
2016-10-27 15:54:49 +02:00
|
|
|
])
|
|
|
|
|
|
|
|
def test_bad_stream_id(self):
|
|
|
|
# type: () -> None
|
2017-05-25 01:44:04 +02:00
|
|
|
email = self.example_email("iago")
|
2016-10-27 15:54:49 +02:00
|
|
|
self.login(email)
|
|
|
|
|
|
|
|
# non-sensible stream id
|
|
|
|
endpoint = '/json/users/me/9999999999/topics'
|
|
|
|
result = self.client_get(endpoint, dict())
|
|
|
|
self.assert_json_error(result, 'Invalid stream id')
|
|
|
|
|
|
|
|
# out of realm
|
|
|
|
bad_stream = self.make_stream(
|
|
|
|
'mit_stream',
|
2017-03-04 09:19:37 +01:00
|
|
|
realm=get_realm('zephyr')
|
2016-10-27 15:54:49 +02:00
|
|
|
)
|
|
|
|
endpoint = '/json/users/me/%s/topics' % (bad_stream.id,)
|
|
|
|
result = self.client_get(endpoint, dict())
|
|
|
|
self.assert_json_error(result, 'Invalid stream id')
|
|
|
|
|
|
|
|
# private stream to which I am not subscribed
|
|
|
|
private_stream = self.make_stream(
|
|
|
|
'private_stream',
|
|
|
|
invite_only=True
|
|
|
|
)
|
|
|
|
endpoint = '/json/users/me/%s/topics' % (private_stream.id,)
|
|
|
|
result = self.client_get(endpoint, dict())
|
|
|
|
self.assert_json_error(result, 'Invalid stream id')
|
|
|
|
|
|
|
|
|
2016-08-23 02:08:42 +02:00
|
|
|
class TestCrossRealmPMs(ZulipTestCase):
|
2016-09-28 00:03:13 +02:00
|
|
|
def make_realm(self, domain):
|
2016-12-04 18:38:56 +01:00
|
|
|
# type: (Text) -> Realm
|
2017-03-14 00:51:51 +01:00
|
|
|
realm = Realm.objects.create(string_id=domain, invite_required=False)
|
2017-03-31 16:20:07 +02:00
|
|
|
RealmDomain.objects.create(realm=realm, domain=domain)
|
2016-09-28 00:03:13 +02:00
|
|
|
return realm
|
|
|
|
|
2015-01-31 07:55:18 +01:00
|
|
|
def setUp(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2015-08-21 18:37:30 +02:00
|
|
|
dep = Deployment()
|
|
|
|
dep.base_api_url = "https://zulip.com/api/"
|
|
|
|
dep.base_site_url = "https://zulip.com/"
|
|
|
|
# We need to save the object before we can access
|
|
|
|
# the many-to-many relationship 'realms'
|
|
|
|
dep.save()
|
2017-01-04 05:30:48 +01:00
|
|
|
dep.realms = [get_realm("zulip")]
|
2015-08-21 18:37:30 +02:00
|
|
|
dep.save()
|
2015-01-31 07:55:18 +01:00
|
|
|
|
2014-01-31 16:44:45 +01:00
|
|
|
def create_user(self, email):
|
2016-12-04 18:38:56 +01:00
|
|
|
# type: (Text) -> UserProfile
|
2017-08-26 01:08:14 +02:00
|
|
|
subdomain = email.split("@")[1]
|
|
|
|
self.register(email, 'test', subdomain=subdomain)
|
|
|
|
return get_user(email, get_realm(subdomain))
|
2014-01-31 16:44:45 +01:00
|
|
|
|
2016-11-02 23:48:47 +01:00
|
|
|
@override_settings(CROSS_REALM_BOT_EMAILS=['feedback@zulip.com',
|
|
|
|
'support@3.example.com'])
|
2016-11-02 16:23:26 +01:00
|
|
|
def test_realm_scenarios(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2016-09-28 00:03:13 +02:00
|
|
|
r1 = self.make_realm('1.example.com')
|
2016-11-02 16:23:26 +01:00
|
|
|
r2 = self.make_realm('2.example.com')
|
|
|
|
r3 = self.make_realm('3.example.com')
|
2014-01-31 16:44:45 +01:00
|
|
|
deployment = Deployment.objects.filter()[0]
|
|
|
|
deployment.realms.add(r1)
|
2016-11-02 16:23:26 +01:00
|
|
|
deployment.realms.add(r2)
|
|
|
|
deployment.realms.add(r3)
|
|
|
|
|
2016-11-02 16:34:32 +01:00
|
|
|
def assert_message_received(to_user, from_user):
|
|
|
|
# type: (UserProfile, UserProfile) -> None
|
|
|
|
messages = get_user_messages(to_user)
|
2016-12-16 02:01:34 +01:00
|
|
|
self.assertEqual(messages[-1].sender.pk, from_user.pk)
|
2016-11-02 16:34:32 +01:00
|
|
|
|
2017-08-18 12:26:43 +02:00
|
|
|
def assert_invalid_email():
|
2016-11-02 16:23:26 +01:00
|
|
|
# type: () -> Any
|
2016-12-16 02:11:42 +01:00
|
|
|
return self.assertRaisesRegex(
|
2016-11-02 16:23:26 +01:00
|
|
|
JsonableError,
|
2017-08-18 12:26:43 +02:00
|
|
|
'Invalid email ')
|
2014-01-31 16:44:45 +01:00
|
|
|
|
|
|
|
user1_email = 'user1@1.example.com'
|
2016-11-02 16:23:26 +01:00
|
|
|
user1a_email = 'user1a@1.example.com'
|
|
|
|
user2_email = 'user2@2.example.com'
|
|
|
|
user3_email = 'user3@3.example.com'
|
2016-11-02 16:46:46 +01:00
|
|
|
feedback_email = 'feedback@zulip.com'
|
2017-07-11 21:33:49 +02:00
|
|
|
support_email = 'support@3.example.com' # note: not zulip.com
|
2016-11-02 16:46:46 +01:00
|
|
|
|
2014-01-31 16:44:45 +01:00
|
|
|
user1 = self.create_user(user1_email)
|
2016-11-02 16:23:26 +01:00
|
|
|
user1a = self.create_user(user1a_email)
|
2014-01-31 16:44:45 +01:00
|
|
|
user2 = self.create_user(user2_email)
|
2016-11-02 00:51:11 +01:00
|
|
|
self.create_user(user3_email)
|
2017-05-24 02:42:31 +02:00
|
|
|
feedback_bot = get_system_bot(feedback_email)
|
2016-11-02 16:54:37 +01:00
|
|
|
support_bot = self.create_user(support_email)
|
2014-01-31 16:44:45 +01:00
|
|
|
|
2016-11-02 16:38:19 +01:00
|
|
|
# Users can PM themselves
|
2016-11-02 16:36:13 +01:00
|
|
|
self.send_message(user1_email, user1_email, Recipient.PERSONAL)
|
|
|
|
assert_message_received(user1, user1)
|
|
|
|
|
2016-11-02 16:38:19 +01:00
|
|
|
# Users on the same realm can PM each other
|
2016-11-02 16:23:26 +01:00
|
|
|
self.send_message(user1_email, user1a_email, Recipient.PERSONAL)
|
2016-11-02 16:34:32 +01:00
|
|
|
assert_message_received(user1a, user1)
|
2014-01-31 16:44:45 +01:00
|
|
|
|
2016-11-02 16:38:19 +01:00
|
|
|
# Cross-realm bots in the zulip.com realm can PM any realm
|
2017-08-18 12:26:43 +02:00
|
|
|
# (They need lower level APIs to do this.)
|
|
|
|
internal_send_private_message(
|
|
|
|
realm=r2,
|
|
|
|
sender=get_system_bot(feedback_email),
|
|
|
|
recipient_user=get_user(user2_email, r2),
|
|
|
|
content='bla',
|
|
|
|
)
|
2016-11-02 16:46:46 +01:00
|
|
|
assert_message_received(user2, feedback_bot)
|
2014-01-31 16:44:45 +01:00
|
|
|
|
2016-11-02 16:38:19 +01:00
|
|
|
# All users can PM cross-realm bots in the zulip.com realm
|
2016-11-02 16:46:46 +01:00
|
|
|
self.send_message(user1_email, feedback_email, Recipient.PERSONAL)
|
|
|
|
assert_message_received(feedback_bot, user1)
|
2014-01-31 16:44:45 +01:00
|
|
|
|
2016-11-02 16:54:37 +01:00
|
|
|
# Users can PM cross-realm bots on non-zulip realms.
|
|
|
|
# (The support bot represents some theoretical bot that we may
|
|
|
|
# create in the future that does not have zulip.com as its realm.)
|
|
|
|
self.send_message(user1_email, [support_email], Recipient.PERSONAL)
|
|
|
|
assert_message_received(support_bot, user1)
|
|
|
|
|
2016-11-02 00:51:11 +01:00
|
|
|
# Allow sending PMs to two different cross-realm bots simultaneously.
|
|
|
|
# (We don't particularly need this feature, but since users can
|
|
|
|
# already individually send PMs to cross-realm bots, we shouldn't
|
|
|
|
# prevent them from sending multiple bots at once. We may revisit
|
|
|
|
# this if it's a nuisance for huddles.)
|
|
|
|
self.send_message(user1_email, [feedback_email, support_email],
|
|
|
|
Recipient.PERSONAL)
|
|
|
|
assert_message_received(feedback_bot, user1)
|
2016-11-02 17:28:33 +01:00
|
|
|
assert_message_received(support_bot, user1)
|
|
|
|
|
2016-11-02 00:51:11 +01:00
|
|
|
# Prevent old loophole where I could send PMs to other users as long
|
|
|
|
# as I copied a cross-realm bot from the same realm.
|
2017-08-18 12:26:43 +02:00
|
|
|
with assert_invalid_email():
|
2016-11-02 00:51:11 +01:00
|
|
|
self.send_message(user1_email, [user3_email, support_email], Recipient.PERSONAL)
|
2016-11-02 17:08:55 +01:00
|
|
|
|
2016-11-02 16:23:26 +01:00
|
|
|
# Users on three different realms can't PM each other,
|
|
|
|
# even if one of the users is a cross-realm bot.
|
2017-08-18 12:26:43 +02:00
|
|
|
with assert_invalid_email():
|
2016-11-02 16:46:46 +01:00
|
|
|
self.send_message(user1_email, [user2_email, feedback_email],
|
2014-01-31 16:44:45 +01:00
|
|
|
Recipient.PERSONAL)
|
|
|
|
|
2017-08-18 12:26:43 +02:00
|
|
|
with assert_invalid_email():
|
2016-11-05 18:16:56 +01:00
|
|
|
self.send_message(feedback_email, [user1_email, user2_email],
|
|
|
|
Recipient.PERSONAL)
|
|
|
|
|
2016-11-02 16:29:41 +01:00
|
|
|
# Users on the different realms can not PM each other
|
2017-08-18 12:26:43 +02:00
|
|
|
with assert_invalid_email():
|
2016-11-02 16:29:41 +01:00
|
|
|
self.send_message(user1_email, user2_email, Recipient.PERSONAL)
|
|
|
|
|
2016-11-02 17:14:33 +01:00
|
|
|
# Users on non-zulip realms can't PM "ordinary" Zulip users
|
2017-08-18 12:26:43 +02:00
|
|
|
with assert_invalid_email():
|
2017-08-25 05:20:26 +02:00
|
|
|
self.send_message(user1_email, "hamlet@zulip.com", Recipient.PERSONAL)
|
2016-11-02 17:14:33 +01:00
|
|
|
|
2016-11-02 16:29:41 +01:00
|
|
|
# Users on three different realms can not PM each other
|
2017-08-18 12:26:43 +02:00
|
|
|
with assert_invalid_email():
|
2016-11-02 16:29:41 +01:00
|
|
|
self.send_message(user1_email, [user2_email, user3_email], Recipient.PERSONAL)
|
|
|
|
|
2017-03-08 12:48:35 +01:00
|
|
|
class ExtractedRecipientsTest(TestCase):
|
|
|
|
def test_extract_recipients(self):
|
|
|
|
# type: () -> None
|
|
|
|
|
|
|
|
# JSON list w/dups, empties, and trailing whitespace
|
|
|
|
s = ujson.dumps([' alice@zulip.com ', ' bob@zulip.com ', ' ', 'bob@zulip.com'])
|
|
|
|
self.assertEqual(sorted(extract_recipients(s)), ['alice@zulip.com', 'bob@zulip.com'])
|
|
|
|
|
|
|
|
# simple string with one name
|
|
|
|
s = 'alice@zulip.com '
|
|
|
|
self.assertEqual(extract_recipients(s), ['alice@zulip.com'])
|
|
|
|
|
|
|
|
# JSON-encoded string
|
|
|
|
s = '"alice@zulip.com"'
|
|
|
|
self.assertEqual(extract_recipients(s), ['alice@zulip.com'])
|
|
|
|
|
|
|
|
# bare comma-delimited string
|
|
|
|
s = 'bob@zulip.com, alice@zulip.com'
|
|
|
|
self.assertEqual(sorted(extract_recipients(s)), ['alice@zulip.com', 'bob@zulip.com'])
|
|
|
|
|
|
|
|
# JSON-encoded, comma-delimited string
|
|
|
|
s = '"bob@zulip.com,alice@zulip.com"'
|
|
|
|
self.assertEqual(sorted(extract_recipients(s)), ['alice@zulip.com', 'bob@zulip.com'])
|
|
|
|
|
2016-08-23 02:08:42 +02:00
|
|
|
class PersonalMessagesTest(ZulipTestCase):
|
2014-01-31 16:44:45 +01:00
|
|
|
|
|
|
|
def test_auto_subbed_to_personals(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
Newly created users are auto-subbed to the ability to receive
|
|
|
|
personals.
|
|
|
|
"""
|
2017-05-24 02:42:31 +02:00
|
|
|
test_email = self.nonreg_email('test')
|
|
|
|
self.register(test_email, "test")
|
|
|
|
user_profile = self.nonreg_user('test')
|
2014-01-31 16:44:45 +01:00
|
|
|
old_messages_count = message_stream_count(user_profile)
|
2017-05-24 02:42:31 +02:00
|
|
|
self.send_message(test_email, test_email, Recipient.PERSONAL)
|
2014-01-31 16:44:45 +01:00
|
|
|
new_messages_count = message_stream_count(user_profile)
|
|
|
|
self.assertEqual(new_messages_count, old_messages_count + 1)
|
|
|
|
|
|
|
|
recipient = Recipient.objects.get(type_id=user_profile.id,
|
|
|
|
type=Recipient.PERSONAL)
|
2016-09-24 00:01:42 +02:00
|
|
|
message = most_recent_message(user_profile)
|
|
|
|
self.assertEqual(message.recipient, recipient)
|
|
|
|
|
|
|
|
with mock.patch('zerver.models.get_display_recipient', return_value='recip'):
|
|
|
|
self.assertEqual(str(message),
|
2016-12-03 00:04:17 +01:00
|
|
|
u'<Message: recip / / '
|
2017-03-13 17:23:07 +01:00
|
|
|
'<UserProfile: test@zulip.com <Realm: zulip 1>>>')
|
2014-01-31 16:44:45 +01:00
|
|
|
|
2016-10-03 18:29:29 +02:00
|
|
|
user_message = most_recent_usermessage(user_profile)
|
|
|
|
self.assertEqual(str(user_message),
|
2016-12-03 00:04:17 +01:00
|
|
|
u'<UserMessage: recip / test@zulip.com ([])>'
|
2016-12-01 06:16:45 +01:00
|
|
|
)
|
2016-10-03 18:29:29 +02:00
|
|
|
|
2016-07-29 21:06:22 +02:00
|
|
|
@slow("checks several profiles")
|
2014-01-31 16:44:45 +01:00
|
|
|
def test_personal_to_self(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
If you send a personal to yourself, only you see it.
|
|
|
|
"""
|
|
|
|
old_user_profiles = list(UserProfile.objects.all())
|
2017-05-24 02:42:31 +02:00
|
|
|
test_email = self.nonreg_email('test1')
|
|
|
|
self.register(test_email, "test1")
|
2014-01-31 16:44:45 +01:00
|
|
|
|
|
|
|
old_messages = []
|
|
|
|
for user_profile in old_user_profiles:
|
|
|
|
old_messages.append(message_stream_count(user_profile))
|
|
|
|
|
2017-05-24 02:42:31 +02:00
|
|
|
self.send_message(test_email, test_email, Recipient.PERSONAL)
|
2014-01-31 16:44:45 +01:00
|
|
|
|
|
|
|
new_messages = []
|
|
|
|
for user_profile in old_user_profiles:
|
|
|
|
new_messages.append(message_stream_count(user_profile))
|
|
|
|
|
|
|
|
self.assertEqual(old_messages, new_messages)
|
|
|
|
|
2017-05-24 02:42:31 +02:00
|
|
|
user_profile = self.nonreg_user('test1')
|
2014-01-31 16:44:45 +01:00
|
|
|
recipient = Recipient.objects.get(type_id=user_profile.id, type=Recipient.PERSONAL)
|
|
|
|
self.assertEqual(most_recent_message(user_profile).recipient, recipient)
|
|
|
|
|
|
|
|
def assert_personal(self, sender_email, receiver_email, content="test content"):
|
2016-12-04 18:38:56 +01:00
|
|
|
# type: (Text, Text, Text) -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
Send a private message from `sender_email` to `receiver_email` and check
|
|
|
|
that only those two parties actually received the message.
|
|
|
|
"""
|
2017-05-24 02:42:31 +02:00
|
|
|
realm = get_realm('zulip') # Assume realm is always 'zulip'
|
|
|
|
sender = get_user(sender_email, realm)
|
|
|
|
receiver = get_user(receiver_email, realm)
|
2014-01-31 16:44:45 +01:00
|
|
|
|
|
|
|
sender_messages = message_stream_count(sender)
|
|
|
|
receiver_messages = message_stream_count(receiver)
|
|
|
|
|
|
|
|
other_user_profiles = UserProfile.objects.filter(~Q(email=sender_email) &
|
|
|
|
~Q(email=receiver_email))
|
|
|
|
old_other_messages = []
|
|
|
|
for user_profile in other_user_profiles:
|
|
|
|
old_other_messages.append(message_stream_count(user_profile))
|
|
|
|
|
|
|
|
self.send_message(sender_email, receiver_email, Recipient.PERSONAL, content)
|
|
|
|
|
|
|
|
# Users outside the conversation don't get the message.
|
|
|
|
new_other_messages = []
|
|
|
|
for user_profile in other_user_profiles:
|
|
|
|
new_other_messages.append(message_stream_count(user_profile))
|
|
|
|
|
|
|
|
self.assertEqual(old_other_messages, new_other_messages)
|
|
|
|
|
|
|
|
# The personal message is in the streams of both the sender and receiver.
|
|
|
|
self.assertEqual(message_stream_count(sender),
|
|
|
|
sender_messages + 1)
|
|
|
|
self.assertEqual(message_stream_count(receiver),
|
|
|
|
receiver_messages + 1)
|
|
|
|
|
|
|
|
recipient = Recipient.objects.get(type_id=receiver.id, type=Recipient.PERSONAL)
|
|
|
|
self.assertEqual(most_recent_message(sender).recipient, recipient)
|
|
|
|
self.assertEqual(most_recent_message(receiver).recipient, recipient)
|
|
|
|
|
2016-07-29 21:06:22 +02:00
|
|
|
@slow("assert_personal checks several profiles")
|
2014-01-31 16:44:45 +01:00
|
|
|
def test_personal(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
If you send a personal, only you and the recipient see it.
|
|
|
|
"""
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2017-05-25 02:08:35 +02:00
|
|
|
self.assert_personal(self.example_email("hamlet"), self.example_email("othello"))
|
2014-01-31 16:44:45 +01:00
|
|
|
|
2016-07-29 21:06:22 +02:00
|
|
|
@slow("assert_personal checks several profiles")
|
2014-01-31 16:44:45 +01:00
|
|
|
def test_non_ascii_personal(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
Sending a PM containing non-ASCII characters succeeds.
|
|
|
|
"""
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2017-05-25 02:08:35 +02:00
|
|
|
self.assert_personal(self.example_email("hamlet"), self.example_email("othello"), u"hümbüǵ")
|
2014-01-31 16:44:45 +01:00
|
|
|
|
2016-08-23 02:08:42 +02:00
|
|
|
class StreamMessagesTest(ZulipTestCase):
|
2014-01-31 16:44:45 +01:00
|
|
|
|
|
|
|
def assert_stream_message(self, stream_name, subject="test subject",
|
|
|
|
content="test content"):
|
2016-12-04 18:38:56 +01:00
|
|
|
# type: (Text, Text, Text) -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
Check that messages sent to a stream reach all subscribers to that stream.
|
|
|
|
"""
|
2017-01-04 05:30:48 +01:00
|
|
|
realm = get_realm('zulip')
|
2016-12-04 01:04:55 +01:00
|
|
|
subscribers = self.users_subscribed_to_stream(stream_name, realm)
|
2017-05-22 18:17:30 +02:00
|
|
|
|
|
|
|
# Outgoing webhook bots don't store UserMessage rows; they will be processed later.
|
|
|
|
subscribers = [subscriber for subscriber in subscribers
|
|
|
|
if subscriber.bot_type != UserProfile.OUTGOING_WEBHOOK_BOT]
|
|
|
|
|
2014-01-31 16:44:45 +01:00
|
|
|
old_subscriber_messages = []
|
|
|
|
for subscriber in subscribers:
|
|
|
|
old_subscriber_messages.append(message_stream_count(subscriber))
|
|
|
|
|
|
|
|
non_subscribers = [user_profile for user_profile in UserProfile.objects.all()
|
|
|
|
if user_profile not in subscribers]
|
|
|
|
old_non_subscriber_messages = []
|
|
|
|
for non_subscriber in non_subscribers:
|
|
|
|
old_non_subscriber_messages.append(message_stream_count(non_subscriber))
|
|
|
|
|
2016-06-27 20:45:23 +02:00
|
|
|
non_bot_subscribers = [user_profile for user_profile in subscribers
|
|
|
|
if not user_profile.is_bot]
|
|
|
|
a_subscriber_email = non_bot_subscribers[0].email
|
2014-01-31 16:44:45 +01:00
|
|
|
self.login(a_subscriber_email)
|
|
|
|
self.send_message(a_subscriber_email, stream_name, Recipient.STREAM,
|
|
|
|
subject, content)
|
|
|
|
|
|
|
|
# Did all of the subscribers get the message?
|
|
|
|
new_subscriber_messages = []
|
|
|
|
for subscriber in subscribers:
|
2016-12-01 06:20:27 +01:00
|
|
|
new_subscriber_messages.append(message_stream_count(subscriber))
|
2014-01-31 16:44:45 +01:00
|
|
|
|
|
|
|
# Did non-subscribers not get the message?
|
|
|
|
new_non_subscriber_messages = []
|
|
|
|
for non_subscriber in non_subscribers:
|
|
|
|
new_non_subscriber_messages.append(message_stream_count(non_subscriber))
|
|
|
|
|
|
|
|
self.assertEqual(old_non_subscriber_messages, new_non_subscriber_messages)
|
|
|
|
self.assertEqual(new_subscriber_messages, [elt + 1 for elt in old_subscriber_messages])
|
|
|
|
|
2017-09-09 16:55:26 +02:00
|
|
|
def test_performance(self):
|
|
|
|
# type: () -> None
|
|
|
|
'''
|
|
|
|
This test is part of the automated test suite, but
|
|
|
|
it is more intended as an aid to measuring the
|
|
|
|
performance of do_send_messages() with consistent
|
|
|
|
data setup across different commits. You can modify
|
|
|
|
the values below and run just this test, and then
|
|
|
|
comment out the print statement toward the bottom.
|
|
|
|
'''
|
|
|
|
num_messages = 2
|
|
|
|
num_extra_users = 10
|
|
|
|
|
|
|
|
sender = self.example_user('cordelia')
|
|
|
|
realm = sender.realm
|
|
|
|
message_content = 'whatever'
|
|
|
|
stream = get_stream('Denmark', realm)
|
|
|
|
subject = 'lunch'
|
|
|
|
recipient = get_recipient(Recipient.STREAM, stream.id)
|
|
|
|
sending_client = make_client(name="test suite")
|
|
|
|
|
|
|
|
for i in range(num_extra_users):
|
|
|
|
# Make every other user be idle.
|
|
|
|
long_term_idle = i % 2 > 0
|
|
|
|
|
|
|
|
email = 'foo%d@example.com' % (i,)
|
|
|
|
user = UserProfile.objects.create(
|
|
|
|
realm=realm,
|
|
|
|
email=email,
|
|
|
|
pointer=0,
|
|
|
|
long_term_idle=long_term_idle,
|
|
|
|
)
|
|
|
|
Subscription.objects.create(
|
|
|
|
user_profile=user,
|
|
|
|
recipient=recipient
|
|
|
|
)
|
|
|
|
|
|
|
|
def send_test_message():
|
|
|
|
# type: () -> None
|
|
|
|
message = Message(
|
|
|
|
sender=sender,
|
|
|
|
recipient=recipient,
|
|
|
|
subject=subject,
|
|
|
|
content=message_content,
|
|
|
|
pub_date=timezone_now(),
|
|
|
|
sending_client=sending_client,
|
|
|
|
)
|
|
|
|
do_send_messages([dict(message=message)])
|
|
|
|
|
|
|
|
before_um_count = UserMessage.objects.count()
|
|
|
|
|
|
|
|
t = time.time()
|
|
|
|
for i in range(num_messages):
|
|
|
|
send_test_message()
|
|
|
|
|
|
|
|
delay = time.time() - t
|
|
|
|
assert(delay) # quiet down lint
|
|
|
|
# print(delay)
|
|
|
|
|
|
|
|
after_um_count = UserMessage.objects.count()
|
|
|
|
ums_created = after_um_count - before_um_count
|
|
|
|
|
|
|
|
num_active_users = num_extra_users / 2
|
|
|
|
self.assertTrue(ums_created > (num_active_users * num_messages))
|
|
|
|
|
2014-01-31 16:44:45 +01:00
|
|
|
def test_not_too_many_queries(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2017-08-25 06:01:29 +02:00
|
|
|
recipient_list = [self.example_user("hamlet"), self.example_user("iago"),
|
|
|
|
self.example_user("cordelia"), self.example_user("othello")]
|
|
|
|
for user_profile in recipient_list:
|
|
|
|
self.subscribe(user_profile, "Denmark")
|
2014-01-31 16:44:45 +01:00
|
|
|
|
2017-05-07 21:25:59 +02:00
|
|
|
sender = self.example_user('hamlet')
|
2014-01-31 16:44:45 +01:00
|
|
|
message_type_name = "stream"
|
2016-09-13 23:32:35 +02:00
|
|
|
sending_client = make_client(name="test suite")
|
2014-01-31 16:44:45 +01:00
|
|
|
stream = 'Denmark'
|
|
|
|
subject = 'foo'
|
|
|
|
content = 'whatever'
|
|
|
|
realm = sender.realm
|
|
|
|
|
|
|
|
def send_message():
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
check_send_message(sender, sending_client, message_type_name, [stream],
|
|
|
|
subject, content, forwarder_user_profile=sender, realm=realm)
|
|
|
|
|
2017-05-30 09:53:42 +02:00
|
|
|
# To get accurate count of the queries, we should make sure that
|
|
|
|
# caches don't come into play. If we count queries while caches are
|
|
|
|
# filled, we will get a lower count. Caches are not supposed to be
|
|
|
|
# persistent, so our test can also fail if cache is invalidated
|
|
|
|
# during the course of the unit test.
|
2017-03-31 08:28:28 +02:00
|
|
|
flush_per_request_caches()
|
2017-05-30 09:53:42 +02:00
|
|
|
cache_delete(get_stream_cache_key(stream, realm))
|
2014-01-31 16:44:45 +01:00
|
|
|
with queries_captured() as queries:
|
|
|
|
send_message()
|
|
|
|
|
2017-09-15 03:08:15 +02:00
|
|
|
self.assert_length(queries, 12)
|
2014-01-31 16:44:45 +01:00
|
|
|
|
2017-01-05 16:50:34 +01:00
|
|
|
def test_stream_message_dict(self):
|
|
|
|
# type: () -> None
|
2017-05-07 17:21:26 +02:00
|
|
|
user_profile = self.example_user('iago')
|
2017-08-25 06:01:29 +02:00
|
|
|
self.subscribe(user_profile, "Denmark")
|
2017-05-25 01:40:26 +02:00
|
|
|
self.send_message(self.example_email("hamlet"), "Denmark", Recipient.STREAM,
|
2017-01-05 16:50:34 +01:00
|
|
|
content="whatever", subject="my topic")
|
|
|
|
message = most_recent_message(user_profile)
|
|
|
|
row = Message.get_raw_db_rows([message.id])[0]
|
|
|
|
dct = MessageDict.build_dict_from_raw_db_row(row, apply_markdown=True)
|
|
|
|
self.assertEqual(dct['display_recipient'], 'Denmark')
|
|
|
|
|
|
|
|
stream = get_stream('Denmark', user_profile.realm)
|
|
|
|
self.assertEqual(dct['stream_id'], stream.id)
|
|
|
|
|
2016-09-24 00:08:15 +02:00
|
|
|
def test_stream_message_unicode(self):
|
|
|
|
# type: () -> None
|
2017-05-07 17:21:26 +02:00
|
|
|
user_profile = self.example_user('iago')
|
2017-08-25 06:01:29 +02:00
|
|
|
self.subscribe(user_profile, "Denmark")
|
2017-05-25 01:40:26 +02:00
|
|
|
self.send_message(self.example_email("hamlet"), "Denmark", Recipient.STREAM,
|
2016-09-24 00:08:15 +02:00
|
|
|
content="whatever", subject="my topic")
|
|
|
|
message = most_recent_message(user_profile)
|
|
|
|
self.assertEqual(str(message),
|
2016-12-03 00:04:17 +01:00
|
|
|
u'<Message: Denmark / my topic / '
|
2017-03-13 17:23:07 +01:00
|
|
|
'<UserProfile: hamlet@zulip.com <Realm: zulip 1>>>')
|
2016-09-24 00:08:15 +02:00
|
|
|
|
2014-01-31 16:44:45 +01:00
|
|
|
def test_message_mentions(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2017-05-07 17:21:26 +02:00
|
|
|
user_profile = self.example_user('iago')
|
2017-08-25 06:01:29 +02:00
|
|
|
self.subscribe(user_profile, "Denmark")
|
2017-05-25 01:40:26 +02:00
|
|
|
self.send_message(self.example_email("hamlet"), "Denmark", Recipient.STREAM,
|
2014-01-31 16:44:45 +01:00
|
|
|
content="test @**Iago** rules")
|
|
|
|
message = most_recent_message(user_profile)
|
|
|
|
assert(UserMessage.objects.get(user_profile=user_profile, message=message).flags.mentioned.is_set)
|
|
|
|
|
2015-09-20 19:32:01 +02:00
|
|
|
def test_stream_message_mirroring(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2015-09-20 19:32:01 +02:00
|
|
|
from zerver.lib.actions import do_change_is_admin
|
2017-05-07 19:39:30 +02:00
|
|
|
user_profile = self.example_user('iago')
|
|
|
|
email = user_profile.email
|
2015-09-20 19:32:01 +02:00
|
|
|
|
|
|
|
do_change_is_admin(user_profile, True, 'api_super_user')
|
2016-09-27 21:41:42 +02:00
|
|
|
result = self.client_post("/api/v1/messages", {"type": "stream",
|
|
|
|
"to": "Verona",
|
2017-05-25 01:50:35 +02:00
|
|
|
"sender": self.example_email("cordelia"),
|
2016-09-27 21:41:42 +02:00
|
|
|
"client": "test suite",
|
|
|
|
"subject": "announcement",
|
|
|
|
"content": "Everyone knows Iago rules",
|
|
|
|
"forged": "true"},
|
|
|
|
**self.api_auth(email))
|
2015-09-20 19:32:01 +02:00
|
|
|
self.assert_json_success(result)
|
|
|
|
do_change_is_admin(user_profile, False, 'api_super_user')
|
2016-09-27 21:41:42 +02:00
|
|
|
result = self.client_post("/api/v1/messages", {"type": "stream",
|
|
|
|
"to": "Verona",
|
2017-05-25 01:50:35 +02:00
|
|
|
"sender": self.example_email("cordelia"),
|
2016-09-27 21:41:42 +02:00
|
|
|
"client": "test suite",
|
|
|
|
"subject": "announcement",
|
|
|
|
"content": "Everyone knows Iago rules",
|
2016-11-30 21:55:59 +01:00
|
|
|
"forged": "true"},
|
2016-09-27 21:41:42 +02:00
|
|
|
**self.api_auth(email))
|
2015-09-20 19:32:01 +02:00
|
|
|
self.assert_json_error(result, "User not authorized for this query")
|
|
|
|
|
2016-07-29 21:06:22 +02:00
|
|
|
@slow('checks all users')
|
2014-01-31 16:44:45 +01:00
|
|
|
def test_message_to_stream(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
If you send a message to a stream, everyone subscribed to the stream
|
|
|
|
receives the messages.
|
|
|
|
"""
|
|
|
|
self.assert_stream_message("Scotland")
|
|
|
|
|
2016-07-29 21:06:22 +02:00
|
|
|
@slow('checks all users')
|
2014-01-31 16:44:45 +01:00
|
|
|
def test_non_ascii_stream_message(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
Sending a stream message containing non-ASCII characters in the stream
|
|
|
|
name, subject, or message body succeeds.
|
|
|
|
"""
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2014-01-31 16:44:45 +01:00
|
|
|
|
|
|
|
# Subscribe everyone to a stream with non-ASCII characters.
|
|
|
|
non_ascii_stream_name = u"hümbüǵ"
|
2017-01-04 05:30:48 +01:00
|
|
|
realm = get_realm("zulip")
|
2016-10-21 22:59:28 +02:00
|
|
|
stream = self.make_stream(non_ascii_stream_name)
|
2014-01-31 16:44:45 +01:00
|
|
|
for user_profile in UserProfile.objects.filter(realm=realm):
|
2017-08-25 06:01:29 +02:00
|
|
|
self.subscribe(user_profile, stream.name)
|
2014-01-31 16:44:45 +01:00
|
|
|
|
|
|
|
self.assert_stream_message(non_ascii_stream_name, subject=u"hümbüǵ",
|
|
|
|
content=u"hümbüǵ")
|
|
|
|
|
2016-08-23 02:08:42 +02:00
|
|
|
class MessageDictTest(ZulipTestCase):
|
2016-07-29 21:06:22 +02:00
|
|
|
@slow('builds lots of messages')
|
2014-01-31 16:44:45 +01:00
|
|
|
def test_bulk_message_fetching(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2017-05-07 17:21:26 +02:00
|
|
|
sender = self.example_user('othello')
|
|
|
|
receiver = self.example_user('hamlet')
|
2014-01-31 16:44:45 +01:00
|
|
|
pm_recipient = Recipient.objects.get(type_id=receiver.id, type=Recipient.PERSONAL)
|
2016-10-21 22:59:28 +02:00
|
|
|
stream_name = u'Çiğdem'
|
|
|
|
stream = self.make_stream(stream_name)
|
2014-01-31 16:44:45 +01:00
|
|
|
stream_recipient = Recipient.objects.get(type_id=stream.id, type=Recipient.STREAM)
|
2016-09-13 23:32:35 +02:00
|
|
|
sending_client = make_client(name="test suite")
|
2014-01-31 16:44:45 +01:00
|
|
|
|
2017-04-14 18:55:17 +02:00
|
|
|
ids = []
|
2014-01-31 16:44:45 +01:00
|
|
|
for i in range(300):
|
|
|
|
for recipient in [pm_recipient, stream_recipient]:
|
|
|
|
message = Message(
|
|
|
|
sender=sender,
|
|
|
|
recipient=recipient,
|
|
|
|
subject='whatever',
|
|
|
|
content='whatever %d' % i,
|
2017-04-15 04:03:56 +02:00
|
|
|
pub_date=timezone_now(),
|
2014-01-31 16:44:45 +01:00
|
|
|
sending_client=sending_client,
|
2017-04-15 04:03:56 +02:00
|
|
|
last_edit_time=timezone_now(),
|
2014-01-31 16:44:45 +01:00
|
|
|
edit_history='[]'
|
|
|
|
)
|
|
|
|
message.save()
|
2017-04-14 18:55:17 +02:00
|
|
|
ids.append(message.id)
|
2014-01-31 16:44:45 +01:00
|
|
|
|
2016-12-06 07:19:34 +01:00
|
|
|
Reaction.objects.create(user_profile=sender, message=message,
|
|
|
|
emoji_name='simple_smile')
|
|
|
|
|
2014-01-31 16:44:45 +01:00
|
|
|
num_ids = len(ids)
|
|
|
|
self.assertTrue(num_ids >= 600)
|
|
|
|
|
2017-03-31 08:28:28 +02:00
|
|
|
flush_per_request_caches()
|
2014-01-31 16:44:45 +01:00
|
|
|
t = time.time()
|
|
|
|
with queries_captured() as queries:
|
|
|
|
rows = list(Message.get_raw_db_rows(ids))
|
|
|
|
|
|
|
|
for row in rows:
|
2016-10-04 15:52:26 +02:00
|
|
|
MessageDict.build_dict_from_raw_db_row(row, False)
|
2014-01-31 16:44:45 +01:00
|
|
|
|
|
|
|
delay = time.time() - t
|
2017-06-01 09:00:09 +02:00
|
|
|
# Make sure we don't take longer than 1.5ms per message to
|
|
|
|
# extract messages. Note that we increased this from 1ms to
|
|
|
|
# 1.5ms to handle tests running in parallel being a bit
|
|
|
|
# slower.
|
2017-05-31 10:12:25 +02:00
|
|
|
error_msg = "Number of ids: {}. Time delay: {}".format(num_ids, delay)
|
2017-06-01 09:00:09 +02:00
|
|
|
self.assertTrue(delay < 0.0015 * num_ids, error_msg)
|
2017-04-14 18:55:17 +02:00
|
|
|
self.assert_length(queries, 5)
|
2014-01-31 16:44:45 +01:00
|
|
|
self.assertEqual(len(rows), num_ids)
|
|
|
|
|
|
|
|
def test_applying_markdown(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2017-05-07 17:21:26 +02:00
|
|
|
sender = self.example_user('othello')
|
|
|
|
receiver = self.example_user('hamlet')
|
2014-01-31 16:44:45 +01:00
|
|
|
recipient = Recipient.objects.get(type_id=receiver.id, type=Recipient.PERSONAL)
|
2016-09-13 23:32:35 +02:00
|
|
|
sending_client = make_client(name="test suite")
|
2014-01-31 16:44:45 +01:00
|
|
|
message = Message(
|
|
|
|
sender=sender,
|
|
|
|
recipient=recipient,
|
|
|
|
subject='whatever',
|
|
|
|
content='hello **world**',
|
2017-04-15 04:03:56 +02:00
|
|
|
pub_date=timezone_now(),
|
2014-01-31 16:44:45 +01:00
|
|
|
sending_client=sending_client,
|
2017-04-15 04:03:56 +02:00
|
|
|
last_edit_time=timezone_now(),
|
2014-01-31 16:44:45 +01:00
|
|
|
edit_history='[]'
|
|
|
|
)
|
|
|
|
message.save()
|
|
|
|
|
|
|
|
# An important part of this test is to get the message through this exact code path,
|
|
|
|
# because there is an ugly hack we need to cover. So don't just say "row = message".
|
|
|
|
row = Message.get_raw_db_rows([message.id])[0]
|
2016-10-04 15:52:26 +02:00
|
|
|
dct = MessageDict.build_dict_from_raw_db_row(row, apply_markdown=True)
|
2014-01-31 16:44:45 +01:00
|
|
|
expected_content = '<p>hello <strong>world</strong></p>'
|
|
|
|
self.assertEqual(dct['content'], expected_content)
|
|
|
|
message = Message.objects.get(id=message.id)
|
|
|
|
self.assertEqual(message.rendered_content, expected_content)
|
|
|
|
self.assertEqual(message.rendered_content_version, bugdown.version)
|
|
|
|
|
2017-03-18 20:10:37 +01:00
|
|
|
@mock.patch("zerver.lib.message.bugdown.convert")
|
|
|
|
def test_applying_markdown_invalid_format(self, convert_mock):
|
|
|
|
# type: (Any) -> None
|
|
|
|
# pretend the converter returned an invalid message without raising an exception
|
|
|
|
convert_mock.return_value = None
|
2017-05-07 17:21:26 +02:00
|
|
|
sender = self.example_user('othello')
|
|
|
|
receiver = self.example_user('hamlet')
|
2017-03-18 20:10:37 +01:00
|
|
|
recipient = Recipient.objects.get(type_id=receiver.id, type=Recipient.PERSONAL)
|
|
|
|
sending_client = make_client(name="test suite")
|
|
|
|
message = Message(
|
|
|
|
sender=sender,
|
|
|
|
recipient=recipient,
|
|
|
|
subject='whatever',
|
|
|
|
content='hello **world**',
|
2017-04-15 04:03:56 +02:00
|
|
|
pub_date=timezone_now(),
|
2017-03-18 20:10:37 +01:00
|
|
|
sending_client=sending_client,
|
2017-04-15 04:03:56 +02:00
|
|
|
last_edit_time=timezone_now(),
|
2017-03-18 20:10:37 +01:00
|
|
|
edit_history='[]'
|
|
|
|
)
|
|
|
|
message.save()
|
|
|
|
|
|
|
|
# An important part of this test is to get the message through this exact code path,
|
|
|
|
# because there is an ugly hack we need to cover. So don't just say "row = message".
|
|
|
|
row = Message.get_raw_db_rows([message.id])[0]
|
|
|
|
dct = MessageDict.build_dict_from_raw_db_row(row, apply_markdown=True)
|
|
|
|
error_content = '<p>[Zulip note: Sorry, we could not understand the formatting of your message]</p>'
|
|
|
|
self.assertEqual(dct['content'], error_content)
|
|
|
|
|
2016-12-06 07:19:34 +01:00
|
|
|
def test_reaction(self):
|
|
|
|
# type: () -> None
|
2017-05-07 17:21:26 +02:00
|
|
|
sender = self.example_user('othello')
|
|
|
|
receiver = self.example_user('hamlet')
|
2016-12-06 07:19:34 +01:00
|
|
|
recipient = Recipient.objects.get(type_id=receiver.id, type=Recipient.PERSONAL)
|
|
|
|
sending_client = make_client(name="test suite")
|
|
|
|
message = Message(
|
|
|
|
sender=sender,
|
|
|
|
recipient=recipient,
|
|
|
|
subject='whatever',
|
|
|
|
content='hello **world**',
|
2017-04-15 04:03:56 +02:00
|
|
|
pub_date=timezone_now(),
|
2016-12-06 07:19:34 +01:00
|
|
|
sending_client=sending_client,
|
2017-04-15 04:03:56 +02:00
|
|
|
last_edit_time=timezone_now(),
|
2016-12-06 07:19:34 +01:00
|
|
|
edit_history='[]'
|
|
|
|
)
|
|
|
|
message.save()
|
|
|
|
|
|
|
|
reaction = Reaction.objects.create(
|
|
|
|
message=message, user_profile=sender,
|
|
|
|
emoji_name='simple_smile')
|
|
|
|
row = Message.get_raw_db_rows([message.id])[0]
|
2016-12-15 05:37:13 +01:00
|
|
|
msg_dict = MessageDict.build_dict_from_raw_db_row(
|
|
|
|
row, apply_markdown=True)
|
2016-12-06 07:19:34 +01:00
|
|
|
self.assertEqual(msg_dict['reactions'][0]['emoji_name'],
|
|
|
|
reaction.emoji_name)
|
|
|
|
self.assertEqual(msg_dict['reactions'][0]['user']['id'],
|
|
|
|
sender.id)
|
|
|
|
self.assertEqual(msg_dict['reactions'][0]['user']['email'],
|
|
|
|
sender.email)
|
|
|
|
self.assertEqual(msg_dict['reactions'][0]['user']['full_name'],
|
|
|
|
sender.full_name)
|
|
|
|
|
|
|
|
|
|
|
|
class SewMessageAndReactionTest(ZulipTestCase):
|
|
|
|
def test_sew_messages_and_reaction(self):
|
|
|
|
# type: () -> None
|
2017-05-07 17:21:26 +02:00
|
|
|
sender = self.example_user('othello')
|
|
|
|
receiver = self.example_user('hamlet')
|
2016-12-06 07:19:34 +01:00
|
|
|
pm_recipient = Recipient.objects.get(type_id=receiver.id, type=Recipient.PERSONAL)
|
|
|
|
stream_name = u'Çiğdem'
|
|
|
|
stream = self.make_stream(stream_name)
|
|
|
|
stream_recipient = Recipient.objects.get(type_id=stream.id, type=Recipient.STREAM)
|
|
|
|
sending_client = make_client(name="test suite")
|
|
|
|
|
|
|
|
needed_ids = []
|
|
|
|
for i in range(5):
|
|
|
|
for recipient in [pm_recipient, stream_recipient]:
|
|
|
|
message = Message(
|
|
|
|
sender=sender,
|
|
|
|
recipient=recipient,
|
|
|
|
subject='whatever',
|
|
|
|
content='whatever %d' % i,
|
2017-04-15 04:03:56 +02:00
|
|
|
pub_date=timezone_now(),
|
2016-12-06 07:19:34 +01:00
|
|
|
sending_client=sending_client,
|
2017-04-15 04:03:56 +02:00
|
|
|
last_edit_time=timezone_now(),
|
2016-12-06 07:19:34 +01:00
|
|
|
edit_history='[]'
|
|
|
|
)
|
|
|
|
message.save()
|
|
|
|
needed_ids.append(message.id)
|
|
|
|
reaction = Reaction(user_profile=sender, message=message,
|
|
|
|
emoji_name='simple_smile')
|
|
|
|
reaction.save()
|
|
|
|
|
|
|
|
messages = Message.objects.filter(id__in=needed_ids).values(
|
|
|
|
*['id', 'content'])
|
|
|
|
reactions = Reaction.get_raw_db_rows(needed_ids)
|
|
|
|
tied_data = sew_messages_and_reactions(messages, reactions)
|
|
|
|
for data in tied_data:
|
|
|
|
self.assertEqual(len(data['reactions']), 1)
|
|
|
|
self.assertEqual(data['reactions'][0]['emoji_name'],
|
|
|
|
'simple_smile')
|
|
|
|
self.assertTrue(data['id'])
|
|
|
|
self.assertTrue(data['content'])
|
|
|
|
|
|
|
|
|
2016-08-23 02:08:42 +02:00
|
|
|
class MessagePOSTTest(ZulipTestCase):
|
2014-01-31 16:44:45 +01:00
|
|
|
|
|
|
|
def test_message_to_self(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
Sending a message to a stream to which you are subscribed is
|
|
|
|
successful.
|
|
|
|
"""
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2016-07-28 00:30:22 +02:00
|
|
|
result = self.client_post("/json/messages", {"type": "stream",
|
2015-12-01 00:18:33 +01:00
|
|
|
"to": "Verona",
|
|
|
|
"client": "test suite",
|
|
|
|
"content": "Test message",
|
|
|
|
"subject": "Test subject"})
|
2014-01-31 16:44:45 +01:00
|
|
|
self.assert_json_success(result)
|
|
|
|
|
|
|
|
def test_api_message_to_self(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
Same as above, but for the API view
|
|
|
|
"""
|
2017-05-25 01:40:26 +02:00
|
|
|
email = self.example_email("hamlet")
|
2016-09-27 21:41:42 +02:00
|
|
|
result = self.client_post("/api/v1/messages", {"type": "stream",
|
|
|
|
"to": "Verona",
|
|
|
|
"client": "test suite",
|
|
|
|
"content": "Test message",
|
|
|
|
"subject": "Test subject"},
|
|
|
|
**self.api_auth(email))
|
2014-01-31 16:44:45 +01:00
|
|
|
self.assert_json_success(result)
|
|
|
|
|
2014-02-06 23:12:34 +01:00
|
|
|
def test_api_message_with_default_to(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-02-06 23:12:34 +01:00
|
|
|
"""
|
|
|
|
Sending messages without a to field should be sent to the default
|
|
|
|
stream for the user_profile.
|
|
|
|
"""
|
2017-05-07 19:39:30 +02:00
|
|
|
user_profile = self.example_user('hamlet')
|
|
|
|
email = user_profile.email
|
2014-02-06 23:12:34 +01:00
|
|
|
user_profile.default_sending_stream = get_stream('Verona', user_profile.realm)
|
|
|
|
user_profile.save()
|
2016-09-27 21:41:42 +02:00
|
|
|
result = self.client_post("/api/v1/messages", {"type": "stream",
|
|
|
|
"client": "test suite",
|
|
|
|
"content": "Test message no to",
|
|
|
|
"subject": "Test subject"},
|
|
|
|
**self.api_auth(email))
|
2014-02-06 23:12:34 +01:00
|
|
|
self.assert_json_success(result)
|
|
|
|
|
2016-04-13 23:59:08 +02:00
|
|
|
sent_message = self.get_last_message()
|
2014-02-06 23:12:34 +01:00
|
|
|
self.assertEqual(sent_message.content, "Test message no to")
|
|
|
|
|
2014-01-31 16:44:45 +01:00
|
|
|
def test_message_to_nonexistent_stream(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
Sending a message to a nonexistent stream fails.
|
|
|
|
"""
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2014-01-31 16:44:45 +01:00
|
|
|
self.assertFalse(Stream.objects.filter(name="nonexistent_stream"))
|
2016-07-28 00:30:22 +02:00
|
|
|
result = self.client_post("/json/messages", {"type": "stream",
|
2015-12-01 00:18:33 +01:00
|
|
|
"to": "nonexistent_stream",
|
|
|
|
"client": "test suite",
|
|
|
|
"content": "Test message",
|
|
|
|
"subject": "Test subject"})
|
2016-12-13 19:18:08 +01:00
|
|
|
self.assert_json_error(result, "Stream 'nonexistent_stream' does not exist")
|
|
|
|
|
|
|
|
def test_message_to_nonexistent_stream_with_bad_characters(self):
|
|
|
|
# type: () -> None
|
|
|
|
"""
|
|
|
|
Nonexistent stream name with bad characters should be escaped properly.
|
|
|
|
"""
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2016-12-13 19:18:08 +01:00
|
|
|
self.assertFalse(Stream.objects.filter(name="""&<"'><non-existent>"""))
|
|
|
|
result = self.client_post("/json/messages", {"type": "stream",
|
|
|
|
"to": """&<"'><non-existent>""",
|
|
|
|
"client": "test suite",
|
|
|
|
"content": "Test message",
|
|
|
|
"subject": "Test subject"})
|
|
|
|
self.assert_json_error(result, "Stream '&<"'><non-existent>' does not exist")
|
2014-01-31 16:44:45 +01:00
|
|
|
|
|
|
|
def test_personal_message(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
Sending a personal message to a valid username is successful.
|
|
|
|
"""
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2016-07-28 00:30:22 +02:00
|
|
|
result = self.client_post("/json/messages", {"type": "private",
|
2015-12-01 00:18:33 +01:00
|
|
|
"content": "Test message",
|
|
|
|
"client": "test suite",
|
2017-05-25 02:08:35 +02:00
|
|
|
"to": self.example_email("othello")})
|
2014-01-31 16:44:45 +01:00
|
|
|
self.assert_json_success(result)
|
|
|
|
|
2017-07-29 00:49:27 +02:00
|
|
|
def test_personal_message_copying_self(self):
|
|
|
|
# type: () -> None
|
|
|
|
"""
|
|
|
|
Sending a personal message to yourself plus another user is succesful,
|
|
|
|
and counts as a message just to that user.
|
|
|
|
"""
|
|
|
|
self.login(self.example_email("hamlet"))
|
|
|
|
result = self.client_post("/json/messages", {
|
|
|
|
"type": "private",
|
|
|
|
"content": "Test message",
|
|
|
|
"client": "test suite",
|
|
|
|
"to": ujson.dumps([self.example_email("othello"),
|
|
|
|
self.example_email("hamlet")])})
|
|
|
|
self.assert_json_success(result)
|
|
|
|
msg = self.get_last_message()
|
|
|
|
# Verify that we're not actually on the "recipient list"
|
|
|
|
self.assertNotIn("Hamlet", str(msg.recipient))
|
|
|
|
|
2014-01-31 16:44:45 +01:00
|
|
|
def test_personal_message_to_nonexistent_user(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
Sending a personal message to an invalid email returns error JSON.
|
|
|
|
"""
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2016-07-28 00:30:22 +02:00
|
|
|
result = self.client_post("/json/messages", {"type": "private",
|
2015-12-01 00:18:33 +01:00
|
|
|
"content": "Test message",
|
|
|
|
"client": "test suite",
|
|
|
|
"to": "nonexistent"})
|
2014-01-31 16:44:45 +01:00
|
|
|
self.assert_json_error(result, "Invalid email 'nonexistent'")
|
|
|
|
|
2017-07-29 00:52:27 +02:00
|
|
|
def test_personal_message_to_deactivated_user(self):
|
|
|
|
# type: () -> None
|
|
|
|
"""
|
|
|
|
Sending a personal message to a deactivated user returns error JSON.
|
|
|
|
"""
|
|
|
|
target_user_profile = self.example_user("othello")
|
|
|
|
do_deactivate_user(target_user_profile)
|
|
|
|
self.login(self.example_email("hamlet"))
|
|
|
|
result = self.client_post("/json/messages", {
|
|
|
|
"type": "private",
|
|
|
|
"content": "Test message",
|
|
|
|
"client": "test suite",
|
|
|
|
"to": self.example_email("othello")})
|
|
|
|
self.assert_json_error(result, "'othello@zulip.com' is no longer using Zulip.")
|
|
|
|
|
|
|
|
result = self.client_post("/json/messages", {
|
|
|
|
"type": "private",
|
|
|
|
"content": "Test message",
|
|
|
|
"client": "test suite",
|
|
|
|
"to": ujson.dumps([self.example_email("othello"),
|
|
|
|
self.example_email("cordelia")])})
|
|
|
|
self.assert_json_error(result, "'othello@zulip.com' is no longer using Zulip.")
|
|
|
|
|
2014-01-31 16:44:45 +01:00
|
|
|
def test_invalid_type(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
Sending a message of unknown type returns error JSON.
|
|
|
|
"""
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2016-07-28 00:30:22 +02:00
|
|
|
result = self.client_post("/json/messages", {"type": "invalid type",
|
2015-12-01 00:18:33 +01:00
|
|
|
"content": "Test message",
|
|
|
|
"client": "test suite",
|
2017-05-25 02:08:35 +02:00
|
|
|
"to": self.example_email("othello")})
|
2014-01-31 16:44:45 +01:00
|
|
|
self.assert_json_error(result, "Invalid message type")
|
|
|
|
|
|
|
|
def test_empty_message(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
Sending a message that is empty or only whitespace should fail
|
|
|
|
"""
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2016-07-28 00:30:22 +02:00
|
|
|
result = self.client_post("/json/messages", {"type": "private",
|
2015-12-01 00:18:33 +01:00
|
|
|
"content": " ",
|
|
|
|
"client": "test suite",
|
2017-05-25 02:08:35 +02:00
|
|
|
"to": self.example_email("othello")})
|
2014-01-31 16:44:45 +01:00
|
|
|
self.assert_json_error(result, "Message must not be empty")
|
|
|
|
|
|
|
|
def test_mirrored_huddle(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
Sending a mirrored huddle message works
|
|
|
|
"""
|
2017-05-24 21:21:35 +02:00
|
|
|
self.login(self.mit_email("starnine"))
|
2016-07-28 00:30:22 +02:00
|
|
|
result = self.client_post("/json/messages", {"type": "private",
|
2017-05-24 21:21:35 +02:00
|
|
|
"sender": self.mit_email("sipbtest"),
|
2015-12-01 00:18:33 +01:00
|
|
|
"content": "Test message",
|
|
|
|
"client": "zephyr_mirror",
|
2017-05-24 21:21:35 +02:00
|
|
|
"to": ujson.dumps([self.mit_email("starnine"),
|
2017-08-26 01:08:14 +02:00
|
|
|
self.mit_email("espuser")])},
|
|
|
|
subdomain="zephyr")
|
2014-01-31 16:44:45 +01:00
|
|
|
self.assert_json_success(result)
|
|
|
|
|
|
|
|
def test_mirrored_personal(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
Sending a mirrored personal message works
|
|
|
|
"""
|
2017-05-24 21:21:35 +02:00
|
|
|
self.login(self.mit_email("starnine"))
|
2016-07-28 00:30:22 +02:00
|
|
|
result = self.client_post("/json/messages", {"type": "private",
|
2017-05-24 21:21:35 +02:00
|
|
|
"sender": self.mit_email("sipbtest"),
|
2015-12-01 00:18:33 +01:00
|
|
|
"content": "Test message",
|
|
|
|
"client": "zephyr_mirror",
|
2017-08-26 01:08:14 +02:00
|
|
|
"to": self.mit_email("starnine")},
|
|
|
|
subdomain="zephyr")
|
2014-01-31 16:44:45 +01:00
|
|
|
self.assert_json_success(result)
|
|
|
|
|
2017-07-29 00:55:06 +02:00
|
|
|
def test_mirrored_personal_to_someone_else(self):
|
|
|
|
# type: () -> None
|
|
|
|
"""
|
|
|
|
Sending a mirrored personal message to someone else is not allowed.
|
|
|
|
"""
|
|
|
|
self.login(self.mit_email("starnine"))
|
|
|
|
result = self.client_post("/json/messages", {"type": "private",
|
|
|
|
"sender": self.mit_email("sipbtest"),
|
|
|
|
"content": "Test message",
|
|
|
|
"client": "zephyr_mirror",
|
2017-08-26 01:08:14 +02:00
|
|
|
"to": self.mit_email("espuser")},
|
|
|
|
subdomain="zephyr")
|
2017-07-29 00:55:06 +02:00
|
|
|
self.assert_json_error(result, "User not authorized for this query")
|
|
|
|
|
2014-01-31 16:44:45 +01:00
|
|
|
def test_duplicated_mirrored_huddle(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
Sending two mirrored huddles in the row return the same ID
|
|
|
|
"""
|
|
|
|
msg = {"type": "private",
|
2017-05-24 21:21:35 +02:00
|
|
|
"sender": self.mit_email("sipbtest"),
|
2014-01-31 16:44:45 +01:00
|
|
|
"content": "Test message",
|
|
|
|
"client": "zephyr_mirror",
|
2017-05-24 21:21:35 +02:00
|
|
|
"to": ujson.dumps([self.mit_email("espuser"),
|
|
|
|
self.mit_email("starnine")])}
|
2014-01-31 16:44:45 +01:00
|
|
|
|
2016-06-01 05:30:10 +02:00
|
|
|
with mock.patch('DNS.dnslookup', return_value=[['starnine:*:84233:101:Athena Consulting Exchange User,,,:/mit/starnine:/bin/bash']]):
|
2017-05-24 21:21:35 +02:00
|
|
|
self.login(self.mit_email("starnine"))
|
2017-08-26 01:08:14 +02:00
|
|
|
result1 = self.client_post("/json/messages", msg,
|
|
|
|
subdomain="zephyr")
|
2016-06-27 20:42:36 +02:00
|
|
|
with mock.patch('DNS.dnslookup', return_value=[['espuser:*:95494:101:Esp Classroom,,,:/mit/espuser:/bin/athena/bash']]):
|
2017-05-24 21:21:35 +02:00
|
|
|
self.login(self.mit_email("espuser"))
|
2017-08-26 01:08:14 +02:00
|
|
|
result2 = self.client_post("/json/messages", msg,
|
|
|
|
subdomain="zephyr")
|
2014-01-31 16:44:45 +01:00
|
|
|
self.assertEqual(ujson.loads(result1.content)['id'],
|
|
|
|
ujson.loads(result2.content)['id'])
|
|
|
|
|
2017-02-12 04:22:13 +01:00
|
|
|
def test_strip_message(self):
|
|
|
|
# type: () -> None
|
|
|
|
"""
|
|
|
|
Sending a message longer than the maximum message length succeeds but is
|
|
|
|
truncated.
|
|
|
|
"""
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2017-02-12 04:22:13 +01:00
|
|
|
post_data = {"type": "stream", "to": "Verona", "client": "test suite",
|
2017-02-18 23:47:18 +01:00
|
|
|
"content": " I like whitespace at the end! \n\n \n", "subject": "Test subject"}
|
2017-02-12 04:22:13 +01:00
|
|
|
result = self.client_post("/json/messages", post_data)
|
|
|
|
self.assert_json_success(result)
|
|
|
|
sent_message = self.get_last_message()
|
2017-02-18 23:47:18 +01:00
|
|
|
self.assertEqual(sent_message.content, " I like whitespace at the end!")
|
2017-02-12 04:22:13 +01:00
|
|
|
|
2014-01-31 16:44:45 +01:00
|
|
|
def test_long_message(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
Sending a message longer than the maximum message length succeeds but is
|
|
|
|
truncated.
|
|
|
|
"""
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2014-01-31 16:44:45 +01:00
|
|
|
long_message = "A" * (MAX_MESSAGE_LENGTH + 1)
|
|
|
|
post_data = {"type": "stream", "to": "Verona", "client": "test suite",
|
|
|
|
"content": long_message, "subject": "Test subject"}
|
2016-07-28 00:30:22 +02:00
|
|
|
result = self.client_post("/json/messages", post_data)
|
2014-01-31 16:44:45 +01:00
|
|
|
self.assert_json_success(result)
|
|
|
|
|
2016-04-13 23:59:08 +02:00
|
|
|
sent_message = self.get_last_message()
|
2016-12-16 02:01:34 +01:00
|
|
|
self.assertEqual(sent_message.content,
|
2016-12-16 05:51:27 +01:00
|
|
|
"A" * (MAX_MESSAGE_LENGTH - 3) + "...")
|
2014-01-31 16:44:45 +01:00
|
|
|
|
|
|
|
def test_long_topic(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
Sending a message with a topic longer than the maximum topic length
|
|
|
|
succeeds, but the topic is truncated.
|
|
|
|
"""
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2014-01-31 16:44:45 +01:00
|
|
|
long_topic = "A" * (MAX_SUBJECT_LENGTH + 1)
|
|
|
|
post_data = {"type": "stream", "to": "Verona", "client": "test suite",
|
|
|
|
"content": "test content", "subject": long_topic}
|
2016-07-28 00:30:22 +02:00
|
|
|
result = self.client_post("/json/messages", post_data)
|
2014-01-31 16:44:45 +01:00
|
|
|
self.assert_json_success(result)
|
|
|
|
|
2016-04-13 23:59:08 +02:00
|
|
|
sent_message = self.get_last_message()
|
2016-12-16 02:01:34 +01:00
|
|
|
self.assertEqual(sent_message.topic_name(),
|
2016-12-16 05:51:27 +01:00
|
|
|
"A" * (MAX_SUBJECT_LENGTH - 3) + "...")
|
2014-01-31 16:44:45 +01:00
|
|
|
|
2016-06-21 19:45:25 +02:00
|
|
|
def test_send_forged_message_as_not_superuser(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2016-07-28 00:30:22 +02:00
|
|
|
result = self.client_post("/json/messages", {"type": "stream",
|
2016-06-21 19:45:25 +02:00
|
|
|
"to": "Verona",
|
|
|
|
"client": "test suite",
|
|
|
|
"content": "Test message",
|
|
|
|
"subject": "Test subject",
|
|
|
|
"forged": True})
|
|
|
|
self.assert_json_error(result, "User not authorized for this query")
|
|
|
|
|
|
|
|
def test_send_message_as_not_superuser_to_different_domain(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2016-07-28 00:30:22 +02:00
|
|
|
result = self.client_post("/json/messages", {"type": "stream",
|
2016-06-21 19:45:25 +02:00
|
|
|
"to": "Verona",
|
|
|
|
"client": "test suite",
|
|
|
|
"content": "Test message",
|
|
|
|
"subject": "Test subject",
|
2017-01-03 06:57:03 +01:00
|
|
|
"realm_str": "mit"})
|
2016-06-21 19:45:25 +02:00
|
|
|
self.assert_json_error(result, "User not authorized for this query")
|
|
|
|
|
|
|
|
def test_send_message_as_superuser_to_domain_that_dont_exist(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2017-05-24 02:42:31 +02:00
|
|
|
user = get_system_bot(settings.EMAIL_GATEWAY_BOT)
|
2016-06-21 19:45:25 +02:00
|
|
|
password = "test_password"
|
|
|
|
user.set_password(password)
|
|
|
|
user.is_api_super_user = True
|
|
|
|
user.save()
|
2017-05-24 02:42:31 +02:00
|
|
|
self.login(user.email, password)
|
2016-07-28 00:30:22 +02:00
|
|
|
result = self.client_post("/json/messages", {"type": "stream",
|
2016-06-21 19:45:25 +02:00
|
|
|
"to": "Verona",
|
|
|
|
"client": "test suite",
|
|
|
|
"content": "Test message",
|
|
|
|
"subject": "Test subject",
|
2017-01-03 06:57:03 +01:00
|
|
|
"realm_str": "non-existing"})
|
2016-06-21 19:45:25 +02:00
|
|
|
user.is_api_super_user = False
|
|
|
|
user.save()
|
2017-01-03 06:57:03 +01:00
|
|
|
self.assert_json_error(result, "Unknown realm non-existing")
|
2016-06-21 19:45:25 +02:00
|
|
|
|
|
|
|
def test_send_message_when_sender_is_not_set(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2017-05-24 21:21:35 +02:00
|
|
|
self.login(self.mit_email("starnine"))
|
2016-07-28 00:30:22 +02:00
|
|
|
result = self.client_post("/json/messages", {"type": "private",
|
2016-06-21 19:45:25 +02:00
|
|
|
"content": "Test message",
|
|
|
|
"client": "zephyr_mirror",
|
2017-08-26 01:08:14 +02:00
|
|
|
"to": self.mit_email("starnine")},
|
|
|
|
subdomain="zephyr")
|
2016-06-21 19:45:25 +02:00
|
|
|
self.assert_json_error(result, "Missing sender")
|
|
|
|
|
|
|
|
def test_send_message_as_not_superuser_when_type_is_not_private(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2017-05-24 21:21:35 +02:00
|
|
|
self.login(self.mit_email("starnine"))
|
2016-07-28 00:30:22 +02:00
|
|
|
result = self.client_post("/json/messages", {"type": "not-private",
|
2017-05-24 21:21:35 +02:00
|
|
|
"sender": self.mit_email("sipbtest"),
|
2016-06-21 19:45:25 +02:00
|
|
|
"content": "Test message",
|
|
|
|
"client": "zephyr_mirror",
|
2017-08-26 01:08:14 +02:00
|
|
|
"to": self.mit_email("starnine")},
|
|
|
|
subdomain="zephyr")
|
2016-06-21 19:45:25 +02:00
|
|
|
self.assert_json_error(result, "User not authorized for this query")
|
|
|
|
|
|
|
|
@mock.patch("zerver.views.messages.create_mirrored_message_users")
|
|
|
|
def test_send_message_create_mirrored_message_user_returns_invalid_input(self, create_mirrored_message_users_mock):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: (Any) -> None
|
2016-06-21 19:45:25 +02:00
|
|
|
create_mirrored_message_users_mock.return_value = (False, True)
|
2017-05-24 21:21:35 +02:00
|
|
|
self.login(self.mit_email("starnine"))
|
2016-07-28 00:30:22 +02:00
|
|
|
result = self.client_post("/json/messages", {"type": "private",
|
2017-05-24 21:21:35 +02:00
|
|
|
"sender": self.mit_email("sipbtest"),
|
2016-06-21 19:45:25 +02:00
|
|
|
"content": "Test message",
|
|
|
|
"client": "zephyr_mirror",
|
2017-08-26 01:08:14 +02:00
|
|
|
"to": self.mit_email("starnine")},
|
|
|
|
subdomain="zephyr")
|
2016-06-21 19:45:25 +02:00
|
|
|
self.assert_json_error(result, "Invalid mirrored message")
|
|
|
|
|
|
|
|
@mock.patch("zerver.views.messages.create_mirrored_message_users")
|
2017-03-04 09:19:37 +01:00
|
|
|
def test_send_message_when_client_is_zephyr_mirror_but_string_id_is_not_zephyr(self, create_mirrored_message_users_mock):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: (Any) -> None
|
2016-06-21 19:45:25 +02:00
|
|
|
create_mirrored_message_users_mock.return_value = (True, True)
|
2017-05-23 02:33:53 +02:00
|
|
|
user = self.mit_user("starnine")
|
|
|
|
email = user.email
|
2017-03-04 09:19:37 +01:00
|
|
|
user.realm.string_id = 'not_zephyr'
|
2016-06-21 19:45:25 +02:00
|
|
|
user.realm.save()
|
2017-05-23 02:33:53 +02:00
|
|
|
self.login(email)
|
2016-07-28 00:30:22 +02:00
|
|
|
result = self.client_post("/json/messages", {"type": "private",
|
2017-05-24 21:21:35 +02:00
|
|
|
"sender": self.mit_email("sipbtest"),
|
2016-06-21 19:45:25 +02:00
|
|
|
"content": "Test message",
|
|
|
|
"client": "zephyr_mirror",
|
2017-05-23 02:33:53 +02:00
|
|
|
"to": email}, name='gownooo')
|
2016-06-21 19:45:25 +02:00
|
|
|
self.assert_json_error(result, "Invalid mirrored realm")
|
|
|
|
|
2017-09-16 11:48:01 +02:00
|
|
|
@override_settings(REALMS_HAVE_SUBDOMAINS=True)
|
2017-02-23 00:56:46 +01:00
|
|
|
def test_send_message_irc_mirror(self):
|
|
|
|
# type: () -> None
|
2017-05-24 02:42:31 +02:00
|
|
|
self.login(self.example_email('hamlet'))
|
2017-02-23 00:56:46 +01:00
|
|
|
bot_info = {
|
|
|
|
'full_name': 'IRC bot',
|
|
|
|
'short_name': 'irc',
|
|
|
|
}
|
|
|
|
result = self.client_post("/json/bots", bot_info)
|
|
|
|
self.assert_json_success(result)
|
|
|
|
|
2017-03-05 04:17:12 +01:00
|
|
|
email = "irc-bot@zulip.testserver"
|
2017-05-24 02:42:31 +02:00
|
|
|
user = get_user(email, get_realm('zulip'))
|
2017-02-23 00:56:46 +01:00
|
|
|
user.is_api_super_user = True
|
|
|
|
user.save()
|
2017-05-24 02:42:31 +02:00
|
|
|
user = get_user(email, get_realm('zulip'))
|
2017-08-25 06:01:29 +02:00
|
|
|
self.subscribe(user, "#IRCland")
|
2017-02-23 00:56:46 +01:00
|
|
|
result = self.client_post("/api/v1/messages",
|
|
|
|
{"type": "stream",
|
|
|
|
"forged": "true",
|
|
|
|
"sender": "irc-user@irc.zulip.com",
|
|
|
|
"content": "Test message",
|
|
|
|
"client": "irc_mirror",
|
|
|
|
"subject": "from irc",
|
|
|
|
"to": "IRCLand"},
|
|
|
|
**self.api_auth(email))
|
|
|
|
self.assert_json_error(result, "IRC stream names must start with #")
|
|
|
|
result = self.client_post("/api/v1/messages",
|
|
|
|
{"type": "stream",
|
|
|
|
"forged": "true",
|
|
|
|
"sender": "irc-user@irc.zulip.com",
|
|
|
|
"content": "Test message",
|
|
|
|
"client": "irc_mirror",
|
|
|
|
"subject": "from irc",
|
|
|
|
"to": "#IRCLand"},
|
|
|
|
**self.api_auth(email))
|
|
|
|
self.assert_json_success(result)
|
|
|
|
|
2016-08-23 02:08:42 +02:00
|
|
|
class EditMessageTest(ZulipTestCase):
|
2014-01-31 16:44:45 +01:00
|
|
|
def check_message(self, msg_id, subject=None, content=None):
|
2016-12-04 18:38:56 +01:00
|
|
|
# type: (int, Optional[Text], Optional[Text]) -> Message
|
2014-01-31 16:44:45 +01:00
|
|
|
msg = Message.objects.get(id=msg_id)
|
2016-10-04 15:52:26 +02:00
|
|
|
cached = message_to_dict(msg, False)
|
|
|
|
uncached = MessageDict.to_dict_uncached_helper(msg, False)
|
2014-01-31 16:44:45 +01:00
|
|
|
self.assertEqual(cached, uncached)
|
|
|
|
if subject:
|
2016-07-14 17:48:11 +02:00
|
|
|
self.assertEqual(msg.topic_name(), subject)
|
2014-01-31 16:44:45 +01:00
|
|
|
if content:
|
|
|
|
self.assertEqual(msg.content, content)
|
|
|
|
return msg
|
|
|
|
|
|
|
|
def test_save_message(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
|
|
|
"""This is also tested by a client test, but here we can verify
|
|
|
|
the cache against the database"""
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
|
|
|
msg_id = self.send_message(self.example_email("hamlet"), "Scotland", Recipient.STREAM,
|
2016-12-03 00:04:17 +01:00
|
|
|
subject="editing", content="before edit")
|
2016-12-22 10:17:49 +01:00
|
|
|
result = self.client_patch("/json/messages/" + str(msg_id), {
|
2014-01-31 16:44:45 +01:00
|
|
|
'message_id': msg_id,
|
|
|
|
'content': 'after edit'
|
|
|
|
})
|
|
|
|
self.assert_json_success(result)
|
|
|
|
self.check_message(msg_id, content="after edit")
|
|
|
|
|
2016-12-22 10:17:49 +01:00
|
|
|
result = self.client_patch("/json/messages/" + str(msg_id), {
|
2014-01-31 16:44:45 +01:00
|
|
|
'message_id': msg_id,
|
|
|
|
'subject': 'edited'
|
|
|
|
})
|
|
|
|
self.assert_json_success(result)
|
|
|
|
self.check_message(msg_id, subject="edited")
|
|
|
|
|
2016-09-19 21:24:49 +02:00
|
|
|
def test_fetch_raw_message(self):
|
|
|
|
# type: () -> None
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2017-05-25 01:50:35 +02:00
|
|
|
msg_id = self.send_message(self.example_email("hamlet"), self.example_email("cordelia"), Recipient.PERSONAL,
|
2016-09-19 21:24:49 +02:00
|
|
|
subject="editing", content="**before** edit")
|
2016-12-21 20:31:16 +01:00
|
|
|
result = self.client_get('/json/messages/' + str(msg_id))
|
2016-09-19 21:24:49 +02:00
|
|
|
self.assert_json_success(result)
|
2017-08-16 09:50:47 +02:00
|
|
|
self.assertEqual(result.json()['raw_content'], '**before** edit')
|
2016-09-19 21:24:49 +02:00
|
|
|
|
|
|
|
# Test error cases
|
2016-12-21 20:31:16 +01:00
|
|
|
result = self.client_get('/json/messages/999999')
|
2016-10-12 02:14:08 +02:00
|
|
|
self.assert_json_error(result, 'Invalid message(s)')
|
2016-09-19 21:24:49 +02:00
|
|
|
|
2017-05-25 01:50:35 +02:00
|
|
|
self.login(self.example_email("cordelia"))
|
2016-12-21 20:31:16 +01:00
|
|
|
result = self.client_get('/json/messages/' + str(msg_id))
|
2016-09-22 18:48:42 +02:00
|
|
|
self.assert_json_success(result)
|
|
|
|
|
2017-05-25 02:08:35 +02:00
|
|
|
self.login(self.example_email("othello"))
|
2016-12-21 20:31:16 +01:00
|
|
|
result = self.client_get('/json/messages/' + str(msg_id))
|
2016-10-12 02:14:08 +02:00
|
|
|
self.assert_json_error(result, 'Invalid message(s)')
|
2016-09-22 18:48:42 +02:00
|
|
|
|
|
|
|
def test_fetch_raw_message_stream_wrong_realm(self):
|
|
|
|
# type: () -> None
|
2017-08-25 06:01:29 +02:00
|
|
|
user_profile = self.example_user("hamlet")
|
|
|
|
self.login(user_profile.email)
|
2016-10-21 22:59:28 +02:00
|
|
|
stream = self.make_stream('public_stream')
|
2017-08-25 06:01:29 +02:00
|
|
|
self.subscribe(user_profile, stream.name)
|
|
|
|
msg_id = self.send_message(user_profile.email, stream.name, Recipient.STREAM,
|
2016-09-22 18:48:42 +02:00
|
|
|
subject="test", content="test")
|
2016-12-21 20:31:16 +01:00
|
|
|
result = self.client_get('/json/messages/' + str(msg_id))
|
2016-09-22 18:48:42 +02:00
|
|
|
self.assert_json_success(result)
|
|
|
|
|
2017-05-24 21:21:35 +02:00
|
|
|
self.login(self.mit_email("sipbtest"))
|
2017-08-26 01:08:14 +02:00
|
|
|
result = self.client_get('/json/messages/' + str(msg_id), subdomain="zephyr")
|
2016-10-12 02:14:08 +02:00
|
|
|
self.assert_json_error(result, 'Invalid message(s)')
|
2016-09-22 18:48:42 +02:00
|
|
|
|
|
|
|
def test_fetch_raw_message_private_stream(self):
|
|
|
|
# type: () -> None
|
2017-08-25 06:01:29 +02:00
|
|
|
user_profile = self.example_user("hamlet")
|
|
|
|
self.login(user_profile.email)
|
2016-10-21 22:59:28 +02:00
|
|
|
stream = self.make_stream('private_stream', invite_only=True)
|
2017-08-25 06:01:29 +02:00
|
|
|
self.subscribe(user_profile, stream.name)
|
|
|
|
msg_id = self.send_message(user_profile.email, stream.name, Recipient.STREAM,
|
2016-09-22 18:48:42 +02:00
|
|
|
subject="test", content="test")
|
2016-12-21 20:31:16 +01:00
|
|
|
result = self.client_get('/json/messages/' + str(msg_id))
|
2016-09-22 18:48:42 +02:00
|
|
|
self.assert_json_success(result)
|
2017-05-25 02:08:35 +02:00
|
|
|
self.login(self.example_email("othello"))
|
2016-12-21 20:31:16 +01:00
|
|
|
result = self.client_get('/json/messages/' + str(msg_id))
|
2016-10-12 02:14:08 +02:00
|
|
|
self.assert_json_error(result, 'Invalid message(s)')
|
2016-09-19 21:24:49 +02:00
|
|
|
|
2017-02-23 00:27:50 +01:00
|
|
|
def test_edit_message_no_permission(self):
|
|
|
|
# type: () -> None
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2017-05-25 01:44:04 +02:00
|
|
|
msg_id = self.send_message(self.example_email("iago"), "Scotland", Recipient.STREAM,
|
2017-02-23 00:27:50 +01:00
|
|
|
subject="editing", content="before edit")
|
|
|
|
result = self.client_patch("/json/messages/" + str(msg_id), {
|
|
|
|
'message_id': msg_id,
|
|
|
|
'content': 'content after edit',
|
|
|
|
})
|
|
|
|
self.assert_json_error(result, "You don't have permission to edit this message")
|
|
|
|
|
2016-07-11 02:55:57 +02:00
|
|
|
def test_edit_message_no_changes(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
|
|
|
msg_id = self.send_message(self.example_email("hamlet"), "Scotland", Recipient.STREAM,
|
2016-07-11 02:55:57 +02:00
|
|
|
subject="editing", content="before edit")
|
2016-12-22 10:17:49 +01:00
|
|
|
result = self.client_patch("/json/messages/" + str(msg_id), {
|
2016-07-11 02:55:57 +02:00
|
|
|
'message_id': msg_id,
|
|
|
|
})
|
|
|
|
self.assert_json_error(result, "Nothing to change")
|
|
|
|
|
|
|
|
def test_edit_message_no_topic(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
|
|
|
msg_id = self.send_message(self.example_email("hamlet"), "Scotland", Recipient.STREAM,
|
2016-07-11 02:55:57 +02:00
|
|
|
subject="editing", content="before edit")
|
2016-12-22 10:17:49 +01:00
|
|
|
result = self.client_patch("/json/messages/" + str(msg_id), {
|
2016-07-11 02:55:57 +02:00
|
|
|
'message_id': msg_id,
|
|
|
|
'subject': ' '
|
|
|
|
})
|
|
|
|
self.assert_json_error(result, "Topic can't be empty")
|
|
|
|
|
2016-07-11 02:58:23 +02:00
|
|
|
def test_edit_message_no_content(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
|
|
|
msg_id = self.send_message(self.example_email("hamlet"), "Scotland", Recipient.STREAM,
|
2016-07-11 02:58:23 +02:00
|
|
|
subject="editing", content="before edit")
|
2016-12-22 10:17:49 +01:00
|
|
|
result = self.client_patch("/json/messages/" + str(msg_id), {
|
2016-07-11 02:58:23 +02:00
|
|
|
'message_id': msg_id,
|
|
|
|
'content': ' '
|
|
|
|
})
|
2016-11-09 22:16:12 +01:00
|
|
|
self.assert_json_success(result)
|
|
|
|
content = Message.objects.filter(id=msg_id).values_list('content', flat = True)[0]
|
|
|
|
self.assertEqual(content, "(deleted)")
|
2016-07-11 02:58:23 +02:00
|
|
|
|
2017-07-16 11:00:44 +02:00
|
|
|
def test_edit_message_history_disabled(self):
|
|
|
|
# type: () -> None
|
|
|
|
user_profile = self.example_user("hamlet")
|
|
|
|
do_set_realm_property(user_profile.realm, "allow_edit_history", False)
|
|
|
|
self.login(self.example_email("hamlet"))
|
|
|
|
|
|
|
|
# Single-line edit
|
|
|
|
msg_id_1 = self.send_message(self.example_email("hamlet"),
|
|
|
|
"Denmark",
|
|
|
|
Recipient.STREAM,
|
|
|
|
subject="editing",
|
|
|
|
content="content before edit")
|
|
|
|
|
|
|
|
new_content_1 = 'content after edit'
|
|
|
|
result_1 = self.client_patch("/json/messages/" + str(msg_id_1), {
|
|
|
|
'message_id': msg_id_1, 'content': new_content_1
|
|
|
|
})
|
|
|
|
self.assert_json_success(result_1)
|
|
|
|
|
|
|
|
result = self.client_get(
|
|
|
|
"/json/messages/" + str(msg_id_1) + "/history")
|
|
|
|
self.assert_json_error(result, "Message edit history is disabled in this organization")
|
|
|
|
|
|
|
|
# Now verify that if we fetch the message directly, there's no
|
|
|
|
# edit history data attached.
|
|
|
|
messages_result = self.client_get("/json/messages",
|
|
|
|
{"anchor": msg_id_1, "num_before": 0, "num_after": 10})
|
|
|
|
self.assert_json_success(messages_result)
|
|
|
|
json_messages = ujson.loads(
|
|
|
|
messages_result.content.decode('utf-8'))
|
|
|
|
for msg in json_messages['messages']:
|
|
|
|
self.assertNotIn("edit_history", msg)
|
|
|
|
|
2017-02-20 00:19:29 +01:00
|
|
|
def test_edit_message_history(self):
|
|
|
|
# type: () -> None
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2017-02-20 00:19:29 +01:00
|
|
|
|
2017-06-10 01:49:12 +02:00
|
|
|
# Single-line edit
|
|
|
|
msg_id_1 = self.send_message(self.example_email("hamlet"),
|
|
|
|
"Scotland",
|
|
|
|
Recipient.STREAM,
|
|
|
|
subject="editing",
|
|
|
|
content="content before edit")
|
|
|
|
new_content_1 = 'content after edit'
|
|
|
|
result_1 = self.client_patch("/json/messages/" + str(msg_id_1), {
|
|
|
|
'message_id': msg_id_1, 'content': new_content_1
|
2017-02-20 00:19:29 +01:00
|
|
|
})
|
2017-06-10 01:49:12 +02:00
|
|
|
self.assert_json_success(result_1)
|
2017-02-20 00:19:29 +01:00
|
|
|
|
2017-06-10 01:49:12 +02:00
|
|
|
message_edit_history_1 = self.client_get(
|
|
|
|
"/json/messages/" + str(msg_id_1) + "/history")
|
|
|
|
json_response_1 = ujson.loads(
|
|
|
|
message_edit_history_1.content.decode('utf-8'))
|
|
|
|
message_history_1 = json_response_1['message_history']
|
2017-02-20 00:19:29 +01:00
|
|
|
|
|
|
|
# Check content of message after edit.
|
2017-06-10 01:49:12 +02:00
|
|
|
self.assertEqual(message_history_1[0]['rendered_content'],
|
2017-02-20 01:46:07 +01:00
|
|
|
'<p>content before edit</p>')
|
2017-06-10 01:49:12 +02:00
|
|
|
self.assertEqual(message_history_1[1]['rendered_content'],
|
2017-02-20 00:19:29 +01:00
|
|
|
'<p>content after edit</p>')
|
2017-06-10 01:49:12 +02:00
|
|
|
self.assertEqual(message_history_1[1]['content_html_diff'],
|
2017-06-06 01:46:41 +02:00
|
|
|
('<p>content '
|
|
|
|
'<span class="highlight_text_deleted">before</span>'
|
|
|
|
'<span class="highlight_text_inserted">after</span>'
|
|
|
|
' edit</p>'))
|
2017-02-20 00:19:29 +01:00
|
|
|
# Check content of message before edit.
|
2017-06-10 01:49:12 +02:00
|
|
|
self.assertEqual(message_history_1[1]['prev_rendered_content'],
|
2017-02-20 00:19:29 +01:00
|
|
|
'<p>content before edit</p>')
|
|
|
|
|
2017-06-10 01:49:12 +02:00
|
|
|
# Edits on new lines
|
|
|
|
msg_id_2 = self.send_message(self.example_email("hamlet"),
|
|
|
|
"Scotland",
|
|
|
|
Recipient.STREAM,
|
|
|
|
subject="editing",
|
|
|
|
content=('content before edit, line 1\n'
|
|
|
|
'\n'
|
|
|
|
'content before edit, line 3'))
|
|
|
|
new_content_2 = ('content before edit, line 1\n'
|
|
|
|
'content after edit, line 2\n'
|
|
|
|
'content before edit, line 3')
|
|
|
|
result_2 = self.client_patch("/json/messages/" + str(msg_id_2), {
|
|
|
|
'message_id': msg_id_2, 'content': new_content_2
|
|
|
|
})
|
|
|
|
self.assert_json_success(result_2)
|
|
|
|
|
|
|
|
message_edit_history_2 = self.client_get(
|
|
|
|
"/json/messages/" + str(msg_id_2) + "/history")
|
|
|
|
json_response_2 = ujson.loads(
|
|
|
|
message_edit_history_2.content.decode('utf-8'))
|
|
|
|
message_history_2 = json_response_2['message_history']
|
|
|
|
|
|
|
|
self.assertEqual(message_history_2[0]['rendered_content'],
|
|
|
|
('<p>content before edit, line 1</p>\n'
|
|
|
|
'<p>content before edit, line 3</p>'))
|
|
|
|
self.assertEqual(message_history_2[1]['rendered_content'],
|
|
|
|
('<p>content before edit, line 1<br>\n'
|
|
|
|
'content after edit, line 2<br>\n'
|
|
|
|
'content before edit, line 3</p>'))
|
|
|
|
self.assertEqual(message_history_2[1]['content_html_diff'],
|
|
|
|
('<p>content before edit, line 1</p>'
|
|
|
|
'<span class="highlight_text_deleted">\n'
|
|
|
|
'</span><p><span class="highlight_text_inserted">\n'
|
|
|
|
'content after edit, line 2</span><br>'
|
|
|
|
'<span class="highlight_text_inserted">\n'
|
|
|
|
'</span>content before edit, line 3</p>'))
|
|
|
|
self.assertEqual(message_history_2[1]['prev_rendered_content'],
|
|
|
|
('<p>content before edit, line 1</p>\n'
|
|
|
|
'<p>content before edit, line 3</p>'))
|
|
|
|
|
2017-02-20 00:49:34 +01:00
|
|
|
def test_edit_cases(self):
|
|
|
|
# type: () -> None
|
|
|
|
"""This test verifies the accuracy of construction of Zulip's edit
|
|
|
|
history data structures."""
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
2017-05-07 17:21:26 +02:00
|
|
|
hamlet = self.example_user('hamlet')
|
2017-05-25 01:40:26 +02:00
|
|
|
msg_id = self.send_message(self.example_email("hamlet"), "Scotland", Recipient.STREAM,
|
2017-02-20 00:49:34 +01:00
|
|
|
subject="subject 1", content="content 1")
|
|
|
|
result = self.client_patch("/json/messages/" + str(msg_id), {
|
|
|
|
'message_id': msg_id,
|
|
|
|
'content': 'content 2',
|
|
|
|
})
|
|
|
|
self.assert_json_success(result)
|
|
|
|
history = ujson.loads(Message.objects.get(id=msg_id).edit_history)
|
|
|
|
self.assertEqual(history[0]['prev_content'], 'content 1')
|
|
|
|
self.assertEqual(history[0]['user_id'], hamlet.id)
|
|
|
|
self.assertEqual(set(history[0].keys()),
|
|
|
|
{u'timestamp', u'prev_content', u'user_id',
|
|
|
|
u'prev_rendered_content', u'prev_rendered_content_version'})
|
|
|
|
|
|
|
|
result = self.client_patch("/json/messages/" + str(msg_id), {
|
|
|
|
'message_id': msg_id,
|
|
|
|
'subject': 'subject 2',
|
|
|
|
})
|
|
|
|
self.assert_json_success(result)
|
|
|
|
history = ujson.loads(Message.objects.get(id=msg_id).edit_history)
|
|
|
|
self.assertEqual(history[0]['prev_subject'], 'subject 1')
|
|
|
|
self.assertEqual(history[0]['user_id'], hamlet.id)
|
|
|
|
self.assertEqual(set(history[0].keys()), {u'timestamp', u'prev_subject', u'user_id'})
|
|
|
|
|
|
|
|
result = self.client_patch("/json/messages/" + str(msg_id), {
|
|
|
|
'message_id': msg_id,
|
|
|
|
'content': 'content 3',
|
|
|
|
'subject': 'subject 3',
|
|
|
|
})
|
|
|
|
self.assert_json_success(result)
|
|
|
|
history = ujson.loads(Message.objects.get(id=msg_id).edit_history)
|
|
|
|
self.assertEqual(history[0]['prev_content'], 'content 2')
|
|
|
|
self.assertEqual(history[0]['prev_subject'], 'subject 2')
|
|
|
|
self.assertEqual(history[0]['user_id'], hamlet.id)
|
|
|
|
self.assertEqual(set(history[0].keys()),
|
|
|
|
{u'timestamp', u'prev_subject', u'prev_content', u'user_id',
|
|
|
|
u'prev_rendered_content', u'prev_rendered_content_version'})
|
|
|
|
|
|
|
|
result = self.client_patch("/json/messages/" + str(msg_id), {
|
|
|
|
'message_id': msg_id,
|
|
|
|
'content': 'content 4',
|
|
|
|
})
|
|
|
|
self.assert_json_success(result)
|
|
|
|
history = ujson.loads(Message.objects.get(id=msg_id).edit_history)
|
|
|
|
self.assertEqual(history[0]['prev_content'], 'content 3')
|
|
|
|
self.assertEqual(history[0]['user_id'], hamlet.id)
|
|
|
|
|
2017-05-25 01:44:04 +02:00
|
|
|
self.login(self.example_email("iago"))
|
2017-02-20 00:49:34 +01:00
|
|
|
result = self.client_patch("/json/messages/" + str(msg_id), {
|
|
|
|
'message_id': msg_id,
|
|
|
|
'subject': 'subject 4',
|
|
|
|
})
|
|
|
|
self.assert_json_success(result)
|
|
|
|
history = ujson.loads(Message.objects.get(id=msg_id).edit_history)
|
|
|
|
self.assertEqual(history[0]['prev_subject'], 'subject 3')
|
2017-05-07 17:21:26 +02:00
|
|
|
self.assertEqual(history[0]['user_id'], self.example_user('iago').id)
|
2017-02-20 00:49:34 +01:00
|
|
|
|
|
|
|
history = ujson.loads(Message.objects.get(id=msg_id).edit_history)
|
|
|
|
self.assertEqual(history[0]['prev_subject'], 'subject 3')
|
|
|
|
self.assertEqual(history[2]['prev_subject'], 'subject 2')
|
|
|
|
self.assertEqual(history[3]['prev_subject'], 'subject 1')
|
|
|
|
self.assertEqual(history[1]['prev_content'], 'content 3')
|
|
|
|
self.assertEqual(history[2]['prev_content'], 'content 2')
|
|
|
|
self.assertEqual(history[4]['prev_content'], 'content 1')
|
|
|
|
|
2017-02-20 00:19:29 +01:00
|
|
|
# Now, we verify that the edit history data sent back has the
|
|
|
|
# correct filled-out fields
|
|
|
|
message_edit_history = self.client_get("/json/messages/" + str(msg_id) + "/history")
|
|
|
|
|
|
|
|
json_response = ujson.loads(message_edit_history.content.decode('utf-8'))
|
2017-02-20 01:46:07 +01:00
|
|
|
|
|
|
|
# We reverse the message history view output so that the IDs line up with the above.
|
|
|
|
message_history = list(reversed(json_response['message_history']))
|
2017-02-20 00:19:29 +01:00
|
|
|
i = 0
|
|
|
|
for entry in message_history:
|
|
|
|
expected_entries = {u'content', u'rendered_content', u'topic', u'timestamp', u'user_id'}
|
|
|
|
if i in {0, 2, 3}:
|
|
|
|
expected_entries.add('prev_topic')
|
|
|
|
if i in {1, 2, 4}:
|
|
|
|
expected_entries.add('prev_content')
|
|
|
|
expected_entries.add('prev_rendered_content')
|
|
|
|
expected_entries.add('content_html_diff')
|
|
|
|
i += 1
|
|
|
|
self.assertEqual(expected_entries, set(entry.keys()))
|
2017-02-20 01:44:12 +01:00
|
|
|
self.assertEqual(len(message_history), 6)
|
2017-02-20 00:19:29 +01:00
|
|
|
self.assertEqual(message_history[0]['prev_topic'], 'subject 3')
|
|
|
|
self.assertEqual(message_history[0]['topic'], 'subject 4')
|
|
|
|
self.assertEqual(message_history[1]['topic'], 'subject 3')
|
|
|
|
self.assertEqual(message_history[2]['topic'], 'subject 3')
|
|
|
|
self.assertEqual(message_history[2]['prev_topic'], 'subject 2')
|
|
|
|
self.assertEqual(message_history[3]['topic'], 'subject 2')
|
|
|
|
self.assertEqual(message_history[3]['prev_topic'], 'subject 1')
|
|
|
|
self.assertEqual(message_history[4]['topic'], 'subject 1')
|
|
|
|
|
|
|
|
self.assertEqual(message_history[0]['content'], 'content 4')
|
|
|
|
self.assertEqual(message_history[1]['content'], 'content 4')
|
|
|
|
self.assertEqual(message_history[1]['prev_content'], 'content 3')
|
|
|
|
self.assertEqual(message_history[2]['content'], 'content 3')
|
|
|
|
self.assertEqual(message_history[2]['prev_content'], 'content 2')
|
|
|
|
self.assertEqual(message_history[3]['content'], 'content 2')
|
|
|
|
self.assertEqual(message_history[4]['content'], 'content 2')
|
|
|
|
self.assertEqual(message_history[4]['prev_content'], 'content 1')
|
|
|
|
|
2017-02-20 01:44:12 +01:00
|
|
|
self.assertEqual(message_history[5]['content'], 'content 1')
|
|
|
|
self.assertEqual(message_history[5]['topic'], 'subject 1')
|
|
|
|
|
2016-07-08 02:25:55 +02:00
|
|
|
def test_edit_message_content_limit(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2016-07-08 02:25:55 +02:00
|
|
|
def set_message_editing_params(allow_message_editing,
|
|
|
|
message_content_edit_limit_seconds):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: (bool, int) -> None
|
2016-07-08 02:25:55 +02:00
|
|
|
result = self.client_patch("/json/realm", {
|
|
|
|
'allow_message_editing': ujson.dumps(allow_message_editing),
|
|
|
|
'message_content_edit_limit_seconds': message_content_edit_limit_seconds
|
|
|
|
})
|
|
|
|
self.assert_json_success(result)
|
|
|
|
|
|
|
|
def do_edit_message_assert_success(id_, unique_str, topic_only = False):
|
2016-12-04 18:38:56 +01:00
|
|
|
# type: (int, Text, bool) -> None
|
2016-07-08 02:25:55 +02:00
|
|
|
new_subject = 'subject' + unique_str
|
|
|
|
new_content = 'content' + unique_str
|
2016-12-02 06:22:52 +01:00
|
|
|
params_dict = {'message_id': id_, 'subject': new_subject}
|
2016-07-08 02:25:55 +02:00
|
|
|
if not topic_only:
|
|
|
|
params_dict['content'] = new_content
|
2016-12-22 10:17:49 +01:00
|
|
|
result = self.client_patch("/json/messages/" + str(id_), params_dict)
|
2016-07-08 02:25:55 +02:00
|
|
|
self.assert_json_success(result)
|
|
|
|
if topic_only:
|
|
|
|
self.check_message(id_, subject=new_subject)
|
|
|
|
else:
|
|
|
|
self.check_message(id_, subject=new_subject, content=new_content)
|
|
|
|
|
|
|
|
def do_edit_message_assert_error(id_, unique_str, error, topic_only = False):
|
2016-12-04 18:38:56 +01:00
|
|
|
# type: (int, Text, Text, bool) -> None
|
2016-07-08 02:25:55 +02:00
|
|
|
message = Message.objects.get(id=id_)
|
2016-07-16 18:18:08 +02:00
|
|
|
old_subject = message.topic_name()
|
2016-07-08 02:25:55 +02:00
|
|
|
old_content = message.content
|
|
|
|
new_subject = 'subject' + unique_str
|
|
|
|
new_content = 'content' + unique_str
|
2016-12-02 06:22:52 +01:00
|
|
|
params_dict = {'message_id': id_, 'subject': new_subject}
|
2016-07-08 02:25:55 +02:00
|
|
|
if not topic_only:
|
|
|
|
params_dict['content'] = new_content
|
2016-12-22 10:17:49 +01:00
|
|
|
result = self.client_patch("/json/messages/" + str(id_), params_dict)
|
2016-07-08 02:25:55 +02:00
|
|
|
message = Message.objects.get(id=id_)
|
|
|
|
self.assert_json_error(result, error)
|
|
|
|
self.check_message(id_, subject=old_subject, content=old_content)
|
|
|
|
|
2017-05-25 01:44:04 +02:00
|
|
|
self.login(self.example_email("iago"))
|
2016-07-08 02:25:55 +02:00
|
|
|
# send a message in the past
|
2017-05-25 01:44:04 +02:00
|
|
|
id_ = self.send_message(self.example_email("iago"), "Scotland", Recipient.STREAM,
|
2016-07-08 02:25:55 +02:00
|
|
|
content="content", subject="subject")
|
|
|
|
message = Message.objects.get(id=id_)
|
|
|
|
message.pub_date = message.pub_date - datetime.timedelta(seconds=180)
|
|
|
|
message.save()
|
|
|
|
|
|
|
|
# test the various possible message editing settings
|
|
|
|
# high enough time limit, all edits allowed
|
|
|
|
set_message_editing_params(True, 240)
|
|
|
|
do_edit_message_assert_success(id_, 'A')
|
|
|
|
|
|
|
|
# out of time, only topic editing allowed
|
|
|
|
set_message_editing_params(True, 120)
|
|
|
|
do_edit_message_assert_success(id_, 'B', True)
|
|
|
|
do_edit_message_assert_error(id_, 'C', "The time limit for editing this message has past")
|
|
|
|
|
|
|
|
# infinite time, all edits allowed
|
|
|
|
set_message_editing_params(True, 0)
|
|
|
|
do_edit_message_assert_success(id_, 'D')
|
|
|
|
|
|
|
|
# without allow_message_editing, nothing is allowed
|
|
|
|
set_message_editing_params(False, 240)
|
2017-05-24 00:27:16 +02:00
|
|
|
do_edit_message_assert_error(id_, 'E', "Your organization has turned off message editing", True)
|
2016-07-08 02:25:55 +02:00
|
|
|
set_message_editing_params(False, 120)
|
2017-05-24 00:27:16 +02:00
|
|
|
do_edit_message_assert_error(id_, 'F', "Your organization has turned off message editing", True)
|
2016-07-08 02:25:55 +02:00
|
|
|
set_message_editing_params(False, 0)
|
2017-05-24 00:27:16 +02:00
|
|
|
do_edit_message_assert_error(id_, 'G', "Your organization has turned off message editing", True)
|
2016-07-08 02:25:55 +02:00
|
|
|
|
2014-01-31 16:44:45 +01:00
|
|
|
def test_propagate_topic_forward(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
|
|
|
id1 = self.send_message(self.example_email("hamlet"), "Scotland", Recipient.STREAM,
|
2016-12-03 00:04:17 +01:00
|
|
|
subject="topic1")
|
2017-05-25 01:44:04 +02:00
|
|
|
id2 = self.send_message(self.example_email("iago"), "Scotland", Recipient.STREAM,
|
2016-12-03 00:04:17 +01:00
|
|
|
subject="topic1")
|
2017-05-25 01:44:04 +02:00
|
|
|
id3 = self.send_message(self.example_email("iago"), "Rome", Recipient.STREAM,
|
2016-12-03 00:04:17 +01:00
|
|
|
subject="topic1")
|
2017-05-25 01:40:26 +02:00
|
|
|
id4 = self.send_message(self.example_email("hamlet"), "Scotland", Recipient.STREAM,
|
2016-12-03 00:04:17 +01:00
|
|
|
subject="topic2")
|
2017-05-25 01:44:04 +02:00
|
|
|
id5 = self.send_message(self.example_email("iago"), "Scotland", Recipient.STREAM,
|
2016-12-03 00:04:17 +01:00
|
|
|
subject="topic1")
|
2014-01-31 16:44:45 +01:00
|
|
|
|
2016-12-22 10:17:49 +01:00
|
|
|
result = self.client_patch("/json/messages/" + str(id1), {
|
2014-01-31 16:44:45 +01:00
|
|
|
'message_id': id1,
|
|
|
|
'subject': 'edited',
|
|
|
|
'propagate_mode': 'change_later'
|
|
|
|
})
|
|
|
|
self.assert_json_success(result)
|
|
|
|
|
|
|
|
self.check_message(id1, subject="edited")
|
|
|
|
self.check_message(id2, subject="edited")
|
|
|
|
self.check_message(id3, subject="topic1")
|
|
|
|
self.check_message(id4, subject="topic2")
|
|
|
|
self.check_message(id5, subject="edited")
|
|
|
|
|
|
|
|
def test_propagate_all_topics(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
|
|
|
id1 = self.send_message(self.example_email("hamlet"), "Scotland", Recipient.STREAM,
|
2016-12-03 00:04:17 +01:00
|
|
|
subject="topic1")
|
2017-05-25 01:40:26 +02:00
|
|
|
id2 = self.send_message(self.example_email("hamlet"), "Scotland", Recipient.STREAM,
|
2016-12-03 00:04:17 +01:00
|
|
|
subject="topic1")
|
2017-05-25 01:44:04 +02:00
|
|
|
id3 = self.send_message(self.example_email("iago"), "Rome", Recipient.STREAM,
|
2016-12-03 00:04:17 +01:00
|
|
|
subject="topic1")
|
2017-05-25 01:40:26 +02:00
|
|
|
id4 = self.send_message(self.example_email("hamlet"), "Scotland", Recipient.STREAM,
|
2016-12-03 00:04:17 +01:00
|
|
|
subject="topic2")
|
2017-05-25 01:44:04 +02:00
|
|
|
id5 = self.send_message(self.example_email("iago"), "Scotland", Recipient.STREAM,
|
2016-12-03 00:04:17 +01:00
|
|
|
subject="topic1")
|
2017-05-25 01:44:04 +02:00
|
|
|
id6 = self.send_message(self.example_email("iago"), "Scotland", Recipient.STREAM,
|
2016-12-03 00:04:17 +01:00
|
|
|
subject="topic3")
|
2014-01-31 16:44:45 +01:00
|
|
|
|
2016-12-22 10:17:49 +01:00
|
|
|
result = self.client_patch("/json/messages/" + str(id2), {
|
2014-01-31 16:44:45 +01:00
|
|
|
'message_id': id2,
|
|
|
|
'subject': 'edited',
|
|
|
|
'propagate_mode': 'change_all'
|
|
|
|
})
|
|
|
|
self.assert_json_success(result)
|
|
|
|
|
|
|
|
self.check_message(id1, subject="edited")
|
|
|
|
self.check_message(id2, subject="edited")
|
|
|
|
self.check_message(id3, subject="topic1")
|
|
|
|
self.check_message(id4, subject="topic2")
|
|
|
|
self.check_message(id5, subject="edited")
|
|
|
|
self.check_message(id6, subject="topic3")
|
|
|
|
|
2017-05-07 17:21:26 +02:00
|
|
|
class MirroredMessageUsersTest(ZulipTestCase):
|
2016-07-27 03:31:17 +02:00
|
|
|
def test_invalid_sender(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2017-05-07 17:21:26 +02:00
|
|
|
user = self.example_user('hamlet')
|
2017-07-11 21:33:49 +02:00
|
|
|
recipients = [] # type: List[Text]
|
2017-08-05 22:10:34 +02:00
|
|
|
|
|
|
|
Request = namedtuple('Request', ['POST'])
|
|
|
|
request = Request(POST=dict()) # no sender
|
2016-07-27 03:31:17 +02:00
|
|
|
|
|
|
|
(valid_input, mirror_sender) = \
|
|
|
|
create_mirrored_message_users(request, user, recipients)
|
|
|
|
|
|
|
|
self.assertEqual(valid_input, False)
|
|
|
|
self.assertEqual(mirror_sender, None)
|
|
|
|
|
|
|
|
def test_invalid_client(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2017-07-11 21:33:49 +02:00
|
|
|
client = get_client(name='banned_mirror') # Invalid!!!
|
2016-07-27 03:31:17 +02:00
|
|
|
|
2017-05-07 17:21:26 +02:00
|
|
|
user = self.example_user('hamlet')
|
2016-07-27 03:31:17 +02:00
|
|
|
sender = user
|
|
|
|
|
2017-07-11 21:33:49 +02:00
|
|
|
recipients = [] # type: List[Text]
|
2017-08-05 22:10:34 +02:00
|
|
|
|
|
|
|
Request = namedtuple('Request', ['POST', 'client'])
|
|
|
|
request = Request(POST = dict(sender=sender.email, type='private'),
|
|
|
|
client = client)
|
2016-07-27 03:31:17 +02:00
|
|
|
|
|
|
|
(valid_input, mirror_sender) = \
|
|
|
|
create_mirrored_message_users(request, user, recipients)
|
|
|
|
|
|
|
|
self.assertEqual(valid_input, False)
|
|
|
|
self.assertEqual(mirror_sender, None)
|
|
|
|
|
|
|
|
def test_invalid_email(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2016-07-27 03:31:17 +02:00
|
|
|
invalid_email = 'alice AT example.com'
|
|
|
|
recipients = [invalid_email]
|
|
|
|
|
|
|
|
# We use an MIT user here to maximize code coverage
|
2017-05-23 02:33:53 +02:00
|
|
|
user = self.mit_user('starnine')
|
2016-07-27 03:31:17 +02:00
|
|
|
sender = user
|
|
|
|
|
2017-08-05 22:10:34 +02:00
|
|
|
Request = namedtuple('Request', ['POST', 'client'])
|
|
|
|
|
2016-07-27 03:31:17 +02:00
|
|
|
for client_name in ['zephyr_mirror', 'irc_mirror', 'jabber_mirror']:
|
|
|
|
client = get_client(name=client_name)
|
|
|
|
|
2017-08-05 22:10:34 +02:00
|
|
|
request = Request(POST = dict(sender=sender.email, type='private'),
|
|
|
|
client = client)
|
2016-07-27 03:31:17 +02:00
|
|
|
|
|
|
|
(valid_input, mirror_sender) = \
|
|
|
|
create_mirrored_message_users(request, user, recipients)
|
|
|
|
|
|
|
|
self.assertEqual(valid_input, False)
|
|
|
|
self.assertEqual(mirror_sender, None)
|
|
|
|
|
2016-11-03 21:24:54 +01:00
|
|
|
@mock.patch('DNS.dnslookup', return_value=[['sipbtest:*:20922:101:Fred Sipb,,,:/mit/sipbtest:/bin/athena/tcsh']])
|
|
|
|
def test_zephyr_mirror_new_recipient(self, ignored):
|
|
|
|
# type: (Any) -> None
|
2016-07-27 03:31:17 +02:00
|
|
|
"""Test mirror dummy user creation for PM recipients"""
|
|
|
|
client = get_client(name='zephyr_mirror')
|
|
|
|
|
2017-05-23 02:33:53 +02:00
|
|
|
user = self.mit_user('starnine')
|
2017-05-23 01:27:31 +02:00
|
|
|
sender = self.mit_user('sipbtest')
|
2016-07-27 03:31:17 +02:00
|
|
|
new_user_email = 'bob_the_new_user@mit.edu'
|
2017-05-24 02:42:31 +02:00
|
|
|
new_user_realm = get_realm("zephyr")
|
2016-07-27 03:31:17 +02:00
|
|
|
|
|
|
|
recipients = [user.email, new_user_email]
|
|
|
|
|
|
|
|
# Now make the request.
|
2017-08-05 22:10:34 +02:00
|
|
|
Request = namedtuple('Request', ['POST', 'client'])
|
|
|
|
request = Request(POST = dict(sender=sender.email, type='private'),
|
|
|
|
client = client)
|
2016-07-27 03:31:17 +02:00
|
|
|
|
|
|
|
(valid_input, mirror_sender) = \
|
|
|
|
create_mirrored_message_users(request, user, recipients)
|
|
|
|
|
|
|
|
self.assertTrue(valid_input)
|
|
|
|
self.assertEqual(mirror_sender, sender)
|
|
|
|
|
|
|
|
realm_users = UserProfile.objects.filter(realm=sender.realm)
|
|
|
|
realm_emails = {user.email for user in realm_users}
|
|
|
|
self.assertIn(user.email, realm_emails)
|
|
|
|
self.assertIn(new_user_email, realm_emails)
|
|
|
|
|
2017-05-24 02:42:31 +02:00
|
|
|
bob = get_user(new_user_email, new_user_realm)
|
2016-07-27 03:31:17 +02:00
|
|
|
self.assertTrue(bob.is_mirror_dummy)
|
|
|
|
|
2016-11-03 21:24:54 +01:00
|
|
|
@mock.patch('DNS.dnslookup', return_value=[['sipbtest:*:20922:101:Fred Sipb,,,:/mit/sipbtest:/bin/athena/tcsh']])
|
|
|
|
def test_zephyr_mirror_new_sender(self, ignored):
|
|
|
|
# type: (Any) -> None
|
2016-07-27 03:31:17 +02:00
|
|
|
"""Test mirror dummy user creation for sender when sending to stream"""
|
|
|
|
client = get_client(name='zephyr_mirror')
|
|
|
|
|
2017-05-23 02:33:53 +02:00
|
|
|
user = self.mit_user('starnine')
|
2016-07-27 03:31:17 +02:00
|
|
|
sender_email = 'new_sender@mit.edu'
|
|
|
|
|
|
|
|
recipients = ['stream_name']
|
|
|
|
|
|
|
|
# Now make the request.
|
2017-08-05 22:10:34 +02:00
|
|
|
Request = namedtuple('Request', ['POST', 'client'])
|
|
|
|
request = Request(POST = dict(sender=sender_email, type='stream'),
|
|
|
|
client = client)
|
2016-07-27 03:31:17 +02:00
|
|
|
|
|
|
|
(valid_input, mirror_sender) = \
|
|
|
|
create_mirrored_message_users(request, user, recipients)
|
|
|
|
|
2017-05-24 04:21:29 +02:00
|
|
|
assert(mirror_sender is not None)
|
2016-07-27 03:31:17 +02:00
|
|
|
self.assertTrue(valid_input)
|
|
|
|
self.assertEqual(mirror_sender.email, sender_email)
|
|
|
|
self.assertTrue(mirror_sender.is_mirror_dummy)
|
|
|
|
|
|
|
|
def test_irc_mirror(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2016-07-27 03:31:17 +02:00
|
|
|
client = get_client(name='irc_mirror')
|
|
|
|
|
2017-05-07 17:21:26 +02:00
|
|
|
sender = self.example_user('hamlet')
|
2016-07-27 03:31:17 +02:00
|
|
|
user = sender
|
|
|
|
|
2017-05-24 02:42:31 +02:00
|
|
|
recipients = [self.nonreg_email('alice'), 'bob@irc.zulip.com', self.nonreg_email('cordelia')]
|
2016-07-27 03:31:17 +02:00
|
|
|
|
|
|
|
# Now make the request.
|
2017-08-05 22:10:34 +02:00
|
|
|
Request = namedtuple('Request', ['POST', 'client'])
|
|
|
|
request = Request(POST = dict(sender=sender.email, type='private'),
|
|
|
|
client = client)
|
2016-07-27 03:31:17 +02:00
|
|
|
|
|
|
|
(valid_input, mirror_sender) = \
|
|
|
|
create_mirrored_message_users(request, user, recipients)
|
|
|
|
|
|
|
|
self.assertEqual(valid_input, True)
|
|
|
|
self.assertEqual(mirror_sender, sender)
|
|
|
|
|
|
|
|
realm_users = UserProfile.objects.filter(realm=sender.realm)
|
|
|
|
realm_emails = {user.email for user in realm_users}
|
2017-05-24 02:42:31 +02:00
|
|
|
self.assertIn(self.nonreg_email('alice'), realm_emails)
|
2016-07-27 03:31:17 +02:00
|
|
|
self.assertIn('bob@irc.zulip.com', realm_emails)
|
|
|
|
|
2017-05-24 02:42:31 +02:00
|
|
|
bob = get_user('bob@irc.zulip.com', sender.realm)
|
2016-07-27 03:31:17 +02:00
|
|
|
self.assertTrue(bob.is_mirror_dummy)
|
|
|
|
|
|
|
|
def test_jabber_mirror(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2016-07-27 03:31:17 +02:00
|
|
|
client = get_client(name='jabber_mirror')
|
|
|
|
|
2017-05-07 17:21:26 +02:00
|
|
|
sender = self.example_user('hamlet')
|
2016-07-27 03:31:17 +02:00
|
|
|
user = sender
|
|
|
|
|
2017-05-24 02:42:31 +02:00
|
|
|
recipients = [self.nonreg_email('alice'), self.nonreg_email('bob'), self.nonreg_email('cordelia')]
|
2016-07-27 03:31:17 +02:00
|
|
|
|
|
|
|
# Now make the request.
|
2017-08-05 22:10:34 +02:00
|
|
|
Request = namedtuple('Request', ['POST', 'client'])
|
|
|
|
request = Request(POST = dict(sender=sender.email, type='private'),
|
|
|
|
client = client)
|
2016-07-27 03:31:17 +02:00
|
|
|
|
|
|
|
(valid_input, mirror_sender) = \
|
|
|
|
create_mirrored_message_users(request, user, recipients)
|
|
|
|
|
|
|
|
self.assertEqual(valid_input, True)
|
|
|
|
self.assertEqual(mirror_sender, sender)
|
|
|
|
|
|
|
|
realm_users = UserProfile.objects.filter(realm=sender.realm)
|
|
|
|
realm_emails = {user.email for user in realm_users}
|
2017-05-24 02:42:31 +02:00
|
|
|
self.assertIn(self.nonreg_email('alice'), realm_emails)
|
|
|
|
self.assertIn(self.nonreg_email('bob'), realm_emails)
|
2016-07-27 03:31:17 +02:00
|
|
|
|
2017-05-24 02:42:31 +02:00
|
|
|
bob = get_user(self.nonreg_email('bob'), sender.realm)
|
2016-07-27 03:31:17 +02:00
|
|
|
self.assertTrue(bob.is_mirror_dummy)
|
|
|
|
|
2016-08-23 02:08:42 +02:00
|
|
|
class StarTests(ZulipTestCase):
|
2014-01-31 16:44:45 +01:00
|
|
|
|
2017-08-26 01:08:14 +02:00
|
|
|
def change_star(self, messages, add=True, **kwargs):
|
|
|
|
# type: (List[int], bool, **Any) -> HttpResponse
|
2016-07-28 00:30:22 +02:00
|
|
|
return self.client_post("/json/messages/flags",
|
2014-01-31 16:44:45 +01:00
|
|
|
{"messages": ujson.dumps(messages),
|
|
|
|
"op": "add" if add else "remove",
|
2017-08-26 01:08:14 +02:00
|
|
|
"flag": "starred"},
|
|
|
|
**kwargs)
|
2014-01-31 16:44:45 +01:00
|
|
|
|
|
|
|
def test_change_star(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
You can set a message as starred/un-starred through
|
2016-04-01 12:16:09 +02:00
|
|
|
POST /json/messages/flags.
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
|
|
|
message_ids = [self.send_message(self.example_email("hamlet"), self.example_email("hamlet"),
|
2014-01-31 16:44:45 +01:00
|
|
|
Recipient.PERSONAL, "test")]
|
|
|
|
|
|
|
|
# Star a message.
|
|
|
|
result = self.change_star(message_ids)
|
|
|
|
self.assert_json_success(result)
|
|
|
|
|
2017-03-24 07:51:46 +01:00
|
|
|
for msg in self.get_messages():
|
2014-01-31 16:44:45 +01:00
|
|
|
if msg['id'] in message_ids:
|
|
|
|
self.assertEqual(msg['flags'], ['starred'])
|
|
|
|
else:
|
|
|
|
self.assertEqual(msg['flags'], ['read'])
|
|
|
|
|
|
|
|
result = self.change_star(message_ids, False)
|
|
|
|
self.assert_json_success(result)
|
|
|
|
|
|
|
|
# Remove the stars.
|
2017-03-24 07:51:46 +01:00
|
|
|
for msg in self.get_messages():
|
2014-01-31 16:44:45 +01:00
|
|
|
if msg['id'] in message_ids:
|
|
|
|
self.assertEqual(msg['flags'], [])
|
|
|
|
|
2016-10-12 02:35:13 +02:00
|
|
|
def test_change_star_public_stream_historical(self):
|
|
|
|
# type: () -> None
|
|
|
|
"""
|
|
|
|
You can set a message as starred/un-starred through
|
|
|
|
POST /json/messages/flags.
|
|
|
|
"""
|
|
|
|
stream_name = "new_stream"
|
2017-08-25 06:01:29 +02:00
|
|
|
self.subscribe(self.example_user("hamlet"), stream_name)
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
|
|
|
message_ids = [self.send_message(self.example_email("hamlet"), stream_name,
|
2016-10-12 02:35:13 +02:00
|
|
|
Recipient.STREAM, "test")]
|
2017-03-08 05:43:36 +01:00
|
|
|
# Send a second message so we can verify it isn't modified
|
2017-05-25 01:40:26 +02:00
|
|
|
other_message_ids = [self.send_message(self.example_email("hamlet"), stream_name,
|
2017-03-10 06:42:03 +01:00
|
|
|
Recipient.STREAM, "test_unused")]
|
2017-05-25 01:50:35 +02:00
|
|
|
received_message_ids = [self.send_message(self.example_email("hamlet"), [self.example_email("cordelia")],
|
2017-03-28 20:41:42 +02:00
|
|
|
Recipient.PERSONAL, "test_received")]
|
2016-10-12 02:35:13 +02:00
|
|
|
|
|
|
|
# Now login as another user who wasn't on that stream
|
2017-05-25 01:50:35 +02:00
|
|
|
self.login(self.example_email("cordelia"))
|
2017-03-28 20:41:42 +02:00
|
|
|
# Send a message to yourself to make sure we have at least one with the read flag
|
2017-05-25 01:50:35 +02:00
|
|
|
sent_message_ids = [self.send_message(self.example_email("cordelia"), [self.example_email("cordelia")],
|
2017-03-28 20:41:42 +02:00
|
|
|
Recipient.PERSONAL, "test_read_message")]
|
|
|
|
result = self.client_post("/json/messages/flags",
|
|
|
|
{"messages": ujson.dumps(sent_message_ids),
|
|
|
|
"op": "add",
|
|
|
|
"flag": "read"})
|
2016-10-12 02:35:13 +02:00
|
|
|
|
|
|
|
# We can't change flags other than "starred" on historical messages:
|
|
|
|
result = self.client_post("/json/messages/flags",
|
|
|
|
{"messages": ujson.dumps(message_ids),
|
|
|
|
"op": "add",
|
|
|
|
"flag": "read"})
|
|
|
|
self.assert_json_error(result, 'Invalid message(s)')
|
|
|
|
|
|
|
|
# Trying to change a list of more than one historical message fails
|
|
|
|
result = self.change_star(message_ids * 2)
|
|
|
|
self.assert_json_error(result, 'Invalid message(s)')
|
|
|
|
|
|
|
|
# Confirm that one can change the historical flag now
|
|
|
|
result = self.change_star(message_ids)
|
|
|
|
self.assert_json_success(result)
|
|
|
|
|
2017-03-24 07:51:46 +01:00
|
|
|
for msg in self.get_messages():
|
2017-03-28 20:41:42 +02:00
|
|
|
if msg['id'] in message_ids:
|
2016-10-12 02:35:13 +02:00
|
|
|
self.assertEqual(set(msg['flags']), {'starred', 'historical', 'read'})
|
2017-03-28 20:41:42 +02:00
|
|
|
elif msg['id'] in received_message_ids:
|
|
|
|
self.assertEqual(msg['flags'], [])
|
2016-10-12 02:35:13 +02:00
|
|
|
else:
|
|
|
|
self.assertEqual(msg['flags'], ['read'])
|
2017-03-28 20:41:42 +02:00
|
|
|
self.assertNotIn(msg['id'], other_message_ids)
|
2016-10-12 02:35:13 +02:00
|
|
|
|
|
|
|
result = self.change_star(message_ids, False)
|
|
|
|
self.assert_json_success(result)
|
|
|
|
|
|
|
|
# But it still doesn't work if you're in another realm
|
2017-05-24 21:21:35 +02:00
|
|
|
self.login(self.mit_email("sipbtest"))
|
2017-08-26 01:08:14 +02:00
|
|
|
result = self.change_star(message_ids, subdomain="zephyr")
|
2016-10-12 02:35:13 +02:00
|
|
|
self.assert_json_error(result, 'Invalid message(s)')
|
|
|
|
|
|
|
|
def test_change_star_private_message_security(self):
|
|
|
|
# type: () -> None
|
|
|
|
"""
|
|
|
|
You can set a message as starred/un-starred through
|
|
|
|
POST /json/messages/flags.
|
|
|
|
"""
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
|
|
|
message_ids = [self.send_message(self.example_email("hamlet"), self.example_email("hamlet"),
|
2016-10-12 02:35:13 +02:00
|
|
|
Recipient.PERSONAL, "test")]
|
|
|
|
|
|
|
|
# Starring private messages you didn't receive fails.
|
2017-05-25 01:50:35 +02:00
|
|
|
self.login(self.example_email("cordelia"))
|
2016-10-12 02:35:13 +02:00
|
|
|
result = self.change_star(message_ids)
|
|
|
|
self.assert_json_error(result, 'Invalid message(s)')
|
|
|
|
|
|
|
|
def test_change_star_private_stream_security(self):
|
|
|
|
# type: () -> None
|
|
|
|
stream_name = "private_stream"
|
2016-10-21 22:59:28 +02:00
|
|
|
self.make_stream(stream_name, invite_only=True)
|
2017-08-25 06:01:29 +02:00
|
|
|
self.subscribe(self.example_user("hamlet"), stream_name)
|
2017-05-25 01:40:26 +02:00
|
|
|
self.login(self.example_email("hamlet"))
|
|
|
|
message_ids = [self.send_message(self.example_email("hamlet"), stream_name,
|
2016-10-12 02:35:13 +02:00
|
|
|
Recipient.STREAM, "test")]
|
|
|
|
|
|
|
|
# Starring private stream messages you received works
|
|
|
|
result = self.change_star(message_ids)
|
|
|
|
self.assert_json_success(result)
|
|
|
|
|
|
|
|
# Starring private stream messages you didn't receive fails.
|
2017-05-25 01:50:35 +02:00
|
|
|
self.login(self.example_email("cordelia"))
|
2016-10-12 02:35:13 +02:00
|
|
|
result = self.change_star(message_ids)
|
|
|
|
self.assert_json_error(result, 'Invalid message(s)')
|
|
|
|
|
2014-01-31 16:44:45 +01:00
|
|
|
def test_new_message(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-01-31 16:44:45 +01:00
|
|
|
"""
|
|
|
|
New messages aren't starred.
|
|
|
|
"""
|
2017-05-24 02:42:31 +02:00
|
|
|
test_email = self.example_email('hamlet')
|
2014-01-31 16:44:45 +01:00
|
|
|
self.login(test_email)
|
|
|
|
content = "Test message for star"
|
|
|
|
self.send_message(test_email, "Verona", Recipient.STREAM,
|
|
|
|
content=content)
|
|
|
|
|
|
|
|
sent_message = UserMessage.objects.filter(
|
2017-05-24 02:42:31 +02:00
|
|
|
user_profile=self.example_user('hamlet')
|
2017-01-24 06:34:26 +01:00
|
|
|
).order_by("id").reverse()[0]
|
2014-01-31 16:44:45 +01:00
|
|
|
self.assertEqual(sent_message.message.content, content)
|
|
|
|
self.assertFalse(sent_message.flags.starred)
|
|
|
|
|
2016-08-23 02:08:42 +02:00
|
|
|
class AttachmentTest(ZulipTestCase):
|
2014-02-21 21:18:38 +01:00
|
|
|
def test_basics(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2014-02-21 21:18:38 +01:00
|
|
|
self.assertFalse(Message.content_has_attachment('whatever'))
|
|
|
|
self.assertFalse(Message.content_has_attachment('yo http://foo.com'))
|
|
|
|
self.assertTrue(Message.content_has_attachment('yo\n https://staging.zulip.com/user_uploads/'))
|
|
|
|
self.assertTrue(Message.content_has_attachment('yo\n /user_uploads/1/wEAnI-PEmVmCjo15xxNaQbnj/photo-10.jpg foo'))
|
|
|
|
|
|
|
|
self.assertFalse(Message.content_has_image('whatever'))
|
|
|
|
self.assertFalse(Message.content_has_image('yo http://foo.com'))
|
|
|
|
self.assertFalse(Message.content_has_image('yo\n /user_uploads/1/wEAnI-PEmVmCjo15xxNaQbnj/photo-10.pdf foo'))
|
|
|
|
for ext in [".bmp", ".gif", ".jpg", "jpeg", ".png", ".webp", ".JPG"]:
|
|
|
|
content = 'yo\n /user_uploads/1/wEAnI-PEmVmCjo15xxNaQbnj/photo-10.%s foo' % (ext,)
|
|
|
|
self.assertTrue(Message.content_has_image(content))
|
|
|
|
|
|
|
|
self.assertFalse(Message.content_has_link('whatever'))
|
|
|
|
self.assertTrue(Message.content_has_link('yo\n http://foo.com'))
|
|
|
|
self.assertTrue(Message.content_has_link('yo\n https://example.com?spam=1&eggs=2'))
|
|
|
|
self.assertTrue(Message.content_has_link('yo /user_uploads/1/wEAnI-PEmVmCjo15xxNaQbnj/photo-10.pdf foo'))
|
|
|
|
|
2016-03-24 20:24:01 +01:00
|
|
|
def test_claim_attachment(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
|
|
|
|
2016-03-24 20:24:01 +01:00
|
|
|
# Create dummy DB entry
|
2017-05-07 21:25:59 +02:00
|
|
|
user_profile = self.example_user('hamlet')
|
2017-02-26 11:03:45 +01:00
|
|
|
sample_size = 10
|
2016-03-24 20:24:01 +01:00
|
|
|
dummy_files = [
|
2017-02-26 11:03:45 +01:00
|
|
|
('zulip.txt', '1/31/4CBjtTLYZhk66pZrF8hnYGwc/zulip.txt', sample_size),
|
|
|
|
('temp_file.py', '1/31/4CBjtTLYZhk66pZrF8hnYGwc/temp_file.py', sample_size),
|
|
|
|
('abc.py', '1/31/4CBjtTLYZhk66pZrF8hnYGwc/abc.py', sample_size)
|
2017-01-24 07:06:13 +01:00
|
|
|
]
|
2016-03-24 20:24:01 +01:00
|
|
|
|
2017-02-26 11:03:45 +01:00
|
|
|
for file_name, path_id, size in dummy_files:
|
|
|
|
create_attachment(file_name, path_id, user_profile, size)
|
2016-03-24 20:24:01 +01:00
|
|
|
|
|
|
|
# Send message referring the attachment
|
2017-08-25 06:01:29 +02:00
|
|
|
self.subscribe(user_profile, "Denmark")
|
2016-03-24 20:24:01 +01:00
|
|
|
|
|
|
|
body = "Some files here ...[zulip.txt](http://localhost:9991/user_uploads/1/31/4CBjtTLYZhk66pZrF8hnYGwc/zulip.txt)" + \
|
2016-12-02 08:15:16 +01:00
|
|
|
"http://localhost:9991/user_uploads/1/31/4CBjtTLYZhk66pZrF8hnYGwc/temp_file.py.... Some more...." + \
|
|
|
|
"http://localhost:9991/user_uploads/1/31/4CBjtTLYZhk66pZrF8hnYGwc/abc.py"
|
2016-03-24 20:24:01 +01:00
|
|
|
|
2017-08-25 06:01:29 +02:00
|
|
|
self.send_message(user_profile.email, "Denmark", Recipient.STREAM, body, "test")
|
2016-03-24 20:24:01 +01:00
|
|
|
|
2017-02-26 11:03:45 +01:00
|
|
|
for file_name, path_id, size in dummy_files:
|
2016-03-24 20:24:01 +01:00
|
|
|
attachment = Attachment.objects.get(path_id=path_id)
|
|
|
|
self.assertTrue(attachment.is_claimed())
|
|
|
|
|
2017-09-05 20:50:25 +02:00
|
|
|
class MissedMessageTest(ZulipTestCase):
|
|
|
|
def test_missed_message_userids(self):
|
|
|
|
# type: () -> None
|
|
|
|
UserPresence.objects.all().delete()
|
|
|
|
|
|
|
|
sender = self.example_user('cordelia')
|
|
|
|
realm = sender.realm
|
|
|
|
hamlet = self.example_user('hamlet')
|
|
|
|
othello = self.example_user('othello')
|
2017-09-09 04:14:28 +02:00
|
|
|
recipient_ids = {hamlet.id, othello.id}
|
2017-09-05 20:50:25 +02:00
|
|
|
message_type = 'stream'
|
|
|
|
user_flags = {} # type: Dict[int, List[str]]
|
|
|
|
|
|
|
|
def assert_missing(user_ids):
|
|
|
|
# type: (List[int]) -> None
|
|
|
|
missed_message_userids = get_userids_for_missed_messages(
|
|
|
|
realm=realm,
|
|
|
|
sender_id=sender.id,
|
|
|
|
message_type=message_type,
|
2017-09-09 04:14:28 +02:00
|
|
|
active_user_ids=recipient_ids,
|
2017-09-05 20:50:25 +02:00
|
|
|
user_flags=user_flags,
|
|
|
|
)
|
|
|
|
self.assertEqual(sorted(user_ids), sorted(missed_message_userids))
|
|
|
|
|
|
|
|
def set_presence(user_id, client_name, ago):
|
|
|
|
# type: (int, Text, int) -> None
|
|
|
|
when = timezone_now() - datetime.timedelta(seconds=ago)
|
|
|
|
UserPresence.objects.create(
|
|
|
|
user_profile_id=user_id,
|
|
|
|
client=get_client(client_name),
|
|
|
|
timestamp=when,
|
|
|
|
)
|
|
|
|
|
|
|
|
message_type = 'private'
|
|
|
|
assert_missing([hamlet.id, othello.id])
|
|
|
|
|
|
|
|
message_type = 'stream'
|
|
|
|
user_flags[hamlet.id] = ['mentioned']
|
|
|
|
assert_missing([hamlet.id])
|
|
|
|
|
|
|
|
set_presence(hamlet.id, 'iPhone', ago=5000)
|
|
|
|
assert_missing([hamlet.id])
|
|
|
|
|
|
|
|
set_presence(hamlet.id, 'webapp', ago=15)
|
|
|
|
assert_missing([])
|
|
|
|
|
|
|
|
message_type = 'private'
|
|
|
|
assert_missing([othello.id])
|
|
|
|
|
2016-08-23 02:08:42 +02:00
|
|
|
class LogDictTest(ZulipTestCase):
|
2016-07-16 05:51:12 +02:00
|
|
|
def test_to_log_dict(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2017-05-24 02:42:31 +02:00
|
|
|
email = self.example_email('hamlet')
|
2016-07-16 05:51:12 +02:00
|
|
|
stream_name = 'Denmark'
|
|
|
|
topic_name = 'Copenhagen'
|
|
|
|
content = 'find me some good coffee shops'
|
2017-05-25 01:40:26 +02:00
|
|
|
# self.login(self.example_email("hamlet"))
|
2016-07-16 05:51:12 +02:00
|
|
|
message_id = self.send_message(email, stream_name,
|
2016-12-03 00:04:17 +01:00
|
|
|
message_type=Recipient.STREAM,
|
|
|
|
subject=topic_name,
|
|
|
|
content=content)
|
2016-07-16 05:51:12 +02:00
|
|
|
message = Message.objects.get(id=message_id)
|
|
|
|
dct = message.to_log_dict()
|
|
|
|
|
|
|
|
self.assertTrue('timestamp' in dct)
|
|
|
|
|
|
|
|
self.assertEqual(dct['content'], 'find me some good coffee shops')
|
|
|
|
self.assertEqual(dct['id'], message.id)
|
|
|
|
self.assertEqual(dct['recipient'], 'Denmark')
|
2017-03-14 23:31:05 +01:00
|
|
|
self.assertEqual(dct['sender_realm_str'], 'zulip')
|
2017-05-25 01:40:26 +02:00
|
|
|
self.assertEqual(dct['sender_email'], self.example_email("hamlet"))
|
2016-07-16 05:51:12 +02:00
|
|
|
self.assertEqual(dct['sender_full_name'], 'King Hamlet')
|
2017-05-24 02:42:31 +02:00
|
|
|
self.assertEqual(dct['sender_id'], self.example_user('hamlet').id)
|
2016-07-16 05:51:12 +02:00
|
|
|
self.assertEqual(dct['sender_short_name'], 'hamlet')
|
|
|
|
self.assertEqual(dct['sending_client'], 'test suite')
|
|
|
|
self.assertEqual(dct['subject'], 'Copenhagen')
|
|
|
|
self.assertEqual(dct['type'], 'stream')
|
|
|
|
|
2016-08-23 02:08:42 +02:00
|
|
|
class CheckMessageTest(ZulipTestCase):
|
2014-01-31 16:44:45 +01:00
|
|
|
def test_basic_check_message_call(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
2017-05-07 17:21:26 +02:00
|
|
|
sender = self.example_user('othello')
|
2016-09-13 23:32:35 +02:00
|
|
|
client = make_client(name="test suite")
|
2016-10-21 22:59:28 +02:00
|
|
|
stream_name = u'España y Francia'
|
|
|
|
self.make_stream(stream_name)
|
2014-01-31 16:44:45 +01:00
|
|
|
subject_name = 'issue'
|
|
|
|
message_content = 'whatever'
|
2017-04-27 22:48:06 +02:00
|
|
|
addressee = Addressee.for_stream(stream_name, subject_name)
|
|
|
|
ret = check_message(sender, client, addressee, message_content)
|
2017-05-25 02:08:35 +02:00
|
|
|
self.assertEqual(ret['message'].sender.email, self.example_email("othello"))
|
2014-01-31 16:44:45 +01:00
|
|
|
|
|
|
|
def test_bot_pm_feature(self):
|
2016-07-30 06:01:08 +02:00
|
|
|
# type: () -> None
|
|
|
|
"""We send a PM to a bot's owner if their bot sends a message to
|
|
|
|
an unsubscribed stream"""
|
2017-05-07 17:21:26 +02:00
|
|
|
parent = self.example_user('othello')
|
2014-01-31 16:44:45 +01:00
|
|
|
bot = do_create_user(
|
2017-01-24 07:06:13 +01:00
|
|
|
email='othello-bot@zulip.com',
|
|
|
|
password='',
|
|
|
|
realm=parent.realm,
|
|
|
|
full_name='',
|
|
|
|
short_name='',
|
|
|
|
active=True,
|
|
|
|
bot_type=UserProfile.DEFAULT_BOT,
|
|
|
|
bot_owner=parent
|
2014-01-31 16:44:45 +01:00
|
|
|
)
|
|
|
|
bot.last_reminder = None
|
|
|
|
|
|
|
|
sender = bot
|
2016-09-13 23:32:35 +02:00
|
|
|
client = make_client(name="test suite")
|
2016-10-21 22:59:28 +02:00
|
|
|
stream_name = u'Россия'
|
2014-01-31 16:44:45 +01:00
|
|
|
subject_name = 'issue'
|
2017-04-27 22:48:06 +02:00
|
|
|
addressee = Addressee.for_stream(stream_name, subject_name)
|
2014-01-31 16:44:45 +01:00
|
|
|
message_content = 'whatever'
|
|
|
|
old_count = message_stream_count(parent)
|
2016-11-15 05:32:47 +01:00
|
|
|
|
|
|
|
# Try sending to stream that doesn't exist sends a reminder to
|
|
|
|
# the sender
|
|
|
|
with self.assertRaises(JsonableError):
|
2017-04-27 22:48:06 +02:00
|
|
|
check_message(sender, client, addressee, message_content)
|
|
|
|
|
2016-11-15 05:32:47 +01:00
|
|
|
new_count = message_stream_count(parent)
|
|
|
|
self.assertEqual(new_count, old_count + 1)
|
|
|
|
self.assertIn("that stream does not yet exist.", most_recent_message(parent).content)
|
|
|
|
|
|
|
|
# Try sending to stream that exists with no subscribers soon
|
|
|
|
# after; due to rate-limiting, this should send nothing.
|
|
|
|
self.make_stream(stream_name)
|
2017-04-27 22:48:06 +02:00
|
|
|
ret = check_message(sender, client, addressee, message_content)
|
2014-01-31 16:44:45 +01:00
|
|
|
new_count = message_stream_count(parent)
|
|
|
|
self.assertEqual(new_count, old_count + 1)
|
2016-11-15 05:32:47 +01:00
|
|
|
|
|
|
|
# Try sending to stream that exists with no subscribers longer
|
|
|
|
# after; this should send an error to the bot owner that the
|
|
|
|
# stream doesn't exist
|
2017-05-24 04:21:29 +02:00
|
|
|
assert(sender.last_reminder is not None)
|
2016-11-15 05:32:47 +01:00
|
|
|
sender.last_reminder = sender.last_reminder - datetime.timedelta(hours=1)
|
|
|
|
sender.save(update_fields=["last_reminder"])
|
2017-04-27 22:48:06 +02:00
|
|
|
ret = check_message(sender, client, addressee, message_content)
|
|
|
|
|
2016-11-15 05:32:47 +01:00
|
|
|
new_count = message_stream_count(parent)
|
|
|
|
self.assertEqual(new_count, old_count + 2)
|
2014-01-31 16:44:45 +01:00
|
|
|
self.assertEqual(ret['message'].sender.email, 'othello-bot@zulip.com')
|
2016-11-15 05:32:47 +01:00
|
|
|
self.assertIn("there are no subscribers to that stream", most_recent_message(parent).content)
|
2017-05-14 21:14:26 +02:00
|
|
|
|
|
|
|
|
|
|
|
class DeleteMessageTest(ZulipTestCase):
|
|
|
|
|
|
|
|
def test_delete_message_by_owner(self):
|
|
|
|
# type: () -> None
|
|
|
|
self.login("hamlet@zulip.com")
|
|
|
|
msg_id = self.send_message("hamlet@zulip.com", "Scotland", Recipient.STREAM)
|
|
|
|
result = self.client_delete('/json/messages/{msg_id}'.format(msg_id=msg_id))
|
|
|
|
self.assert_json_error(result, "You don't have permission to edit this message")
|
|
|
|
|
|
|
|
def test_delete_message_by_realm_admin(self):
|
|
|
|
# type: () -> None
|
|
|
|
self.login("iago@zulip.com")
|
|
|
|
msg_id = self.send_message("hamlet@zulip.com", "Scotland", Recipient.STREAM)
|
|
|
|
result = self.client_delete('/json/messages/{msg_id}'.format(msg_id=msg_id))
|
|
|
|
self.assert_json_success(result)
|
|
|
|
|
|
|
|
def test_delete_message_second_time(self):
|
|
|
|
# type: () -> None
|
|
|
|
self.login("iago@zulip.com")
|
|
|
|
msg_id = self.send_message("hamlet@zulip.com", "Scotland", Recipient.STREAM,
|
|
|
|
subject="editing", content="before edit")
|
|
|
|
self.client_delete('/json/messages/{msg_id}'.format(msg_id=msg_id))
|
|
|
|
result = self.client_delete('/json/messages/{msg_id}'.format(msg_id=msg_id))
|
|
|
|
self.assert_json_error(result, "Invalid message(s)")
|
2017-07-13 16:39:01 +02:00
|
|
|
|
|
|
|
class SoftDeactivationMessageTest(ZulipTestCase):
|
|
|
|
|
2017-07-16 09:41:38 +02:00
|
|
|
def test_maybe_catch_up_soft_deactivated_user(self):
|
|
|
|
# type: () -> None
|
2017-08-25 06:01:29 +02:00
|
|
|
recipient_list = [self.example_user("hamlet"), self.example_user("iago")]
|
|
|
|
for user_profile in recipient_list:
|
|
|
|
self.subscribe(user_profile, "Denmark")
|
2017-07-16 09:41:38 +02:00
|
|
|
|
|
|
|
sender = self.example_email('iago')
|
|
|
|
stream_name = 'Denmark'
|
|
|
|
subject = 'foo'
|
|
|
|
|
|
|
|
def last_realm_audit_log_entry(event_type):
|
|
|
|
# type: (str) -> RealmAuditLog
|
|
|
|
return RealmAuditLog.objects.filter(
|
|
|
|
event_type=event_type
|
|
|
|
).order_by('-event_time')[0]
|
|
|
|
|
|
|
|
long_term_idle_user = self.example_user('hamlet')
|
2017-08-18 10:09:54 +02:00
|
|
|
# We are sending this message to ensure that long_term_idle_user has
|
|
|
|
# at least one UserMessage row.
|
|
|
|
self.send_message(long_term_idle_user.email, stream_name,
|
|
|
|
Recipient.STREAM)
|
2017-07-16 09:41:38 +02:00
|
|
|
do_soft_deactivate_users([long_term_idle_user])
|
|
|
|
|
|
|
|
message = 'Test Message 1'
|
|
|
|
self.send_message(sender, stream_name, Recipient.STREAM,
|
|
|
|
message, subject)
|
|
|
|
idle_user_msg_list = get_user_messages(long_term_idle_user)
|
|
|
|
idle_user_msg_count = len(idle_user_msg_list)
|
|
|
|
self.assertNotEqual(idle_user_msg_list[-1].content, message)
|
|
|
|
with queries_captured() as queries:
|
|
|
|
maybe_catch_up_soft_deactivated_user(long_term_idle_user)
|
|
|
|
self.assert_length(queries, 8)
|
|
|
|
self.assertFalse(long_term_idle_user.long_term_idle)
|
|
|
|
self.assertEqual(last_realm_audit_log_entry(
|
|
|
|
'user_soft_activated').modified_user, long_term_idle_user)
|
|
|
|
idle_user_msg_list = get_user_messages(long_term_idle_user)
|
|
|
|
self.assertEqual(len(idle_user_msg_list), idle_user_msg_count + 1)
|
|
|
|
self.assertEqual(idle_user_msg_list[-1].content, message)
|
|
|
|
|
2017-07-13 16:39:01 +02:00
|
|
|
def test_add_missing_messages(self):
|
|
|
|
# type: () -> None
|
2017-08-25 06:01:29 +02:00
|
|
|
recipient_list = [self.example_user("hamlet"), self.example_user("iago")]
|
|
|
|
for user_profile in recipient_list:
|
|
|
|
self.subscribe(user_profile, "Denmark")
|
2017-07-13 16:39:01 +02:00
|
|
|
|
|
|
|
sender = self.example_user('iago')
|
|
|
|
realm = sender.realm
|
|
|
|
sending_client = make_client(name="test suite")
|
|
|
|
stream_name = 'Denmark'
|
|
|
|
stream = get_stream(stream_name, realm)
|
|
|
|
subject = 'foo'
|
|
|
|
|
|
|
|
def send_fake_message(message_content, stream):
|
|
|
|
# type: (str, Stream) -> Message
|
|
|
|
recipient = get_recipient(Recipient.STREAM, stream.id)
|
|
|
|
return Message.objects.create(sender = sender,
|
|
|
|
recipient = recipient,
|
|
|
|
subject = subject,
|
|
|
|
content = message_content,
|
|
|
|
pub_date = timezone_now(),
|
|
|
|
sending_client = sending_client)
|
|
|
|
|
|
|
|
long_term_idle_user = self.example_user('hamlet')
|
2017-08-18 10:09:54 +02:00
|
|
|
self.send_message(long_term_idle_user.email, stream_name,
|
|
|
|
Recipient.STREAM)
|
2017-07-13 16:39:01 +02:00
|
|
|
do_soft_deactivate_users([long_term_idle_user])
|
|
|
|
|
|
|
|
# Test that add_missing_messages() in simplest case of adding a
|
|
|
|
# message for which UserMessage row doesn't exist for this user.
|
|
|
|
sent_message = send_fake_message('Test Message 1', stream)
|
|
|
|
idle_user_msg_list = get_user_messages(long_term_idle_user)
|
|
|
|
idle_user_msg_count = len(idle_user_msg_list)
|
|
|
|
self.assertNotEqual(idle_user_msg_list[-1], sent_message)
|
|
|
|
with queries_captured() as queries:
|
|
|
|
add_missing_messages(long_term_idle_user)
|
|
|
|
self.assert_length(queries, 5)
|
|
|
|
idle_user_msg_list = get_user_messages(long_term_idle_user)
|
|
|
|
self.assertEqual(len(idle_user_msg_list), idle_user_msg_count + 1)
|
|
|
|
self.assertEqual(idle_user_msg_list[-1], sent_message)
|
|
|
|
|
|
|
|
# Test that add_missing_messages() only adds messages that aren't
|
|
|
|
# already present in the UserMessage table. This test works on the
|
|
|
|
# fact that previous test just above this added a message but didn't
|
|
|
|
# updated the last_active_message_id field for the user.
|
|
|
|
sent_message = send_fake_message('Test Message 2', stream)
|
|
|
|
idle_user_msg_list = get_user_messages(long_term_idle_user)
|
|
|
|
idle_user_msg_count = len(idle_user_msg_list)
|
|
|
|
self.assertNotEqual(idle_user_msg_list[-1], sent_message)
|
|
|
|
with queries_captured() as queries:
|
|
|
|
add_missing_messages(long_term_idle_user)
|
|
|
|
self.assert_length(queries, 5)
|
|
|
|
idle_user_msg_list = get_user_messages(long_term_idle_user)
|
|
|
|
self.assertEqual(len(idle_user_msg_list), idle_user_msg_count + 1)
|
|
|
|
self.assertEqual(idle_user_msg_list[-1], sent_message)
|
|
|
|
|
|
|
|
# Test UserMessage rows are created correctly in case of stream
|
|
|
|
# Subscription was altered by admin while user was away.
|
|
|
|
|
|
|
|
# Test for a public stream.
|
|
|
|
sent_message_list = []
|
|
|
|
sent_message_list.append(send_fake_message('Test Message 3', stream))
|
|
|
|
# Alter subscription to stream.
|
2017-08-25 06:23:11 +02:00
|
|
|
self.unsubscribe(long_term_idle_user, stream_name)
|
2017-07-13 16:39:01 +02:00
|
|
|
send_fake_message('Test Message 4', stream)
|
2017-08-25 06:01:29 +02:00
|
|
|
self.subscribe(long_term_idle_user, stream_name)
|
2017-07-13 16:39:01 +02:00
|
|
|
sent_message_list.append(send_fake_message('Test Message 5', stream))
|
|
|
|
sent_message_list.reverse()
|
|
|
|
idle_user_msg_list = get_user_messages(long_term_idle_user)
|
|
|
|
idle_user_msg_count = len(idle_user_msg_list)
|
|
|
|
for sent_message in sent_message_list:
|
|
|
|
self.assertNotEqual(idle_user_msg_list.pop(), sent_message)
|
|
|
|
with queries_captured() as queries:
|
|
|
|
add_missing_messages(long_term_idle_user)
|
|
|
|
self.assert_length(queries, 5)
|
|
|
|
idle_user_msg_list = get_user_messages(long_term_idle_user)
|
|
|
|
self.assertEqual(len(idle_user_msg_list), idle_user_msg_count + 2)
|
|
|
|
for sent_message in sent_message_list:
|
|
|
|
self.assertEqual(idle_user_msg_list.pop(), sent_message)
|
|
|
|
|
|
|
|
# Test consecutive subscribe/unsubscribe in a public stream
|
|
|
|
sent_message_list = []
|
|
|
|
|
|
|
|
sent_message_list.append(send_fake_message('Test Message 6', stream))
|
|
|
|
# Unsubscribe from stream and then immediately subscribe back again.
|
2017-08-25 06:23:11 +02:00
|
|
|
self.unsubscribe(long_term_idle_user, stream_name)
|
2017-08-25 06:01:29 +02:00
|
|
|
self.subscribe(long_term_idle_user, stream_name)
|
2017-07-13 16:39:01 +02:00
|
|
|
sent_message_list.append(send_fake_message('Test Message 7', stream))
|
|
|
|
# Again unsubscribe from stream and send a message.
|
|
|
|
# This will make sure that if initially in a unsubscribed state
|
|
|
|
# a consecutive subscribe/unsubscribe doesn't misbehave.
|
2017-08-25 06:23:11 +02:00
|
|
|
self.unsubscribe(long_term_idle_user, stream_name)
|
2017-07-13 16:39:01 +02:00
|
|
|
send_fake_message('Test Message 8', stream)
|
|
|
|
# Do a subscribe and unsubscribe immediately.
|
2017-08-25 06:01:29 +02:00
|
|
|
self.subscribe(long_term_idle_user, stream_name)
|
2017-08-25 06:23:11 +02:00
|
|
|
self.unsubscribe(long_term_idle_user, stream_name)
|
2017-07-13 16:39:01 +02:00
|
|
|
|
|
|
|
sent_message_list.reverse()
|
|
|
|
idle_user_msg_list = get_user_messages(long_term_idle_user)
|
|
|
|
idle_user_msg_count = len(idle_user_msg_list)
|
|
|
|
for sent_message in sent_message_list:
|
|
|
|
self.assertNotEqual(idle_user_msg_list.pop(), sent_message)
|
|
|
|
with queries_captured() as queries:
|
|
|
|
add_missing_messages(long_term_idle_user)
|
|
|
|
self.assert_length(queries, 5)
|
|
|
|
idle_user_msg_list = get_user_messages(long_term_idle_user)
|
|
|
|
self.assertEqual(len(idle_user_msg_list), idle_user_msg_count + 2)
|
|
|
|
for sent_message in sent_message_list:
|
|
|
|
self.assertEqual(idle_user_msg_list.pop(), sent_message)
|
|
|
|
# Note: At this point in this test we have long_term_idle_user
|
|
|
|
# unsubscribed from the 'Denmark' stream.
|
|
|
|
|
|
|
|
# Test for a Private Stream.
|
|
|
|
stream_name = "Core"
|
|
|
|
private_stream = self.make_stream('Core', invite_only=True)
|
2017-08-25 06:01:29 +02:00
|
|
|
self.subscribe(self.example_user("iago"), stream_name)
|
2017-07-13 16:39:01 +02:00
|
|
|
sent_message_list = []
|
|
|
|
send_fake_message('Test Message 9', private_stream)
|
2017-08-25 06:01:29 +02:00
|
|
|
self.subscribe(self.example_user("hamlet"), stream_name)
|
2017-07-13 16:39:01 +02:00
|
|
|
sent_message_list.append(send_fake_message('Test Message 10', private_stream))
|
2017-08-25 06:23:11 +02:00
|
|
|
self.unsubscribe(long_term_idle_user, stream_name)
|
2017-07-13 16:39:01 +02:00
|
|
|
send_fake_message('Test Message 11', private_stream)
|
2017-08-25 06:01:29 +02:00
|
|
|
self.subscribe(long_term_idle_user, stream_name)
|
2017-07-13 16:39:01 +02:00
|
|
|
sent_message_list.append(send_fake_message('Test Message 12', private_stream))
|
|
|
|
sent_message_list.reverse()
|
|
|
|
idle_user_msg_list = get_user_messages(long_term_idle_user)
|
|
|
|
idle_user_msg_count = len(idle_user_msg_list)
|
|
|
|
for sent_message in sent_message_list:
|
|
|
|
self.assertNotEqual(idle_user_msg_list.pop(), sent_message)
|
|
|
|
with queries_captured() as queries:
|
|
|
|
add_missing_messages(long_term_idle_user)
|
|
|
|
self.assert_length(queries, 5)
|
|
|
|
idle_user_msg_list = get_user_messages(long_term_idle_user)
|
|
|
|
self.assertEqual(len(idle_user_msg_list), idle_user_msg_count + 2)
|
|
|
|
for sent_message in sent_message_list:
|
|
|
|
self.assertEqual(idle_user_msg_list.pop(), sent_message)
|
2017-07-18 16:03:47 +02:00
|
|
|
|
|
|
|
def test_user_message_filter(self):
|
|
|
|
# type: () -> None
|
|
|
|
# In this test we are basically testing out the logic used out in
|
|
|
|
# do_send_messages() in action.py for filtering the messages for which
|
|
|
|
# UserMessage rows should be created for a soft-deactivated user.
|
|
|
|
recipient_list = [
|
2017-08-25 06:01:29 +02:00
|
|
|
self.example_user("hamlet"),
|
|
|
|
self.example_user("iago"),
|
|
|
|
self.example_user('cordelia')
|
2017-07-18 16:03:47 +02:00
|
|
|
]
|
2017-08-25 06:01:29 +02:00
|
|
|
for user_profile in recipient_list:
|
|
|
|
self.subscribe(user_profile, "Denmark")
|
2017-07-18 16:03:47 +02:00
|
|
|
|
|
|
|
cordelia = self.example_user('cordelia')
|
|
|
|
sender = self.example_email('iago')
|
|
|
|
stream_name = 'Denmark'
|
|
|
|
subject = 'foo'
|
|
|
|
|
|
|
|
def send_stream_message(content):
|
|
|
|
# type: (str) -> None
|
|
|
|
self.send_message(sender, stream_name, Recipient.STREAM,
|
|
|
|
content, subject)
|
|
|
|
|
|
|
|
def send_personal_message(content):
|
|
|
|
# type: (str) -> None
|
|
|
|
self.send_message(sender, self.example_email("hamlet"),
|
|
|
|
Recipient.PERSONAL, content)
|
|
|
|
|
|
|
|
long_term_idle_user = self.example_user('hamlet')
|
2017-08-18 10:09:54 +02:00
|
|
|
self.send_message(long_term_idle_user.email, stream_name,
|
|
|
|
Recipient.STREAM)
|
2017-07-18 16:03:47 +02:00
|
|
|
do_soft_deactivate_users([long_term_idle_user])
|
|
|
|
|
|
|
|
def assert_um_count(user, count):
|
|
|
|
# type: (UserProfile, int) -> None
|
|
|
|
user_messages = get_user_messages(user)
|
|
|
|
self.assertEqual(len(user_messages), count)
|
|
|
|
|
|
|
|
def assert_last_um_content(user, content, negate=False):
|
|
|
|
# type: (UserProfile, Text, bool) -> None
|
|
|
|
user_messages = get_user_messages(user)
|
|
|
|
if negate:
|
|
|
|
self.assertNotEqual(user_messages[-1].content, content)
|
|
|
|
else:
|
|
|
|
self.assertEqual(user_messages[-1].content, content)
|
|
|
|
|
|
|
|
# Test that sending a message to a stream with soft deactivated user
|
|
|
|
# doesn't end up creating UserMessage row for deactivated user.
|
|
|
|
general_user_msg_count = len(get_user_messages(cordelia))
|
|
|
|
soft_deactivated_user_msg_count = len(get_user_messages(long_term_idle_user))
|
|
|
|
message = 'Test Message 1'
|
|
|
|
send_stream_message(message)
|
|
|
|
assert_last_um_content(long_term_idle_user, force_text(message), negate=True)
|
|
|
|
assert_um_count(long_term_idle_user, soft_deactivated_user_msg_count)
|
|
|
|
assert_um_count(cordelia, general_user_msg_count + 1)
|
|
|
|
assert_last_um_content(cordelia, force_text(message))
|
|
|
|
|
|
|
|
# Test sending a private message to soft deactivated user creates
|
|
|
|
# UserMessage row.
|
|
|
|
soft_deactivated_user_msg_count = len(get_user_messages(long_term_idle_user))
|
|
|
|
message = 'Test PM'
|
|
|
|
send_personal_message(message)
|
|
|
|
assert_um_count(long_term_idle_user, soft_deactivated_user_msg_count + 1)
|
|
|
|
assert_last_um_content(long_term_idle_user, force_text(message))
|
|
|
|
|
|
|
|
# Test UserMessage row is created while user is deactivated if
|
|
|
|
# user itself is mentioned.
|
|
|
|
general_user_msg_count = len(get_user_messages(cordelia))
|
|
|
|
soft_deactivated_user_msg_count = len(get_user_messages(long_term_idle_user))
|
|
|
|
message = 'Test @**King Hamlet** mention'
|
|
|
|
send_stream_message(message)
|
|
|
|
assert_last_um_content(long_term_idle_user, force_text(message))
|
|
|
|
assert_um_count(long_term_idle_user, soft_deactivated_user_msg_count + 1)
|
|
|
|
assert_um_count(cordelia, general_user_msg_count + 1)
|
|
|
|
assert_last_um_content(cordelia, force_text(message))
|
|
|
|
|
|
|
|
# Test UserMessage row is not created while user is deactivated if
|
|
|
|
# anyone is mentioned but the user.
|
|
|
|
general_user_msg_count = len(get_user_messages(cordelia))
|
|
|
|
soft_deactivated_user_msg_count = len(get_user_messages(long_term_idle_user))
|
|
|
|
message = 'Test @**Cordelia Lear** mention'
|
|
|
|
send_stream_message(message)
|
|
|
|
assert_last_um_content(long_term_idle_user, force_text(message), negate=True)
|
|
|
|
assert_um_count(long_term_idle_user, soft_deactivated_user_msg_count)
|
|
|
|
assert_um_count(cordelia, general_user_msg_count + 1)
|
|
|
|
assert_last_um_content(cordelia, force_text(message))
|
|
|
|
|
|
|
|
# Test UserMessage row is created while user is deactivated if
|
|
|
|
# there is a wildcard mention such as @all or @everyone
|
|
|
|
general_user_msg_count = len(get_user_messages(cordelia))
|
|
|
|
soft_deactivated_user_msg_count = len(get_user_messages(long_term_idle_user))
|
|
|
|
message = 'Test @**all** mention'
|
|
|
|
send_stream_message(message)
|
|
|
|
assert_last_um_content(long_term_idle_user, force_text(message))
|
|
|
|
assert_um_count(long_term_idle_user, soft_deactivated_user_msg_count + 1)
|
|
|
|
assert_um_count(cordelia, general_user_msg_count + 1)
|
|
|
|
assert_last_um_content(cordelia, force_text(message))
|
|
|
|
|
|
|
|
general_user_msg_count = len(get_user_messages(cordelia))
|
|
|
|
soft_deactivated_user_msg_count = len(get_user_messages(long_term_idle_user))
|
|
|
|
message = 'Test @**everyone** mention'
|
|
|
|
send_stream_message(message)
|
|
|
|
assert_last_um_content(long_term_idle_user, force_text(message))
|
|
|
|
assert_um_count(long_term_idle_user, soft_deactivated_user_msg_count + 1)
|
|
|
|
assert_um_count(cordelia, general_user_msg_count + 1)
|
|
|
|
assert_last_um_content(cordelia, force_text(message))
|
|
|
|
|
2017-08-27 18:55:06 +02:00
|
|
|
# Test UserMessage row is not created while user is deactivated if there
|
2017-07-18 16:03:47 +02:00
|
|
|
# is a alert word in message.
|
|
|
|
do_add_alert_words(long_term_idle_user, ['test_alert_word'])
|
|
|
|
general_user_msg_count = len(get_user_messages(cordelia))
|
|
|
|
soft_deactivated_user_msg_count = len(get_user_messages(long_term_idle_user))
|
|
|
|
message = 'Testing test_alert_word'
|
|
|
|
send_stream_message(message)
|
|
|
|
assert_last_um_content(long_term_idle_user, force_text(message))
|
|
|
|
assert_um_count(long_term_idle_user, soft_deactivated_user_msg_count + 1)
|
|
|
|
assert_um_count(cordelia, general_user_msg_count + 1)
|
|
|
|
assert_last_um_content(cordelia, force_text(message))
|
|
|
|
|
|
|
|
# Test UserMessage row is created while user is deactivated if
|
|
|
|
# message is a me message.
|
|
|
|
general_user_msg_count = len(get_user_messages(cordelia))
|
|
|
|
soft_deactivated_user_msg_count = len(get_user_messages(long_term_idle_user))
|
|
|
|
message = '/me says test'
|
|
|
|
send_stream_message(message)
|
2017-08-27 18:55:06 +02:00
|
|
|
assert_last_um_content(long_term_idle_user, force_text(message), negate=True)
|
|
|
|
assert_um_count(long_term_idle_user, soft_deactivated_user_msg_count)
|
2017-07-18 16:03:47 +02:00
|
|
|
assert_um_count(cordelia, general_user_msg_count + 1)
|
|
|
|
assert_last_um_content(cordelia, force_text(message))
|