diff --git a/conftest.py b/conftest.py index bad250b0a709df8c7aed77c2c922e46cbcb7e46b..a640db7822fd7ec1bc313b40cc103f213cf33403 100644 --- a/conftest.py +++ b/conftest.py @@ -6,6 +6,7 @@ from rest_framework.test import APIClient from django.contrib.auth.models import Permission, User from django.core.files.uploadedfile import SimpleUploadedFile from program.models import ( + CBA, Category, FundingCategory, Image, @@ -24,6 +25,7 @@ from program.models import ( ) from program.tests.factories import ( CategoryFactory, + CBAFactory, CommonUserFactory, FundingCategoryFactory, ImageFactory, @@ -202,6 +204,11 @@ def radio_settings(fallback_show) -> RadioSettings: ) +@pytest.fixture +def cba(common_user1) -> CBA: + return CBAFactory(user=common_user1) + + @pytest.fixture def owned_show(common_user1, show) -> Show: """Show owned by a common user""" diff --git a/program/tests/factories.py b/program/tests/factories.py index 94e577ef0638451edd3d1c431c48534921589bec..c4761815c2f697f5c0c5c49f24403c3575c432a4 100644 --- a/program/tests/factories.py +++ b/program/tests/factories.py @@ -7,6 +7,7 @@ from django.contrib.auth.models import Permission, User from django.db.models import QuerySet from django.utils.timezone import now from program.models import ( + CBA, Category, FundingCategory, Image, @@ -158,3 +159,11 @@ class OwnerFactory(DjangoModelFactory): class RadioSettingsFactory(DjangoModelFactory): class Meta: model = RadioSettings + + +class CBAFactory(DjangoModelFactory): + class Meta: + model = CBA + + username = Sequence(lambda n: "username_%d" % n) + user_token = Sequence(lambda n: "user_token_%d" % n)