Skip to content

Commit 66bf61c

Browse files
committed
refactor: Fix/add previous commits regarding '_private' and 'private_'.
1 parent d520e50 commit 66bf61c

File tree

7 files changed

+19
-21
lines changed

7 files changed

+19
-21
lines changed

tests/conftest.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1104,7 +1104,7 @@ def index_stream(empty_index: Index) -> Index:
11041104
"""
11051105
index = empty_index
11061106
index["stream_msg_ids_by_stream_id"] = defaultdict(set, {205: {537286}})
1107-
index["private_msg_ids"] = {537287, 537288}
1107+
index["direct_msg_ids"] = {537287, 537288}
11081108
return index
11091109

11101110

@@ -1145,8 +1145,8 @@ def index_user(empty_index: Index) -> Index:
11451145
"""
11461146
user_ids = frozenset({5179, 5140})
11471147
index = empty_index
1148-
index["private_msg_ids_by_user_ids"] = defaultdict(set, {user_ids: {537287}})
1149-
index["private_msg_ids"] = {537287, 537288}
1148+
index["direct_msg_ids_by_user_ids"] = defaultdict(set, {user_ids: {537287}})
1149+
index["direct_msg_ids"] = {537287, 537288}
11501150
return index
11511151

11521152

@@ -1158,8 +1158,8 @@ def index_user_multiple(empty_index: Index) -> Index:
11581158
"""
11591159
user_ids = frozenset({5179, 5140, 5180})
11601160
index = empty_index
1161-
index["private_msg_ids_by_user_ids"] = defaultdict(set, {user_ids: {537288}})
1162-
index["private_msg_ids"] = {537287, 537288}
1161+
index["direct_msg_ids_by_user_ids"] = defaultdict(set, {user_ids: {537288}})
1162+
index["direct_msg_ids"] = {537287, 537288}
11631163
return index
11641164

11651165

@@ -1178,7 +1178,7 @@ def index_all_starred(empty_index: Index, request: Any) -> Index:
11781178
msgs_with_stars = request.param
11791179
index = empty_index
11801180
index["starred_msg_ids"] = msgs_with_stars
1181-
index["private_msg_ids"] = {537287, 537288}
1181+
index["direct_msg_ids"] = {537287, 537288}
11821182
for msg_id, msg in index["messages"].items():
11831183
if msg_id in msgs_with_stars and "starred" not in msg["flags"]:
11841184
msg["flags"].append("starred")
@@ -1192,7 +1192,7 @@ def index_all_mentions(
11921192
mentioned_messages, wildcard_mentioned_messages = mentioned_messages_combination
11931193
index = empty_index
11941194
index["mentioned_msg_ids"] = mentioned_messages | wildcard_mentioned_messages
1195-
index["private_msg_ids"] = {537287, 537288}
1195+
index["direct_msg_ids"] = {537287, 537288}
11961196
for msg_id, msg in index["messages"].items():
11971197
if msg_id in mentioned_messages and "mentioned" not in msg["flags"]:
11981198
msg["flags"].append("mentioned")

tests/core/test_core.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -247,7 +247,7 @@ def test_narrow_to_user(
247247
recipients = frozenset([controller.model.user_id, user_id])
248248
assert controller.model.recipients == recipients
249249
widget = controller.view.message_view.log.extend.call_args_list[0][0][0][0]
250-
id_list = index_user["private_msg_ids_by_user_ids"][recipients]
250+
id_list = index_user["direct_msg_ids_by_user_ids"][recipients]
251251
assert {widget.original_widget.message["id"]} == id_list
252252

253253
@pytest.mark.parametrize(
@@ -302,7 +302,7 @@ def test_narrow_to_all_pm(
302302
controller.view.message_view.log.clear.assert_called_once_with()
303303

304304
widgets = controller.view.message_view.log.extend.call_args_list[0][0][0]
305-
id_list = index_user["private_msg_ids"]
305+
id_list = index_user["direct_msg_ids"]
306306
msg_ids = {widget.original_widget.message["id"] for widget in widgets}
307307
assert msg_ids == id_list
308308

tests/model/test_model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -477,16 +477,16 @@ def test_set_narrow_not_already_set(
477477
{"topic_msg_ids": {1: {"BOO": {0, 1}}}},
478478
set(),
479479
),
480-
([["is", "private"]], {"private_msg_ids": {0, 1}}, {0, 1}),
480+
([["is", "private"]], {"direct_msg_ids": {0, 1}}, {0, 1}),
481481
(
482482
[["pm-with", "[email protected]"]],
483-
{"private_msg_ids_by_user_ids": {frozenset({1, 2}): {0, 1}}},
483+
{"direct_msg_ids_by_user_ids": {frozenset({1, 2}): {0, 1}}},
484484
{0, 1},
485485
),
486486
(
487487
[["pm-with", "[email protected]"]],
488488
{ # Covers recipient empty-set case
489-
"private_msg_ids_by_user_ids": {
489+
"direct_msg_ids_by_user_ids": {
490490
frozenset({1, 3}): {0, 1} # NOTE {1,3} not {1,2}
491491
}
492492
},

tests/ui_tools/test_messages.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,8 +85,7 @@ def test_direct_message_to_self(self, mocker):
8585
)
8686
self.model.user_email = "[email protected]"
8787
mocker.patch(
88-
MODULE + ".MessageBox._is_direct_message_to_self", # rebased
89-
return_value=True,
88+
MODULE + ".MessageBox._is_direct_message_to_self", return_value=True
9089
)
9190
mocker.patch.object(MessageBox, "main_view")
9291
msg_box = MessageBox(message, self.model, None)

zulipterminal/helper.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -322,7 +322,7 @@ def index_messages(messages: List[Message], model: Any, index: Index) -> Index:
322322
...
323323
}
324324
},
325-
'private_msg_ids_by_user_ids': {
325+
'direct_msg_ids_by_user_ids': {
326326
(3, 7): { # user_ids frozenset
327327
51234,
328328
56454,
@@ -345,7 +345,7 @@ def index_messages(messages: List[Message], model: Any, index: Index) -> Index:
345345
23423,
346346
...
347347
},
348-
'private_msg_ids': {
348+
'direct_msg_ids': {
349349
22334,
350350
23423,
351351
...
@@ -452,7 +452,7 @@ def index_messages(messages: List[Message], model: Any, index: Index) -> Index:
452452
index["mentioned_msg_ids"].add(msg["id"])
453453

454454
if msg["type"] == "private":
455-
index["private_msg_ids"].add(msg["id"])
455+
index["direct_msg_ids"].add(msg["id"])
456456
recipients = frozenset(
457457
{recipient["id"] for recipient in msg["display_recipient"]}
458458
)
@@ -463,7 +463,7 @@ def index_messages(messages: List[Message], model: Any, index: Index) -> Index:
463463
for email in narrow[0][1].split(", ")
464464
] + [model.user_id]
465465
if recipients == frozenset(narrow_emails):
466-
index["private_msg_ids_by_user_ids"][recipients].add(msg["id"])
466+
index["direct_msg_ids_by_user_ids"][recipients].add(msg["id"])
467467

468468
if msg["type"] == "stream" and msg["stream_id"] == model.stream_id:
469469
index["stream_msg_ids_by_stream_id"][msg["stream_id"]].add(msg["id"])

zulipterminal/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -369,10 +369,10 @@ def get_message_ids_in_current_narrow(self) -> Set[int]:
369369
topic = narrow[1][1]
370370
ids = index["topic_msg_ids"][stream_id].get(topic, set())
371371
elif narrow[0][1] == "private":
372-
ids = index["private_msg_ids"]
372+
ids = index["direct_msg_ids"]
373373
elif narrow[0][0] == "pm-with":
374374
recipients = self.recipients
375-
ids = index["private_msg_ids_by_user_ids"].get(recipients, set())
375+
ids = index["direct_msg_ids_by_user_ids"].get(recipients, set())
376376
elif narrow[0][1] == "starred":
377377
ids = index["starred_msg_ids"]
378378
elif narrow[0][1] == "mentioned":

zulipterminal/ui_tools/messages.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,6 @@ def need_recipient_header(self) -> bool:
146146
else:
147147
raise RuntimeError("Invalid message type")
148148

149-
def _is_direct_message_to_self(self) -> bool:
150149
def _is_direct_message_to_self(self) -> bool:
151150
recipient_list = self.message["display_recipient"]
152151
return (

0 commit comments

Comments
 (0)