feat: ChatMemberFactory and MessageFactory

This commit is contained in:
Egor 2020-07-03 16:09:00 +05:00
parent 3016c8e21e
commit c161b7afce
2 changed files with 32 additions and 0 deletions

View file

@ -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

View file

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