Skip to content

Commit 93aea5f

Browse files
author
Daksh
committed
fix tests
1 parent 23d4086 commit 93aea5f

File tree

5 files changed

+22
-14
lines changed

5 files changed

+22
-14
lines changed

stream_chat/async_chat/client.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -356,11 +356,15 @@ async def update_message_partial(
356356
return await self.put(f"messages/{message_id}", data=data)
357357

358358
async def delete_message(
359-
self, message_id: str, delete_for_me: bool = False, deleted_by: str = None, **options: Any
359+
self,
360+
message_id: str,
361+
delete_for_me: bool = False,
362+
deleted_by: str = None,
363+
**options: Any,
360364
) -> StreamResponse:
361365
if delete_for_me and not deleted_by:
362366
raise ValueError("deleted_by is required when delete_for_me is True")
363-
367+
364368
data = options.copy()
365369
if delete_for_me:
366370
data["delete_for_me"] = True

stream_chat/base/client.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -558,13 +558,17 @@ def update_message_partial(
558558
"""
559559
pass
560560

561-
@abc.abstractmethod
561+
@abc.abstractmethod
562562
def delete_message(
563-
self, message_id: str, delete_for_me: bool = False, deleted_by: str = None, **options: Any
563+
self,
564+
message_id: str,
565+
delete_for_me: bool = False,
566+
deleted_by: str = None,
567+
**options: Any,
564568
) -> Union[StreamResponse, Awaitable[StreamResponse]]:
565569
"""
566570
Deletes a message.
567-
571+
568572
Args:
569573
message_id: The ID of the message to delete
570574
delete_for_me: If True, deletes the message only for the specified user

stream_chat/client.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -343,11 +343,15 @@ def update_message_partial(
343343
return self.put(f"messages/{message_id}", data=data)
344344

345345
def delete_message(
346-
self, message_id: str, delete_for_me: bool = False, deleted_by: str = None, **options: Any
346+
self,
347+
message_id: str,
348+
delete_for_me: bool = False,
349+
deleted_by: str = None,
350+
**options: Any,
347351
) -> StreamResponse:
348352
if delete_for_me and not deleted_by:
349353
raise ValueError("deleted_by is required when delete_for_me is True")
350-
354+
351355
data = options.copy()
352356
if delete_for_me:
353357
data["delete_for_me"] = True

stream_chat/tests/async_chat/test_client.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -473,9 +473,7 @@ async def test_delete_message_for_me(
473473

474474
# Delete message for the user
475475
response = await client.delete_message(
476-
msg_id,
477-
delete_for_me=True,
478-
deleted_by=random_user["id"]
476+
msg_id, delete_for_me=True, deleted_by=random_user["id"]
479477
)
480478

481479
# Verify the request succeeded

stream_chat/tests/test_client.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -518,9 +518,7 @@ def test_delete_message_for_me(
518518

519519
# Delete message for the user
520520
response = client.delete_message(
521-
msg_id,
522-
delete_for_me=True,
523-
deleted_by=random_user["id"]
521+
msg_id, delete_for_me=True, deleted_by=random_user["id"]
524522
)
525523

526524
# Verify the request succeeded
@@ -703,7 +701,7 @@ def test_query_channels_members_in(
703701
def test_create_blocklist(self, client: StreamChat):
704702
client.create_blocklist(name="Foo", words=["fudge", "heck"], type="word")
705703

706-
def test test_list_blocklists(self, client: StreamChat):
704+
def test_list_blocklists(self, client: StreamChat):
707705
response = client.list_blocklists()
708706
# There are now 2 default blocklists + the "Foo" blocklist created by test_create_blocklist
709707
assert len(response["blocklists"]) >= 3

0 commit comments

Comments
 (0)