import wagtail_factories

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


class CompetenceProfilePageFactory(wagtail_factories.PageFactory):
    title = "KompetenzNavi"

    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