from polyfactory import Use
from polyfactory.factories.pydantic_factory import ModelFactory

from app.types import EmailPresaveEvent, NativePresaveEvent


class EmailPresaveEventFactory(ModelFactory[EmailPresaveEvent]):
    email = Use(ModelFactory.__faker__.email)


class NativePresaveEventFactory(ModelFactory[NativePresaveEvent]):
    pass