-
Notifications
You must be signed in to change notification settings - Fork 9
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: add chat_deleted_by_user event
- Loading branch information
Showing
8 changed files
with
321 additions
and
225 deletions.
There are no files selected for viewing
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,57 @@ | ||
from dataclasses import dataclass | ||
from typing import Any, Dict, Literal | ||
from uuid import UUID | ||
|
||
from pydantic import Field | ||
|
||
from pybotx.models.api_base import VerifiedPayloadBaseModel | ||
from pybotx.models.base_command import ( | ||
BaseBotAPIContext, | ||
BotAPIBaseCommand, | ||
BotAPIBaseSystemEventPayload, | ||
BotCommandBase, | ||
) | ||
from pybotx.models.bot_account import BotAccount | ||
from pybotx.models.enums import BotAPISystemEventTypes | ||
|
||
|
||
@dataclass | ||
class ChatDeletedByUserEvent(BotCommandBase): | ||
"""Event `system:chat_deleted_by_user`. | ||
Attributes: | ||
sync_id: Event sync id. | ||
chat_id: Deleted chat id. | ||
huid: huid of the deleter. | ||
""" | ||
|
||
sync_id: UUID | ||
chat_id: UUID | ||
huid: UUID | ||
|
||
|
||
class BotAPIChatDeletedByUserData(VerifiedPayloadBaseModel): | ||
user_huid: UUID | ||
group_chat_id: UUID | ||
|
||
|
||
class BotAPIChatDeletedByUserPayload(BotAPIBaseSystemEventPayload): | ||
body: Literal[BotAPISystemEventTypes.CHAT_DELETED_BY_USER] | ||
data: BotAPIChatDeletedByUserData | ||
|
||
|
||
class BotAPIChatDeletedByUser(BotAPIBaseCommand): | ||
payload: BotAPIChatDeletedByUserPayload = Field(..., alias="command") | ||
sender: BaseBotAPIContext = Field(..., alias="from") | ||
|
||
def to_domain(self, raw_command: Dict[str, Any]) -> ChatDeletedByUserEvent: | ||
return ChatDeletedByUserEvent( | ||
sync_id=self.sync_id, | ||
bot=BotAccount( | ||
id=self.bot_id, | ||
host=self.sender.host, | ||
), | ||
chat_id=self.payload.data.group_chat_id, | ||
huid=self.payload.data.user_huid, | ||
raw_command=raw_command, | ||
) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,90 @@ | ||
from typing import Optional | ||
from uuid import UUID | ||
|
||
import pytest | ||
|
||
from pybotx import ( | ||
Bot, | ||
BotAccount, | ||
BotAccountWithSecret, | ||
HandlerCollector, | ||
lifespan_wrapper, | ||
) | ||
from pybotx.models.system_events.chat_deleted_by_user import ChatDeletedByUserEvent | ||
|
||
pytestmark = [ | ||
pytest.mark.asyncio, | ||
pytest.mark.mock_authorization, | ||
pytest.mark.usefixtures("respx_mock"), | ||
] | ||
|
||
|
||
async def test__chat_deleted_by_user__succeed( | ||
bot_account: BotAccountWithSecret, | ||
) -> None: | ||
# - Arrange - | ||
payload = { | ||
"sync_id": "a465f0f3-1354-491c-8f11-f400164295cb", | ||
"command": { | ||
"body": "system:chat_deleted_by_user", | ||
"data": { | ||
"group_chat_id": "6fa5f1e9-1453-0ad7-2d6d-b791467e382a", | ||
"user_huid": "37b821f5-a2be-5dc1-b107-87807ce97e56", | ||
}, | ||
"command_type": "system", | ||
"metadata": {}, | ||
}, | ||
"async_files": [], | ||
"attachments": [], | ||
"entities": [], | ||
"from": { | ||
"user_huid": None, | ||
"group_chat_id": "6fa5f1e9-1453-0ad7-2d6d-b791467e382a", | ||
"ad_login": None, | ||
"ad_domain": None, | ||
"username": None, | ||
"chat_type": None, | ||
"manufacturer": None, | ||
"device": None, | ||
"device_software": None, | ||
"device_meta": {}, | ||
"platform": None, | ||
"platform_package_id": None, | ||
"is_admin": None, | ||
"is_creator": None, | ||
"app_version": None, | ||
"locale": "en", | ||
"host": "cts.example.com", | ||
}, | ||
"bot_id": "24348246-6791-4ac0-9d86-b948cd6a0e46", | ||
"proto_version": 4, | ||
"source_sync_id": None, | ||
} | ||
|
||
collector = HandlerCollector() | ||
chat_deleted: Optional[ChatDeletedByUserEvent] = None | ||
|
||
@collector.chat_deleted_by_user | ||
async def chat_deleted_handler(event: ChatDeletedByUserEvent, bot: Bot) -> None: | ||
nonlocal chat_deleted | ||
chat_deleted = event | ||
# Drop `raw_command` from asserting | ||
chat_deleted.raw_command = None | ||
|
||
built_bot = Bot(collectors=[collector], bot_accounts=[bot_account]) | ||
|
||
# - Act - | ||
async with lifespan_wrapper(built_bot) as bot: | ||
bot.async_execute_raw_bot_command(payload, verify_request=False) | ||
|
||
# - Assert - | ||
assert chat_deleted == ChatDeletedByUserEvent( | ||
sync_id=UUID("a465f0f3-1354-491c-8f11-f400164295cb"), | ||
bot=BotAccount( | ||
id=UUID("24348246-6791-4ac0-9d86-b948cd6a0e46"), | ||
host="cts.example.com", | ||
), | ||
chat_id=UUID("6fa5f1e9-1453-0ad7-2d6d-b791467e382a"), | ||
huid=UUID("37b821f5-a2be-5dc1-b107-87807ce97e56"), | ||
raw_command=None, | ||
) |