mirror of https://github.com/zulip/zulip.git
288 lines
13 KiB
Python
288 lines
13 KiB
Python
import orjson
|
|
from django.utils.timezone import now as timezone_now
|
|
|
|
from zerver.actions.create_user import do_reactivate_user
|
|
from zerver.actions.realm_settings import do_set_realm_property
|
|
from zerver.actions.user_settings import do_change_user_setting
|
|
from zerver.actions.users import do_deactivate_user
|
|
from zerver.lib.test_classes import ZulipTestCase
|
|
from zerver.lib.user_mutes import add_user_mute, get_mute_object
|
|
from zerver.models import UserMessage, UserProfile
|
|
|
|
|
|
class TestReadReceipts(ZulipTestCase):
|
|
def mark_message_read(self, user: UserProfile, message_id: int) -> None:
|
|
result = self.api_post(
|
|
user,
|
|
"/api/v1/messages/flags",
|
|
{"messages": orjson.dumps([message_id]).decode(), "op": "add", "flag": "read"},
|
|
)
|
|
self.assert_json_success(result)
|
|
|
|
def test_stream_message(self) -> None:
|
|
hamlet = self.example_user("hamlet")
|
|
sender = self.example_user("othello")
|
|
|
|
message_id = self.send_stream_message(sender, "Verona", "read receipts")
|
|
self.login("hamlet")
|
|
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
self.assert_json_success(result)
|
|
self.assertTrue(hamlet.id not in result.json()["user_ids"])
|
|
|
|
self.mark_message_read(hamlet, message_id)
|
|
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
self.assert_json_success(result)
|
|
self.assertTrue(hamlet.id in result.json()["user_ids"])
|
|
self.assertTrue(sender.id not in result.json()["user_ids"])
|
|
|
|
def test_personal_message(self) -> None:
|
|
hamlet = self.example_user("hamlet")
|
|
sender = self.example_user("othello")
|
|
|
|
message_id = self.send_personal_message(sender, hamlet)
|
|
self.login("hamlet")
|
|
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
self.assert_json_success(result)
|
|
self.assertTrue(hamlet.id not in result.json()["user_ids"])
|
|
|
|
self.mark_message_read(hamlet, message_id)
|
|
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
self.assert_json_success(result)
|
|
self.assertTrue(hamlet.id in result.json()["user_ids"])
|
|
self.assertTrue(sender.id not in result.json()["user_ids"])
|
|
|
|
def test_huddle_message(self) -> None:
|
|
hamlet = self.example_user("hamlet")
|
|
sender = self.example_user("othello")
|
|
cordelia = self.example_user("cordelia")
|
|
|
|
message_id = self.send_huddle_message(sender, [hamlet, cordelia])
|
|
self.login("hamlet")
|
|
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
self.assert_json_success(result)
|
|
self.assertTrue(hamlet.id not in result.json()["user_ids"])
|
|
self.assertTrue(sender.id not in result.json()["user_ids"])
|
|
|
|
self.mark_message_read(hamlet, message_id)
|
|
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
self.assert_json_success(result)
|
|
self.assertTrue(hamlet.id in result.json()["user_ids"])
|
|
self.assertTrue(sender.id not in result.json()["user_ids"])
|
|
|
|
def test_inaccessible_stream_message(self) -> None:
|
|
sender = self.example_user("othello")
|
|
|
|
private_stream = "private stream"
|
|
self.make_stream(stream_name=private_stream, invite_only=True)
|
|
self.subscribe(sender, private_stream)
|
|
|
|
message_id = self.send_stream_message(sender, private_stream, "read receipts")
|
|
|
|
self.login("hamlet")
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
self.assert_json_error(result, "Invalid message(s)")
|
|
|
|
self.login_user(sender)
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
self.assert_json_success(result)
|
|
|
|
def test_filter_deactivated_users(self) -> None:
|
|
hamlet = self.example_user("hamlet")
|
|
cordelia = self.example_user("cordelia")
|
|
sender = self.example_user("othello")
|
|
|
|
message_id = self.send_stream_message(sender, "Verona", "read receipts")
|
|
|
|
# Mark message as read for hamlet and cordelia.
|
|
self.mark_message_read(hamlet, message_id)
|
|
self.mark_message_read(cordelia, message_id)
|
|
|
|
# Login as cordelia and make sure hamlet is in read receipts before deactivation.
|
|
self.login("cordelia")
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
self.assert_json_success(result)
|
|
self.assertTrue(hamlet.id in result.json()["user_ids"])
|
|
self.assertTrue(cordelia.id in result.json()["user_ids"])
|
|
|
|
# Deactivate hamlet and verify hamlet is not in read receipts.
|
|
do_deactivate_user(hamlet, acting_user=None)
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
self.assert_json_success(result)
|
|
self.assertTrue(hamlet.id not in result.json()["user_ids"])
|
|
self.assertTrue(cordelia.id in result.json()["user_ids"])
|
|
|
|
# Reactivate hamlet and verify hamlet appears again in read recipts.
|
|
do_reactivate_user(hamlet, acting_user=None)
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
self.assert_json_success(result)
|
|
self.assertTrue(hamlet.id in result.json()["user_ids"])
|
|
self.assertTrue(cordelia.id in result.json()["user_ids"])
|
|
|
|
def test_send_read_receipts_privacy_setting(self) -> None:
|
|
hamlet = self.example_user("hamlet")
|
|
sender = self.example_user("othello")
|
|
cordelia = self.example_user("cordelia")
|
|
|
|
message_id = self.send_stream_message(sender, "Verona", "read receipts")
|
|
|
|
self.mark_message_read(hamlet, message_id)
|
|
self.mark_message_read(cordelia, message_id)
|
|
|
|
self.login("aaron")
|
|
do_set_realm_property(sender.realm, "enable_read_receipts", False, acting_user=None)
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
self.assert_json_error(result, "Read receipts are disabled in this organization.")
|
|
|
|
do_set_realm_property(sender.realm, "enable_read_receipts", True, acting_user=None)
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
self.assert_json_success(result)
|
|
self.assertIn(hamlet.id, result.json()["user_ids"])
|
|
self.assertIn(cordelia.id, result.json()["user_ids"])
|
|
|
|
# Disable read receipts setting; confirm Cordelia no longer appears.
|
|
do_change_user_setting(cordelia, "send_read_receipts", False, acting_user=cordelia)
|
|
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
self.assert_json_success(result)
|
|
self.assertIn(hamlet.id, result.json()["user_ids"])
|
|
self.assertNotIn(cordelia.id, result.json()["user_ids"])
|
|
|
|
def test_send_read_receipts_privacy_setting_bot(self) -> None:
|
|
hamlet = self.example_user("hamlet")
|
|
sender = self.example_user("othello")
|
|
bot = self.example_user("default_bot")
|
|
|
|
message_id = self.send_stream_message(sender, "Verona", "read receipts")
|
|
|
|
self.mark_message_read(hamlet, message_id)
|
|
self.mark_message_read(bot, message_id)
|
|
|
|
self.login("aaron")
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
self.assert_json_success(result)
|
|
self.assertIn(hamlet.id, result.json()["user_ids"])
|
|
self.assertIn(bot.id, result.json()["user_ids"])
|
|
|
|
# Disable read receipts setting; confirm bot no longer appears.
|
|
do_change_user_setting(bot, "send_read_receipts", False, acting_user=bot)
|
|
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
self.assert_json_success(result)
|
|
self.assertIn(hamlet.id, result.json()["user_ids"])
|
|
self.assertNotIn(bot.id, result.json()["user_ids"])
|
|
|
|
def test_historical_usermessages_read_flag_not_considered(self) -> None:
|
|
"""
|
|
Ensure UserMessage rows with historical flag are also
|
|
considered for read receipts.
|
|
"""
|
|
hamlet = self.example_user("hamlet")
|
|
cordelia = self.example_user("cordelia")
|
|
|
|
stream_name = "test stream"
|
|
self.subscribe(cordelia, stream_name)
|
|
|
|
message_id = self.send_stream_message(cordelia, stream_name, content="foo")
|
|
|
|
self.login("hamlet")
|
|
|
|
# Have hamlet react to the message to
|
|
# create a historical UserMessage row.
|
|
reaction_info = {
|
|
"emoji_name": "smile",
|
|
}
|
|
result = self.client_post(f"/json/messages/{message_id}/reactions", reaction_info)
|
|
self.assert_json_success(result)
|
|
|
|
# Ensure UserMessage row with historical and read flags exists
|
|
user_message = UserMessage.objects.get(user_profile=hamlet, message_id=message_id)
|
|
self.assertTrue(user_message.flags.historical)
|
|
self.assertTrue(user_message.flags.read)
|
|
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
self.assert_json_success(result)
|
|
self.assertIn(hamlet.id, result.json()["user_ids"])
|
|
|
|
def test_filter_muted_users(self) -> None:
|
|
hamlet = self.example_user("hamlet")
|
|
cordelia = self.example_user("cordelia")
|
|
othello = self.example_user("othello")
|
|
iago = self.example_user("iago")
|
|
|
|
# Hamlet mutes Cordelia
|
|
add_user_mute(hamlet, cordelia, date_muted=timezone_now())
|
|
# Cordelia mutes Othello
|
|
add_user_mute(cordelia, othello, date_muted=timezone_now())
|
|
|
|
# Iago sends a message
|
|
message_id = self.send_stream_message(iago, "Verona", "read receipts")
|
|
|
|
# Mark message as read for users.
|
|
self.mark_message_read(hamlet, message_id)
|
|
self.mark_message_read(cordelia, message_id)
|
|
self.mark_message_read(othello, message_id)
|
|
|
|
# Login as Iago and make sure all three users are in read receipts.
|
|
self.login("iago")
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
response_dict = self.assert_json_success(result)
|
|
self.assert_length(response_dict["user_ids"], 3)
|
|
self.assertTrue(hamlet.id in response_dict["user_ids"])
|
|
self.assertTrue(cordelia.id in response_dict["user_ids"])
|
|
self.assertTrue(othello.id in response_dict["user_ids"])
|
|
|
|
# Login as Hamlet and make sure Cordelia is not in read receipts.
|
|
self.login("hamlet")
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
response_dict = self.assert_json_success(result)
|
|
self.assert_length(response_dict["user_ids"], 2)
|
|
self.assertTrue(hamlet.id in response_dict["user_ids"])
|
|
self.assertFalse(cordelia.id in response_dict["user_ids"])
|
|
self.assertTrue(othello.id in response_dict["user_ids"])
|
|
|
|
# Login as Othello and make sure Cordelia is not in in read receipts.
|
|
self.login("othello")
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
response_dict = self.assert_json_success(result)
|
|
self.assert_length(response_dict["user_ids"], 2)
|
|
self.assertTrue(hamlet.id in response_dict["user_ids"])
|
|
self.assertFalse(cordelia.id in response_dict["user_ids"])
|
|
self.assertTrue(othello.id in response_dict["user_ids"])
|
|
|
|
# Login as Cordelia and make sure Hamlet and Othello are not in read receipts.
|
|
self.login("cordelia")
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
response_dict = self.assert_json_success(result)
|
|
self.assert_length(response_dict["user_ids"], 1)
|
|
self.assertFalse(hamlet.id in response_dict["user_ids"])
|
|
self.assertTrue(cordelia.id in response_dict["user_ids"])
|
|
self.assertFalse(othello.id in response_dict["user_ids"])
|
|
|
|
# Cordelia unmutes Othello
|
|
mute_object = get_mute_object(cordelia, othello)
|
|
assert mute_object is not None
|
|
mute_object.delete()
|
|
|
|
# Now Othello should appear in her read receipts, but not Hamlet.
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
response_dict = self.assert_json_success(result)
|
|
self.assert_length(response_dict["user_ids"], 2)
|
|
self.assertFalse(hamlet.id in response_dict["user_ids"])
|
|
self.assertTrue(cordelia.id in response_dict["user_ids"])
|
|
self.assertTrue(othello.id in response_dict["user_ids"])
|
|
|
|
# Login as Othello and make sure all three users are in read receipts.
|
|
self.login("othello")
|
|
result = self.client_get(f"/json/messages/{message_id}/read_receipts")
|
|
response_dict = self.assert_json_success(result)
|
|
self.assert_length(response_dict["user_ids"], 3)
|
|
self.assertTrue(hamlet.id in response_dict["user_ids"])
|
|
self.assertTrue(cordelia.id in response_dict["user_ids"])
|
|
self.assertTrue(othello.id in response_dict["user_ids"])
|