From 2c519fd4320d0c9df8817a44bbb8b0ef41129345 Mon Sep 17 00:00:00 2001
From: Ernesto Rico Schmidt <ernesto@helsinki.at>
Date: Wed, 20 Mar 2024 20:11:45 -0400
Subject: [PATCH] test: clean-up test factories
---
program/tests/factories.py | 39 +++++++++++++++++++-------------------
1 file changed, 20 insertions(+), 19 deletions(-)
diff --git a/program/tests/factories.py b/program/tests/factories.py
index 335b76c2..1ce0368c 100644
--- a/program/tests/factories.py
+++ b/program/tests/factories.py
@@ -1,6 +1,7 @@
from datetime import timedelta
-import factory
+from factory import Sequence
+from factory.django import DjangoModelFactory
from django.contrib.auth.models import User
from django.utils.timezone import now
@@ -17,56 +18,56 @@ from program.models import (
)
-class CommonUserFactory(factory.django.DjangoModelFactory):
+class CommonUserFactory(DjangoModelFactory):
class Meta:
model = User
password = "password"
- username = factory.Sequence(lambda n: "common_%d" % n)
+ username = Sequence(lambda n: "common_%d" % n)
-class HostFactory(factory.django.DjangoModelFactory):
+class HostFactory(DjangoModelFactory):
class Meta:
model = Host
- name = factory.Sequence(lambda n: "host %d" % n)
+ name = Sequence(lambda n: "host %d" % n)
-class ImageFactory(factory.django.DjangoModelFactory):
+class ImageFactory(DjangoModelFactory):
class Meta:
model = Image
-class FundingCategoryFactory(factory.django.DjangoModelFactory):
+class FundingCategoryFactory(DjangoModelFactory):
class Meta:
model = FundingCategory
- name = factory.Sequence(lambda n: "funding category %d" % n)
- slug = factory.Sequence(lambda n: "fc_%d" % n)
+ name = Sequence(lambda n: "funding category %d" % n)
+ slug = Sequence(lambda n: "fc_%d" % n)
-class TypeFactory(factory.django.DjangoModelFactory):
+class TypeFactory(DjangoModelFactory):
class Meta:
model = Type
- name = factory.Sequence(lambda n: "type %d" % n)
- slug = factory.Sequence(lambda n: "t_%d" % n)
+ name = Sequence(lambda n: "type %d" % n)
+ slug = Sequence(lambda n: "t_%d" % n)
-class ShowFactory(factory.django.DjangoModelFactory):
+class ShowFactory(DjangoModelFactory):
class Meta:
model = Show
- name = factory.Sequence(lambda n: "show %d" % n)
- slug = factory.Sequence(lambda n: "%s_d" % n)
+ name = Sequence(lambda n: "show %d" % n)
+ slug = Sequence(lambda n: "%s_d" % n)
-class RRuleFactory(factory.django.DjangoModelFactory):
+class RRuleFactory(DjangoModelFactory):
class Meta:
model = RRule
-class ScheduleFactory(factory.django.DjangoModelFactory):
+class ScheduleFactory(DjangoModelFactory):
class Meta:
model = Schedule
@@ -75,7 +76,7 @@ class ScheduleFactory(factory.django.DjangoModelFactory):
start_time = "02:00:00"
-class TimeslotFactory(factory.django.DjangoModelFactory):
+class TimeslotFactory(DjangoModelFactory):
class Meta:
model = TimeSlot
@@ -83,7 +84,7 @@ class TimeslotFactory(factory.django.DjangoModelFactory):
start = now()
-class LicenseFactory(factory.django.DjangoModelFactory):
+class LicenseFactory(DjangoModelFactory):
class Meta:
model = License
--
GitLab