import wagtail_factories
from factory.django import DjangoModelFactory

from vbv_lernwelt.course.models import Course, CoursePage


class CourseFactory(DjangoModelFactory):
    class Meta:
        model = Course

    title = "Versicherungsvermittler/-in"
    category_name = "Handlungsfeld"


class CoursePageFactory(wagtail_factories.PageFactory):
    title = "Versicherungsvermittler/-in"

    class Meta:
        model = CoursePage