tests/support.py

"""Test utilities for building fixture data."""

from __future__ import annotations

from datetime import datetime

from vivamais.domain.entities import Message
from vivamais.domain.enums import MessageKind


def build_message(
    contact: str = "Maria",
    kind: MessageKind = MessageKind.TEXT,
    text: str | None = None,
    media_path: str | None = None,
    timestamp: datetime | None = None,
    from_me: bool = False,
) -> Message:
    """Build a Message with sensible defaults for testing."""
    if timestamp is None:
        timestamp = datetime(2026, 6, 5, 10, 0)
    return Message(
        contact=contact,
        kind=kind,
        text=text,
        media_path=media_path,
        timestamp=timestamp,
        from_me=from_me,
    )