diff --git a/gestioncof/tests/utils.py b/gestioncof/tests/utils.py index 7325e350..f2d95c16 100644 --- a/gestioncof/tests/utils.py +++ b/gestioncof/tests/utils.py @@ -1,5 +1,12 @@ from django.contrib.auth import get_user_model +from gestioncof.petits_cours_models import ( + PetitCoursAbility, + PetitCoursAttributionCounter, + PetitCoursDemande, + PetitCoursSubject, +) + User = get_user_model() @@ -66,3 +73,23 @@ def create_root(username, attrs=None): attrs.setdefault("is_staff", True) attrs.setdefault("is_superuser", True) return _create_user(username, attrs=attrs) + + +def create_petitcours_ability(**kwargs): + if "user" not in kwargs: + kwargs["user"] = create_user() + if "matiere" not in kwargs: + kwargs["matiere"] = create_petitcours_subject() + if "niveau" not in kwargs: + kwargs["niveau"] = "college" + ability = PetitCoursAbility.objects.create(**kwargs) + PetitCoursAttributionCounter.get_uptodate(ability.user, ability.matiere) + return ability + + +def create_petitcours_demande(**kwargs): + return PetitCoursDemande.objects.create(**kwargs) + + +def create_petitcours_subject(**kwargs): + return PetitCoursSubject.objects.create(**kwargs)