mirror of
https://github.com/aiogram/aiogram.git
synced 2026-04-08 16:37:47 +00:00
feat: ChatMemberFactory and MessageFactory
This commit is contained in:
parent
3016c8e21e
commit
c161b7afce
2 changed files with 32 additions and 0 deletions
|
|
@ -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
|
||||
|
|
@ -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())
|
||||
Loading…
Add table
Add a link
Reference in a new issue