Skip to content

Commit 77a5c8f

Browse files
committed
Fix failing tests in query threads
1 parent dcd4d96 commit 77a5c8f

File tree

2 files changed

+17
-19
lines changed

2 files changed

+17
-19
lines changed

stream_chat/tests/async_chat/test_query_threads.py

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ async def test_query_threads(
2222
)
2323

2424
# Query threads with filter and sort
25-
filter_conditions = {"parent_id": parent_message["message"]["id"]}
25+
filter_conditions = {"parent_message_id": parent_message["message"]["id"]}
2626
sort_conditions = [{"field": "created_at", "direction": -1}]
2727

2828
response = await client.query_threads(
@@ -43,23 +43,22 @@ async def test_query_threads(
4343
async def test_query_threads_with_options(
4444
self, client: StreamChatAsync, channel, random_user: Dict
4545
):
46-
# Create a thread with multiple messages
47-
parent_message = await channel.send_message(
48-
{"text": "Parent message"}, random_user["id"]
49-
)
50-
thread_messages = []
46+
5147
for i in range(3):
52-
msg = await channel.send_message(
48+
parent_msg = await channel.send_message(
49+
{"text": f"Parent message {i}"}, random_user["id"]
50+
)
51+
52+
thread_msg = await channel.send_message(
5353
{
5454
"text": f"Thread message {i}",
55-
"parent_id": parent_message["message"]["id"],
55+
"parent_id": parent_msg["message"]["id"],
5656
},
5757
random_user["id"],
5858
)
59-
thread_messages.append(msg)
6059

6160
# Query threads with limit and offset
62-
filter_conditions = {"parent_id": parent_message["message"]["id"]}
61+
filter_conditions = {"channel_cid": channel.cid}
6362
sort_conditions = [{"field": "created_at", "direction": -1}]
6463

6564
response = await client.query_threads(

stream_chat/tests/test_query_threads.py

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ def test_query_threads(self, client: StreamChat, channel, random_user: Dict):
1717
random_user["id"],
1818
)
1919

20-
filter_conditions = {"parent_id": parent_message["message"]["id"]}
20+
filter_conditions = {"parent_message_id": parent_message["message"]["id"]}
2121
sort_conditions = [{"field": "created_at", "direction": -1}]
2222

2323
response = client.query_threads(
@@ -36,21 +36,20 @@ def test_query_threads(self, client: StreamChat, channel, random_user: Dict):
3636
def test_query_threads_with_options(
3737
self, client: StreamChat, channel, random_user: Dict
3838
):
39-
parent_message = channel.send_message(
40-
{"text": "Parent message"}, random_user["id"]
41-
)
42-
thread_messages = []
4339
for i in range(3):
44-
msg = channel.send_message(
40+
parent_msg = channel.send_message(
41+
{"text": f"Parent message {i}"}, random_user["id"]
42+
)
43+
44+
thread_msg = channel.send_message(
4545
{
4646
"text": f"Thread message {i}",
47-
"parent_id": parent_message["message"]["id"],
47+
"parent_id": parent_msg["message"]["id"],
4848
},
4949
random_user["id"],
5050
)
51-
thread_messages.append(msg)
5251

53-
filter_conditions = {"parent_id": parent_message["message"]["id"]}
52+
filter_conditions = {"channel_cid": channel.cid}
5453
sort_conditions = [{"field": "created_at", "direction": -1}]
5554

5655
response = client.query_threads(

0 commit comments

Comments
 (0)