import wagtail_factories

from vbv_lernwelt.media_files.models import ContentDocument, UserDocument


class ContentDocumentFactory(wagtail_factories.DocumentFactory):
    link_display_text = "Dokument herunter laden"
    description = ""

    class Meta:
        model = ContentDocument
        django_get_or_create = ("title", "description")


class UserDocumentFactory(wagtail_factories.DocumentFactory):
    class Meta:
        model = UserDocument
        django_get_or_create = ("title",)
