async def setfloodaction(message: Message, chat: dict, strings: dict):
SUPPORTED_ACTIONS = ["kick", "ban", "mute", "tmute", "tban"] # noqa
if (action := message.get_args().lower()) not in SUPPORTED_ACTIONS:
return await message.reply(
strings["invalid_args"].format(
supported_actions=", ".join(SUPPORTED_ACTIONS)
)
)
async def setfloodaction(message: Message, chat: dict, strings: dict):
SUPPORTED_ACTIONS = ['kick', 'ban', 'mute'] # noqa
if (action := message.get_args().lower()) not in SUPPORTED_ACTIONS:
return await message.reply(strings['invalid_args'].format(
supported_actions=", ".join(SUPPORTED_ACTIONS)))
# coding: utf8
import pytest
from app.middlewares.unique_id import UniqueIdMiddleware
from aiogram.dispatcher.middlewares import BaseMiddleware
from aiogram.types.message import Message, Chat
MESSAGE_ID = 123456789
CHAT_ID = 424242
DATA_UNIQUE_ID_KEY = 'unique_id'
message = Message()
message.message_id = MESSAGE_ID
message.chat = Chat
message.chat.id = CHAT_ID
TEMPLATE_UNIQUE_ID = f'{message.chat.id}-{message.message_id}'
@pytest.mark.unit
class TestUniqueId:
def test__unique_id_cls_is_sub_base_middleware(self):
mid_unique_id = UniqueIdMiddleware()
assert isinstance(mid_unique_id, BaseMiddleware)
def test__make_unique_id(self):
mid_unique_id = UniqueIdMiddleware()
unique_id = mid_unique_id._make(message)
assert unique_id == TEMPLATE_UNIQUE_ID