aiogram/tests/factories/message.py

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