From c161b7afce58a06bc0d7e0146898b74a66ac818f Mon Sep 17 00:00:00 2001 From: Egor Date: Fri, 3 Jul 2020 16:09:00 +0500 Subject: [PATCH] feat: ChatMemberFactory and MessageFactory --- tests/factories/chat_member.py | 17 +++++++++++++++++ tests/factories/message.py | 15 +++++++++++++++ 2 files changed, 32 insertions(+) diff --git a/tests/factories/chat_member.py b/tests/factories/chat_member.py index e69de29b..d60732ee 100644 --- a/tests/factories/chat_member.py +++ b/tests/factories/chat_member.py @@ -0,0 +1,17 @@ +import factory + +from aiogram.api.types.chat_member import ChatMemberStatus +from tests.factories.user import UserFactory + + +class ChatMemberFactory(factory.Factory): + user = factory.SubFactory(UserFactory) + + status = ChatMemberStatus.ADMINISTRATOR + can_be_edited = False + can_change_info = True + can_delete_messages = True + can_invite_users = True + can_restrict_members = True + can_pin_messages = True + can_promote_members = False diff --git a/tests/factories/message.py b/tests/factories/message.py index e69de29b..0b1e7682 100644 --- a/tests/factories/message.py +++ b/tests/factories/message.py @@ -0,0 +1,15 @@ +import factory +from datetime import datetime + +from tests.factories import sequences +from tests.factories.chat import ChatFactory +from tests.factories.user import UserFactory + + +class MessageFactory(factory.Factory): + message_id = sequences.id_ + from_user = factory.SubFactory(UserFactory) + chat = factory.SubFactory(ChatFactory) + text = factory.Sequence(lambda n: f"Message text #{n}") + + date = factory.LazyFunction(lambda _: datetime.now().toordinal())