import wagtail_factories

from vbv_lernwelt.learnpath.models import LearningPath, Topic, Circle, LearningSequence, LearningUnit


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

    class Meta:
        model = LearningPath


class TopicFactory(wagtail_factories.PageFactory):
    title = "Gewinnen von Kunden"
    is_visible = True

    class Meta:
        model = Topic


class CircleFactory(wagtail_factories.PageFactory):
    title = "Gewinnen"

    class Meta:
        model = Circle


class LearningSequenceFactory(wagtail_factories.PageFactory):
    title = "Grundlagen"

    class Meta:
        model = LearningSequence


class LearningUnitFactory(wagtail_factories.PageFactory):
    title = "Herzlich Willkommen"

    class Meta:
        model = LearningUnit