test_query_threads.py

from typing import Dict

import pytest

from stream_chat import StreamChat
from stream_chat.types.stream_response import StreamResponse


@pytest.mark.incremental
class TestQueryThreads:
    def test_query_threads(self, client: StreamChat, channel, random_user: Dict):
        parent_message = channel.send_message(
            {"text": "Parent message"}, random_user["id"]
        )
        thread_message = channel.send_message(
            {"text": "Thread message", "parent_id": parent_message["message"]["id"]},
            random_user["id"],
        )

        filter_conditions = {"parent_message_id": parent_message["message"]["id"]}
        sort_conditions = [{"field": "created_at", "direction": -1}]

        response = client.query_threads(
            filter=filter_conditions, sort=sort_conditions, user_id=random_user["id"]
        )

        assert isinstance(response, StreamResponse)
        assert "threads" in response
        assert len(response["threads"]) > 0

        thread = response["threads"][0]
        assert "latest_replies" in thread
        assert len(thread["latest_replies"]) > 0
        assert thread["latest_replies"][0]["text"] == thread_message["message"]["text"]

    def test_query_threads_with_options(
        self, client: StreamChat, channel, random_user: Dict
    ):
        for i in range(3):
            parent_msg = channel.send_message(
                {"text": f"Parent message {i}"}, random_user["id"]
            )

            channel.send_message(
                {
                    "text": f"Thread message {i}",
                    "parent_id": parent_msg["message"]["id"],
                },
                random_user["id"],
            )

        filter_conditions = {"channel_cid": channel.cid}
        sort_conditions = [{"field": "created_at", "direction": -1}]

        response = client.query_threads(
            filter=filter_conditions,
            sort=sort_conditions,
            limit=1,
            user_id=random_user["id"],
        )

        assert isinstance(response, StreamResponse)
        assert "threads" in response
        assert len(response["threads"]) == 1
        assert "next" in response