mirror of
https://github.com/aiogram/aiogram.git
synced 2026-04-08 16:37:47 +00:00
27 lines
829 B
Python
27 lines
829 B
Python
from datetime import datetime
|
|
|
|
import factory
|
|
|
|
from aiogram.api.types import Message
|
|
from tests.factories import sequences
|
|
from tests.factories.chat import ChatFactory
|
|
from tests.factories.user import UserFactory
|
|
|
|
|
|
class MessageFactory(factory.Factory):
|
|
class Meta:
|
|
model = Message
|
|
|
|
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())
|
|
|
|
def __new__(cls, *args, **kwargs) -> "MessageFactory.Meta.model":
|
|
"""
|
|
This is a dirty hack for correct type hints
|
|
See https://github.com/FactoryBoy/factory_boy/issues/468#issuecomment-505646794
|
|
"""
|
|
return super().__new__(*args, **kwargs)
|