diff --git a/program/tests/factories.py b/program/tests/factories.py index 738c6ddcd28637f228ba27c85e41b3367405b2bc..fcf52b6b5cbad578dcdeda75799a506c19082b19 100644 --- a/program/tests/factories.py +++ b/program/tests/factories.py @@ -7,15 +7,20 @@ from django.contrib.auth.models import Permission, User from django.db.models import QuerySet from django.utils.timezone import now from program.models import ( + Category, FundingCategory, Host, Image, + Language, License, + LinkType, + MusicFocus, Note, RRule, Schedule, Show, TimeSlot, + Topic, Type, ) @@ -117,3 +122,33 @@ class LicenseFactory(DjangoModelFactory): class NoteFactory(DjangoModelFactory): class Meta: model = Note + + +class CategoryFactory(DjangoModelFactory): + class Meta: + model = Category + + +class LanguageFactory(DjangoModelFactory): + class Meta: + model = Language + + +class LinkTypeFactory(DjangoModelFactory): + class Meta: + model = LinkType + + +class MusicFocusFactory(DjangoModelFactory): + class Meta: + model = MusicFocus + + +class TopicFactory(DjangoModelFactory): + class Meta: + model = Topic + + +class OwnerFactory(DjangoModelFactory): + class Meta: + model = User