mirror of
https://github.com/zulip/zulip.git
synced 2025-11-15 11:22:04 +00:00
69730a78cc
Automatically generated by the following script, based on the output
of lint with flake8-comma:
import re
import sys
last_filename = None
last_row = None
lines = []
for msg in sys.stdin:
m = re.match(
r"\x1b\[35mflake8 \|\x1b\[0m \x1b\[1;31m(.+):(\d+):(\d+): (\w+)", msg
)
if m:
filename, row_str, col_str, err = m.groups()
row, col = int(row_str), int(col_str)
if filename == last_filename:
assert last_row != row
else:
if last_filename is not None:
with open(last_filename, "w") as f:
f.writelines(lines)
with open(filename) as f:
lines = f.readlines()
last_filename = filename
last_row = row
line = lines[row - 1]
if err in ["C812", "C815"]:
lines[row - 1] = line[: col - 1] + "," + line[col - 1 :]
elif err in ["C819"]:
assert line[col - 2] == ","
lines[row - 1] = line[: col - 2] + line[col - 1 :].lstrip(" ")
if last_filename is not None:
with open(last_filename, "w") as f:
f.writelines(lines)
Signed-off-by: Anders Kaseorg <anders@zulipchat.com>
180 lines
6.7 KiB
Python
180 lines
6.7 KiB
Python
from django.utils.timezone import now as timezone_now
|
|
from datetime import timedelta, datetime, timezone
|
|
from typing import Any, Dict
|
|
from unittest import mock
|
|
|
|
from zerver.lib.test_classes import ZulipTestCase
|
|
from zerver.lib.stream_topic import StreamTopicTarget
|
|
|
|
from zerver.models import (
|
|
get_stream,
|
|
UserProfile,
|
|
MutedTopic,
|
|
)
|
|
|
|
from zerver.lib.topic_mutes import (
|
|
add_topic_mute,
|
|
get_topic_mutes,
|
|
remove_topic_mute,
|
|
topic_is_muted,
|
|
)
|
|
|
|
class MutedTopicsTests(ZulipTestCase):
|
|
def test_user_ids_muting_topic(self) -> None:
|
|
hamlet = self.example_user('hamlet')
|
|
cordelia = self.example_user('cordelia')
|
|
realm = hamlet.realm
|
|
stream = get_stream('Verona', realm)
|
|
recipient = stream.recipient
|
|
topic_name = 'teST topic'
|
|
|
|
stream_topic_target = StreamTopicTarget(
|
|
stream_id=stream.id,
|
|
topic_name=topic_name,
|
|
)
|
|
|
|
user_ids = stream_topic_target.user_ids_muting_topic()
|
|
self.assertEqual(user_ids, set())
|
|
|
|
def mute_user(user: UserProfile) -> None:
|
|
add_topic_mute(
|
|
user_profile=user,
|
|
stream_id=stream.id,
|
|
recipient_id=recipient.id,
|
|
topic_name='test TOPIC',
|
|
date_muted=timezone_now(),
|
|
)
|
|
|
|
mute_user(hamlet)
|
|
user_ids = stream_topic_target.user_ids_muting_topic()
|
|
self.assertEqual(user_ids, {hamlet.id})
|
|
hamlet_date_muted = MutedTopic.objects.filter(user_profile=hamlet)[0].date_muted
|
|
self.assertTrue(timezone_now() - hamlet_date_muted <= timedelta(seconds=100))
|
|
|
|
mute_user(cordelia)
|
|
user_ids = stream_topic_target.user_ids_muting_topic()
|
|
self.assertEqual(user_ids, {hamlet.id, cordelia.id})
|
|
cordelia_date_muted = MutedTopic.objects.filter(user_profile=cordelia)[0].date_muted
|
|
self.assertTrue(timezone_now() - cordelia_date_muted <= timedelta(seconds=100))
|
|
|
|
def test_add_muted_topic(self) -> None:
|
|
user = self.example_user('hamlet')
|
|
self.login_user(user)
|
|
|
|
stream = get_stream('Verona', user.realm)
|
|
|
|
url = '/api/v1/users/me/subscriptions/muted_topics'
|
|
|
|
payloads = [
|
|
{'stream': stream.name, 'topic': 'Verona3', 'op': 'add'},
|
|
{'stream_id': stream.id, 'topic': 'Verona3', 'op': 'add'},
|
|
]
|
|
|
|
mock_date_muted = datetime(2020, 1, 1, tzinfo=timezone.utc).timestamp()
|
|
for data in payloads:
|
|
with mock.patch('zerver.views.muting.timezone_now',
|
|
return_value=datetime(2020, 1, 1, tzinfo=timezone.utc)):
|
|
result = self.api_patch(user, url, data)
|
|
self.assert_json_success(result)
|
|
|
|
self.assertIn([stream.name, 'Verona3', mock_date_muted], get_topic_mutes(user))
|
|
self.assertTrue(topic_is_muted(user, stream.id, 'Verona3'))
|
|
self.assertTrue(topic_is_muted(user, stream.id, 'verona3'))
|
|
|
|
remove_topic_mute(
|
|
user_profile=user,
|
|
stream_id=stream.id,
|
|
topic_name='Verona3',
|
|
)
|
|
|
|
def test_remove_muted_topic(self) -> None:
|
|
user = self.example_user('hamlet')
|
|
realm = user.realm
|
|
self.login_user(user)
|
|
|
|
stream = get_stream('Verona', realm)
|
|
recipient = stream.recipient
|
|
|
|
url = '/api/v1/users/me/subscriptions/muted_topics'
|
|
payloads = [
|
|
{'stream': stream.name, 'topic': 'vERONA3', 'op': 'remove'},
|
|
{'stream_id': stream.id, 'topic': 'vEroNA3', 'op': 'remove'},
|
|
]
|
|
mock_date_muted = datetime(2020, 1, 1, tzinfo=timezone.utc).timestamp()
|
|
|
|
for data in payloads:
|
|
add_topic_mute(
|
|
user_profile=user,
|
|
stream_id=stream.id,
|
|
recipient_id=recipient.id,
|
|
topic_name='Verona3',
|
|
date_muted=datetime(2020, 1, 1, tzinfo=timezone.utc),
|
|
)
|
|
self.assertIn([stream.name, 'Verona3', mock_date_muted], get_topic_mutes(user))
|
|
|
|
result = self.api_patch(user, url, data)
|
|
|
|
self.assert_json_success(result)
|
|
self.assertNotIn([stream.name, 'Verona3', mock_date_muted], get_topic_mutes(user))
|
|
self.assertFalse(topic_is_muted(user, stream.id, 'verona3'))
|
|
|
|
def test_muted_topic_add_invalid(self) -> None:
|
|
user = self.example_user('hamlet')
|
|
realm = user.realm
|
|
self.login_user(user)
|
|
|
|
stream = get_stream('Verona', realm)
|
|
recipient = stream.recipient
|
|
add_topic_mute(
|
|
user_profile=user,
|
|
stream_id=stream.id,
|
|
recipient_id=recipient.id,
|
|
topic_name='Verona3',
|
|
date_muted=timezone_now(),
|
|
)
|
|
|
|
url = '/api/v1/users/me/subscriptions/muted_topics'
|
|
|
|
data: Dict[str, Any] = {'stream': stream.name, 'topic': 'Verona3', 'op': 'add'}
|
|
result = self.api_patch(user, url, data)
|
|
self.assert_json_error(result, "Topic already muted")
|
|
|
|
data = {'stream_id': 999999999, 'topic': 'Verona3', 'op': 'add'}
|
|
result = self.api_patch(user, url, data)
|
|
self.assert_json_error(result, "Invalid stream id")
|
|
|
|
data = {'topic': 'Verona3', 'op': 'add'}
|
|
result = self.api_patch(user, url, data)
|
|
self.assert_json_error(result, "Please supply 'stream'.")
|
|
|
|
data = {'stream': stream.name, 'stream_id': stream.id, 'topic': 'Verona3', 'op': 'add'}
|
|
result = self.api_patch(user, url, data)
|
|
self.assert_json_error(result, "Please choose one: 'stream' or 'stream_id'.")
|
|
|
|
def test_muted_topic_remove_invalid(self) -> None:
|
|
user = self.example_user('hamlet')
|
|
realm = user.realm
|
|
self.login_user(user)
|
|
stream = get_stream('Verona', realm)
|
|
|
|
url = '/api/v1/users/me/subscriptions/muted_topics'
|
|
data: Dict[str, Any] = {'stream': 'BOGUS', 'topic': 'Verona3', 'op': 'remove'}
|
|
result = self.api_patch(user, url, data)
|
|
self.assert_json_error(result, "Topic is not muted")
|
|
|
|
data = {'stream': stream.name, 'topic': 'BOGUS', 'op': 'remove'}
|
|
result = self.api_patch(user, url, data)
|
|
self.assert_json_error(result, "Topic is not muted")
|
|
|
|
data = {'stream_id': 999999999, 'topic': 'BOGUS', 'op': 'remove'}
|
|
result = self.api_patch(user, url, data)
|
|
self.assert_json_error(result, "Topic is not muted")
|
|
|
|
data = {'topic': 'Verona3', 'op': 'remove'}
|
|
result = self.api_patch(user, url, data)
|
|
self.assert_json_error(result, "Please supply 'stream'.")
|
|
|
|
data = {'stream': stream.name, 'stream_id': stream.id, 'topic': 'Verona3', 'op': 'remove'}
|
|
result = self.api_patch(user, url, data)
|
|
self.assert_json_error(result, "Please choose one: 'stream' or 'stream_id'.")
|