import wagtail_factories

from vbv_lernwelt.competence.models import CompetenceProfilePage, PerformanceCriteria, CompetencePage


class CompetenceProfilePageFactory(wagtail_factories.PageFactory):
    title = 'Kompetenzprofil'

    class Meta:
        model = CompetenceProfilePage


class CompetencePageFactory(wagtail_factories.PageFactory):
    competence_id = 'A1'
    title = 'Weiterempfehlung für Neukunden generieren'

    class Meta:
        model = CompetencePage


class PerformanceCriteriaFactory(wagtail_factories.PageFactory):
    competence_id = 'A1.1'
    title = 'Bestehende Kunden so zu beraten, dass sie von diesen weiterempfohlen werden'

    class Meta:
        model = PerformanceCriteria