From 2e21156e568cd1c1ec0cd992141bdeb1b060b2fe Mon Sep 17 00:00:00 2001
From: Ernesto Rico Schmidt <ernesto@helsinki.at>
Date: Tue, 17 Sep 2024 12:20:19 -0400
Subject: [PATCH] test: add CBA factory and fixture
---
conftest.py | 7 +++++++
program/tests/factories.py | 9 +++++++++
2 files changed, 16 insertions(+)
diff --git a/conftest.py b/conftest.py
index bad250b0..a640db78 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 94e577ef..c4761815 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)
--
GitLab