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())