from factory import Factory, Faker
from scheduler.models import BoefjeMeta, RawData


class RawDataFactory(Factory):
    class Meta:
        model = RawData

    id: str = Faker("uuid4")
    boefje_meta: BoefjeMeta = None
    mime_types: list[dict[str, str]] = [{"value": "text/plain"}, {"value": "text/html"}, {"value": "text/xml"}]
    secure_hash: str = ""
    hash_retrieval_link: str = ""
