import wagtail_factories
from factory.django import DjangoModelFactory

from vbv_lernwelt.course.models import CoursePage, Course


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

    name = 'Versicherungsvermittler/in'
    category_name = 'Handlungsfeld'


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

    class Meta:
        model = CoursePage