[PBE-1321] use SortOrder

GetStream · Feb 21, 2024 · 1dae3c3 · 1dae3c3
1 parent 06d1575
commit 1dae3c3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/stream_chat/tests/async_chat/test_segment.py b/stream_chat/tests/async_chat/test_segment.py
@@ -3,6 +3,7 @@
 import pytest
 
 from stream_chat.async_chat.client import StreamChatAsync
+from stream_chat.types.base import SortOrder
 from stream_chat.types.segment import SegmentType
 
 
@@ -70,12 +71,12 @@ async def test_segment_targets(self, client: StreamChatAsync):
         assert "next" in query_targets_1
         assert len(query_targets_1["targets"]) == 3
 
-        query_targets_2 = segment.query_targets(
+        query_targets_2 = await segment.query_targets(
             filter_conditions={"target_id": {"$lte": "<user_id>"}},
             options={
                 "limit": 3,
                 "next": query_targets_1["next"],
-                "sort": [{"field": "target_id", "direction": -1}],
+                "sort": [{"field": "target_id", "direction": SortOrder.DESC}],
             },
         )
         assert query_targets_2.is_ok()

diff --git a/stream_chat/tests/test_segment.py b/stream_chat/tests/test_segment.py
@@ -3,6 +3,7 @@
 import pytest
 
 from stream_chat import StreamChat
+from stream_chat.types.base import SortOrder
 from stream_chat.types.segment import SegmentType
 
 
@@ -75,7 +76,7 @@ def test_segment_targets(self, client: StreamChat):
             options={
                 "limit": 3,
                 "next": query_targets_1["next"],
-                "sort": [{"field": "target_id", "direction": -1}],
+                "sort": [{"field": "target_id", "direction": SortOrder.DESC}],
             },
         )
         assert query_targets_2.is_ok()