from random import choice

from polyfactory.factories.pydantic_factory import ModelFactory

from app.models import Event, WebhookEvent


class EventFactory(ModelFactory[Event]):
    __model__ = Event

    path = lambda: choice(["/shopify", "/facebook-ads", "/ad-reporting"])  # noqa: E731
    http_method = "POST"


class WebhookEventFactory(ModelFactory[WebhookEvent]):
    __model__ = WebhookEvent

    destination_group_id = "test"